Application: gvSIG desktop

package com.iver.gvsig;

import java.io.File;

import java.io.IOException;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

import com.iver.andami.PluginServices;

import com.iver.andami.plugins.Extension;

import com.iver.cit.gvsig.fmap.DriverException;

import com.iver.cit.gvsig.fmap.FMap;

import com.iver.cit.gvsig.fmap.ViewPort;

import com.iver.cit.gvsig.fmap.core.DefaultFeature;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.core.IFeature;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.cit.gvsig.fmap.drivers.DriverIOException;

import com.iver.cit.gvsig.fmap.drivers.FieldDescription;

import com.iver.cit.gvsig.fmap.drivers.SHPLayerDefinition;

import com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver;

import com.iver.cit.gvsig.fmap.drivers.shp.IndexedShpDriver;

import com.iver.cit.gvsig.fmap.edition.DefaultRowEdited;

import com.iver.cit.gvsig.fmap.edition.EditionException;

import com.iver.cit.gvsig.fmap.edition.IRowEdited;

import com.iver.cit.gvsig.fmap.edition.writers.shp.ShpWriter;

import com.iver.cit.gvsig.fmap.layers.FLayer;

import com.iver.cit.gvsig.fmap.layers.FLayers;

import com.iver.cit.gvsig.fmap.layers.FLyrVect;

import com.iver.cit.gvsig.fmap.layers.ReadableVectorial;

import com.iver.cit.gvsig.gui.View;

import com.iver.cit.gvsig.project.ProjectView;

import com.iver.gvsig.measure.Operations;

/**

 * Extension responsible for calculating the perimeter of the geometries selected,

 * only when these they be of type line.

 * When there is not an active layer, or some of the geometries selected be not of type line

 * this tool will remain in not visible way.

 *

 * @author Vicente Caballero Navarro

 */

public class PerimeterExtension extends Extension {

        private FMap map;

        private FLyrVect lv;

        /**

         * @see com.iver.andami.plugins.IExtension#initialize()

         */

        public void initialize() {

        }

        /**

         * @see com.iver.andami.plugins.IExtension#execute(java.lang.String)

         */

        public void execute(String actionCommand) {

                ViewPort vp=map.getViewPort();

                Operations operations=new Operations();

                ShpWriter writer=((IndexedShpDriver)lv.getSource().getDriver()).getShpWriter();

                SHPLayerDefinition lyrDef;

                try {

                        lyrDef = Operations.createLayerDefinition(lv);

                        FieldDescription[] fD=lyrDef.getFieldsDesc();

                        FieldDescription[] newFD = operations.getPerimeterFields(fD);

                        lyrDef.setFieldsDesc(newFD);

                        ReadableVectorial adapter = lv.getSource();

                        File newFile = ((VectorialFileDriver)adapter.getDriver()).getFile();

                        writer.setFile(newFile);

                        writer.initialize(lyrDef);

                        writer.preProcess();

                        for (int i=0;i<adapter.getShapeCount();i++){

                                IFeature feat=adapter.getFeature(i);

                                IGeometry geom=feat.getGeometry().cloneGeometry();

                                Value[] values=feat.getAttributes();

                                Value[] newValues=new Value[values.length];

                                for (int j=0;j<values.length-1;j++){

                                        newValues[j]=values[j];

                                }

                                double perimeter=operations.getPerimeter(geom,vp);

                                newValues[values.length-1]=ValueFactory.createValue(perimeter);

                                DefaultFeature df=new DefaultFeature(geom,newValues);

                                IRowEdited edRow = new DefaultRowEdited(df, IRowEdited.STATUS_ADDED, i);

                                writer.process(edRow);

                        }

                        writer.postProcess();

                        adapter.getDriver().reLoad();

                } catch (EditionException e) {

                        e.printStackTrace();

                } catch (DriverException e1) {

                        e1.printStackTrace();

                } catch (com.hardcode.gdbms.engine.data.driver.DriverException e1) {

                        e1.printStackTrace();

                } catch (DriverIOException e) {

                        e.printStackTrace();

                } catch (IOException e) {

                        e.printStackTrace();

                }

        }

        /**

         * @see com.iver.andami.plugins.IExtension#isEnabled()

         */

        public boolean isEnabled() {

                return true;

        }

        /**

         * @see com.iver.andami.plugins.IExtension#isVisible()

         */

        public boolean isVisible() {

                com.iver.andami.ui.mdiManager.View f = PluginServices.getMDIManager()

                                .getActiveView();

                if (f == null) {

                        return false;

                }

                if (f instanceof View) {

                        View vista = (View) f;

                        ProjectView model = vista.getModel();

                        map = model.getMapContext();

                        FLayers layers = map.getLayers();

                        FLayer[] layersActives = layers.getActives();

                        if (layersActives.length == 1) {

                                if (layersActives[0] instanceof FLyrVect) {

                                        lv = (FLyrVect) layersActives[0];

                                        try {

                                                int type = lv.getShapeType();

                                                if (FShape.ARC == type || FShape.LINE == type) {

                                                        return true;

                                                }

                                        } catch (DriverException e) {

                                                return false;

                                        }

                                }

                        }

                }

                return false;

        }

}