Application: gvSIG desktop

/* 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 org.gvsig.labeling.symbol;

import java.awt.Color;

import java.awt.Font;

import java.awt.Graphics2D;

import java.awt.Rectangle;

import java.awt.Shape;

import java.awt.font.FontRenderContext;

import java.awt.font.GlyphVector;

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.awt.image.BufferedImage;

import org.gvsig.compat.print.PrintAttributes;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.Geometry.SUBTYPES;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.rendering.symbols.CartographicSupport;

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol;

import org.gvsig.labeling.lang.LabelClassUtils;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.marker.IMarkerSymbol_v2;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.marker.impl.AbstractMarkerSymbol;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.style.IMask;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.persistence.PersistenceManager;

import org.gvsig.tools.persistence.PersistentState;

import org.gvsig.tools.persistence.exception.PersistenceException;

import org.gvsig.tools.task.Cancellable;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Allows to use a source of TrueType characters  to define the marker that will

 * substitute the symbol.If the picture is defined in a source, the performance is

 * more agile.

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

 */

public class CharacterMarkerSymbol extends AbstractMarkerSymbol implements IMarkerSymbol_v2 {

        private static Logger logger =

                        LoggerFactory.getLogger(CharacterMarkerSymbol.class);

        public static final String CHARACTER_MARKER_SYMBOL_PERSISTENCE_NAME =

                        "CHARACTER_MARKER_SYMBOL_PERSISTENCE_NAME";

        private Font font = new Font("Arial", Font.PLAIN, 20);

        private int unicode;

        private ISymbol selectionSymbol;

        private VisualCorrection visualCorrection;

//        private boolean isShapeVisible = true;

//        private String desc = "";

//        private double size;

//        private Color color = Color.BLACK;

//        private Point2D offset = new Point2D.Double();

//        private IMask mask;

//        private double rotation_radians = 0;

//        private int unit = -1;

//        private int referenceSystem = CartographicSupport.WORLD;

        private int alpha = 255;

        /*

         *

         */

        private static GeometryManager geoman = GeometryLocator.getGeometryManager();

        /**

         * Creates a new instance of CharacterMarker with default values

         *

         */

        public CharacterMarkerSymbol() {

                super();

        }

        /**

         * Creates a new instance of CharacterMarker specifying the marker source

         * font, the character code corresponding to the symbol, and the color that

         * will be used in rendering time.

         *

         * @param font -

         *            src Font

         * @param charCode -

         *            character code of the symbol for this font

         * @param color -

         *            color to be used in when rendering.

         */

        public CharacterMarkerSymbol(Font font, int charCode, Color color) {

                super();

                this.font = font;

                unicode = charCode;

                setColor(color);

        }

        /**

         * Returns the font that will be used to define the symbol

         * @return font

         */

        public Font getFont() {

            if(font == null){

                font = new Font("Arial", Font.PLAIN, (int) Math.round(getSize()));

            }

            return font;

        }

        /**

         * Sets the font that will be used to define the symbol

         * @return font

         */

        public void setFont(Font font) {

                this.font = font;

                this.setSize(this.font.getSize());

        }

        @Override

    public ISymbol getSymbolForSelection(Color selectionColor) {

        if (selectionSymbol == null) {

            selectionSymbol = (ISymbol) LabelClassUtils.clone(this);

            selectionSymbol.setColor(selectionColor);

        } else {

            selectionSymbol.setColor(selectionColor);

        }

        if (selectionSymbol instanceof CartographicSupport) {

            ((CartographicSupport) selectionSymbol).setUnit(this.getUnit());

        }

        return selectionSymbol;

    }

    @Override

        public void draw(Graphics2D g,

                        AffineTransform affineTransform, Geometry geom,

                        Feature feat, Cancellable cancel, Rectangle r) {

                double size_d = getCartographicSize();

//                size_d = this.getPrintSize(properties, size_d);

                size_d = this.getAdjustedSize(r, size_d);

                double theta = getEfectiveRotationInRadians(feat);

                int xOffset = (int) getEfectiveOffset(feat).getX(); 

                int yOffset = (int) getEfectiveOffset(feat).getY(); 

                if (size_d < 0.0001) {

                        return;

                }

                org.gvsig.fmap.geom.primitive.Point cen = null;

                try {

                        cen = geom.centroid();

                } catch (Exception e) {

                        logger.warn("While getting centroid", e);

                        return;

                }

        g.setColor(getColor());

                Point2D p = new Point2D.Double(cen.getX(), cen.getY());

                if (isVisuallyCorrected()) {

                        size_d *= visualCorrection.sizeScale;

                        p.setLocation(p.getX() - xOffset*size_d*visualCorrection.xOffsetScale, p.getY() - yOffset*size_d*visualCorrection.yOffsetScale);

                }

                g.setFont(getFont().deriveFont((float)size_d));

                g.translate((int) (p.getX() + xOffset), (int) (p.getY()  + yOffset));

                if (theta != 0)        {

                    g.rotate(theta);

                }

                char[] text = new char[] { (char) unicode };

                IMask mask = getMask();

                if (mask != null) {

                        FontRenderContext frc = g.getFontRenderContext();

                        GlyphVector gv = font.createGlyphVector(frc, text );

                        Shape markerShape = gv.getOutline(0, 0);

                        mask.getFillSymbol().

                                         draw(g, null, mask.getHaloShape(markerShape), feat, cancel);

                }

                g.drawChars(text, 0, text.length, - (int) (size_d*0.4), (int) (size_d*0.4));

                if (theta!=0) {

                    g.rotate(-theta);

                }

                g.translate(-(int) (p.getX() + xOffset), - (int) (p.getY() + yOffset));

        }

        /**

         * Sets the unicode for a symbol represented by a character

         * @param symbol, int

         */

        public void setUnicode(int symbol) {

                this.unicode = symbol;

        }

        /**

         * Obtains the unicode for a symbol

         * @return unicode, int

         */

        public int getUnicode() {

                return unicode;

        }

        /*

        public void setXMLEntity(XMLEntity xml) {

                setColor(StringUtilities.string2Color(xml.getStringProperty("color")));

                Point p = new Point();

                p.setLocation(xml.getDoubleProperty("xOffset"), xml.getDoubleProperty("yOffset"));

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

                size = xml.getDoubleProperty("size");

                font = new Font(xml.getStringProperty("font"),

                                xml.getIntProperty("fontStyle"),

                                (int) size);

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

                unicode = xml.getIntProperty("symbolCode");

                setOffset(p);

                setRotation(xml.getDoubleProperty("rotation"));

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

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

        }

        */

//        public double getSize() {

//                return size;

//        }

    @Override

        public void setSize(double size) {

                super.setSize(size);

                //NOTE: No hay que llamar al setFont

                font = new Font(font.getName(), font.getStyle(), (int) Math.round(size));

        }

        /**

         * Returns true of false depending if the character marker symbol selected is visually

         * corrected.That is, if the character has been properly scaled to use a specific number

         * of pixels to be represented.

         *

         * @return boolean

         */

        public boolean isVisuallyCorrected() {

                return visualCorrection != null;

        }

        /**

         * Sets the visual correction for a character in order to scale it if it is necessary.

         *

         * @return boolean

         */

        public void setVisuallyCorrected(boolean visuallyCorrected) {

                if (visuallyCorrected && visualCorrection == null) {

                        CharacterMarkerSymbol clone =

                                        (CharacterMarkerSymbol) LabelClassUtils.clone(this);

                        int frameSize = 200;

                        double symbolSize = 100;

                        BufferedImage bi = new BufferedImage(frameSize, frameSize, BufferedImage.TYPE_INT_ARGB);

                        Graphics2D aGraphics = bi.createGraphics();

                        clone.setOffset(new Point2D.Double(0,0));

                        clone.setSize(symbolSize);

                        clone.setColor(Color.PINK);

                        // draw it at center of the image

                        org.gvsig.fmap.geom.primitive.Point pCenter = null;

                        try {

                                pCenter = geoman.createPoint(

                                                frameSize/2,frameSize/2, SUBTYPES.GEOM2D);

                        } catch (Exception e) {

                                logger.error("While creating point", e);

                        }

                        clone.draw(aGraphics, new AffineTransform(), pCenter, null, null, null);

                        int realTop    = -1;

                        int realBottom = -1;

                        int realLeft   = -1;

                        int realRight  = -1;

                        // let's see where the highest pixel is in Y-axis

                        boolean done = false;

                        for (int j = 0; !done && j < bi.getHeight(); j++) {

                                for (int i = 0; !done && i < bi.getWidth(); i++) {

                                        if (bi.getRGB(i, j) != 0) {

                                                realTop = j;

                                                done = true;

                                        }

                                }

                        }

                        // let's see where the lowest pixel is in Y-axis

                        done = false;

                        for (int j =  bi.getHeight()-1; !done &&  j >= 0; j--) {

                                for (int i = 0; !done &&  i < bi.getWidth(); i++) {

                                        if (bi.getRGB(i, j) != 0) {

                                                realBottom = i;

                                                done = true;

                                        }

                                }

                        }

                        // let's see where the first pixel at left is in X-axis

                        done = false;

                        for (int i = 0; !done &&  i < bi.getWidth(); i++) {

                                for (int j = 0; !done &&  j < bi.getHeight(); j++) {

                                        if (bi.getRGB(i, j) != 0) {

                                                realLeft = i;

                                                done = true;

                                        }

                                }

                        }

                        // let's see where the first pixel at right is in X-axis

                        done = false;

                        for (int i = bi.getWidth()-1; !done &&  i >=0 ; i--) {

                                for (int j = 0; !done &&  j < bi.getHeight(); j++) {

                                        if (bi.getRGB(i, j) != 0) {

                                                realRight = i;

                                                done = true;

                                        }

                                }

                        }

                        int realWidth = bi.getWidth() - realRight - realLeft;

                        int realHeight = bi.getHeight() - realBottom - realTop;

                        visualCorrection = new VisualCorrection();

                        if (realBottom!=-1 && realTop!=-1 && realLeft!=-1 && realRight != -1) {

                                double correctingSize = Math.max(realHeight, realWidth);

                                visualCorrection.sizeScale = clone.getSize() / correctingSize;

                        }

                        if (realLeft!=-1 && realRight!=-1) {

                                double correctingCenterX = (((realWidth)*0.5)+realLeft);

                                double correctingCenterY = (((realHeight)*0.5)+realBottom);

                                visualCorrection.xOffsetScale = (pCenter.getX() - correctingCenterX) / frameSize;

                                visualCorrection.yOffsetScale = (pCenter.getY() - correctingCenterY) / frameSize;

                        }

                } else {

                        visualCorrection = null;

                }

        }

    @Override

        public Object clone() throws CloneNotSupportedException {

                return LabelClassUtils.clone(this);

        }

        public void getPixExtentPlus(

                        Geometry geom, float[] distances,

                        ViewPort viewPort, int dpi) {

//                float cs = (float) getCartographicSize(viewPort, dpi, geom);

                float cs = (float) this.getEfectiveSize(null);

                distances[0] = cs;

                distances[1] = cs;

        }

        public boolean isOneDotOrPixel(Geometry geom,

                        double[] positionOfDotOrPixel, ViewPort viewPort, int dpi) {

                int type = geom.getType();

                switch (type) {

                case Geometry.TYPES.NULL:

                case Geometry.TYPES.POINT:

                case Geometry.TYPES.MULTIPOINT:

                        return false;

                default:

                        org.gvsig.fmap.geom.primitive.Envelope geomBounds = geom

                        .getEnvelope();

                        double dist1Pixel = viewPort.getDist1pixel();

                        float[] distances = new float[2];

                        this.getPixExtentPlus(geom, distances, viewPort, dpi);

                        boolean onePoint =

                                        (geomBounds.getLength(0) + distances[0] <= dist1Pixel && geomBounds

                                        .getLength(1)

                                        + distances[1] <= dist1Pixel);

                        if (onePoint) {

                                Envelope bounds = geom.getEnvelope();

                                positionOfDotOrPixel[0] = bounds.getMinimum(0);

                                positionOfDotOrPixel[1] = bounds.getMinimum(1);

                        }

                        return onePoint;

                }

        }

        public int getOnePointRgb() {

                return this.getColor().getRGB();

        }

//        @Override

//        public String getDescription() {

//                return desc;

//        }

//

//        public boolean isShapeVisible() {

//                return isShapeVisible;

//        }

//

//        public void setDescription(String d) {

//                desc = d;

//        }

//        public int getSymbolType() {

//                return Geometry.TYPES.POINT;

//        }

        @Override

        public boolean isSuitableFor(Geometry geom) {

                return geom.getType() == Geometry.TYPES.POINT

                                || geom.getType() == Geometry.TYPES.MULTIPOINT;

        }

//        public void drawInsideRectangle(Graphics2D g,

//                        AffineTransform scaleInstance, Rectangle r,

//                        PrintAttributes properties) throws SymbolDrawingException {

//

//

//                org.gvsig.fmap.geom.primitive.Point center = null;

//                try {

//                        center = geoman.createPoint(

//                                        r.getCenterX(), r.getCenterY(), SUBTYPES.GEOM2D);

//                } catch (CreateGeometryException e) {

//                        throw new SymbolDrawingException(TYPES.POINT);

//                }

//

//                if (properties==null) {

//                        draw(g, scaleInstance, center, null, null);

//                } else {

//                        double originalSize = getSize();

//                        double size=originalSize;

//                        int pq = properties.getPrintQuality();

//                        if (pq == PrintAttributes.PRINT_QUALITY_HIGH) {

//                                size *= (double) 600/72;

//                        } else {

//                                if (pq == PrintAttributes.PRINT_QUALITY_NORMAL) {

//                                        size *= (double) 300/72;

//                                } else {

////                                        unitFactor *= 72; (which is the same than doing nothing)

//                                }

//                        }

//                        setSize(size);

//                        print(g, scaleInstance, center, properties);

//                        setSize(originalSize);

//                }

//

//        }

    @Override

        public void print(Graphics2D g, AffineTransform at, Geometry geom,

                        PrintAttributes properties) {

                /* TODO Use this?

                double dpi = 100;

                int pq = properties.getPrintQuality();

                if (pq == PrintAttributes.PRINT_QUALITY_NORMAL){

                        dpi = 300;

                } else if (pq == PrintAttributes.PRINT_QUALITY_HIGH){

                        dpi = 600;

                } else if (pq == PrintAttributes.PRINT_QUALITY_DRAFT){

                        dpi = 72;

                }

                */

                draw(g,at,geom,null,null);

        }

//        public void setUnit(int unitIndex) {

//                unit = unitIndex;

//        }

//

//        public int getUnit() {

//                return unit;

//        }

//

//        public int getReferenceSystem() {

//                return this.referenceSystem;

//        }

//

//        public void setReferenceSystem(int rs) {

//                this.referenceSystem = rs;

//        }

//

//        public double toCartographicSize(ViewPort viewPort, double dpi,

//                        Geometry geom) {

//

//                double oldSize = getSize();

//                setCartographicSize(getCartographicSize(viewPort, dpi, geom), geom);

//                return oldSize;        }

//

//        public void setCartographicSize(double cartographicSize, Geometry geom) {

//                setSize(cartographicSize);

//        }

//

//        public double getCartographicSize(ViewPort viewPort, double dpi,

//                        Geometry geom) {

//

//                return SymbolUtils.getCartographicLength(this,

//                                                                          getSize(),

//                                                                          viewPort,

//                                                                          dpi);

//        }

//        public double getRotation() {

//                return rotation_radians;

//        }

//

//        public void setRotation(double rot) {

//                rotation_radians = rot;

//        }

//

//        public Point2D getOffset() {

//                return offset;

//        }

//

//        public void setOffset(Point2D off) {

//                offset = off;

//        }

//

//        public Color getColor() {

//                return color;

//        }

//

//        public void setColor(Color co) {

//                color = co;

//        }

    @Override

        public void setAlpha(int a) {

                alpha = a;

        }

//        public IMask getMask() {

//                return mask;

//        }

//

//        public void setMask(IMask m) {

//                mask = m;

//        }

        // =========================================

    @Override

        public void loadFromState(PersistentState state) throws PersistenceException {

            super.loadFromState(state);

//                if (state.hasValue("mask")) {

//                        this.mask = (IMask) state.get("mask");

//                } else {

//                        this.mask = null;

//                }

//                // ==============================================

//                this.color = (Color) state.get("color");

//                double aux = state.getDouble("size");

//                this.setSize(aux);

//                this.rotation_radians = state.getDouble("rotation");

//                this.desc = state.getString("desc");

//                this.isShapeVisible = state.getBoolean("isShapeVisible");

//                state.set("unit", getUnit());

//                state.set("referenceSystem", getReferenceSystem());

                if(state.isNull("offset")) {

                    double x = state.getDouble("xOffset");

                    double y = state.getDouble("yOffset");

                    this.setOffset(new Point2D.Double(x, y));

                }

                if (state.hasValue("vc_xOffsetScale")) {

                        this.visualCorrection = new VisualCorrection();

                        this.visualCorrection.xOffsetScale = state.getDouble("vc_xOffsetScale");

                        this.visualCorrection.yOffsetScale = state.getDouble("vc_yOffsetScale");

                        this.visualCorrection.sizeScale = state.getDouble("vc_sizeScale");

                } else {

                        this.visualCorrection = null;

                }

                this.font = (Font) state.get("font");

                this.unicode = state.getInt("unicode");

        }

    @Override

        public void saveToState(PersistentState state) throws PersistenceException {

            super.saveToState(state);

//                if (this.mask != null) {

//                        state.set("mask", this.mask);

//                }

//                state.set("color", this.color);

//                state.set("size", this.size);

//                state.set("rotation", this.rotation_radians);

//                state.set("desc", this.desc);

//                state.set("isShapeVisible", this.isShapeVisible);

//                setUnit(state.getInt("unit"));

//                setReferenceSystem(state.getInt("referenceSystem"));

                //Mantenemos estas dos l?neas por compatibilidad pero deber?an eliminarse

                state.set("xOffset", this.getOffset().getX());

                state.set("yOffset", this.getOffset().getY());

                if (this.visualCorrection != null) {

                        state.set("vc_xOffsetScale", this.visualCorrection.xOffsetScale);

                        state.set("vc_yOffsetScale", this.visualCorrection.yOffsetScale);

                        state.set("vc_sizeScale", this.visualCorrection.sizeScale);

                }

                // ==================================================

                state.set("font", this.font);

                state.set("unicode", this.unicode);

    }

        public static void registerPersistent() {

                PersistenceManager manager = ToolsLocator.getPersistenceManager();

                if( manager.getDefinition(CHARACTER_MARKER_SYMBOL_PERSISTENCE_NAME)==null ) {

                        DynStruct definition = manager.addDefinition(

                                        CharacterMarkerSymbol.class,

                                        CHARACTER_MARKER_SYMBOL_PERSISTENCE_NAME,

                                        CHARACTER_MARKER_SYMBOL_PERSISTENCE_NAME+" Persistence definition",

                                        null,

                                        null

                        );

                        definition.extend(manager.getDefinition(MARKER_SYMBOL_PERSISTENCE_DEFINITION_NAME));

//                        definition.addDynFieldObject("color").setClassOfValue(Color.class).setMandatory(true);

//                        definition.addDynFieldDouble("size").setMandatory(true);

//                        definition.addDynFieldDouble("rotation").setMandatory(true);

//                        definition.addDynFieldInt("unit").setMandatory(true);

//                        definition.addDynFieldInt("referenceSystem").setMandatory(true);

//                        definition.addDynFieldObject("mask").setClassOfValue(IMask.class).setMandatory(false);

//                        definition.addDynFieldString("desc").setMandatory(true);

//                        definition.addDynFieldBoolean("isShapeVisible").setMandatory(true);

                        //Por compatibilidad mantenemos estas dos l?neas aunque deber?an eliminarse

                        definition.addDynFieldDouble("xOffset").setMandatory(true);

                        definition.addDynFieldDouble("yOffset").setMandatory(true);

                        definition.addDynFieldObject("font").setClassOfValue(Font.class).setMandatory(true);

                        definition.addDynFieldInt("unicode").setMandatory(true);

                        // =====================

                        definition.addDynFieldDouble("vc_xOffsetScale").setMandatory(false);

                        definition.addDynFieldDouble("vc_yOffsetScale").setMandatory(false);

                        definition.addDynFieldDouble("vc_sizeScale").setMandatory(false);

                }

        }

        /**

         * Class to be used for the methods that control the visual correction of a character.

         * This visual correction has the responsibility of modify the dimensions of the

         * character to be used as a symbol(in case that the user wants to use more or less

         * pixels to represent it )

         *

         */

        private class VisualCorrection {

                double xOffsetScale = 1;

                double yOffsetScale = 1;

                double sizeScale = 1;

        }

}