Application: gvSIG desktop » gvSIG 3D

package org.gvsig.view3d.vector.lib.impl.vector;

import java.awt.Color;

import java.util.ArrayList;

import java.util.List;

import org.cresques.cts.ICoordTrans;

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

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

import org.gvsig.fmap.geom.Geometry;

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

import org.gvsig.fmap.geom.aggregate.MultiPolygon;

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

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

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

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.rendering.legend.IVectorLegend;

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

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.fill.IFillSymbol;

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

import org.gvsig.view3d.vector.lib.api.VectorElevationMode;

import org.gvsig.view3d.vector.lib.api.VectorLoaderParameters;

import org.gvsig.view3d.vector.lib.impl.AbstractRenderableLayer;

import gov.nasa.worldwind.geom.Position;

import gov.nasa.worldwind.render.BasicShapeAttributes;

import gov.nasa.worldwind.render.Material;

import gov.nasa.worldwind.render.Renderable;

public class DefaultVectorPolygonRenderableLayer extends AbstractRenderableLayer {

    private BasicShapeAttributes normalShapeAttributes;

    public DefaultVectorPolygonRenderableLayer(VectorLoaderParameters parameters, FeatureStore featureStore,

        IVectorLegend legend) {

        super(parameters, featureStore);

        this.normalShapeAttributes = getNormalBasicShapeAttributes(legend);

    }

    @Override

    protected List<Renderable> getRenderables(Feature feature) {

        Geometry geom = feature.getDefaultGeometry();

        List<Renderable> renderables = new ArrayList<>();

        if (geom.getGeometryType().isTypeOf(TYPES.MULTIPOLYGON)) {

            MultiPolygon multiPolygon = (MultiPolygon) geom;

            for (int i = 0; i < multiPolygon.getPrimitivesNumber(); i++) {

                Polygon polygon = (Polygon) multiPolygon.getSurfaceAt(i);

                if (polygon.getNumVertices() >= 4) {

                    renderables.add(getRenderable(polygon));

                }

            }

        } else {

            Polygon polygon = (Polygon) geom;

            if (polygon.getNumVertices() >= 4) {

                renderables.add(getRenderable(polygon));

            }

        }

        return renderables;

    }

    private gov.nasa.worldwind.render.Polygon getRenderable(Polygon polygon) {

        gov.nasa.worldwind.render.Polygon renderable = new gov.nasa.worldwind.render.Polygon();

        boolean hasZ = polygon.getDimension() > 2;

        double h = 0.0;

        if (VectorElevationMode.CLAMP_TO_GROUND.equals(parameters.getVectorElevationMode())) {

            hasZ = false;

        } else {

            if (VectorElevationMode.RELATIVE_TO_GROUND.equals(parameters.getVectorElevationMode())) {

                hasZ = false;

            }

            if (parameters.getConstantHeight() != null) {

                h = parameters.getConstantHeight();

            }

        }

        ICoordTrans coordTrans = getCoordTrans(this.parameters.getLayer().getProjection());

        List<Position> verticesList = getVertices(polygon, h, hasZ, coordTrans);

        renderable.setOuterBoundary(verticesList);

        int numInteriorRings = polygon.getNumInteriorRings();

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

            renderable.addInnerBoundary(getInteriorRing(polygon.getInteriorRing(i), hasZ, h));

        }

        renderable.setAltitudeMode(getAltitudeMode(this.parameters.getVectorElevationMode()));

        renderable.setAttributes(normalShapeAttributes);

        return renderable;

    }

    private List<Position> getInteriorRing(Ring interiorRing, boolean hasZ, double h) {

        int internalNumVertices = interiorRing.getNumVertices();

        List<Position> internalVerticesList = new ArrayList<>(internalNumVertices);

        for (int j = 0; j < internalNumVertices; j++) {

            Point vertex = interiorRing.getVertex(j);

            if (hasZ) {

                double z = vertex.getCoordinateAt(2);

                internalVerticesList.add(Position.fromDegrees(vertex.getY(), vertex.getX(), z + h));

            } else {

                internalVerticesList.add(Position.fromDegrees(vertex.getY(), vertex.getX(), h));

            }

        }

        return internalVerticesList;

    }

    private BasicShapeAttributes getNormalBasicShapeAttributes(IVectorLegend legend) {

        Material fillMaterial = Material.LIGHT_GRAY;

        Material outlineMaterial = Material.BLACK;

        double fillOpacity = 0.7;

        double outlineWidth = 2;

        ISymbol symbol = legend.getDefaultSymbol();

        if (symbol instanceof IFillSymbol) {

            IFillSymbol fillSymbol = (IFillSymbol) symbol;

            Color fillColor = fillSymbol.getFillColor();

            int fillAlpha = fillSymbol.getFillAlpha();

            ILineSymbol outline = fillSymbol.getOutline();

            Color lineColor = outline.getColor();

            fillMaterial = new Material(fillColor);

            outlineMaterial = new Material(lineColor);

            fillOpacity = fillAlpha / 255.0;

            outlineWidth = outline.getLineWidth();

        } else {

            outlineMaterial = new Material(this.parameters.getDefaultColor());

            fillMaterial = new Material(this.parameters.getDefaultColor());

            fillOpacity = 0.7;

            outlineWidth = 2;

        }

        BasicShapeAttributes attr = new BasicShapeAttributes();

        attr.setInteriorMaterial(fillMaterial);

        attr.setInteriorOpacity(fillOpacity);

        attr.setOutlineMaterial(outlineMaterial);

        attr.setOutlineWidth(outlineWidth);

        return attr;

    }

}