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.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.dal.feature.FeatureType;

import org.gvsig.fmap.geom.Geometry;

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

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

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

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

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

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

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

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

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

import gov.nasa.worldwind.geom.Position;

import gov.nasa.worldwind.render.Material;

import gov.nasa.worldwind.render.PointPlacemark;

import gov.nasa.worldwind.render.PointPlacemarkAttributes;

import gov.nasa.worldwind.render.Renderable;

public class DefaultVectorExtrusionPointRenderableLayer extends AbstractRenderableLayer {

    private PointPlacemarkAttributes markAttributes;

    public DefaultVectorExtrusionPointRenderableLayer(VectorExtrusionLoaderParameters parameters, FeatureStore featureStore,

        IVectorLegend legend) {

        super(parameters, featureStore);

        this.markAttributes = getBasicMarkerAttributes(legend);

    }

    @Override

    protected List<Renderable> getRenderable(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.MULTIPOINT)) {

            MultiPoint multiPoint = (MultiPoint) geom;

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

                renderables.add(getRenderable(multiPoint.getPointAt(i), heightFieldValue));

            }

        } else {

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

        }

        return renderables;

    }

    private PointPlacemark getRenderable(Point point, double heightFieldValue) {

        boolean hasZ = point.getDimension() > 2;

        double h = 0.0;

        double y = point.getY();

        double x = point.getX();

        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();

        PointPlacemark marker;

        if (hasZ) {

            double z = point.getCoordinateAt(2);

            marker = new PointPlacemark(Position.fromDegrees(y, x, z + h));

        } else {

            Position position = Position.fromDegrees(y, x, h);

            marker = new PointPlacemark(position);

        }

        marker.setAltitudeMode(getAltitudeMode(parameters.getVectorElevationMode()));

        marker.setLineEnabled(true);

        marker.setAttributes(this.markAttributes);

        return marker;

    }

    @Override

    protected FeatureQuery getFeatureQuery() throws DataException {

        FeatureType featureType = featureStore.getDefaultFeatureType();

        String geomAtt = featureType.getDefaultGeometryAttributeName();

        FeatureQuery featureQuery = featureStore.createFeatureQuery();

        featureQuery.addAttributeName(geomAtt);

        // Add field height to feature query

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

        if (StringUtils.isNotBlank(fieldHeight)) {

            featureQuery.addAttributeName(fieldHeight);

        }

        return featureQuery;

    }

    private PointPlacemarkAttributes getBasicMarkerAttributes(IVectorLegend legend) {

        if (this.markAttributes != null) {

            return this.markAttributes;

        }

        double markerSize = 0;

        Material fillMaterial = null;

        if (legend == null) {

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

            markerSize = 0.3;

        } else {

            ISymbol symbol = legend.getDefaultSymbol();

            if (symbol instanceof ISimpleMarkerSymbol) {

                ISimpleMarkerSymbol markerSymbol = (ISimpleMarkerSymbol) symbol;

                Color fillColor = markerSymbol.getColor();

                markerSize = markerSymbol.getSize();

                fillMaterial = new Material(fillColor);

            } else {

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

                markerSize = 4;

            }

        }

        markAttributes = new PointPlacemarkAttributes();

        Color color = fillMaterial.getDiffuse();

        markAttributes.setImageColor(color);

        markAttributes.setLineMaterial(fillMaterial);

        markAttributes.setLineWidth(1d);

        markAttributes.setUsePointAsDefaultImage(true);

        markAttributes.setScale(markerSize);

        return markAttributes;

    }

}