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

import java.awt.BorderLayout;

import java.awt.Color;

import java.awt.FlowLayout;

import java.awt.Image;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.LinkedHashMap;

import java.util.List;

import javax.swing.JButton;

import javax.swing.JPanel;

import javax.swing.JToggleButton;

import javax.swing.border.EmptyBorder;

import javax.swing.border.LineBorder;

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.grid.Grid;

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

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.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.fmap.roi.VectorialROI;

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

import org.gvsig.raster.tools.app.basic.tool.roi.ROIFeatureManager;

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

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

/**

 * Panel con la tabla de ROIs.

 * 

 * @author Diego Guerrero Sevilla (diego.guerrero@uclm.es)

 * 

 */

public class ROIsTablePanel extends JPanel {

        private static final long          serialVersionUID    = 2138437448356198224L;

        private ROIsManagerPanel            managerPanel       = null;

        private ROIsTablePanelListener      listener           = null;

        private JPanel                      controlPanel       = null;

        private JPanel                      tableButtonsPanel  = null;

        private JPanel                      toolsPanel         = null;

        private TableContainer              tableContainer     = null;

        private JButton                     newButton          = null;

        private JButton                     deleteButton       = null;

        private JButton                     exportButton       = null;

        private JButton                     importButton       = null;

        private JToggleButton               pointToolButton    = null;

        private JToggleButton               lineToolButton     = null;

        private JToggleButton               polygonToolButton  = null;

        private FLayer                      fLayer             = null;

        private MapControl                  mapControl         = null;

        private AbstractViewPanel           view               = null;

        private GraphicLayer                graphicLayer       = null;

        private LinkedHashMap<String, ROI>  rois               = null;

        private Grid                        grid               = null;

        private Image                       curImage           = null;

        private String                      previousTool       = null;

        private MapContextManager           mapContextManager  = MapContextLocator.getMapContextManager();

        private ROIFeatureManager           roiFeatureManager  = null;

        public ROIsTablePanel(ROIsManagerPanel managerPanel) {

                super();

                this.managerPanel = managerPanel;

                setBorder(new LineBorder(Color.GRAY));

                setLayout(new BorderLayout());

                add(getTable(), BorderLayout.CENTER);

                add(getControlPanel(), BorderLayout.SOUTH);

                initialize();

        }

        private void initialize() {

                listener = new ROIsTablePanelListener(this);

                getNewButton().addActionListener(listener);

                getDeleteButton().addActionListener(listener);

                getImportButton().addActionListener(listener);

                getExportButton().addActionListener(listener);

                getPointToolButton().addActionListener(listener);

                getLineToolButton().addActionListener(listener);

                getPolygonToolButton().addActionListener(listener);

                getTable().getTable().getJTable().getSelectionModel()

                                .addListSelectionListener(listener);

                getTable().getTable().getJTable().getModel().addTableModelListener(

                                listener);

                getPolygonToolButton().setSelected(true);

                setToolsEnabled(false);

        }

        public ROIFeatureManager getRoiFeatureManager() {

                return roiFeatureManager;

        }

        public ROIsManagerPanel getManagerPanel() {

                return managerPanel;

        }

        public void setToolsEnabled(boolean b) {

                getPointToolButton().setEnabled(b);

                getLineToolButton().setEnabled(b);

                getPolygonToolButton().setEnabled(b);

        }

        public TableContainer getTable() {

                if (tableContainer == null) {

                        String[] columnNames = { PluginServices.getText(this, "nombre"),

                                        PluginServices.getText(this, "poligonos"),

                                        PluginServices.getText(this, "lineas"),

                                        PluginServices.getText(this, "puntos"),

                                        PluginServices.getText(this, "color")};

                        int[] columnWidths = { 50, 25, 25, 25, 45 };

                        tableContainer = new TableContainer(columnNames, columnWidths);

                        tableContainer.setModel("ROIsTableModel");

                        tableContainer.initialize();

                        tableContainer.setControlVisible(false);

                        tableContainer.setBorder(new EmptyBorder(5, 5, 0, 5));

                }

                return tableContainer;

        }

        public JPanel getControlPanel() {

                if (controlPanel == null) {

                        controlPanel = new JPanel();

                        controlPanel.setLayout(new FlowLayout(FlowLayout.LEFT));

                        controlPanel.add(getTableButtonsPanel());

                        controlPanel.add(getToolsPanel());

                }

                return controlPanel;

        }

        public JButton getDeleteButton() {

                if (deleteButton == null) {

                        deleteButton = new JButton(PluginServices.getText(this, "borrar"));

                }

                return deleteButton;

        }

        public JButton getNewButton() {

                if (newButton == null) {

                        newButton = new JButton(PluginServices.getText(this, "nuevo"));

                }

                return newButton;

        }

        public FLayer getFLayer() {

                return fLayer;

        }

        public void setFLayer(FLayer layer) throws GridException {

                fLayer = layer;

                /*

                FLyrRasterSE rasterLayer = (FLyrRasterSE) layer;

                IRasterDataSource dsetCopy = null;

                dsetCopy = rasterLayer.getDataSource().newDataset();

                BufferFactory bufferFactory = new BufferFactory(dsetCopy);

                if (!RasterBuffer.loadInMemory(dsetCopy))

                        bufferFactory.setReadOnly(true);

                bufferFactory.setAllDrawableBands();

                int bands[] = null;

                bands = new int[rasterLayer.getBandCount()];

                for (int i = 0; i < rasterLayer.getBandCount(); i++)

                        bands[i] = i;

                try {

                        grid = new Grid(bufferFactory, bands);

                } catch (RasterBufferInvalidException e) {

                        e.printStackTrace();

                }

                */

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

                }

                /*

                 * Caragar las ROIs asociadas a la capa, si las hay:

                 */

                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 {

                        getTable().removeAllRows();

                } catch (NotInitializeException e) {

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

                }

                boolean repaint = roiFeatureManager.clear(graphicLayer);

                if (repaint)

                        getMapControl().rePaintDirtyLayers();

                getRois().clear();

        }

        /**

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

         * 

         * @throws GridException

         * 

         */

        private void loadROIs() throws GridException {

                ArrayList<ROI> roisArray = ((FLyrRaster) fLayer).getRois();

                loadROIs(roisArray);

        }

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

                                        //row[5] = new Integer((int)roi.getValues());

                                        ((DefaultTableModel) 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 setROIs(ArrayList<ROI> roisArray) throws GridException {

                loadROIs(roisArray);

        }

        public MapControl getMapControl() {

                return mapControl;

        }

        public void selectDrawRoiTool() {

                if (mapControl != null)

                        if (getPolygonToolButton().isSelected()) {

                                Image img = IconThemeHelper.getImageIcon("poligoncursor-icon").getImage();

                                curImage = img;

                                mapControl.setTool("drawPolygonROI");

                        } else if (getLineToolButton().isSelected()) {

                                Image img = IconThemeHelper.getImageIcon("linecursor-icon").getImage();

                                curImage = img;

                                mapControl.setTool("drawLineROI");

                        } else if (getPointToolButton().isSelected()) {

                                Image img = IconThemeHelper.getImageIcon("pointcursor-icon").getImage();

                                curImage = img;

                                mapControl.setTool("drawPointROI");

                        }

        }

        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;

                        }

                        /*IRasterDataSource dsetCopy = null; 

                        dsetCopy = rasterLayer.getDataSource().newDataset();

                        BufferFactory bufferFactory = new BufferFactory(dsetCopy);

                        bufferFactory.setReadOnly(true);

                        bufferFactory.setAllDrawableBands();

                        int bands[] = null;

                        bands = new int[rasterLayer.getBandCount()];

                        for (int i = 0; i < rasterLayer.getBandCount(); i++)

                                bands[i] = i;

                        try {

                                grid = RasterLocator.getManager().createGrid(datasource, bands);

                                //new Grid(bufferFactory, bands);

                        } catch (RasterBufferInvalidException e) {

                                e.printStackTrace();

                        }*/

                }                

                return grid;

        }

        private LinkedHashMap<String, ROI> getRois() {

                if (rois == null)

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

                return rois;

        }

        public JPanel getToolsPanel() {

                if (toolsPanel == null) {

                        toolsPanel = new JPanel();

                        toolsPanel.setLayout(new FlowLayout(FlowLayout.RIGHT));

                        toolsPanel.add(getPointToolButton());

                        toolsPanel.add(getLineToolButton());

                        toolsPanel.add(getPolygonToolButton());

                        toolsPanel.add(getExportButton());

                        toolsPanel.add(getImportButton());

                }

                return toolsPanel;

        }

        public JPanel getTableButtonsPanel() {

                if (tableButtonsPanel == null) {

                        tableButtonsPanel = new JPanel();

                        tableButtonsPanel.setLayout(new FlowLayout(FlowLayout.LEFT));

                        tableButtonsPanel.add(getNewButton());

                        tableButtonsPanel.add(getDeleteButton());

                }

                return tableButtonsPanel;

        }

        public JToggleButton getPointToolButton() {

                if (pointToolButton == null) {

                        pointToolButton = new JToggleButton();

                        pointToolButton.setIcon(IconThemeHelper.getImageIcon("point-icon"));

                }

                return pointToolButton;

        }

        public JToggleButton getLineToolButton() {

                if (lineToolButton == null) {

                        lineToolButton = new JToggleButton();

                        lineToolButton.setIcon(IconThemeHelper.getImageIcon("line-icon"));

                }

                return lineToolButton;

        }

        public JToggleButton getPolygonToolButton() {

                if (polygonToolButton == null) {

                        polygonToolButton = new JToggleButton();

                        polygonToolButton.setIcon(IconThemeHelper.getImageIcon("layer-modify-roi"));

                }

                return polygonToolButton;

        }

        public JButton getImportButton() {

                if (importButton == null) {

                        importButton = new JButton();

                        importButton.setIcon(IconThemeHelper.getImageIcon(

                                        "tfwload-icon"));

                        importButton.setToolTipText(PluginServices.getText(this,

                                        "cargar_rois"));

                }

                return importButton;

        }

        public JButton getExportButton() {

                if (exportButton == null) {

                        exportButton = new JButton();

                        exportButton.setIcon(IconThemeHelper.getImageIcon("layer-saveas"));

                        exportButton.setToolTipText(PluginServices.getText(this, "salvar_rois"));

                }

                return exportButton;

        }

        public Image getToolImage() {

                return curImage;

        }

        public AbstractViewPanel getView() {

                return view;

        }

        public void addROI(ROI roi) {

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

        }

        public ROI getROI(String roiName) {

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

        }

        /**

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

                }

        }

        /**

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

                // TODO: add support to remove unused symbols??

//                if (graphicLayer.getNumGraphics() == 0) {

//                        graphicLayer.clearSymbolsGraphics();

//                }

        }

        public ArrayList<ROI> getROIs() {

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

        }

        public void setPreviousTool() {

                if (previousTool != null)

                        getMapControl().setTool(previousTool);

        }

        public void changeRoiName(String currentName, String newName)

                        throws DataException {

                getROI(currentName).setName(newName);

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

                roiFeatureManager.changeROIName(currentName, newName, graphicLayer);

        }

        public void sortTable() {

        }

        public GraphicLayer getGraphicLayer() {

                return graphicLayer;

        }

}