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

    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 crsObject = (IProjection) vectorLayer.getFeatureStore().getDynValue(DataStore.METADATA_CRS);

            // TODO check CRS compatibility

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

            }

        }

        ILegend legend = vectorLayer.getLegend();

        changeLegend(legend);

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

                    }

                    PointPlacemark marker = makeShape(pointFeature);

                    if (marker != null)

                        renderablesList.add(marker);

                }

            } 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 PointPlacemark makeShape(Feature pointFeature) {

        try {

            Geometry geometry = pointFeature.getDefaultGeometry();

            if (geometry instanceof Point) {

                Point point = (Point) geometry;

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

                return marker;

            }

        } catch (Exception e) {

            e.printStackTrace();

        }

        return null;

    }

    @Override

    public void changeLegend(ILegend legend) {

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

    }

}