Application: gvSIG desktop

package org.gvsig.data.datastores.vectorial.file.shp;

import java.io.File;

import java.util.Iterator;

import junit.framework.TestCase;

import org.gvsig.data.DataManager;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.datastores.vectorial.file.DataStoreTest;

import org.gvsig.data.datastores.vectorial.file.shp.Register;

import org.gvsig.data.datastores.vectorial.file.shp.SHPStore;

import org.gvsig.data.datastores.vectorial.file.shp.SHPStoreParameters;

import org.gvsig.data.datastores.vectorial.file.shp.utils.SHP;

import org.gvsig.data.exception.CloseException;

import org.gvsig.data.exception.InitializeException;

import org.gvsig.data.exception.OpenException;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.vectorial.FeatureManager;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureAttributeDescriptor;

import org.gvsig.data.vectorial.IFeatureCollection;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.data.vectorial.expansionadapter.MemoryExpansionAdapter;

import com.iver.cit.gvsig.fmap.drivers.dbf.TestDbf;

public class SHPTest extends TestCase {

        private File fileNormal = new File(DataStoreTest.class.getResource("data/prueba.shp").getFile());

        private File fileGeometryNull = new File(DataStoreTest.class.getResource("data/pruebaNull.shp").getFile());

        public static void main(String[] args) {

                junit.textui.TestRunner.run(TestDbf.class);

        }

        protected void setUp() throws Exception {

                super.setUp();

        }

        public void testInitialize() {

                Register.selfRegister();

                shp(fileNormal);

                shp(fileGeometryNull);

        }

        private void shp(File file){

                DataManager manager = DataManager.getManager();

                SHPStore store = null;

                SHPStoreParameters shpParameters=null;

                shpParameters=(SHPStoreParameters)manager.createDataStoreParameters(SHPStore.DATASTORE_NAME);

                shpParameters.setFile(file);

//                shpParameters.setDBFFile(SHP.getDbfFile(file));

//                shpParameters.setSHXFile(SHP.getShxFile(file));

                try {

                        store= (SHPStore)manager.createDataStore((IDataStoreParameters)shpParameters);

                } catch (InitializeException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                try {

                        store.open();

                } catch (OpenException e2) {

                        // TODO Auto-generated catch block

                        e2.printStackTrace();

                }

                IFeatureType ft=store.getDefaultFeatureType();

                IFeatureCollection featureCollection=null;

                try {

                        featureCollection = (IFeatureCollection)store.getDataCollection(ft,null,null);

                } catch (ReadException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                }

                Iterator iter = featureCollection.iterator();

                while (iter.hasNext()) {

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

                        System.out.println("SHP Feature ------------------- ");

                        Iterator iterator=ft.iterator();

                        while (iterator.hasNext()) {

                                IFeatureAttributeDescriptor descriptor = (IFeatureAttributeDescriptor) iterator.next();

                                int i=descriptor.ordinal();

                                String type=descriptor.getDataType();

                                if (type.equals(IFeatureAttributeDescriptor.TYPE_BOOLEAN)){

                                        System.out.println("Boolean ----- "+ feature.getBoolean(i));

                                }else if (type.equals(IFeatureAttributeDescriptor.TYPE_BYTE)){

                                        System.out.println("Byte ----- "+ feature.getByte(i));

                                }else if (type.equals(IFeatureAttributeDescriptor.TYPE_DATE)){

                                        System.out.println("Date ----- "+ feature.getDate(i));

                                }else if (type.equals(IFeatureAttributeDescriptor.TYPE_DOUBLE)){

                                        System.out.println("Double ----- "+ feature.getDouble(i));

                                }else if (type.equals(IFeatureAttributeDescriptor.TYPE_FLOAT)){

                                        System.out.println("Float ----- "+ feature.getFloat(i));

                                }else if (type.equals(IFeatureAttributeDescriptor.TYPE_INT)){

                                        System.out.println("Integer ----- "+ feature.getInt(i));

                                }else if (type.equals(IFeatureAttributeDescriptor.TYPE_LONG)){

                                        System.out.println("Long ----- "+ feature.getLong(i));

                                }else if (type.equals(IFeatureAttributeDescriptor.TYPE_STRING)){

                                        System.out.println("String ----- "+ feature.getString(i));

                                }else if (type.equals(IFeatureAttributeDescriptor.TYPE_GEOMETRY)){

                                        System.out.println("Geometry ----- "+ feature.getGeometry(i));

                                }

                        }

                }

                iter=null;

                featureCollection.dispose();

                try {

                        store.close();

                        store.dispose();

                } catch (CloseException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

}