| gvSIG desktop 1

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2005 IVER T.I. and Generalitat Valenciana.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

package com.iver.cit.gvsig.fmap.core.symbols;

import java.awt.BasicStroke;

import java.awt.Color;

import java.awt.Graphics2D;

import java.awt.Shape;

import java.awt.geom.AffineTransform;

import java.awt.geom.PathIterator;

import java.awt.geom.Point2D;

import java.awt.image.BufferedImage;

import java.io.File;

import java.io.IOException;

import java.net.URL;

import java.util.Random;

import javax.imageio.ImageIO;

import javax.print.attribute.PrintRequestAttributeSet;

import org.apache.batik.ext.awt.geom.PathLength;

import org.apache.log4j.Logger;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.iver.cit.gvsig.fmap.Messages;

import com.iver.cit.gvsig.fmap.core.FPolyline2D;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.core.SymbologyFactory;

import com.iver.utiles.XMLEntity;

/**

 * PictureLineSymbol allows to use any symbol defined as an image (by an image file)

 * supported  by gvSIG.This symbol will be used as an initial object.The line will be

 * painted as a succession of puntual symbols through the path defined by it(the line).

 *

 * @author jaume dominguez faus - jaume.dominguez@iver.es

 */

public class PictureLineSymbol extends AbstractLineSymbol  {

        transient private BufferedImage img;

        transient private BufferedImage selImg;

        transient private PictureLineSymbol selectionSym;

        private String selImagePath;

        private double width;

        private String imagePath;

        private boolean selected;

        private double xScale = 1;

        private double yScale = 1;

        /**

         * Constructor method

         *

         */

        public PictureLineSymbol() {

                super();

        }

        /**

         * Constructor method

         * @param imageURL, URL of the normal image

         * @param selImageURL, URL of the image when it is selected in the map

         * @throws IOException

         */

        public PictureLineSymbol(URL imageURL, URL selImageURL) throws IOException {

                this(new File(imageURL.getFile()), selImageURL !=null ? new File(selImageURL.getFile()) : null);

        }

        /**

         * Constructor method

         * @param imageFile, File of the normal image

         * @param selImageFile, File of the image when it is selected in the map

         * @throws IOException

         */

        public PictureLineSymbol(File imageFile, File selImageFile) throws IOException {

                setImage(imageFile);

                if (selImageFile!=null)

                        setSelImage(selImageFile);

        }

        /**

         * Sets the file for the image to be used as a picture line symbol

         * @param imageFile, File

         * @throws IOException

         */

        public void setImage(File imageFile) throws IOException{

                img = ImageIO.read(imageFile);

                imagePath = imageFile.getAbsolutePath();

//                imageOutline = null;

        }

        /**

         * Sets the file for the image to be used as a picture line symbol (when it is selected in the map)

         * @param imageFile, File

         * @throws IOException

         */

        public void setSelImage(File imageFile) throws IOException{

                if (imageFile!=null && imageFile.exists()) {

                        selImg = ImageIO.read(imageFile);

                        selImagePath = imageFile.getAbsolutePath();

                } else {

                        selImg = img;

                        selImagePath = imagePath;

                }

//                imageOutline = null;

        }

        public void setLineWidth(double width) {

                this.width = width;

        }

        public double getLineWidth() {

                return width;

        }

        public ISymbol getSymbolForSelection() {

                if (selectionSym == null) {

                        selectionSym = (PictureLineSymbol) SymbologyFactory.createSymbolFromXML(getXMLEntity(), getDescription());

                        selectionSym.selected=true;

                        selectionSym.selectionSym = selectionSym; // avoid too much lazy creations

                }

                return selectionSym;

        }

        /**

         * Returns the image that will substitute the picture line symbol when it is selected

         * in the map

         * @return img, BufferedImage

         * @throws IOException

         */

        private BufferedImage getImg() throws IOException {

                if (img == null) {

                        img = ImageIO.read(new File(imagePath));

                }

                return img;

        }

        /**

         * Returns the image that will substitute the picture line symbol when it is selected

         * in the map

         * @return selImg, BufferedImage

         * @throws IOException

         */

        private BufferedImage getSelImg() throws IOException {

                if (selImg == null) {

                        if (selImagePath!= null) {

                                selImg = ImageIO.read(new File(selImagePath));

                        } else return getImg();

                }

                return selImg;

        }

        public void draw(Graphics2D g, AffineTransform affineTransform, FShape shp) {

                g.setClip(new BasicStroke((float) width, BasicStroke.CAP_ROUND,BasicStroke.CAP_ROUND).createStrokedShape(shp));

                try{

                        BufferedImage img = selected ? getSelImg() : getImg();

                        if (xScale<=0 &&  yScale<=0)

                                return;

                        AffineTransform scaleInstance = AffineTransform.getScaleInstance(xScale, yScale);

                        final double imageWidth  = img.getWidth()  * xScale;

                        final double imageHeight = img.getHeight() * yScale;

                        if (imageWidth==0 || imageHeight==0) return;

                        double halfImageHeight = imageHeight*.5;

                        PathLength pl = new PathLength(shp);

                        PathIterator iterator = ((FPolyline2D) shp).getPathIterator(null, 0.8);

                        double[] theData = new double[6];

                        Point2D firstPoint = null, startPoint = null, endPoint = null;

                        if (!iterator.isDone()) {

                                if ( iterator.currentSegment(theData) != PathIterator.SEG_CLOSE) {

                                        firstPoint = new Point2D.Double(theData[0], theData[1]);

                                }

                        }

                        float currentPathLength = 1;

                        while (!iterator.isDone()) {

                                int theType = iterator.currentSegment(theData);

                                switch (theType) {

                                case PathIterator.SEG_MOVETO:

                                        startPoint = new Point2D.Double(theData[0], theData[1]);

                                        endPoint = null;

                                        iterator.next();

                                        continue;

                                case PathIterator.SEG_LINETO:

                                case PathIterator.SEG_QUADTO:

                                case PathIterator.SEG_CUBICTO:

                                        endPoint = new Point2D.Double(theData[0], theData[1]);

                                        break;

                                case PathIterator.SEG_CLOSE:

                                        endPoint = startPoint;

                                        startPoint = firstPoint;

                                        break;

                                }

                                double a = endPoint.getX() - startPoint.getX();

                                double b = endPoint.getY() - startPoint.getY();

                                double theta = pl.angleAtLength(currentPathLength);

                                double x = startPoint.getX();

                                double y = startPoint.getY();

                                // Theorem of Pythagoras

                                float segmentLength = (float) Math.sqrt(a*a + b*b);

                                // compute how many times the image has to be drawn

                                // to completely cover this segment's length

                                int count = (int) Math.ceil(segmentLength/imageWidth);

                                for (int i = 0; i < count; i++) {

                                        g.translate(x, y);

                                        g.rotate(theta);

                                        double xOffsetTranslation = imageWidth*i;

                                                g.translate(xOffsetTranslation, -halfImageHeight);

                                                g.drawImage(img, scaleInstance, null);

                                                g.translate(-xOffsetTranslation, halfImageHeight);

                                        g.rotate(-theta);

                                        g.translate(-x, -y);

                                }

                                startPoint = endPoint;

                                currentPathLength += segmentLength;

                                iterator.next();

                        }

                } catch (IOException e) {

                        Logger.getLogger(this.getClass()).error(Messages.getString("failed_to_read_image_file"), e);

                }

                g.setClip(null);

        }

        public int getPixExtentPlus(Graphics2D g, AffineTransform affineTransform,

                        Shape shp) {

                // TODO Implement it

                throw new Error("Not yet implemented!");

        }

        public XMLEntity getXMLEntity() {

                XMLEntity xml = new XMLEntity();

                xml.putProperty("className", getClassName());

                xml.putProperty("isShapeVisible", isShapeVisible());

                xml.putProperty("desc", getDescription());

                xml.putProperty("imagePath", imagePath);

                xml.putProperty("selImagePath", selImagePath);

                xml.putProperty("lineWidth", getLineWidth());

                xml.putProperty("xScale", getXScale());

                xml.putProperty("yScale", getYScale());

                // measure unit

                xml.putProperty("unit", getUnit());

                // reference system

                xml.putProperty("referenceSystem", getReferenceSystem());

                return xml;

        }

        public void setXMLEntity(XMLEntity xml) {

                setDescription(xml.getStringProperty("desc"));

                setIsShapeVisible(xml.getBooleanProperty("isShapeVisible"));

                imagePath = xml.getStringProperty("imagePath");

                selImagePath = xml.getStringProperty("selImagePath");

                setLineWidth(xml.getDoubleProperty("lineWidth"));

                setXScale(xml.getDoubleProperty("xScale"));

                setYScale(xml.getDoubleProperty("yScale"));

                setReferenceSystem(xml.getIntProperty("referenceSystem"));

                setUnit(xml.getIntProperty("unit"));

        }

        /**

         * Sets the yscale for the picture line symbol

         * @param yScale

         */

        public void setYScale(double yScale) {

                this.yScale = yScale;

        }

        /**

         * Sets the xscale for the picture line symbol

         * @param xScale

         */

        public void setXScale(double xScale) {

                this.xScale = xScale;

        }

        public String getClassName() {

                return getClass().getName();

        }

        public void print(Graphics2D g, AffineTransform at, FShape shape,

                        PrintRequestAttributeSet properties) throws ReadDriverException {

                // TODO Implement it

                throw new Error("Not yet implemented!");

        }

        /**

         * Returns the path of the image that is used as a picture line symbol (when it

         * is selected in the map)

         * @return selimagePath,String

         */

        public String getSelImagePath() {

                return selImagePath;

        }

        /**

         * Returns the path of the image that is used as a picture line symbol

         * @return imagePath,String

         */

        public String getImagePath() {

                return imagePath;

        }

        /**

         * Returns the xscale for the picture line symbol

         * @param xScale

         */

        public double getXScale() {

                return xScale;

        }

        /**

         * Returns the yscale for the picture line symbol

         * @param yScale

         */

        public double getYScale() {

                return yScale;

        }

}