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.io.File;

import java.util.Iterator;

import org.gvsig.fmap.dal.DALFileLibrary;

import org.gvsig.fmap.dal.DataStoreParameters;

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

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

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

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

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

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

import org.gvsig.fmap.dal.index.spatial.jts.JTSIndexLibrary;

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

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

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

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

import org.gvsig.fmap.geom.impl.DefaultGeometryLibrary;

public class TestSHP extends BaseTestEditableFeatureStore {

        protected boolean testSHPInitialized = false;

        private FilesystemServerExplorer myExplorer;

        public static final File file_prueba = new File(TestSHP.class.getResource(

                        "data/prueba.shp").getFile());

        public static final File file_prueba_destino = new File(file_prueba

                        .getParent()

                        + "/pruebaTemp.shp");

        public static final File file_pruebaNull = new File(TestSHP.class

                        .getResource("data/pruebaNull.shp").getFile());

        public static final File file_poly_valencia = new File(TestSHP.class

                        .getResource("data/poly-valencia.shp").getFile());

        protected void setUp() throws Exception {

                super.setUp();

                if (testSHPInitialized) {

                        return;

                }

                DALFileLibrary libFile = new DALFileLibrary();

                libFile.initialize();

                libFile.postInitialize();

                DefaultGeometryLibrary lib = new DefaultGeometryLibrary();

                lib.initialize();

                lib.postInitialize();

                SHPLibrary shpLib = new SHPLibrary();

                shpLib.initialize();

                shpLib.postInitialize();

                JTSIndexLibrary jtsIndex = new JTSIndexLibrary();

                jtsIndex.initialize();

                jtsIndex.postInitialize();

                testSHPInitialized = true;

        }

        public void testSimpleIteration() {

                FeatureStore store = null;

                SHPStoreParameters shpParameters = null;

                try {

                        shpParameters = (SHPStoreParameters) dataManager

                                        .createStoreParameters(SHPStoreProvider.NAME);

                        //                        shpParameters.setFile(file_poly_valencia);

                        shpParameters.setFile(file_prueba);

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

                        FeatureSet set;

                        FeatureType type = store.getDefaultFeatureType();

                        set = store.getFeatureSet();

                        System.out.println("Num:" + set.getSize());

                        Iterator it = set.iterator();

                        Iterator ftIt;

                        FeatureAttributeDescriptor desc;

                        int i = 0;

                        Feature feature;

                        while (it.hasNext()) {

                                ftIt = type.iterator();

                                feature = (Feature) it.next();

                                //                                while (ftIt.hasNext()) {

                                //                                        desc = (FeatureAttributeDescriptor) ftIt.next();

                                //                                        System.out.println(desc.getName() + ":"

                                //                                                        + feature.get(desc.getIndex()));

                                //

                                //                                }

                                //                                System.out.println(feature.get("NOMBRE"));

                                System.out.print(feature.getDefaultEnvelope() + "\t");

                                //                                System.out.print(feature.getDefaultGeometry() + "\t");

                                //                                System.out.println(feature.get("NOMBRE"));

                                System.out.println("row:" + i);

                                i++;

                        }

                        set.dispose();

                        store.dispose();

                } catch (DataException e3) {

                        e3.printStackTrace();

                        fail();

                        return;

                }

        }

        public void testEditing(Object x) {

                FeatureStore store = null;

                SHPStoreParameters shpParameters = null;

                try {

                        shpParameters = (SHPStoreParameters) dataManager

                                        .createStoreParameters(SHPStoreProvider.NAME);

                        shpParameters.setFile(file_poly_valencia);

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

                        FeatureSet set;

                        FeatureType type = store.getDefaultFeatureType();

                        System.err.println("Antes de la edici?n");

                        set = store.getFeatureSet();

                        System.out.println("Num:" + set.getSize());

                        Iterator it = set.iterator();

                        Iterator ftIt;

                        FeatureAttributeDescriptor desc;

                        int i = 0;

                        Feature feature;

                        while (it.hasNext()) {

                                ftIt = type.iterator();

                                feature = (Feature) it.next();

                                //                                while (ftIt.hasNext()) {

                                //                                        desc = (FeatureAttributeDescriptor) ftIt.next();

                                //                                        System.out.println(desc.getName() + ":"

                                //                                                        + feature.get(desc.getIndex()));

                                //

                                //                                }

                                //                                System.out.println(feature.get("NOMBRE"));

                                System.out.print(feature.getDefaultEnvelope() + "\t");

                                //                                System.out.print(feature.getDefaultGeometry() + "\t");

                                //                                System.out.println(feature.get("NOMBRE"));

                                System.out.println("row:" + i);

                                i++;

                        }

                        set.dispose();

                        store.edit();

                        EditableFeature ef = store.createNewFeature();

                        store.insert(ef);

                        store.finishEditing();

                        System.err.println("Despu?s de la edici?n");

                        set = store.getFeatureSet();

                        System.out.println("Num:" + set.getSize());

                        it = set.iterator();

                        i = 0;

                        while (it.hasNext()) {

                                ftIt = type.iterator();

                                feature = (Feature) it.next();

                                //                                while (ftIt.hasNext()) {

                                //                                        desc = (FeatureAttributeDescriptor) ftIt.next();

                                //                                        System.out.println(desc.getName() + ":"

                                //                                                        + feature.get(desc.getIndex()));

                                //

                                //                                }

                                //                                System.out.println(feature.get("NOMBRE"));

                                System.out.print(feature.getDefaultEnvelope() + "\t");

                                //                                System.out.print(feature.getDefaultGeometry() + "\t");

                                //                                System.out.println(feature.get("NOMBRE"));

                                System.out.println("row:" + i);

                                i++;

                        }

                        set.dispose();

                        store.dispose();

                } catch (DataException e3) {

                        e3.printStackTrace();

                        fail();

                        return;

                }

        }

        public void testExport(Object x) {

                DBFStoreParameters dbfParameters = null;

                try {

                        dbfParameters = (DBFStoreParameters) dataManager

                                        .createStoreParameters(DBFStoreProvider.NAME);

                        dbfParameters.setFile(file_prueba);

                        FeatureStore store = (FeatureStore) dataManager

                                        .createStore(dbfParameters);

                        FilesystemServerExplorerParameters explorerParams=(FilesystemServerExplorerParameters) dataManager.createServerExplorerParameters(FilesystemServerExplorerParameters.DYNCLASS_NAME);

                        explorerParams.setRoot(file_prueba.getParent());

                        FilesystemServerExplorer explorer=(FilesystemServerExplorer) dataManager.createServerExplorer(explorerParams);

                        NewFeatureStoreParameters newParams = (NewFeatureStoreParameters) explorer.getAddParameters(file_prueba_destino);

                        store.export(explorer, newParams);

                        FeatureStore result = (FeatureStore)dataManager.createStore(newParams);

                        FeatureSet set = result.getFeatureSet();

                        FeatureSet originalSet=store.getFeatureSet();

                        assertEquals(set.getSize(), originalSet.getSize());

                        Iterator originalIter=originalSet.iterator();

                        Iterator iter = set.iterator();

                        while (iter.hasNext()) {

                                assertEquals(originalIter.next(),iter.next());

                        }

                        result.dispose();

                        explorer.remove(newParams);

                } catch (DataException e3) {

                        e3.printStackTrace();

                        fail();

                        return;

                }

        }

        /*

         * (non-Javadoc)

         *

         * @see

         * org.gvsig.fmap.dal.feature.BaseTestFeatureStore#getDefaultDataStoreParameters

         * ()

         */

        public DataStoreParameters getDefaultDataStoreParameters()

                        throws DataException {

                SHPStoreParameters shpParameters = (SHPStoreParameters) dataManager

                                .createStoreParameters(SHPStoreProvider.NAME);

                //                        shpParameters.setFile(file_poly_valencia);

                shpParameters.setFile(file_prueba);

                return shpParameters;

        }

        /*

         * (non-Javadoc)

         *

         * @see org.gvsig.fmap.dal.feature.BaseTestFeatureStore#hasExplorer()

         */

        public boolean hasExplorer() {

                return true;

        }

        /*

         * (non-Javadoc)

         *

         * @seeorg.gvsig.fmap.dal.feature.BaseTestEditableFeatureStore#

         * getDefaultNewDataStoreParameters()

         */

        public NewFeatureStoreParameters getDefaultNewDataStoreParameters()

                        throws DataException {

                if (this.myExplorer == null) {

                        SHPStoreParameters parameters = (SHPStoreParameters) this

                                        .getDefaultDataStoreParameters();

                        FeatureStore store = (FeatureStore) dataManager

                                        .createStore(parameters);

                        myExplorer = (FilesystemServerExplorer) store.getExplorer();

                        store.dispose();

                }

                return (NewFeatureStoreParameters) myExplorer

                                .getAddParameters(file_prueba_destino);

        }

        public boolean usesResources() {

                return true;

        }

}