Application: gvSIG desktop » gvSIG 3D

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

import java.awt.Color;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.cresques.cts.ICoordTrans;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.dal.exception.DataException;

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

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

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.MultiLine;

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

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

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

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.line.ILineSymbol;

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

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

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.Path;

import gov.nasa.worldwind.render.Renderable;

public class DefaultVectorExtrusionLineRenderableLayer extends AbstractRenderableLayer {

    private BasicShapeAttributes normalShapeAttributes;

    public DefaultVectorExtrusionLineRenderableLayer(VectorLoaderParameters parameters, FeatureStore featureStore,

        IVectorLegend legend) {

        super(parameters, featureStore);

        this.normalShapeAttributes = getNormalShapAttributes(legend);

    }

    @Override

    protected List<Renderable> getRenderables(Feature feature) {

        double heightFieldValue = 0;

        String heightField = ((VectorExtrusionLoaderParameters) parameters).getHeightField();

        if (StringUtils.isNoneBlank(heightField)) {

            heightFieldValue = feature.getDouble(heightField);

        }

        Geometry geom = feature.getDefaultGeometry();

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

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

            MultiLine multiLine = (MultiLine) geom;

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

                if (multiLine.getCurveAt(i).getNumVertices() >= 2) {

                    renderables.add(getRenderable(multiLine.getCurveAt(i), heightFieldValue));

                }

            }

        } else {

            if (((Line) geom).getNumVertices() >= 2) {

                renderables.add(getRenderable((Line) geom, heightFieldValue));

            }

        }

        return renderables;

    }

    private Renderable getRenderable(Curve curve, double heightFieldValue) {

        boolean hasZ = curve.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();

            }

        }

        h += heightFieldValue;

        h *= ((VectorExtrusionLoaderParameters) parameters).getVerticalExaggeration();

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

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

        Path path = new Path(verticesList);

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

        path.setAttributes(normalShapeAttributes);

        path.setExtrude(true);

        return path;

    }

    private BasicShapeAttributes getNormalShapAttributes(IVectorLegend legend) {

        Material outlineMaterial;

        double outlineOpacity = 1.0;

        double outlineWidth = 3.0;

        ISymbol symbol = legend.getDefaultSymbol();

        if (symbol instanceof ILineSymbol) {

            ILineSymbol lineSymbol = (ILineSymbol) symbol;

            Color lineColor = lineSymbol.getColor();

            int lineAlpha = lineSymbol.getAlpha();

            outlineMaterial = new Material(lineColor);

            outlineOpacity = lineAlpha / 255.0;

            outlineWidth = lineSymbol.getLineWidth();

        } else {

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

            outlineOpacity = 1.0;

            outlineWidth = 3.0;

        }

        normalShapeAttributes = new BasicShapeAttributes();

        normalShapeAttributes.setOutlineMaterial(outlineMaterial);

        normalShapeAttributes.setOutlineWidth(outlineWidth);

        normalShapeAttributes.setOutlineOpacity(outlineOpacity);

        return normalShapeAttributes;

    }

    @Override

    protected FeatureQuery getFeatureQuery() throws DataException {

        FeatureQuery featureQuery = super.getFeatureQuery();

        // Add field height to feature query

        String fieldHeight = ((VectorExtrusionLoaderParameters) parameters).getHeightField();

        if (StringUtils.isNotBlank(fieldHeight)) {

            featureQuery.addAttributeName(fieldHeight);

        }

        return featureQuery;

    }

}