Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 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 3

 * 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.fmap.dal.store.shp;

import java.io.File;

import java.io.IOException;

import org.gvsig.basicformats.FormatsFile;

import org.gvsig.basicformats.PRJFile;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.NewDataStoreParameters;

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

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

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

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

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

import org.gvsig.fmap.dal.resource.ResourceAction;

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

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

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

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.AbsolutePathRequiredException;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * @author jmvivo

 *

 */

public class SHPFilesystemServerProvider extends DBFFilesystemServerProvider {

        protected static final Logger LOGGER = LoggerFactory.getLogger(SHPFilesystemServerProvider.class);

        @Override

        public boolean accept(File pathname) {

                return pathname.getName().toLowerCase().endsWith(".shp");

        }

        @Override

        public int getMode() {

                return DataServerExplorer.MODE_GEOMETRY|DataServerExplorer.MODE_FEATURE;

        }

        @Override

        public boolean canCreate() {

                return true;

        }

        private ResourceProvider createResource(File shpFile, File[] otherFiles)

                        throws InitializeException {

                Object[] shpParams = new Object[] { shpFile.getAbsolutePath() };

                MultiResource resource =

                                (MultiResource) serverExplorer.getServerExplorerProviderServices()

                                                .createResource(MultiResource.TYPE_NAME, shpParams);

                resource.addResource(FileResource.NAME, shpParams, true);

                for (int i = 0; i < otherFiles.length; i++) {

                        resource.addResource(FileResource.NAME,

                                        new Object[] { otherFiles[i].getAbsolutePath() }, false);

                }

                resource.addConsumer(this);

                return resource;

        }

        // private FileResource createResource(File file) throws InitializeException

        // {

        // FileResource resource;

        // resource =

        // (FileResource) this.serverExplorer.getServerExplorerProviderServices()

        // .createResource(FileResource.NAME,

        // new Object[] { file.getAbsolutePath() });

        // resource.addConsumer(this);

        // return resource;

        // }

        @Override

        public boolean canCreate(NewDataStoreParameters parameters) {

                if (!super.canCreate(parameters)) {

                        return false;

                }

                SHPNewStoreParameters params = (SHPNewStoreParameters) parameters;

                if (params.getSHPFile().getParentFile().canWrite()) {

                        return false;

                }

                if (params.getSHPFile().exists()) {

                        if (!params.getSHPFile().canWrite()) {

                                return false;

                        }

                }

                if (params.getSHXFile().getParentFile().canWrite()) {

                        return false;

                }

                if (params.getSHXFile().exists()) {

                        if (!params.getSHXFile().canWrite()) {

                                return false;

                        }

                }

                return true;

        }

        @Override

        public boolean closeResourceRequested(ResourceProvider resource) {

                // while it is using a resource anyone can't close it

                return false;

        }

        @Override

        public void create(NewDataStoreParameters parameters, boolean overwrite)

                        throws CreateException {

                final SHPNewStoreParameters params = (SHPNewStoreParameters) parameters;

                final EditableFeatureType fType  = params.getDefaultFeatureType();

                File dbfFile = params.getDBFFile();

                File shpFile = params.getSHPFile();

                File shxFile = params.getSHXFile();

                if (!shpFile.isAbsolute()) {

                        throw new AbsolutePathRequiredException(shpFile.getPath());

                }

                if (!dbfFile.isAbsolute()) {

                        throw new AbsolutePathRequiredException(dbfFile.getPath());

                }

                if (!shxFile.isAbsolute()) {

                        throw new AbsolutePathRequiredException(shxFile.getPath());

                }

                // File[] files = new File[] { shpFile, shxFile, dbfFile };

                // TODO .prj file

                // FileResource[] resources;

                ResourceProvider resource;

                try {

                        // resources =

                        resource = createResource(shpFile, new File[] { shxFile, dbfFile });

                        resource.closeRequest();

                        // closeResources(resources);

                } catch (DataException e1) {

                        throw new CreateException(shpFile.getPath(), e1);

                }

                if (shpFile.exists()) {

                        if (overwrite) {

                                // FIXME

                                // if (!shpFile.delete()) {

                                // throw new CreateException(this.getDataStoreProviderName(),

                                // new IOException(

                                // "cannot delete file: " + shpFile.getPath()));

                                // }

                                // if (shxFile.exists()) {

                                // if (!shxFile.delete()) {

                                // throw new CreateException(this.getDataStoreProviderName(),

                                // new IOException("cannot delete file: "

                                // + shxFile.getPath()));

                                // }

                                //

                                // }

                        } else {

                                throw new CreateException(this.getDataStoreProviderName(),

                                                new IOException("file already exist"));

                        }

                }

                try {

//                        beginResources(resources);

                        resource.execute(new ResourceAction() {

                                @Override

                                public Object run() throws Exception {

                                        SHPFeatureWriter writer =

                                                        new SHPFeatureWriter(getDataStoreProviderName());

                                        writer.begin(params, fType, 0);

                                        PRJFile prjFile = FormatsFile.createPRJFile();

                                        prjFile.setCRS(params.getCRS());

                                        prjFile.writeQuietly(params.getFile());

                                        writer.end();

                                        return null;

                                }

                        });

//                        notifyChangesResources(resources);

                        resource.notifyChanges();

                } catch (Exception e) {

                        throw new CreateException(this.getDataStoreProviderName(), e);

                } finally {

                        // endResources(resources);

                        // removeConsumer(resources);

                        resource.removeConsumer(this);

                }

        }

        private void removeConsumer(FileResource[] resources) {

            for (FileResource resource : resources) {

                resource.removeConsumer(this);

            }

        }

        // private void endResources(FileResource[] resources) {

        // for (int i = 0; i < resources.length; i++) {

        // resources[i].end();

        // }

        // }

        private void notifyChangesResources(FileResource[] resources)

                        throws ResourceNotifyChangesException {

            for (FileResource resource : resources) {

                resource.notifyChanges();

            }

        }

        @Override

        public NewDataStoreParameters getCreateParameters() {

                SHPNewStoreParameters params =

                                (SHPNewStoreParameters) super.getCreateParameters();

                EditableFeatureType fType =

                                (EditableFeatureType) params.getDefaultFeatureType();

                // SHPStoreProvider.addGeometryColumn(fType);

                // params.setDefaultFeatureType(fType);

                return params;

        }

        @Override

        protected NewDataStoreParameters createInstanceNewDataStoreParameters() {

                return new SHPNewStoreParameters();

        }

        @Override

        public String getDataStoreProviderName() {

                return SHPStoreProvider.NAME;

        }

        @Override

        public String getDescription() {

                return SHPStoreProvider.DESCRIPTION;

        }

        @Override

        public DataStoreParameters getParameters(File file) throws DataException {

                DataManager manager = DALLocator.getDataManager();

                SHPStoreParameters params =

                                (SHPStoreParameters) manager.createStoreParameters(this.getDataStoreProviderName());

                params.setSHPFile(file.getPath()); 

                return params;

        }

        @Override

        public void remove(DataStoreParameters parameters) throws RemoveException {

                SHPStoreParameters params = (SHPStoreParameters) parameters;

                final File dbfFile = params.getDBFFile();

                final File shpFile = params.getSHPFile();

                final File shxFile = params.getSHXFile();

                // TODO .prj file

                // File[] files = new File[] { shpFile, shxFile, dbfFile };

                //

                // FileResource[] resources;

                File[] otherFiles = new File[] { shxFile, dbfFile };

                final ResourceProvider resource;

                try {

                        resource = createResource(shpFile, otherFiles);

                        resource.closeRequest();

                        resource.closeRequest();

                } catch (DataException e1) {

                        throw new RemoveException(shpFile.getPath(), e1);

                }

                try {

                        resource.execute(new ResourceAction() {

                                @Override

                                public Object run() throws Exception {

                                        deleteFile(shpFile);

                                        deleteFile(dbfFile);

                                        deleteFile(shxFile);

                                        resource.notifyChanges();

                                        return null;

                                }

                        });

                } catch (Exception e) {

                        throw new RemoveException(this.getDataStoreProviderName(), e);

                } finally {

                        resource.removeConsumer(this);

                }

        }

        private void deleteFile(File file) throws RemoveException {

                if (file.exists() && !file.delete()) {

                        throw new RemoveException(this.getDataStoreProviderName(),

                                        new IOException()); // FIXME Exception

                }

        }

        @Override

        public String getResourceRootPathName(DataStore dataStore) {

                SHPStoreParameters shpParams = (SHPStoreParameters) dataStore.getParameters();

                return removeFileExtension(shpParams.getSHPFile());

        }

}