Application: gvSIG desktop

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

import java.awt.Rectangle;

import java.awt.Shape;

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.util.Iterator;

import org.apache.log4j.Logger;

import org.gvsig.data.DataCollection;

import org.gvsig.data.DataManager;

import org.gvsig.data.DataStoreParameters;

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

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

import org.gvsig.data.spatialprovisionalold.IExtent;

import org.gvsig.data.vectorial.Feature;

import org.gvsig.data.vectorial.FeatureType;

import org.gvsig.fmap.geom.Geometry;

public class PerformanceMeasure {

        public static final int SHP_DRIVER_PSEUDO_NIO = 0;

        public static final int SHP_DRIVER_IN_MEMORY = 1;

        public static final int SHP_DRIVER_JNI_SHAPELIB = 2;

        private Logger logger = null;

        private File shpFile = null;

        private long layerID = 0;

        private Rectangle2D fullExtent = null;

        private IGeometricDataStore store = null;

        private String storeName = "";

        private String name = "";

        public PerformanceMeasure(File f, Logger _logger, String st_name) throws Exception {

                logger = _logger;

                name = f.getName();

                shpFile = f;

                storeName = st_name;

                layerID = System.currentTimeMillis();

                initStore(st_name);

        }

        private void initStore(String store_name) {

                DataManager manager = DataManager.getManager();

                logger.debug("PERFORMANCE_TEST LAYER " + name + " STORE " + store_name + " DATAMANAGER_OBTAINED " + (System.currentTimeMillis() - layerID));

                long t = System.currentTimeMillis();

                SHPStoreParameters shpParameters = null;

                shpParameters = (SHPStoreParameters)

                manager.createDataStoreParameters(store_name);

                shpParameters.setSHPFile(shpFile);

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

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

                shpParameters.setDataStoreName(store_name);

                logger.debug("PERFORMANCE_TEST LAYER " + name + " STORE " + store_name + " DATASTORE_PARAMETERS_SET " + (System.currentTimeMillis() - t));

                t = System.currentTimeMillis();

                try {

                        store = (IGeometricDataStore) manager.createDataStore((DataStoreParameters) shpParameters);

                        store.setOnlyGeometry(true);

                        logger.debug("PERFORMANCE_TEST LAYER " + name + " STORE " + store_name + " DATASTORE_CREATED " + (System.currentTimeMillis() - t));

                        t = System.currentTimeMillis();

                        store.doOpen();

                        logger.debug("PERFORMANCE_TEST LAYER " + name + " STORE " + store_name + " DATASTORE_OPENED " + (System.currentTimeMillis() - t));

                        t = System.currentTimeMillis();

                        IExtent iext = store.getFullExtent();

                        logger.debug("PERFORMANCE_TEST LAYER " + name + " STORE " + store_name + " FULLEXTENT_CALCULATED " + (System.currentTimeMillis() - t));

                        t = System.currentTimeMillis();

                        double xmin = iext.getMin(0);

                        double ymin = iext.getMin(1);

                        double xmax = iext.getMax(0);

                        double ymax = iext.getMax(1);

                        fullExtent = new Rectangle2D.Double(xmin, ymin, xmax - xmin, ymax

                                        - ymin);

                } catch (Exception ex) {

                        logger.error("While initializing layer: " + ex.getMessage());

                }

        }

        public long getId() {

                return layerID;

        }

        public Rectangle2D getFullExtent()  {

                return fullExtent;

        }

        public void drawStore() {

                long draw_start = System.currentTimeMillis();

                FeatureType ft = store.getDefaultFeatureType();

                DataCollection featureCollection = null;

                try {

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

                } catch (Exception e1) {

                        logger.error("While getting feat. collection for drawing: " + e1.getMessage());

                        return;

                }

                Iterator iter = featureCollection.iterator();

                while (iter.hasNext()) { // d

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

                        Geometry geom = (Geometry) feature.getDefaultGeometry();

                        Shape shp = geom.getInternalShape();

                        Rectangle bounds = shp.getBounds();

                }

                iter = null;

                featureCollection.dispose();

                featureCollection = null;

                logger.debug("PERFORMANCE_TEST LAYER " + name + " STORE " + storeName + " DRAWTIME " + (System.currentTimeMillis() - draw_start));

        }

        public boolean isWritable() {

                // TODO Auto-generated method stub

                return false;

        }

        public void wakeUp(boolean driver_is_already_open) {

                // TODO Auto-generated method stub

        }

        public StringBuffer toXML(String name, String proyectPath) {

                // TODO Auto-generated method stub

                return null;

        }

        public void clean() {

                        long t = System.currentTimeMillis();

                        try {

                                store.doClose();

                                store.doDispose();

                                store = null;

                                logger.debug("PERFORMANCE_TEST LAYER " + name + " STORE " + storeName + " CLOSED_DATASTORE " + (System.currentTimeMillis() - t));                        

                        } catch (Exception ex) {

                                logger.error("While cleaning vector layer: " + ex.getMessage());

                        }

        }

}