Application: gvSIG desktop » gvSIG legends: base legends

package org.gvsig.legend.urbanverticalsignage.lib.impl;

import java.awt.Graphics2D;

import java.awt.Rectangle;

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.awt.image.BufferedImage;

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

import org.gvsig.fmap.geom.Geometry;

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

import org.gvsig.fmap.mapcontext.MapContext;

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

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.impl.CartographicSupportToolkit;

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

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

import org.gvsig.tools.task.Cancellable;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class ImageMarkerSymbol extends AbstractMarkerSymbol implements  IMarkerSymbol {

    private ImageMarkerSymbol selectionSym;

    private boolean selected;

    private BufferedImage image;

    private BufferedImage imagesel;

    void setImage(BufferedImage img) {

        this.image = img;

    }

    void setImageSel(BufferedImage img) {

        this.imagesel = img;

    }

    @Override

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

        Point p;

        try {

            p = geom.centroid();

        } catch (Exception ex) {

            return;

        }

        if (affineTransform != null) {

            p.transform(affineTransform);

        }

        double x, y;

        int size = (int) Math.round(getEfectiveSize(f));

        double halfSize = getEfectiveSize(f) / 2;

        x = p.getX() - halfSize;

        y = p.getY() - halfSize;

        Point2D theOffset = this.getEfectiveOffset(f);

        int xOffset = (int) theOffset.getX();

        int yOffset = (int) theOffset.getY();

        if (size > 0) {

            if( isDrawLineToOffset() ) {

                g.setColor(this.getEfectiveLineToOffsetColor(f));

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

            }

            Rectangle rect = new Rectangle(size, size);

            g.translate(x + xOffset, y + yOffset);

            double auxRotation = getEfectiveRotationInRadians(f);

            g.rotate(auxRotation, halfSize, halfSize);

            this.drawInsideRectangle(g, rect, true);

            g.rotate(-auxRotation, halfSize, halfSize);

            g.translate(-(x + xOffset), -(y + yOffset));

        }

    }

    private void drawInsideRectangle(Graphics2D g, Rectangle r, boolean keepAspectRatio) {

        BufferedImage theImage = (!selected) ? this.image : this.imagesel;

        if (theImage != null) {

            double xOffset = 0;

            double yOffset = 0;

            @SuppressWarnings("UnusedAssignment")

            double xScale = 1;

            @SuppressWarnings("UnusedAssignment")

            double yScale = 1;

            if (keepAspectRatio) {

                double scale;

                if (theImage.getWidth() > theImage.getHeight()) {

                    scale = r.getWidth() / theImage.getWidth();

                    yOffset = 0.5 * (r.getHeight() - theImage.getHeight() * scale);

                } else {

                    scale = r.getHeight() / theImage.getHeight();

                    xOffset = 0.5 * (r.getWidth() - theImage.getWidth() * scale);

                }

                xScale = yScale = scale;

            } else {

                xScale = r.getWidth() / theImage.getWidth();

                yScale = r.getHeight() / theImage.getHeight();

                yOffset = theImage.getHeight() * 0.5 * yScale;

            }

            AffineTransform at = AffineTransform.getTranslateInstance(xOffset, yOffset);

            at.concatenate(AffineTransform.getScaleInstance(xScale, yScale));

            g.drawRenderedImage(theImage, at);

        }

    }

    @Override

    public ISymbol getSymbolForSelection() {

        if (selectionSym == null) {

            try {

                selectionSym = (ImageMarkerSymbol) this.clone();

            } catch (CloneNotSupportedException e) {

//                LOG.warn("Error creating the selection symbol for the symbol "+ this, e);

            }

            selectionSym.selected = true;

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

        } else {

            selectionSym.setColor(MapContext.getSelectionColor());

        }

        return selectionSym;

    }

}