Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 */

package org.gvsig.raster.tools.app.basic.tool.roi;

import java.awt.Image;

import java.io.File;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.LinkedHashMap;

import java.util.List;

import javax.swing.JComboBox;

import javax.swing.table.DefaultTableModel;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.ui.mdiManager.IWindow;

import org.gvsig.app.project.documents.view.gui.AbstractViewPanel;

import org.gvsig.app.project.documents.view.toolListeners.StatusBarListener;

import org.gvsig.fmap.dal.coverage.exception.GridException;

import org.gvsig.fmap.dal.coverage.exception.ROIException;

import org.gvsig.fmap.dal.coverage.grid.Grid;

import org.gvsig.fmap.dal.coverage.store.RasterDataStore;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.EditableFeature;

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontext.layers.vectorial.GraphicLayer;

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol;

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.tools.Behavior.Behavior;

import org.gvsig.fmap.mapcontrol.tools.Behavior.MouseMovementBehavior;

import org.gvsig.fmap.mapcontrol.tools.Behavior.PointBehavior;

import org.gvsig.fmap.mapcontrol.tools.Behavior.PolygonBehavior;

import org.gvsig.fmap.mapcontrol.tools.Behavior.PolylineBehavior;

import org.gvsig.gui.beans.table.TableContainer;

import org.gvsig.gui.beans.table.exceptions.NotInitializeException;

import org.gvsig.gui.beans.table.models.IModel;

import org.gvsig.metadata.exceptions.MetadataException;

import org.gvsig.raster.fmap.layers.FLyrRaster;

import org.gvsig.raster.roi.ROI;

import org.gvsig.raster.roi.VectorialROI;

import org.gvsig.raster.swing.roi.ROIPanel;

import org.gvsig.raster.swing.roi.ROIPanelDataModel;

import org.gvsig.raster.tools.app.basic.RasterToolsUtil;

import org.gvsig.raster.tools.app.basic.tool.roi.ui.listener.DrawMouseViewListener;

/**

 * Data model for ROIs panel

 * 

 * @author Nacho Brodin nachobrodin@gmail.com

 */

public class ROIDataModel implements ROIPanelDataModel {

        private FLayer                      fLayer             = null;

        private MapControl                  mapControl         = null;

        private AbstractViewPanel           view               = null;

        private GraphicLayer                graphicLayer       = null;

        private String                      previousTool       = null;

        private MapContextManager           mapContextManager  = MapContextLocator.getMapContextManager();

        private ROIFeatureManager           roiFeatureManager  = null;

        private Image                       curImage           = null;

        private LinkedHashMap<String, ROI>  rois               = null;

        private ROIPanel                    panel              = null;

        private Grid                        grid               = null;

        private List<FLyrVect>              fLyrsList          = null;

        private File                        roiFile            = null;

        public class LayerElement {

                private FLyrVect     lyr  = null;

                public LayerElement(FLyrVect lyr) {

                        this.lyr = lyr;

                }

                public FLyrVect getLayer() {

                        return lyr;

                }

                public String toString() {

                        return lyr.getName();

                }

        }

        public ROIDataModel(List<FLyrVect> vectLayers) {

                this.fLyrsList = vectLayers;

        }

        public void setROIPanel(ROIPanel panel) {

                this.panel = panel;

                JComboBox list = panel.getLayerList(ROIPanel.FLYRS_COMBO);

                list.removeAllItems();

                for (int i = 0; i < fLyrsList.size(); i++) {

                        list.addItem(new LayerElement(fLyrsList.get(i)));

                }

        }

        public ROIPanel getROIPanel() {

                return panel;

        }

        public RasterDataStore getRasterDataStore() {

                return ((FLyrRaster)fLayer).getDataStore();

        }

        /**

         * Sets the layer 

         * @param layer

         * @throws GridException 

         */

        public void setLayer(FLayer layer) throws GridException {

                fLayer = layer;

                if (view == null) {

                        IWindow[] list = PluginServices.getMDIManager().getAllWindows();

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

                                if (list[i] instanceof AbstractViewPanel)

                                        view = (AbstractViewPanel) list[i];

                        }

                        if (view == null)

                                return;

                        mapControl = view.getMapControl();

                        graphicLayer = view.getMapControl().getMapContext().getGraphicsLayer();

                        /*

                         * Guardar la herramienta actual para recuperarla m?s tarde, por

                         * ejemplo al cerrar el contenedor del este panel.

                         */

                        previousTool = mapControl.getCurrentTool();

                        // Listener de eventos de movimiento que pone las coordenadas del

                        // rat?n en la barra de estado

                        StatusBarListener sbl = new StatusBarListener(mapControl);

                        DrawMouseViewListener drawMouseViewListener = new DrawMouseViewListener(this);

                        mapControl.addBehavior("drawPolygonROI", new Behavior[] {

                                        new PolygonBehavior(drawMouseViewListener),

                                        new MouseMovementBehavior(sbl) });

                        mapControl.addBehavior("drawLineROI", new Behavior[] {

                                        new PolylineBehavior(drawMouseViewListener),

                                        new MouseMovementBehavior(sbl) });

                        mapControl.addBehavior("drawPointROI", new Behavior[] {

                                        new PointBehavior(drawMouseViewListener),

                                        new MouseMovementBehavior(sbl) });

                }

                try {

                        roiFeatureManager = new ROIFeatureManager(fLayer.getMetadataID().toString());

                } catch (MetadataException e) {

                        throw new GridException("Can't create ROI",e); 

                }

                clearROIs();

                loadROIs();

        }

        /**

         * Elimina todas las filas de la tabla de rois as? como la ROI y objetos

         * FGraphic asociados (borr?ndolos tambi?n de la vista) a cada una.

         * 

         */

        public void clearROIs() {

                try {

                        panel.getTable().removeAllRows();

                } catch (NotInitializeException e) {

                        RasterToolsUtil.messageBoxError("error_tabla_rois", this, e);

                }

                boolean repaint = roiFeatureManager.clear(graphicLayer);

                if (repaint)

                        getMapControl().rePaintDirtyLayers();

                getRois().clear();

        }

        private LinkedHashMap<String, ROI> getRois() {

                if (rois == null)

                        rois = new LinkedHashMap<String, ROI>();

                return rois;

        }

        /**

         * Cargar las ROIs asociadas a la capa, si las hay.

         * 

         * @throws GridException

         * 

         */

        private void loadROIs() throws GridException {

                List<ROI> roisArray = null;

                try {

                        roisArray = ((FLyrRaster) fLayer).getRois();

                } catch (ROIException e) {

                        RasterToolsUtil.messageBoxError("problems_loading_rois", this, e);

                }

                loadROIs(roisArray);

        }

        @SuppressWarnings("deprecation")

        private void loadROIs(List<ROI> roisArray) throws GridException {

                if (roisArray != null) {

                        ISymbol symbol = null;

                        int nPoints, nPolygons, nLines;

                        FeatureStore store = graphicLayer.getFeatureStore();

                        try {

                                store.edit();

                                for (Iterator<ROI> iter = roisArray.iterator(); iter.hasNext();) {

                                        ROI roi = (ROI) iter.next();

                                        nPoints = 0;

                                        nPolygons = 0;

                                        nLines = 0;

                                        Object row[] = ((IModel) panel.getTable().getModel()).getNewLine();

                                        row[0] = roi.getName();

                                        row[4] = roi.getColor();

                                        addROI(roi);

                                        VectorialROI vectorialROI = (VectorialROI) roi;

                                        for (Iterator<Geometry> iterator = vectorialROI.getGeometries().iterator(); iterator.hasNext();) {

                                                Geometry geometry = (Geometry) iterator.next();

                                                symbol = mapContextManager.getSymbolManager().createSymbol(geometry.getType(), roi.getColor());

                                                switch (geometry.getType()) {

                                                case Geometry.TYPES.POINT:

                                                        nPoints++;

                                                        break;

                                                case Geometry.TYPES.SURFACE:

                                                        nPolygons++;

                                                        break;

                                                case Geometry.TYPES.CURVE:

                                                        nLines++;

                                                        break;

                                                }

                                                EditableFeature feature = roiFeatureManager.addROIFeature(vectorialROI, graphicLayer, geometry, symbol);

                                                store.insert(feature);

                                        }

                                        row[1] = new Integer(nPolygons);

                                        row[2] = new Integer(nLines);

                                        row[3] = new Integer(nPoints);

                                        ((DefaultTableModel)panel.getTable().getModel()).addRow(row);

                                }

                                store.finishEditing();

                        } catch (DataException e) {

                                throw new GridException("Error loading ROIs from layer " + fLayer.getName(), e);

                        }

                        selectDrawRoiTool();

                        getMapControl().drawGraphics();

                }

        }

        public void addROI(ROI roi) {

                getRois().put(roi.getName(), roi);

        }

        public void selectDrawRoiTool() {

                if (mapControl != null)

                        if (panel.getButton(ROIPanel.POLYGON_BUTTON).isSelected()) {

                                Image img = IconThemeHelper.getImageIcon("polygon-cursor").getImage();

                                curImage = img;

                                mapControl.setTool("drawPolygonROI");

                        } else if (panel.getButton(ROIPanel.LINE_BUTTON).isSelected()) {

                                Image img = IconThemeHelper.getImageIcon("line-cursor").getImage();

                                curImage = img;

                                mapControl.setTool("drawLineROI");

                        } else if (panel.getButton(ROIPanel.POINT_BUTTON).isSelected()) {

                                Image img = IconThemeHelper.getImageIcon("point-cursor").getImage();

                                curImage = img;

                                mapControl.setTool("drawPointROI");

                        }

        }

        public void changeRoiName(String currentName, String newName)

        throws DataException {

                getROI(currentName).setName(newName);

                getRois().put(newName, getRois().remove(currentName));

                roiFeatureManager.changeROIName(currentName, newName, graphicLayer);

        }

        /**

         * Elimina la ROI correspondiente al nombre <code>roiName</code> as? como

         * sus objetos FGraphic asociados (borr?ndolos tambi?n de la vista). (NO

         * elimina la fila correspondiente en la tabla del panel).

         * 

         * @param roiName

         *            Nombre de la ROI a eliminar.

         */

        public void removeROI(String roiName) {

                getRois().remove(roiName);

                boolean repaint = roiFeatureManager.removeROI(roiName, graphicLayer);

                if (repaint) {

                        getMapControl().rePaintDirtyLayers();

                }

        }

        public Grid getGrid() {

                if(grid == null) {

                        FLyrRaster rasterLayer = (FLyrRaster) getFLayer();

                        try {

                                grid = rasterLayer.getReadOnlyFullGrid(false);

                        } catch (GridException e) {

                                return null;

                        } catch (InterruptedException e) {

                                return null;

                        }

                }                

                return grid;

        }

        /**

         * Borra de capa GraphicsLayer de mapContext los gr?ficos pertenecientes a

         * las rois del panel. Si la GraphicsLayer se queda sin gr?ficos se eliminan

         * tambien los s?mbolos de esta.

         */

        public void clearRoiGraphics() {

                roiFeatureManager.clearFeaturesFromGraphicLayer(graphicLayer);

        }

        public MapControl getMapControl() {

                return mapControl;

        }

        public TableContainer getTable() {

                return panel.getTable();

        }

        public ROI getROI(String roiName) {

                return (ROI) getRois().get(roiName);

        }

        public GraphicLayer getGraphicLayer() {

                return graphicLayer;

        }

        public void setPreviousTool() {

                if (previousTool != null)

                        getMapControl().setTool(previousTool);

        }

        public List<ROI> getROIs() {

                return new ArrayList<ROI>(getRois().values());

        }

        public void setROIs(List<ROI> roisArray) throws GridException {

                loadROIs(roisArray);

        }

        public FLayer getFLayer() {

                return fLayer;

        }

        public ROIFeatureManager getRoiFeatureManager() {

                return roiFeatureManager;

        }

        public Image getToolImage() {

                return curImage;

        }

        public void setROIFile(File file) {

                this.roiFile = file;

        }

        public File getROIFile() {

                return this.roiFile;

        }

}