Application: gvSIG desktop » gvSIG 3D

package org.gvsig.gvsig3d.impl.symbology3d.marker.impl;

import java.awt.BasicStroke;

import java.awt.Graphics2D;

import java.awt.Rectangle;

import java.awt.geom.AffineTransform;

import org.gvsig.compat.print.PrintAttributes;

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

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

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

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

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

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

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

import org.gvsig.fmap.mapcontext.MapContextLocator;

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

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

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

import org.gvsig.gvsig3d.symbology3d.marker.ISimpleMarker3DSymbol;

import org.gvsig.osgvp.exceptions.node.NodeException;

import org.gvsig.osgvp.geometry.GeoPointArray3D;

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.locator.LocatorException;

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.gvsig.tools.util.Callable;

/**

 * 

 * @author jtorres

 *

 */

public class SimpleMarker3DSymbol extends AbstractMarkerSymbol implements ISimpleMarker3DSymbol{

        public static final String SIMPLE_MARKER3D_SYMBOL_PERSISTENCE_DEFINITION_NAME = "SimpleMarker3DSymbol";

        public GeoPointArray3D _geometry;

        protected float _height = 100;

        public SimpleMarker3DSymbol() 

        {

                super();

                try {

                        _geometry = new GeoPointArray3D();

                } catch (NodeException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

        public ISymbol getSymbolForSelection() {

                //TODO 

                return null;

        }

        public GeoPointArray3D getGeometry(){

                return _geometry;

        }

        public void previewSymbol(Graphics2D g, AffineTransform affineTransform,

                        Geometry geom, Feature f, Cancellable cancel) {

                int x, y;

                org.gvsig.fmap.geom.primitive.Point p = (org.gvsig.fmap.geom.primitive.Point)geom.cloneGeometry();

                p.transform(affineTransform);

                int size = (int) getSize();

                int halfSize = size/2;

                x = ((int) (p.getX() + getOffset().getX()) - halfSize);

                y = ((int) (p.getY() + getOffset().getY()) - halfSize);

                g.setColor(getColor());

                g.setStroke(new BasicStroke(1));

                // IMask mask = getMask();

                // if (mask != null) {

                // IFillSymbol maskShape = mask.getFillSymbol();

                // // maskShape.draw(g, null, mask.getHaloShape(shp));

                // }

                g.fillOval(x, y, size, size);

        }

        public void drawInsideRectangle(Graphics2D g, AffineTransform scaleInstance, Rectangle r, PrintAttributes properties) throws SymbolDrawingException {

                //SimpleMarkerSymbol smarker = new SimpleMarkerSymbol();

                try {

                        if (properties==null)

                        this.previewSymbol(g, scaleInstance, GeometryLocator.getGeometryManager().createPoint(r.getCenterX(), r.getCenterY(), SUBTYPES.GEOM2D), null, null);

                        else

                        {

                                double originalSize = getSize();

                                double size=originalSize;

                                int pq = properties.getPrintQuality();

                                if (pq == PrintAttributes.PRINT_QUALITY_NORMAL){

                                        size *= (double) 300/72;

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

                                        size *= (double) 600/72;

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

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

                                }

                                setSize(size);

                                print(g, scaleInstance,GeometryLocator.getGeometryManager().createPoint(r.getCenterX(), r.getCenterY(), SUBTYPES.GEOM2D), properties);

                                setSize(originalSize);

                        }

                } catch (CreateGeometryException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                } catch (LocatorException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

                public void loadFromState(PersistentState state)

        throws PersistenceException {

                // Set parent fill symbol properties

                super.loadFromState(state);

                // Set own properties sample

                //thi.setStyle(state.getInt(FIELD_MARKERSTYLE));

        }

        public void saveToState(PersistentState state) throws PersistenceException {

                // Save parent fill symbol properties

                super.saveToState(state);

        // Save own properties

        //        state.set(FIELD_MARKERSTYLE, getStyle());

        }

        public static class RegisterPersistence implements Callable {

                public Object call() throws Exception {

                        PersistenceManager manager = ToolsLocator.getPersistenceManager();

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

                                DynStruct definition = manager.addDefinition(

                                                SimpleMarker3DSymbol.class,

                                                SIMPLE_MARKER3D_SYMBOL_PERSISTENCE_DEFINITION_NAME,

                                                SIMPLE_MARKER3D_SYMBOL_PERSISTENCE_DEFINITION_NAME+" Persistence definition",

                                                null, 

                                                null

                                );

                                // Extend the MarkerSymbol base definition

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

                                //Add the fields to save                                

                                // Marker style

                                //definition.addDynFieldInt(FIELD_MARKERSTYLE).setMandatory(true);

                        }

                        return Boolean.TRUE;

                }

        }

        public static class RegisterSymbol implements Callable {

                public Object call() throws Exception {

                        int[] shapeTypes;

                        SymbolManager manager = MapContextLocator.getSymbolManager();

                        shapeTypes =

                                new int[] { Geometry.TYPES.POINT, Geometry.TYPES.MULTIPOINT};

                        manager.registerSymbol(ISimpleMarker3DSymbol.SYMBOL_NAME,

                                        shapeTypes,

                                        SimpleMarker3DSymbol.class,"project.document.view3d");

                        return Boolean.TRUE;

                }

        }

        public void draw(Graphics2D g, AffineTransform affineTransform,

                        Geometry geom, Feature f, Cancellable cancel) {

                        org.gvsig.fmap.geom.primitive.Point point = (org.gvsig.fmap.geom.primitive.Point)geom.cloneGeometry();

                        //TODO: Transform coordinates for spherical projection

                        if (point.getGeometryType().isSubTypeOf(Geometry.SUBTYPES.GEOM3D))

                                _geometry.addPoint(point.getX(),point.getY(),point.getCoordinateAt(2));

                        else

                                _geometry.addPoint(point.getCoordinateAt(0),point.getCoordinateAt(1),_height);

//                        int x, y;

//                        org.gvsig.fmap.geom.primitive.Point p = (org.gvsig.fmap.geom.primitive.Point)geom.cloneGeometry();

//                        p.transform(affineTransform);

//                        

//                        int size = (int) getSize();

//

//                        int halfSize = size/2;

//                        x = ((int) (p.getX() + getOffset().getX()) - halfSize);

//                        y = ((int) (p.getY() + getOffset().getY()) - halfSize);

//

//                        g.setColor(getColor());

//                        g.setStroke(new BasicStroke(1));

//                        // IMask mask = getMask();

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

//                        // IFillSymbol maskShape = mask.getFillSymbol();

//                        // // maskShape.draw(g, null, mask.getHaloShape(shp));

//                        // }

//                        g.fillOval(x, y, size, size); 

        }

        public void setFixedHeight(float height) {

                _height = height;                

        }

}