Application: gvSIG desktop

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

import java.io.File;

import java.util.Date;

import junit.framework.TestCase;

import org.apache.log4j.FileAppender;

import org.apache.log4j.Level;

import org.apache.log4j.Logger;

import org.apache.log4j.PatternLayout;

import org.gvsig.compatible.StringUtil;

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

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

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

import es.prodevelop.gvsig.mobile.fmap.driver.vect.shp.ShpReader;

//import es.prodevelop.gvsig.mobile.fmap.layer.newmodel.FPerformanceTestVectorLayer;

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

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

//import es.prodevelop.gvsig.mobile.fmap.util.string.StringUtilities;

/**

 * This class tests the performance of 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 PerformanceTest extends TestCase {

        private static Logger logger;

        public static String SD_FOLDER = "Tarjeta SD";

        private static String resourceDir = "";

        static {

                String base = System.getProperty("java.class.path");

                int ind = base.indexOf(File.pathSeparator);

                if (ind != -1) base = base.substring(0, ind);

                ind = base.lastIndexOf(File.separator);

                base = base.substring(0, ind);

                //logger.debug("Base directory: " + base);

                ind = base.lastIndexOf(File.separator);

                resourceDir = base.substring(0, ind) + File.separator + "resources";

        }

        /**

         * Number of sample executions

         */

        public static final int REPEAT_TIMES = 5;

        public static final String[] SIZE_TEST_BASE_FILE_NAMES = {

                "point_", "polygon_" };

        public static final int[] SIZE_TEST_FILE_SIZES = {

                10,

                200,

                1000,

                8000,

                40000,

                200000 };

        public static final int[] SIZE_TEST_FILE_SIZES_JNI = {

                10,

                200,

                1000,

                8000,

                40000 };

        public static final int[] SIZE_TEST_FILE_SIZES_MEM = {

                10,

                200,

                1000,

                8000 };

        public static final String[] FEATURE_SIZE_TEST_FEATURE_TYPES = {

                "few", "medium", "many" };

        public static final String FEATURE_SIZE_TEST_BASE_FILE_NAME = "VerticesPerFeature.shp";

        public PerformanceTest() {

                ShpReader.touch();

                initLogger();

        }

        public static void main(String[] args) {

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

        }

        protected void setUp() throws Exception {

                super.setUp();

        }

        public void testInitialize() {

                new SHPStore();

                doit(SHPStore.DATASTORE_NAME);

                new MemSHPStore();

                doit(MemSHPStore.DATASTORE_NAME);

                new JNISHPStore();

                doit(JNISHPStore.DATASTORE_NAME);

                logger.debug("FIN.");

        }

        private void doit(String storename) {

                File shpFile = null;

                int[] sizes = SIZE_TEST_FILE_SIZES;

                if (storename.compareTo(JNISHPStore.DATASTORE_NAME) == 0) {

                        sizes = SIZE_TEST_FILE_SIZES_JNI;

                } else {

                        if (storename.compareTo(MemSHPStore.DATASTORE_NAME) == 0) {

                                sizes = SIZE_TEST_FILE_SIZES_MEM;

                        }

                }

                // ---------- size test

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

                        int size = sizes[i];

                        for (int j=0; j<SIZE_TEST_BASE_FILE_NAMES.length; j++) {

                                String base_name = SIZE_TEST_BASE_FILE_NAMES[j];

                                String filename = base_name + size + ".shp";

                                if (Utils.USING_PDA) {

                                        shpFile = getPdaTestFile(SD_FOLDER, "testdata", filename);

                                } else {

                                        shpFile = getResourceFile("testdata", filename);

                                }

                                performance(shpFile, storename);

                        }

                }

                // ---------- feature size test

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

                        String feat_type = FEATURE_SIZE_TEST_FEATURE_TYPES[i];

                        String filename = feat_type + FEATURE_SIZE_TEST_BASE_FILE_NAME;

                        if (Utils.USING_PDA) {

                                shpFile = getPdaTestFile(SD_FOLDER, "testdata", filename);

                        } else {

                                shpFile = getResourceFile("testdata", filename);

                        }

                        performance(shpFile, storename);

                }

        }

        private void initLogger() {

                Date now = new Date(System.currentTimeMillis());

                String nowstr = now.toString();

                nowstr = StringUtil.replaceAllString(nowstr, " ", "_");

                nowstr = StringUtil.replaceAllString(nowstr, ":", "_") + ".txt";

                String outlogpath = "";

                if (Utils.USING_PDA) {

                        outlogpath = getPdaTestFile(SD_FOLDER, "testdata", "performance_test_" + nowstr).getAbsolutePath();

                } else {

                        outlogpath = getResourceFile("testdata", "performance_test_" + nowstr).getAbsolutePath();

                }

                System.out.println("Log file: " + outlogpath);

                try {

                    PatternLayout l = new PatternLayout("%5p [%t] - %m%n");

                    FileAppender fa = new FileAppender(l, outlogpath, false);

                    // ConsoleAppender ca = new ConsoleAppender(l);

                    Logger.getRootLogger().setLevel(Level.DEBUG);

                    Logger.getRootLogger().addAppender(fa);

                    // Logger.getRootLogger().addAppender(ca);

                } catch (Exception ex) {

                        System.err.println("Error while initializing log4j: " + ex.getMessage());

                }

                logger = Logger.getLogger(PerformanceTest.class);

        }

        private void performance(File file, String st_name) {

                if (!file.exists()) return;

                PerformanceMeasure lyr = null;

                try {

                        lyr = new PerformanceMeasure(file, logger, st_name);

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

                                // just to see the progress bar

                                assertTrue(true);

                                lyr.drawStore();

                        }

                } catch (Exception e) {

                        logger.debug("PERFORMANCE_TEST LAYER " + file.getName() + " STORE " + st_name + " ERROR " + e.getMessage());

                }

                if (lyr != null) lyr.clean();

                System.gc();

                lyr = null;

                System.gc();

        }

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

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

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

        }

        /**

         * Gets a file from the resource folder 

         * @param dir

         * @param file_name

         * @return the file object

         */

        public static File getResourceFile(String dir, String file_name) {

                String filePath = resourceDir + File.separator + dir + File.separator + file_name;

                File resp = new File(filePath);

                return resp;

        }

        /**

         * Gets a file from the pda sd card folder 

         * @param dir

         * @param file_name

         * @return the file object

         */

        public static File getPdaTestFile(String sdfolder, String dir, String file_name) {

                String filePath = File.separator + sdfolder + File.separator + dir + File.separator + file_name;

                File resp = new File(filePath);

                return resp;

        }        

}