| gvSIG desktop 1

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

import java.awt.Shape;

import java.io.File;

import java.text.DateFormat;

import java.text.ParseException;

import java.util.Date;

import java.util.Iterator;

import junit.framework.TestCase;

import org.apache.log4j.Logger;

import org.gvsig.data.DataManager;

import org.gvsig.data.IDataCollection;

import org.gvsig.data.IDataStoreParameters;

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

import org.gvsig.data.datastores.vectorial.file.shp_jni.JNISHPStore;

import org.gvsig.data.datastores.vectorial.file.shp_mem.MemSHPStore;

import org.gvsig.data.datastores.vectorial.file.shp_util.IGeometricDataStore;

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

import org.gvsig.data.vectorial.IFeatureType;

import es.prodevelop.geodetic.utils.conversion.GeoUtils;

import es.prodevelop.gvsig.mobile.fmap.core.FGeometry;

import es.prodevelop.gvsig.mobile.fmap.core.FMultiPoint2D;

import es.prodevelop.gvsig.mobile.fmap.core.FPoint2D;

import es.prodevelop.gvsig.mobile.fmap.core.FPolygon2D;

import es.prodevelop.gvsig.mobile.fmap.core.FPolyline2D;

import es.prodevelop.gvsig.mobile.fmap.util.ResourceReader;

import es.prodevelop.gvsig.mobile.fmap.util.Utils;

import es.prodevelop.gvsig.mobile.fmap.util.graph2d.Graph2DUtilities;

/**

 * This class tests the three SHP drivers implemented for gvSIG Mobile.

 * These drivers are compatible with the new data access model (june 2008)

 * and will share a lot of code with gvsig Desktop.

 * 

 * The right testdata folder is needed.

 * 

 *  

 * @author jldominguez

 *

 */

public class SHPTest extends TestCase {

        private static Logger logger = Logger.getLogger(SHPTest.class);

        /**

         * Test IDs

         */

        public static final int TEST_PUNTO = 0; 

        public static final int TEST_MPUNTO = 1; 

        public static final int TEST_LINEA = 2; 

        public static final int TEST_POLIGONO = 3; 

        public static final int TEST_NULL = 4; 

        public static final int TEST_ATT = 5; 

        private File filePunto, fileMPunto, fileLinea, filePoligono, fileNull, fileAtt;

        public static void main(String[] args) {

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

        }

        protected void setUp() throws Exception {

                super.setUp();

        }

        public void testInitialize() {

                if (Utils.USING_PDA) {

                        filePunto = ResourceReader.getPdaTestFile(PerformanceTest.SD_FOLDER, "testdata", "puntos_grande.shp");

                        fileMPunto = ResourceReader.getPdaTestFile(PerformanceTest.SD_FOLDER, "testdata", "multipuntos_grande.shp");

                        fileLinea = ResourceReader.getPdaTestFile(PerformanceTest.SD_FOLDER, "testdata", "lineas_grande.shp");

                        filePoligono = ResourceReader.getPdaTestFile(PerformanceTest.SD_FOLDER, "testdata", "poligonos_grande.shp");

                        fileNull = ResourceReader.getPdaTestFile(PerformanceTest.SD_FOLDER, "testdata", "pruebaNull.shp");

                        fileAtt = ResourceReader.getPdaTestFile(PerformanceTest.SD_FOLDER, "testdata", "atttest.shp");

                } else {

                        filePunto = ResourceReader.getResourceFile("testdata", "puntos_grande.shp");

                        fileMPunto = ResourceReader.getResourceFile("testdata", "multipuntos_grande.shp");

                        fileLinea = ResourceReader.getResourceFile("testdata", "lineas_grande.shp");

                        filePoligono = ResourceReader.getResourceFile("testdata", "poligonos_grande.shp");

                        fileNull = ResourceReader.getResourceFile("testdata", "pruebaNull.shp");

                        fileAtt = ResourceReader.getResourceFile("testdata", "atttest.shp");

                }

                new SHPStore();

                doTest(SHPStore.DATASTORE_NAME);

                new MemSHPStore();

                doTest(MemSHPStore.DATASTORE_NAME);

                new JNISHPStore();

                doTest(JNISHPStore.DATASTORE_NAME);

        }

        /**

         * Performs a test with a given data sorce name

         * 

         * @see SHPStore

         * @see MemSHPStore

         * @see JNISHPStore

         * 

         * @param ds_name

         */

        public void doTest(String ds_name) {

                logger.debug("DATASOURCE: " + ds_name + ", FILE = " + filePunto.getName() + "...");

                shp(filePunto, TEST_PUNTO, ds_name);

                if (ds_name.compareTo(JNISHPStore.DATASTORE_NAME) != 0) {

                        logger.debug("DATASOURCE: " + ds_name + ", FILE = " + fileMPunto.getName() + "...");

                        shp(fileMPunto, TEST_MPUNTO, ds_name);

                }

                logger.debug("DATASOURCE: " + ds_name + ", FILE = " + fileLinea.getName() + "...");

                shp(fileLinea, TEST_LINEA, ds_name);

                logger.debug("DATASOURCE: " + ds_name + ", FILE = " + filePoligono.getName() + "...");

                shp(filePoligono, TEST_POLIGONO, ds_name);

                logger.debug("DATASOURCE: " + ds_name + ", FILE = " + fileNull.getName() + "...");

                shp(fileNull, TEST_NULL, ds_name);

                logger.debug("DATASOURCE: " + ds_name + ", FILE = " + fileAtt.getName() + "...");

                shp(fileAtt, TEST_ATT, ds_name);

                // shp(fileGeometryNull);

        }

        private void shp(File file, int fileid, String datastore_name) {

                DataManager manager = DataManager.getManager();

                IGeometricDataStore store = null;

                SHPStoreParameters shpParameters=null;

                shpParameters =(SHPStoreParameters) manager.createDataStoreParameters(datastore_name);

                shpParameters.setFile(file);

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

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

                try {

                        store = (IGeometricDataStore)

                        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();

                IDataCollection dataCollection = null;

                try {

                        dataCollection = store.getDataCollection(ft,null,null);

                } catch (ReadException e1) {

                        e1.printStackTrace();

                }

                Iterator iter = dataCollection.iterator();

                int feature_count = 0;

                while (iter.hasNext()) {

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

                        boolean simple = (datastore_name.compareTo(JNISHPStore.DATASTORE_NAME) == 0);

                        check(feature, ft, fileid, simple);

                        feature_count++;

                }

                iter=null;

                dataCollection.dispose();

                try {

                        store.close();

                        store.dispose();

                } catch (CloseException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

        private void check(IFeature feature, IFeatureType att_desc, int fileid, boolean simpletypes) {

                double expected = 0;

                switch (fileid) {

                case TEST_PUNTO:

                        expected = feature.getDouble("X");

                        FPoint2D p =  (FPoint2D) ((FGeometry) feature.getDefaultGeometry()).getInternalShape();

                        checkPoint(feature, p, expected, simpletypes);

                        break;

                case TEST_MPUNTO:

                        checkMPoint(feature, simpletypes);

                        break;

                case TEST_LINEA:

                        expected = feature.getDouble("LEN");

                        FPolyline2D pl =  (FPolyline2D) ((FGeometry) feature.getDefaultGeometry()).getInternalShape();

                        checkLineLength(feature, pl, expected, simpletypes);

                        break;

                case TEST_POLIGONO:

                        expected = feature.getDouble("AREA");

                        FPolygon2D pg =  (FPolygon2D) ((FGeometry) feature.getDefaultGeometry()).getInternalShape();

                        checkPolygonArea(feature, pg, expected, simpletypes);

                        break;

                case TEST_NULL:

                        checkNullGeometry(feature);

                        break;

                case TEST_ATT:

                        checkAttributes(feature, simpletypes);

                        break;

                }

        }

        private void checkAttributes(IFeature feat, boolean simpl) {

                if (! simpl) {

                        Date d = feat.getDate("ATT_DATE");

                        String str = d.toString().toLowerCase();

                        boolean boo =

                                ((str.indexOf("9999") != -1)

                                        && (

                                                        (str.indexOf("12") != -1)

                                                        || (str.indexOf("dec") != -1)

                                                        || (str.indexOf("dic") != -1))

                                        && (str.indexOf("31") != -1));

                        assertTrue("Found a date different to 9999 12 31", boo);

                }

                String str = feat.getString("ATT_STRING");

                assertEquals(str, "DFGDFGDFG");

                double real = feat.getDouble("ATT_REAL");

                boolean boo = ((real == 4.4444) || (real == 4));

                assertTrue("REAL = " + real + ", EXP = 4, 4.4444", boo);

                boolean bool = feat.getBoolean("ATT_BOOLEA");

                assertTrue("EXP = FALSE, RET = " + (!bool), !bool);

                if (simpl) {

                        double d = feat.getDouble("ATT_INTEGE");

                        assertTrue("REAL = " + d + ", EXP = 44444", d == 44444.0);

                } else {

                        int integ = feat.getInt("ATT_INTEGE");

                        assertTrue("REAL = " + integ + ", EXP = 44444", integ == 44444);

                }

        }

        private void checkPolygonArea(IFeature feature, FPolygon2D poly, double expected, boolean simpl) {

                int id = 0;

                if (simpl) {

                        id = (int) Math.round(feature.getDouble("GUIA_ID"));

                } else {

                        id = feature.getInt("GUIA_ID");

                }

                double ar = Graph2DUtilities.getPathIteratorArea(poly.getPathIterator(null));

                assertTrue("AREA = " + ar + ", EXP = " + expected, acceptablySimilar(expected, ar, 0.1));

                switch (id) {

                case 174996:

                        assertTrue("AREA = " + ar + ", EXP = " + 357.7, acceptablySimilar(ar, 357.7, 0.1));

                        break;

                case 569303133:

                        assertTrue("AREA = " + ar + ", EXP = " + 409.46, acceptablySimilar(ar, 409.46, 0.1));

                        break;

                case 1229222137:

                        assertTrue("AREA = " + ar + ", EXP = " + 49.285, acceptablySimilar(ar, 49.285, 0.1));

                        break;

                case 1795370359:

                        assertTrue("AREA = " + ar + ", EXP = " + 98.165, acceptablySimilar(ar, 98.165, 0.1));

                        break;

                }

        }

        private void checkLineLength(IFeature feature, FPolyline2D pl, double expected, boolean simpl) {

                int color = 0;

                int eleva = 0;

                if (simpl) {

                        color = (int) Math.round(feature.getDouble("COLOR"));

                        eleva = (int) Math.round(feature.getDouble("ELEVACION"));

                } else {

                        color = feature.getInt("COLOR");

                        eleva = feature.getInt("ELEVACION");

                }

                double len = GeoUtils.getIteratorLengthAndLast(pl.getPathIterator(null))[0];

                assertTrue("Not good: " + len  + " and " + expected, acceptablySimilar(expected, len, 0.05));

                switch (color) {

                case 888830465:

                        assertEquals(eleva, -1519669334);

                        break;

                case 135691458:

                        assertEquals(eleva, -1594772521);

                        break;

                case 631009463:

                        assertEquals(eleva, -409958687);

                        break;

                case 601164227:

                        assertEquals(eleva, -1632831996);

                        break;

                case 1766834083:

                        assertEquals(eleva, -671796992);

                        break;

                }

        }

        private void checkMPoint(IFeature feature, boolean simpl) {

                int tipocont = 0;

                if (simpl) {

                        tipocont = (int) Math.round(feature.getDouble("TIPOCONT"));

                } else {

                        tipocont = feature.getInt("GUIA_ID");

                }

                String nomb = feature.getString("FIRST_NOMB");

                FMultiPoint2D mp = (FMultiPoint2D) feature.getDefaultGeometry(); 

                switch (tipocont) {

                case 116:

                        assertEquals(nomb, "MARIN (EL)");

                        assertTrue("Checking presence of coordinate X: ", oneOfXInMPoint(mp, 272463.83));

                        break;

                case 118:

                        assertEquals(nomb, "SALAMANCA");

                        assertTrue("Checking presence of coordinate X: ", oneOfXInMPoint(mp, 276024.43));

                        break;

                case 15035:

                        assertEquals(nomb, "ORBISO");

                        assertTrue("Checking presence of coordinate X: ", oneOfXInMPoint(mp, 554478.48));

                        break;

                case 19586:

                        assertEquals(nomb, "VALDEZORRAS");

                        assertTrue("Checking presence of coordinate X: ", oneOfXInMPoint(mp, 241343.25));

                        break;

                case 30000:

                         assertEquals(nomb, "ORZOLA");

                         assertTrue("Checking presence of coordinate X: ", oneOfXInMPoint(mp, -619386.36));

                        break;

                }

        }

        private boolean oneOfXInMPoint(Shape pointorpoints, double d) {

                if (pointorpoints instanceof FMultiPoint2D) {

                        FMultiPoint2D mp = (FMultiPoint2D) pointorpoints;

                        int np = mp.getNumPoints();

                        for (int i=0; i<np; i++) {

                                double itemx = mp.getPoint(i).getX();

                                if (acceptablySimilar(d, itemx, 0.01)) return true; 

                        }

                        return false;

                } else {

                        FPoint2D p = (FPoint2D) pointorpoints;

                        return acceptablySimilar(d, p.getX(), 0.01); 

                }

        }

        private void checkNullGeometry(IFeature feature) {

                Object obj = feature.getDefaultGeometry();

                assertNotNull("Checking not NULL geometry: ", obj);

        }

        private void checkPoint(IFeature feature, FPoint2D point, double expected, boolean simpl) {

                int id = 0;

                if (simpl) {

                        id = (int) Math.round(feature.getDouble("NUMEROENT"));

                } else {

                        id = feature.getInt("NUMEROENT");

                }

                String nomb = feature.getString("NOMBRE");

                assertTrue(acceptablySimilar(expected, point.getX(), 0.01));

                switch (id) {

                case 58:

                        assertEquals(nomb, "MARIN (EL)");

                        break;

                case 59:

                        assertEquals(nomb, "SALAMANCA");

                        break;

                case 2825:

                        assertEquals(nomb, "NAVAHONDILLA");

                        break;

                case 6474:

                        assertEquals(nomb, "CENTENERA");

                        break;

                case 27761:

                        assertEquals(nomb, "EVAN DE ARRIBA");

                        break;

                }

        }

        private boolean acceptablySimilar(double expec, double area, double tol) {

                double diff = Math.abs(expec - area);

                return ((diff / Math.abs((expec + area) / 2)) < tol);

        }

//        public void test_Resources(){

//

//                DataManager manager = DataManager.getManager();

//                Register.selfRegister();

//

//                SHPStoreParameters shpParameters;

//

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

//                shpParameters.setFile(fileNormal);

//

//

//                DataStoreTest.doFileResourceTest(shpParameters);

//        }

}