Application: gvSIG desktop

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

* 

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

* 

*/

/*

* AUTHORS (In addition to CIT):

* 2009 {Iver T.I.}   {Task}

*/

package org.gvsig.fmap.mapcontext.layers.vectorial.impl;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

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.dal.store.memory.MemoryStoreProvider;

import org.gvsig.fmap.geom.Geometry;

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

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

import org.gvsig.fmap.mapcontext.exceptions.LegendLayerException;

import org.gvsig.fmap.mapcontext.exceptions.LoadLayerException;

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

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

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

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

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

import org.gvsig.tools.exception.BaseException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * @author <a href="mailto:jpiera@gvsig.org">Jorge Piera</a>

 */

public class DefaultGraphicLayer extends FLyrVect implements GraphicLayer{

        protected static final Logger logger = LoggerFactory.getLogger(DefaultGraphicLayer.class);

        private static DataManager dataManager = DALLocator.getDataManager();

        private IVectorialUniqueValueLegend legend = null;

        private FeatureStore store = null;

        private long ids = 0;

        private int symbolId = -1;

        public DefaultGraphicLayer() {

                super();        

        }

        public void initialize(IProjection projection) throws ValidateDataParametersException, DataException, LoadLayerException { 

                DataStoreParameters parameters = dataManager

                .createStoreParameters(MemoryStoreProvider.NAME);

                store = (FeatureStore) dataManager.createStore(parameters);

                store.edit();

                EditableFeatureType editableFeatureType = store.getDefaultFeatureType().getEditable();

                editableFeatureType.add(GraphicLayer.FEATURE_ATTR_ID, DataTypes.STRING);

                EditableFeatureAttributeDescriptor geometryDescriptor = editableFeatureType.add(GraphicLayer.FEATURE_ATTR_GEOMETRY, DataTypes.GEOMETRY);

                geometryDescriptor.setGeometryType(TYPES.GEOMETRY);

                geometryDescriptor.setGeometrySubType(SUBTYPES.GEOM2D);

                geometryDescriptor.setSRS(projection);

                editableFeatureType.setDefaultGeometryAttributeName(GraphicLayer.FEATURE_ATTR_GEOMETRY);

                editableFeatureType.add(GraphicLayer.FEATURE_ATTR_IDSYMBOL, DataTypes.INT);

                editableFeatureType.add(GraphicLayer.FEATURE_ATTR_LABEL, DataTypes.STRING);

                EditableFeatureAttributeDescriptor featureIdDescriptor = 

                        editableFeatureType.add(GraphicLayer.FEATURE_ATTR_FEATUREID, DataTypes.LONG);

                featureIdDescriptor.setIsPrimaryKey(true);

                editableFeatureType.setHasOID(true);

                store.update(editableFeatureType);        

                store.finishEditing();                

                this.setDataStore(store);

        }

        public void addGraphic(String id, Geometry geom, int idsym) {

                EditableFeature feature;

                try {

                        if (!store.isEditing()){

                                store.edit();

                        }

                        feature = store.createNewFeature().getEditable();

                        feature.set(GraphicLayer.FEATURE_ATTR_ID, id);

                        feature.set(GraphicLayer.FEATURE_ATTR_GEOMETRY, geom);

                        feature.set(GraphicLayer.FEATURE_ATTR_IDSYMBOL, new Integer(idsym));                

                        feature.set(GraphicLayer.FEATURE_ATTR_FEATUREID, new Long(ids));

                        ids++;

                        store.insert(feature);        

                        store.finishEditing();

                } catch (DataException e) {

                        logger.error("Error adding a geometry to the graphic layer", e);

                }

        }

        public void addGraphic(Geometry geom, int idsym) {

                addGraphic("0", geom, idsym);                

        }

        public void addGraphic(Geometry geom, int idsym, String label) {

                addGraphic("0", geom, idsym, label);                

        }

        public void addGraphic(String id, Geometry geom, int idsym, String label) {

                /*

                 * El id, puede ser duplicado. Una aplicacion/extension e?adira todos sus

                 * graficos con el mismo id para poder acceder a ellos de una tacada.

                 */

                try{

                        if (!store.isEditing()){

                                store.edit();

                        }

                        EditableFeature feature = store.createNewFeature().getEditable();

                        feature.set(GraphicLayer.FEATURE_ATTR_ID, id);

                        feature.set(GraphicLayer.FEATURE_ATTR_GEOMETRY, geom);

                        feature.set(GraphicLayer.FEATURE_ATTR_IDSYMBOL, new Integer(idsym));        

                        feature.set(GraphicLayer.FEATURE_ATTR_LABEL, label);        

                        feature.set(GraphicLayer.FEATURE_ATTR_FEATUREID, new Long(ids));

                        ids++;

                        store.insert(feature);

                        store.finishEditing();

                } catch (DataException e) {

                        logger.error("Error adding a geometry to the graphic layer", e);

                }

        }

        public int addSymbol(ISymbol newSymbol) {

                symbolId++;

                legend.addSymbol(new Integer(symbolId), newSymbol);

                return symbolId;

        }

        public ISymbol getSymbol(int symbolPos) {

                return legend.getSymbolByValue(new Integer(symbolPos));                

        }

        public void clearAllGraphics() {

                DisposableIterator iterator = null;

                FeatureSet featureSet = null;

                try{

                        if (!store.isEditing()){

                                store.edit();

                        }

                        featureSet = store.getFeatureSet();

                        for (iterator = featureSet.fastIterator(); iterator.hasNext();) {

                                Feature feature = (Feature) iterator.next();

                                featureSet.delete(feature);

                        }

                        store.finishEditing();                        

                } catch (DataException e) {

                        logger.error("Error clearing all the geometry of the graphic layer", e);

                } finally {

                        if (featureSet != null) {

                                featureSet.dispose();

                        }

                        if (iterator != null) {

                                iterator.dispose();

                        }

                }

        }

        public int clearAllSymbols() {

                legend.clear();

                symbolId = -1;

                return symbolId;

        }

        public void removeGraphics(String id) {

                DisposableIterator iterator = null;

                FeatureSet featureSet = null;

                try{

                        if (!store.isEditing()){

                                store.edit();

                        }

                        featureSet = store.getFeatureSet();

                        for (iterator = featureSet.fastIterator(); iterator.hasNext();) {

                                Feature feature = (Feature) iterator.next();

                                if (feature.get(FEATURE_ATTR_ID).equals(id)){

                                        featureSet.delete(feature);

                                }

                        }

                        store.finishEditing();

                } catch (DataException e) {

                        logger.error("Error clearing all the geometry of the graphic layer", e);

                } finally {

                        if (featureSet != null) {

                                featureSet.dispose();

                        }

                        if (iterator != null) {

                                iterator.dispose();

                        }

                }

        }

        protected void doDispose() throws BaseException {

                super.doDispose();

                if (store != null) {

                        store.dispose();

                        store = null;

                }

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect#setLegend(org.gvsig.fmap.mapcontext.rendering.legend.IVectorLegend)

         */

        public void setLegend(IVectorLegend legend) throws LegendLayerException {

                if (legend instanceof IVectorialUniqueValueLegend){

                        super.setLegend(legend);

                        this.legend = (IVectorialUniqueValueLegend)legend;

                }else{

                        throw new LegendLayerException("The legend for the graphics layer must be a instance of IVectorialUniqueValueLegend", null);

                }

        }        

}