| gvSIG desktop 1

package org.gvsig.data.datastores.vectorial.driver;

import java.awt.geom.Rectangle2D;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.gvsig.data.DataSourceManager;

import org.gvsig.data.IDataCollection;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.commands.implementation.AbstractFeatureCommand;

import org.gvsig.data.spatialprovisional.IExtent;

import org.gvsig.data.vectorial.AbstractFeatureStore;

import org.gvsig.data.vectorial.FeatureStoreNotification;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureCollection;

import org.gvsig.data.vectorial.IFeatureID;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.metadata.IMetadata;

import org.gvsig.metadata.IMetadataManager;

import org.gvsig.metadata.Metadata;

import org.gvsig.metadata.MetadataManager;

/**

 *

 * Los DataStore basados en drivers solo soportan un

 * tipo de feature, no pueden tener varios tipos de feature en ellos.

 *

 * @author Equipo de desarrollo de gvSIG.

 *

 */

public class DriverStore extends AbstractFeatureStore {

    public static String DATASTORE_NAME = "DriverDataSource";

    protected IDriver driver;

    private List featureTypes = new ArrayList();//<IFeatureType>

        private Rectangle2D fullExtent;

        public void init(IDataStoreParameters parameters) {

        super.init(parameters);

            IDriverStoreParameters driverStoreParameters=(IDriverStoreParameters)parameters;

            DataSourceManager dsm = DataSourceManager.getManager();

        this.driver = dsm.createDriver(driverStoreParameters.getDriverParameters());

    }

    public String getName() {

        return DATASTORE_NAME;

    }

    public void open() {

        this.observable.notifyObservers(

                new FeatureStoreNotification(this,FeatureStoreNotification.BEFORE_OPEN)

            );

        driver.open();

        this.observable.notifyObservers(

                new FeatureStoreNotification(this,FeatureStoreNotification.AFTER_OPEN)

            );

    }

    public void close() {

        this.observable.notifyObservers(

                new FeatureStoreNotification(this,FeatureStoreNotification.BEFORE_CLOSE)

            );

        driver.close();

        this.observable.notifyObservers(

                new FeatureStoreNotification(this,FeatureStoreNotification.AFTER_CLOSE)

            );

    }

    public void dispose() {

        this.observable.notifyObservers(

                new FeatureStoreNotification(this,FeatureStoreNotification.BEFORE_DISPOSE)

            );

        driver.dispose();

        this.observable.notifyObservers(

                new FeatureStoreNotification(this,FeatureStoreNotification.AFTER_DISPOSE)

            );

    }

    public IFeature getFeatureByID(IFeatureID id) {

            if (this.alterMode){

                    if (featureManager.contains(id)) {

                            return featureManager.getFeature(id);

                    }

            }

        return driver.getFeatureByID(id);

    }

    public List getFeatureTypes() {

        featureTypes.set(0,driver.getFeatureType());

        return featureTypes;

    }

    public IFeatureType getDefaultFeatureType() {

            IFeatureType ft = driver.getFeatureType();

            if (isEditing()){

//                    Aqu? hay que construir un FeatureType con los cambios que se hayan hecho en la edici?n.

                    return attributeManager.getFeatureType(ft);

            }

        return ft;

    }

    public IDataCollection getDataCollection(IFeatureType type, String filter, String order) {

            IDataCollection dataCollection=driver.getDataCollection(featureManager,type, filter, order);

            this.addObserver(dataCollection);

            return dataCollection;

    }

    private void write(IFeatureCollection collection, IFeatureType featureType) {

        IFeaturesWriter writer = driver.getFeaturesWriter();

        writer.init(driver);

        writer.updateFeatureType(featureType);

        writer.preProcess();

        Iterator iterator=collection.iterator();

        while (iterator.hasNext()) {

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

                        writer.updateFeature(feature);

                }

        writer.postProcess();

    }

    public void finishEditing() {

            Iterator commandsfeatures = null;

            IFeaturesWriter writer = driver.getFeaturesWriter();

            IFeatureType type = getDefaultFeatureType();

            if (writer.isWriteAll()){

                    //?Llamar al driver para escribir una colecci?n?

                    write((IFeatureCollection)getDataCollection(),type);

            }else{

                    writer.init(driver);

                writer.updateFeatureType(type);

                writer.preProcess();

                    commandsfeatures = commands.getCommandsDeleted().iterator();

                    while( commandsfeatures.hasNext() ) {

                            Object obj=commandsfeatures.next();

                            if (obj instanceof AbstractFeatureCommand){

                                    IFeature feature = ((AbstractFeatureCommand)obj).getFeature();

                                    writer.deleteFeature(feature);

                            }

                    }

                    commandsfeatures = commands.getCommandsUpdated().iterator();

                    while( commandsfeatures.hasNext() ) {

                            Object obj=commandsfeatures.next();

                            if (obj instanceof AbstractFeatureCommand){

                                    IFeature feature = ((AbstractFeatureCommand)obj).getFeature();

                                    writer.updateFeature(feature);

                            }

                    }

                    commandsfeatures = commands.getCommandsInserted().iterator();

                    while( commandsfeatures.hasNext() ) {

                            Object obj=commandsfeatures.next();

                            if (obj instanceof AbstractFeatureCommand){

                                    IFeature feature = ((AbstractFeatureCommand)obj).getFeature();

                                    writer.insertFeature(feature);

                            }

                    }

                    writer.postProcess();

            }

    }

    public boolean isEditable() {

            return driver.isEditable();

    }

   public boolean isWithDefaultLegend() {

                return (driver instanceof IDefaultLegend);

        }

        public Object getDefaultLegend() {

                if (driver instanceof IDefaultLegend)

                        return ((IDefaultLegend)driver).getDefaultLegend();

                return null;

        }

        public Object getDefaultLabelingStrategy() {

                if (driver instanceof IDefaultLegend)

                        return ((IDefaultLegend)driver).getDefaultLabelingStrategy();

                return null;

        }

        public boolean canAlterFeatureType() {

                return driver.canAlterFeatureType();

        }

        public IMetadata getMetadata() {

                IMetadata metadata=driver.getMetadata();

                if (this.alterMode){

                        IExtent extent=(IExtent)metadata.get("extent");

                        IFeatureCollection featureCollection=(IFeatureCollection)getDataCollection();

                    if (spatialManager.isFullExtentDirty()){

                            if (!featureCollection.isEmpty()){

                                    Iterator featureIterator=featureCollection.iterator();

                                    extent = ((IFeature)featureIterator.next()).getExtent();

                                    while(featureIterator.hasNext()){

                                            IFeature feature=(IFeature)featureIterator.next();

                                            IExtent boundExtent=feature.getExtent();

                                            if (boundExtent!=null)

                                                    extent.add(boundExtent);

                                    }

                            }

                    }

                    metadata.set("extent",extent);

                }

                return metadata;

        }

}