Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * 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 3

 * 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., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.symbology.fmap.mapcontext.rendering.symbol.line.impl;

import java.awt.Color;

import java.awt.Graphics2D;

import java.awt.Rectangle;

import java.awt.Shape;

import java.awt.geom.AffineTransform;

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.exception.CreateGeometryException;

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

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

import org.gvsig.fmap.mapcontext.MapContextLocator;

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

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

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

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.line.ILineSymbol;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.line.ISimpleLineSymbol;

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

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

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

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

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.swing.api.TransparencySupport;

import org.gvsig.tools.task.Cancellable;

import org.gvsig.tools.util.Callable;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * SimpleLineSymbol is the most basic symbol for the representation of line

 * objects. Allows to define the width of the line, the color and the drawn

 * pattern.

 *

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

 * @author 2009-     <a href="cordinyana@gvsig.org">C?sar Ordi?ana</a> - gvSIG team

 */

public class SimpleLineSymbol extends AbstractLineSymbol implements ISimpleLineSymbol, TransparencySupport {

    private static final Logger LOG = LoggerFactory.getLogger(SimpleLineSymbol.class);

    public static final String SIMPLE_LINE_SYMBOL_PERSISTENCE_DEFINITION_NAME = "SimpleLineSymbol";

    private static final String SELECTION_SYMBOL = "symbolForSelection";

    SimpleLineSymbol symbolForSelection;

    private static final GeometryManager geomManager = GeometryLocator.getGeometryManager();

    public SimpleLineSymbol() {

        super();

        setLineWidth(1d);

   }

    @Override

    public ISymbol getSymbolForSelection(Color selectionColor) {

        if (symbolForSelection == null) {

            symbolForSelection = (SimpleLineSymbol) cloneForSelection(selectionColor);

        } else {

            symbolForSelection.setColor(selectionColor);

        }

        if (symbolForSelection instanceof TransparencySupport) {

            symbolForSelection.setTransparency(this.getTransparency());

        }

        return symbolForSelection;

    }

    @Override

    public void draw(Graphics2D g, AffineTransform affineTransform, Geometry geom, Feature f, Cancellable cancel, Rectangle r) {

        if(r != null){

            geom = this.getSampleGeometry(r);

        }

        Color c = getColor();

        if (true) {

            // Esto deberia ser para optimizar el pintado de geometrias grandes.

            try {

                Geometry env = geom.getEnvelope().getGeometry();

                env.transform(affineTransform);

                Envelope env2 = env.getEnvelope();

                if (env2.getLength(0) < 1.5 && env2.getLength(1) < 1.5) {

                    g.setColor(c);

                    Point upperCorner = env2.getUpperCorner();

                    int x = (int) upperCorner.getX();

                    int y = (int) upperCorner.getY();

                    g.drawLine(x, y, x, y);

                    return;

                }

            } catch (Exception ex) {

                LOG.warn("Error optimizing the drawing of the geometry. Continues with normal drawing.", ex);

                // Do nothing, continue with the draw of the original geometry

            }

        }

        Geometry geomToDraw;

//        try {

//            geomToDraw = geom.toLines();

//        } catch (Exception ex) {

            geomToDraw = geom.cloneGeometry();

//        }

        ILineStyle lineStyle = getLineStyle();

        if (lineStyle != null) {

            if(lineStyle instanceof CartographicSupport && r == null){

                ((CartographicSupport)lineStyle).setCartographicContext(this.getCartographicContext());

                g.setStroke(lineStyle.getCartographicStroke());

            } else {

                g.setStroke(lineStyle.getStroke());

            }

        }

        if(geomToDraw != null) {

            g.setColor(c);

            Shape shape = null;

            geomToDraw.transform(affineTransform);

            double offset = lineStyle.getOffset();

            if(offset != 0) {

                geomToDraw = applyOffset(this.toCartographicUnits(offset), r, geomToDraw);

            }

            if(geomToDraw != null){

                shape = geomToDraw.getShape();

                if (shape != null) {

                    g.draw(shape);

                    if (lineStyle != null) {

                        ArrowDecoratorStyle arrowDecorator = (ArrowDecoratorStyle) lineStyle.getArrowDecorator();

                        if (arrowDecorator != null) {

                            try {

                                if(arrowDecorator instanceof CartographicSupport){

                                    if(r == null){

                                        arrowDecorator.setCartographicContext(this.getCartographicContext());

                                    } else {

                                        arrowDecorator.setCartographicContext((CartographicContext)null);

                                    }

                                }

                                //Pasamos una transformaci?n afin identidad porque la geometr?a ya viene transformada

                                arrowDecorator.draw(g, null, geomToDraw, f);

                            } catch (CreateGeometryException e) {

                                LOG.warn("Error drawing geometry.", e);

                            }

                        }

                    }

                }

            }

        }

    }

    private Geometry applyOffset(double offset, Rectangle r, Geometry geomToDraw) {

        if (offset != 0) {

            try {

                geomToDraw = geomToDraw.offset(offset);

//                if(offset > 0 ){

//                    geomToDraw.flip();

//                }

            } catch (Exception e) {

                LOG.warn("Error creating a polygon with an offset", e);

            }

        }

        return geomToDraw;

    }

//    @Override

//    public int getOnePointRgb() {

//        return getColor().getRGB();

//    }

    @Override

    public void setLineWidth(double width) {

        getLineStyle().setLineWidth((float) width);

    }

    @Override

    public double getLineWidth() {

        return getLineStyle().getLineWidth();

    }

    @Override

    public Object clone() throws CloneNotSupportedException {

        SimpleLineSymbol copy = (SimpleLineSymbol) super.clone();

        if (symbolForSelection != null) {

            copy.symbolForSelection = (SimpleLineSymbol) symbolForSelection

                    .clone();

        }

        return copy;

    }

    @Override

    public void loadFromState(PersistentState state) throws PersistenceException {

        // Set parent style properties

        super.loadFromState(state);

        this.symbolForSelection = (SimpleLineSymbol) state.get(SELECTION_SYMBOL);

    }

    @Override

    public void saveToState(PersistentState state) throws PersistenceException {

        // Save parent fill symbol properties

        super.saveToState(state);

        // Save own properties

        if (this.symbolForSelection != null) {

            state.set(SELECTION_SYMBOL, this.getSymbolForSelection());

        }

    }

    @Override

    public void setTransparency(double transparency) {

        Color c = getColor();

        c = new Color(c.getRed(), c.getGreen(), c.getBlue(), (int) (transparency * 255));

        setColor(c);

        ISymbol selectionSymbol = this.symbolForSelection;

        if (selectionSymbol != null && selectionSymbol instanceof TransparencySupport) {

            ((TransparencySupport) selectionSymbol).setTransparency(transparency);

        }

    }

    @Override

    public double getTransparency() {

        return ((double)this.getAlpha())/255;

    }

    public static class RegisterPersistence implements Callable {

        @Override

        public Object call() throws Exception {

            PersistenceManager manager = ToolsLocator.getPersistenceManager();

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

                DynStruct definition = manager.addDefinition(

                        SimpleLineSymbol.class,

                        SIMPLE_LINE_SYMBOL_PERSISTENCE_DEFINITION_NAME,

                        SIMPLE_LINE_SYMBOL_PERSISTENCE_DEFINITION_NAME + " Persistence definition",

                        null,

                        null

                );

                // Extend the LineSymbol base definition

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

                definition.addDynFieldObject(SELECTION_SYMBOL).setClassOfValue(SimpleLineSymbol.class);

            }

            return Boolean.TRUE;

        }

    }

    public static class RegisterSymbol implements Callable {

        @Override

        public Object call() throws Exception {

            int[] shapeTypes;

            SymbolManager manager = MapContextLocator.getSymbolManager();

            shapeTypes = new int[]{Geometry.TYPES.CURVE, Geometry.TYPES.ARC,

                Geometry.TYPES.MULTICURVE, Geometry.TYPES.CIRCUMFERENCE,

                Geometry.TYPES.PERIELLIPSE, Geometry.TYPES.SPLINE,

                Geometry.TYPES.LINE, Geometry.TYPES.MULTILINE};

            manager.registerSymbol(ILineSymbol.SYMBOL_NAME,

                    shapeTypes,

                    SimpleLineSymbol.class);

            return Boolean.TRUE;

        }

    }

    @Override

    public void setCartographicContext(CartographicContext ctx) {

        super.setCartographicContext(ctx);

        if (getLineStyle() != null

                && getLineStyle().getArrowDecorator() != null) {

            IArrowDecoratorStyle arrowDecorator = getLineStyle().getArrowDecorator();

            if(arrowDecorator != null){

                if (arrowDecorator instanceof CartographicSupport) {

                    ((CartographicSupport) arrowDecorator).setCartographicContext(ctx);

                }

                if (arrowDecorator.getMarker() != null) {

                    IMarkerSymbol marker = arrowDecorator.getMarker();

                    if (marker instanceof CartographicSupport) {

                        ((CartographicSupport) marker).setCartographicContext(ctx);

                    }

                }

            }

        }

    }

}