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

import java.util.ArrayList;

import java.util.Hashtable;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

import org.gvsig.fmap.dal.feature.exception.PerformEditingException;

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

import org.gvsig.fmap.dal.feature.spi.FeatureReferenceProviderServices;

import org.gvsig.fmap.dal.feature.spi.memory.AbstractMemoryStoreProvider;

import org.gvsig.fmap.dal.feature.spi.memory.MemoryResource;

import org.gvsig.fmap.dal.resource.spi.ResourceProvider;

import org.gvsig.fmap.dal.spi.DataStoreProviderServices;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.exception.CreateEnvelopeException;

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

import org.gvsig.metadata.MetadataLocator;

import org.gvsig.metadata.MetadataManager;

import org.gvsig.metadata.exceptions.MetadataException;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.dynobject.DynStruct;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

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

 * 

 *         TODO: take into account the parameter "autoOrderAttributeName" to

 *         order the {@link Feature}s by default by that attribute, in ascending

 *         order.

 */

public class MemoryStoreProvider extends AbstractMemoryStoreProvider{

        private static final Logger logger = LoggerFactory.getLogger(MemoryStoreProvider.class);

        private static GeometryManager geomManager = GeometryLocator.getGeometryManager();

        public static final String NAME = "Memory";

        public static final String DESCRIPTION = "Memory provider";

        public static final String METADATA_DEFINITION_NAME = "MemoryProvider";

        private long counterNewsOIDs = 0;

        private Envelope envelope = null;

        private Map oids = null;

        private Double sourceID = null;

        private ResourceProvider memoryResource;

        public static DynObject newMetadataContainer(String name) {

            return ToolsLocator.getDynObjectManager()

                        .createDynObject(

                                MetadataLocator.getMetadataManager().getDefinition(name)

                        );

        }

        public MemoryStoreProvider(MemoryStoreParameters memoryStoreParameters, 

                        DataStoreProviderServices storeServices) throws InitializeException {

                super(

                                memoryStoreParameters, 

                                storeServices,

                                newMetadataContainer(METADATA_DEFINITION_NAME)

                );

                sourceID = new Double(Math.random());

                data = new ArrayList();

                oids = new Hashtable();

                initialize();

                memoryResource =

                                createResource(MemoryResource.NAME,

                                                new Object[] { memoryStoreParameters.getName() });

        }        

        public String getName() {

                return this.sourceID.toString();

        }

        public String getFullName() {

                return "memory:"+this.sourceID.toString();

        }

        private void initialize() {

                FeatureType featureType = getStoreServices().createFeatureType().getNotEditableCopy();                

                List types = new ArrayList(1);

                types.add(featureType);

                getStoreServices().setFeatureTypes(types, featureType);

        }

        /*

         *         (non-Javadoc)

         * @see org.gvsig.fmap.dal.feature.spi.FeatureStoreProvider#createNewOID()

         */

        public Object createNewOID() {

                return new Long(counterNewsOIDs++);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.feature.spi.FeatureStoreProvider#getName()

         */

        public String getProviderName() {

                return NAME;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.feature.spi.FeatureStoreProvider#getOIDType()

         */

        public int getOIDType() {

                return DataTypes.LONG;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.fmap.dal.spi.DataStoreProvider#getSourceId()

         */

        public Object getSourceId() {

                return sourceID;

        }

        public void open() throws OpenException {

        }

        public static void registerMetadataDefinition() throws MetadataException {

                MetadataManager manager = MetadataLocator.getMetadataManager();

                if( manager.getDefinition(METADATA_DEFINITION_NAME)==null ) {

                        DynStruct metadataDefinition = manager.addDefinition(METADATA_DEFINITION_NAME, "Memory Store");

                        metadataDefinition.extend(manager

                                        .getDefinition(DataStore.SPATIAL_METADATA_DEFINITION_NAME));

                }

        }

        public boolean allowWrite() {

                return true;

        }

        public void performChanges(Iterator deleteds, Iterator inserteds,

                        Iterator updateds, Iterator originalFeatureTypesUpdated)

        throws PerformEditingException {

                FeatureSet set = null;

                DisposableIterator iter = null;

                try {                

                        FeatureType featureType = this.getStoreServices().getDefaultFeatureType();

                        List types = new ArrayList(1);

                        types.add(featureType);                        

                        getStoreServices().setFeatureTypes(types, featureType);        

                        //Sets the envelope                        

                        envelope = 

                                geomManager.createEnvelope(featureType.getDefaultGeometryAttribute().getGeometrySubType());

                        set = this.getFeatureStore().getFeatureSet();

                        iter = set.fastIterator();

                        ArrayList newData = new ArrayList();

                        Hashtable newOids = new Hashtable();

                        while (iter.hasNext()) {

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

                                FeatureProvider featureProvider = this.getStoreServices().getFeatureProviderFromFeature(feature);

                                newData.add(featureProvider);

                                newOids.put(featureProvider.getOID(), featureProvider);

                                envelope.add(feature.getDefaultEnvelope());

                        }

                        data = newData;

                        oids = newOids;

                } catch (DataException e) {

                        logger.error("Error adding features", e);

                } catch (CreateEnvelopeException e) {

                        logger.error("Error creating the envelope", e);

                } finally {

                        dispose(iter);

                        dispose(set);

                }

        }

        public Envelope getEnvelope() throws DataException {

                return envelope;

        }

        protected FeatureProvider internalGetFeatureProviderByReference(

                        FeatureReferenceProviderServices reference) throws DataException {

                return (FeatureProvider) oids.get(reference.getOID());

        }

        public ResourceProvider getResource() {

                return memoryResource;

        }

}