| gvSIG desktop 1

package org.gvsig.fmap.data.feature.file.shp;

import java.io.File;

import java.util.Iterator;

import junit.framework.TestCase;

import org.gvsig.fmap.data.CloseException;

import org.gvsig.fmap.data.DataManager;

import org.gvsig.fmap.data.InitializeException;

import org.gvsig.fmap.data.ReadException;

import org.gvsig.fmap.data.feature.Feature;

import org.gvsig.fmap.data.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.data.feature.FeatureCollection;

import org.gvsig.fmap.data.feature.FeatureStore;

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

import org.gvsig.fmap.data.feature.file.DataStoreTest;

public class SHPTest extends TestCase {

        public static final File fileNormal = new File(DataStoreTest.class

                        .getResource(

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

        public static final File fileGeometryNull = new File(DataStoreTest.class

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

        public static void main(String[] args) {

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

        }

        protected void setUp() throws Exception {

                super.setUp();

        }

        public void testInitialize() {

                Register.selfRegister();

                shp(fileNormal);

                shp(fileGeometryNull);

        }

        public static SHPStoreParameters getParams(File file)

                        throws InitializeException {

                DataManager manager = DataManager.getManager();

                SHPStoreParameters shpParameters = null;

                shpParameters = (SHPStoreParameters) manager

                                .createDataStoreParameters(SHPStore.DATASTORE_NAME);

                shpParameters.setFile(file);

                return shpParameters;

        }

        private void shp(File file){

                DataManager manager = DataManager.getManager();

                FeatureStore store = null;

                SHPStoreParameters shpParameters=null;

                try {

                        shpParameters = getParams(file);

                } catch (InitializeException e3) {

                        e3.printStackTrace();

                        fail();

                }

                try {

                        store= (FeatureStore)manager.createDataStore(shpParameters);

                } catch (InitializeException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                FeatureType ft=store.getDefaultFeatureType();

                FeatureCollection featureCollection=null;

                try {

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

                } catch (ReadException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                }

                Iterator iter = featureCollection.iterator();

                while (iter.hasNext()) {

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

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

                        Iterator iterator=ft.iterator();

                        while (iterator.hasNext()) {

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

                                int i=descriptor.ordinal();

                                String type=descriptor.getDataType();

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

                                }

                        }

                }

                iter=null;

                featureCollection.dispose();

                try {

                        store.dispose();

                } catch (CloseException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

        public void test_Resources(){

                DataManager manager = DataManager.getManager();

                Register.selfRegister();

                SHPStoreParameters shpParameters=null;

                try {

                        shpParameters = getParams(fileNormal);

                } catch (InitializeException e) {

                        e.printStackTrace();

                        fail();

                }

                DataStoreTest.doFileResourceTest(shpParameters);

        }

}