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

* 2008 IVER T.I. S.A.   {{Task}}

*/

package org.gvsig.fmap.dal.store.shp;

import java.util.Iterator;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.resource.exception.ResourceBeginException;

import org.gvsig.fmap.dal.resource.exception.ResourceNotifyCloseException;

import org.gvsig.fmap.dal.resource.exception.ResourceNotifyOpenException;

import org.gvsig.fmap.dal.resource.file.FileResource;

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

import org.gvsig.fmap.dal.store.dbf.DBFStoreProvider;

import org.gvsig.fmap.dal.store.shp.utils.SHPFile;

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

import org.gvsig.tools.persistence.PersistenceException;

import org.gvsig.tools.persistence.PersistentState;

public class SHPStoreProvider extends DBFStoreProvider {

        public static String NAME = "SHPStore";

        public static String DESCRIPTION = "SHP file";

        private SHPFile shpFile;

        private ResourceProvider shpResource;

        private SHPStoreParameters shpParams;

        protected static final String GEOMETRY_ATTIBUTE_NAME = "GEOMETRY";

        public SHPStoreProvider(SHPStoreParameters params)

                        throws InitializeException {

                super(params);

                this.shpParams = params;

                shpResource = this.createResource(FileResource.NAME,

                                new Object[] { this.shpParams.getSHPFileName() });

                shpResource.addConsumer(this);

                //DBFResource tmpResource = new DBFResource(dbfParameters);

                //

                //try {

                //        this.dbf = (DBFResource) this.store.addResource(tmpResource);

                //} catch (DataException e1) {

                //        throw new InitializeException(this.getName(), e1);

                //}

                this.shpFile = new SHPFile(this.shpParams);

        }

        protected static EditableFeatureAttributeDescriptor addGeometryColumn(

                        EditableFeatureType fType) {

                EditableFeatureAttributeDescriptor attr = fType.add(

                                GEOMETRY_ATTIBUTE_NAME, DataTypes.GEOMETRY);

                fType.setDefaultGeometryAttributeName(attr.getName());

                return attr;

        }

        protected static void removeGeometryColumn(

                        EditableFeatureType fType) {

                fType.setDefaultGeometryAttributeName(null);

                fType.remove(GEOMETRY_ATTIBUTE_NAME);

        }

        protected EditableFeatureType getTheFeatureType()

                        throws InitializeException {

                EditableFeatureType fType = super.getTheFeatureType();

                try {

                        this.open();

                        this.shpResource.begin();

                } catch (DataException e) {

                        throw new InitializeException(this.getName(), e);

                }

                try {

                        EditableFeatureAttributeDescriptor attr = addGeometryColumn(

                                        fType).setGeometryType(

                                        this.shpFile.getGeometryType());

                        String srs = this.getSRS(this.shpFile.getSRSParameters());

                        if (srs == null){

                                // TODO petar ??

                                srs = "EPSG:23030";

                        }

                        attr.setSRS(srs);

                        return fType;

                } catch (ReadException e) {

                        throw new InitializeException(e);

                } finally {

                        this.shpResource.end();

                }

        }

        private String getSRS(String srsParameters) {

                // TODO identificar que SRS hay que usar, ya sea

                // el que se recibe de los parametros o el que

                // conicida con el que se ha encontrado en el

                // prg... y si ninguna de las dos que?

                return null;

        }

        protected SHPStoreParameters getShpParameters() {

                return shpParams;

        }

        public String getName() {

                return NAME;

        }

        public boolean allowWrite() {

                return super.allowWrite() && this.shpFile.isEditable();

        }

        /**

         *

         * @param index

         * @param featureType

         * @return

         * @throws ReadException

         */

        protected FeatureData getFeatureDataByIndex(long index,

                        FeatureType featureType) throws DataException {

                this.open();

                this.shpResource.begin();

                try {

                        FeatureData featureData = super.getFeatureDataByIndex(index,

                                        featureType);

                        featureData.setDefaultEnvelope(this.shpFile.getBoundingBox(index));

                        return featureData;

                } catch (DataException e) {

                        throw e;

                } finally {

                        this.shpResource.end();

                }

        }

        protected void initFeatureDataByIndex(FeatureData featureData,

                        long index, FeatureType featureType) throws DataException {

                this.open();

                this.shpResource.begin();

                try {

                        super.initFeatureDataByIndex(featureData, index, featureType);

                        featureData.setDefaultEnvelope(this.shpFile.getBoundingBox(index));

                } finally {

                        this.shpResource.end();

                }

        }

        /**

         *

         * @param featureData

         * @throws DataException

         */

        protected void loadFeatureDataByIndex(FeatureData featureData)

                        throws DataException {

                this.open();

                this.shpResource.begin();

                try {

                        long index = ((Long) featureData.getOID()).longValue();

                        super.loadFeatureDataByIndex(featureData);

                        if (featureData.getType().getDefaultGeometryAttributeIndex() >= 0) {

                                featureData.setDefaultGeometry(this.shpFile.getGeometry(index));

                        }

                } finally {

                        this.shpResource.end();

                }

        }

        protected void loadValue(FeatureData featureData, int rowIndex,

                        FeatureAttributeDescriptor descriptor) throws ReadException {

                if (descriptor.getDataType() == DataTypes.GEOMETRY) {

                        return;

                } else {

                        super.loadValue(featureData, rowIndex, descriptor);

                }

        }

        public FeatureData createFeatureData(FeatureType type) throws DataException {

                FeatureData data = new SHPFeatureData(this, (DefaultFeatureType) type);

                return data;

        }

        public void open() throws OpenException {

                super.open();

                if (this.shpFile.isOpen()) {

                        return;

                }

                try {

                        this.shpResource.begin();

                } catch (ResourceBeginException e) {

                        throw new OpenException(this.getName(), e);

                }

                try {

                        this.shpFile.open();

                        this.shpResource.notifyOpen();

                } catch (UnsupportedVersionException e) {

                        throw new OpenException(this.getName(), e);

                } catch (ResourceNotifyOpenException e) {

                        throw new OpenException(this.getName(), e);

                } catch (FileNotFoundException e) {

                        throw new OpenException(this.getName(), e);

                } catch (DataException e) {

                        throw new OpenException(this.getName(), e);

                } finally {

                        this.shpResource.end();

                }

        }

        public void close() throws CloseException {

                super.close();

                if (!this.shpFile.isOpen()) {

                        return;

                }

                try {

                        this.shpResource.begin();

                } catch (ResourceBeginException e) {

                        throw new CloseException(this.getName(), e);

                }

                try {

                        this.shpFile.close();

                        this.shpResource.notifyClose();

                } catch (ResourceNotifyCloseException e) {

                        throw new CloseException(this.getName(), e);

                } finally {

                        this.shpResource.end();

                }

        }

        public boolean canWriteGeometry(int geometryType) throws DataException {

                this.open();

                this.shpResource.begin();

                try {

                        return this.shpFile.canWriteGeometry(geometryType);

                } finally {

                        this.shpResource.end();

                }

        }

        public void performEditing(Iterator deleteds, Iterator inserteds,

                        Iterator updateds) throws PerformEditingException {

                FeatureType fType;

                try {

                        fType = this.store.getDefaultFeatureType();

                } catch (DataException e) {

                        throw new PerformEditingException(this.getName(), e);

                }

                // TODO Comprobar el campo de geometria

                EditableFeatureType dbfFtype = fType.getEditable();

                removeGeometryColumn(dbfFtype);

                SHPFeatureWriter writer = null;

                try {

                        this.shpResource.begin();

                        FeatureSet set = this.store.getFeatureSet();

                        writer = new SHPFeatureWriter(this.getName());

                        writer.begin(this.shpParams, fType, dbfFtype, set.getSize());

                        Iterator iter = set.iterator();

                        while (iter.hasNext()) {

                                writer.write((Feature) iter.next());

                        }

                        writer.end();

                        shpResource.notifyChanges();

                } catch (Exception e) {

                        throw new PerformEditingException(this.getName(), e);

                } finally {

                        shpResource.end();

                }

        }

        public Envelope getEnvelope() throws DataException {

                this.open();

                this.shpResource.begin();

                try {

                        return this.shpFile.getFullExtent();

                } finally {

                        this.shpResource.end();

                }

        }

        public void setState(PersistentState state) throws PersistenceException {

                // TODO Auto-generated method stub

                super.setState(state);

        }

        public PersistentState getState() throws PersistenceException {

                // TODO Auto-generated method stub

                return super.getState();

        }

        public void append(Feature feature) {

                // TODO Auto-generated method stub

                super.append(feature);

        }

        public void beginAppend() {

                // TODO Auto-generated method stub

                super.beginAppend();

        }

        public void endAppend() {

                // TODO Auto-generated method stub

                super.endAppend();

        }

}