Application: gvSIG desktop » gvSIG 3D

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright © 2007-2016 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

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

import java.awt.Color;

import java.util.ArrayList;

import org.cresques.cts.ICoordTrans;

import org.cresques.cts.IProjection;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.dal.DataStore;

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

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

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

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

import org.gvsig.fmap.geom.Geometry;

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

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

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

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

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

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

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

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.view3d.lib.api.E3DElevationModes;

import org.gvsig.view3d.lib.api.View3DLocator;

import org.gvsig.view3d.lib.api.View3DManager;

import org.gvsig.view3d.lib.api.layers.NwwVectorLayer;

import org.gvsig.view3d.lib.api.properties.E3DLayerLoadingModes;

import org.gvsig.view3d.lib.api.properties.ExtrudedVectorLayerProperties3D;

import org.gvsig.view3d.lib.api.properties.LayerProperties3D;

import org.gvsig.view3d.lib.api.properties.VectorLayerProperties3D;

import gov.nasa.worldwind.avlist.AVKey;

import gov.nasa.worldwind.avlist.AVList;

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;

import gov.nasa.worldwind.util.WWUtil;

/**

 * A default vector points layer created from a gvisg layer.

 * 

 * @author Andrea Antonello andrea.antonello@gmail.com

 */

public class DefaultPointPMLayer extends NwwVectorLayer {

    private static final Logger LOG = LoggerFactory.getLogger(DefaultPointPMLayer.class);

    private FLyrVect vectorLayer;

    private PointPlacemarkAttributes basicMarkerAttributes;

    private LayerProperties3D mLayerProperties;

    private String heightFieldName;

    private double verticalExageration = 1.0;

    private double constantHeight = 1.0;

    private boolean hasConstantHeight = false;

    private boolean applyExtrusion = false;

    private Color defaultColor = Color.RED;

    private Material fillMaterial = Material.LIGHT_GRAY;

    private double markerSize = 1d;

    private E3DElevationModes elevationMode = E3DElevationModes.CLAMP_TO_GROUND;

    private ICoordTrans data2NwwCT;

    public DefaultPointPMLayer( AVList params ) {

        params = params != null ? params : (AVList) this.getValue(AVKey.CONSTRUCTION_PARAMETERS);

        if (params == null) {

            LOG.error("Can not load data, no parameters supplied");

            throw new IllegalArgumentException();

        }

        if (this.getValue(AVKey.CONSTRUCTION_PARAMETERS) == null) {

            this.setValue(AVKey.CONSTRUCTION_PARAMETERS, params.copy());

            params = (AVList) this.getValue(AVKey.CONSTRUCTION_PARAMETERS);

        }

        // if (params.getValue(AVKey.FILE_STORE) == null) {

        // setDataFileStore(WorldWind.getDataFileStore());

        // }

        String datasetName = params.getStringValue(AVKey.DATASET_NAME);

        if (WWUtil.isEmpty(datasetName)) {

            LOG.error("Can not load vector data, missing required parameter: {}", AVKey.DATASET_NAME);

            throw new IllegalArgumentException();

        }

        // TODO this should be checked in the environment that creates this object: to be removed

        // after review

        //

        // mapControl3D = (MapControl3D) params.getValue(DefaultMapControl3D.GVSIG_MAPCONTROL3D);

        //

        // if (mapControl3D == null) {

        // LOG.error("Can not load vector data, missing required parameter: {}",

        // DefaultMapControl3D.GVSIG_MAPCONTROL3D);

        // throw new IllegalArgumentException();

        // }

        FLayer layer = (FLayer) params.getValue(E3DLayerLoadingModes.GVSIG_LAYER);

        if (layer instanceof FLyrVect) {

            vectorLayer = (FLyrVect) layer;

            IProjection sourcePrj = vectorLayer.getProjection();

            if (!sourcePrj.equals(getNwwPrj())) {

                data2NwwCT = sourcePrj.getCT(getNwwPrj());

            }

        } else {

            LOG.error("Can not load vector data, missing layer or invalid layer type.");

            throw new IllegalArgumentException();

        }

        View3DManager manager = View3DLocator.getManager();

        mLayerProperties = manager.getLayerProperties(layer, null);

        if (mLayerProperties instanceof VectorLayerProperties3D) {

            VectorLayerProperties3D properties = (VectorLayerProperties3D) mLayerProperties;

            defaultColor = properties.getDefaultColor();

            elevationMode = E3DElevationModes.getModeFromValue(properties.getElevationMode());

            Double constantExtrusionHeight = properties.getConstantHeight();

            if (constantExtrusionHeight != null) {

                // apply constant height

                hasConstantHeight = true;

                constantHeight = constantExtrusionHeight;

            }

        }

        if (mLayerProperties instanceof ExtrudedVectorLayerProperties3D) {

            ExtrudedVectorLayerProperties3D properties = (ExtrudedVectorLayerProperties3D) mLayerProperties;

            heightFieldName = properties.getExtrusionHeightField();

            applyExtrusion = true;

            if (heightFieldName != null) {

                verticalExageration = properties.getVerticalExaggeration();

            }

        }

        updateLegend();

        try {

            Thread t = new WorkerThread();

            t.start();

        } catch (Exception e) {

            LOG.error("Error loading the data...", e);

        }

    }

    public class WorkerThread extends Thread {

        public void run() {

            addPointsToRender();

        }

    }

    private void addPointsToRender() {

        try {

            ArrayList<Renderable> renderablesList = new ArrayList<>();

            FeatureStore pointsStore = vectorLayer.getFeatureStore();

            FeatureSet pointsSet = null;

            DisposableIterator pointsIterator = null;

            try {

                pointsSet = pointsStore.getFeatureSet();

                pointsIterator = pointsSet.fastIterator();

                while( pointsIterator.hasNext() ) {

                    Feature pointFeature;

                    synchronized (pointsStore) {

                        pointFeature = (Feature) pointsIterator.next();

                    }

                    makeShape(renderablesList, pointFeature);

                }

            } finally {

                DisposeUtils.dispose(pointsIterator);

                DisposeUtils.dispose(pointsSet);

            }

            setRenderables(renderablesList);

        } catch (DataException e) {

            LOG.error("Can not load vector data, missing layer or invalid layer type.");

            e.printStackTrace();

        }

    }

    private void makeShape( ArrayList<Renderable> renderablesList, Feature pointFeature ) {

        try {

            Geometry geometry = pointFeature.getDefaultGeometry().cloneGeometry();

            geometry.reProject(data2NwwCT);

            if (geometry instanceof Point) {

                Point point = (Point) geometry;

                addPointPlaceMark(renderablesList, pointFeature, point);

            } else if (geometry instanceof MultiPoint) {

                MultiPoint multiPoint = (MultiPoint) geometry;

                int pointsCount = multiPoint.getPrimitivesNumber();

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

                    Point point = multiPoint.getPointAt(i);

                    addPointPlaceMark(renderablesList, pointFeature, point);

                }

            }

        } catch (Exception e) {

            e.printStackTrace();

        }

    }

    private void addPointPlaceMark( ArrayList<Renderable> renderablesList, Feature pointFeature, Point point ) {

        boolean hasZ = point.getDimension() > 2;

        double h = 0.0;

        double y = point.getY();

        double x = point.getX();

        switch( elevationMode ) {

        case CLAMP_TO_GROUND:

            hasZ = false;

            break;

        case RELATIVE_TO_GROUND:

            /*

             * we ignore the 3rd coordinate in that case and continue.

             * The altitude mode will the handle the passed values

             * properly as absolute or relative. 

             */

            hasZ = false;

        case ABSOLUTE:

        default:

            if (hasConstantHeight) {

                h = constantHeight;

            }

            if (heightFieldName != null) {

                double tmpH = pointFeature.getDouble(heightFieldName);

                h += tmpH;

            }

            break;

        }

        h *= verticalExageration;

        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(elevationMode.getValue());

        marker.setLineEnabled(applyExtrusion);

        marker.setAttributes(basicMarkerAttributes);

        renderablesList.add(marker);

    }

    @Override

    public void updateLegend( ) {

        ILegend legend = vectorLayer.getLegend();

        ISymbol symbol = legend.getDefaultSymbol();

        if (symbol instanceof ISimpleMarkerSymbol) {

            ISimpleMarkerSymbol markerSymbol = (ISimpleMarkerSymbol) symbol;

            Color fillColor = markerSymbol.getColor();

            markerSize = markerSymbol.getSize();

            fillMaterial = new Material(fillColor);

            // fillOpacity = fillAlpha / 255.0;

        } else {

            fillMaterial = new Material(defaultColor);

            markerSize = 0.3;

        }

        if (basicMarkerAttributes == null) {

            basicMarkerAttributes = new PointPlacemarkAttributes();

        }

        Color color = fillMaterial.getDiffuse();

        basicMarkerAttributes.setImageColor(color);

        Color darkenColor = darkenColor(color);

        basicMarkerAttributes.setLineMaterial(new Material(darkenColor));

        basicMarkerAttributes.setLineWidth(1d);

        basicMarkerAttributes.setUsePointAsDefaultImage(true);

        basicMarkerAttributes.setScale(markerSize);

    }

}