Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2007 Instituto de Desarrollo Regional and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   Instituto de Desarrollo Regional (Universidad de Castilla La-Mancha)

 *   Campus Universitario s/n

 *   02071 Alabacete

 *   Spain

 *

 *   +34 967 599 200

 */

package org.gvsig.rastertools.roi.ui;

import java.awt.BorderLayout;

import java.awt.Color;

import java.awt.Cursor;

import java.awt.FlowLayout;

import java.awt.Graphics2D;

import java.awt.Image;

import java.awt.Point;

import java.awt.Toolkit;

import java.awt.image.BufferedImage;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

import javax.swing.ImageIcon;

import javax.swing.JButton;

import javax.swing.JMenu;

import javax.swing.JMenuBar;

import javax.swing.JMenuItem;

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

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

import org.gvsig.gui.beans.defaultbuttonspanel.DefaultButtonsPanel;

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.raster.buffer.BufferFactory;

import org.gvsig.raster.buffer.RasterBufferInvalidException;

import org.gvsig.raster.grid.Grid;

import org.gvsig.raster.grid.roi.ROI;

import org.gvsig.rastertools.roi.ui.listener.DrawMouseViewListener;

import org.gvsig.rastertools.roi.ui.listener.ROIManagerPanelListener;

import com.iver.andami.PluginServices;

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

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

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

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

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

import com.iver.cit.gvsig.fmap.core.symbols.IFillSymbol;

import com.iver.cit.gvsig.fmap.core.symbols.ILineSymbol;

import com.iver.cit.gvsig.fmap.core.symbols.IMarkerSymbol;

import com.iver.cit.gvsig.fmap.core.symbols.ISymbol;

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

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

import com.iver.cit.gvsig.fmap.rendering.FGraphic;

import com.iver.cit.gvsig.fmap.tools.Behavior.Behavior;

import com.iver.cit.gvsig.fmap.tools.Behavior.MouseMovementBehavior;

import com.iver.cit.gvsig.fmap.tools.Behavior.PointBehavior;

import com.iver.cit.gvsig.fmap.tools.Behavior.PolygonBehavior;

import com.iver.cit.gvsig.fmap.tools.Behavior.PolylineBehavior;

import com.iver.cit.gvsig.project.documents.view.gui.View;

import com.iver.cit.gvsig.project.documents.view.toolListeners.StatusBarListener;

/**

 * Iterfazde usuario para la gesti?n de ROIs.

 * 

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

 *

 */

public class ROIManagerPanel extends DefaultButtonsPanel {

        private static final long serialVersionUID = 2138437448356198224L;

        private ROIManagerDialog                         roiManagerDialog                         = null;

        private ROIManagerPanelListener     listener                                        = null;

        private JMenuBar                                         menuBar                                                 = null;

        private JPanel                                                 tablePanel                                         = 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 JToggleButton                                pointToolButton                                = null;

        private JToggleButton                                lineToolButton                                = null;

        private JToggleButton                                polygonToolButton                        = null;

        private JMenuItem                                         closeMenuItem                                 = null;

        private JMenuItem                                         saveMenuItem                                 = null;

        private JMenuItem                                         loadMenuItem                                 = null;

        private JMenuItem                                         exportMenuItem                                 = null;

        private JMenuItem                                         helpMenuItem                                 = null;

        private JMenuItem                                         preferencesMenuItem                        = null;

        private FLayer                                                fLayer                                                 = null;

        private MapControl                                        mapControl                                        = null;                

        private View                                                view                                                = null;

        private HashMap                                                rois                                                = null;

        private HashMap                                                roiGraphics                                        = null;

        private Grid                                                 grid                                                = null;

        private String                                pathToImages                                  = "images/";

        private Cursor                                                cursor                                                = null;

        public ROIManagerPanel(ROIManagerDialog roiManagerDialog) {

                super();

                this.roiManagerDialog = roiManagerDialog;

                initialize();

        }

        private void initialize() {

                setLayout(new BorderLayout(5,5));

                setBorder(new EmptyBorder(0,15,11,15));                

                add(getMenuBar(),BorderLayout.NORTH);

                add(getTablePanel(),BorderLayout.CENTER);

                listener = new ROIManagerPanelListener(this);

                getCloseMenuItem().addActionListener(listener);

                getNewButton().addActionListener(listener);

                getDeleteButton().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 void setToolsEnabled(boolean b) {

                getPointToolButton().setEnabled(b);

                getLineToolButton().setEnabled(b);

                getPolygonToolButton().setEnabled(b);

        }

        public JMenuBar getMenuBar() {

                if (menuBar==null){

                        menuBar = new JMenuBar();

                        menuBar.setBorder(new EmptyBorder(0,0,0,0));

                        JMenu fileMenu = new JMenu(PluginServices.getText(this, "archivo"));

                        JMenu optionsMenu = new JMenu(PluginServices.getText(this, "opciones"));

                        JMenu helpMenu = new JMenu(PluginServices.getText(this, "help"));

                        fileMenu.add(getSaveMenuItem());

                        fileMenu.add(getLoadMenuItem());

                        fileMenu.add(getExportMenuItem());

                        fileMenu.add(getCloseMenuItem());

                        helpMenu.add(getHelpMenuItem());

                        optionsMenu.add(getPreferencesMenuItem());

                        menuBar.add(fileMenu);

                        menuBar.add(optionsMenu);

                        menuBar.add(helpMenu);

                }

                return menuBar;

        }

        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 = {70,30,30,30, 110};

                        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 getTablePanel() {

                if (tablePanel == null){

                        tablePanel = new JPanel();

                        tablePanel.setBorder(new LineBorder(Color.GRAY));

                        tablePanel.setLayout(new BorderLayout());

                        tablePanel.add(getTable(),BorderLayout.CENTER);

                        tablePanel.add(getControlPanel(),BorderLayout.SOUTH);

                }

                return tablePanel;

        }

        public JPanel getControlPanel() {

                if (controlPanel == null){

                        controlPanel = new JPanel();

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

                        controlPanel.add(getTableButtonsPanel());

                        controlPanel.add(getToolsPanel());

                        //controlPanel.add(getToolBar());

                }

                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 JMenuItem getCloseMenuItem() {

                if (closeMenuItem == null){

                        closeMenuItem = new JMenuItem(PluginServices.getText(this, "cerrar"));

                }

                return closeMenuItem;

        }

        public ROIManagerDialog getRoiManagerDialog() {

                return roiManagerDialog;

        }

        public JMenuItem getExportMenuItem() {

                if (helpMenuItem == null){

                        exportMenuItem = new JMenuItem(PluginServices.getText(this, "exportar"));

                }

                return exportMenuItem;

        }

        public JMenuItem getHelpMenuItem() {

                if (helpMenuItem == null){

                        helpMenuItem = new JMenuItem(PluginServices.getText(this, "help"));

                }

                return helpMenuItem;

        }

        public JMenuItem getLoadMenuItem() {

                if (loadMenuItem == null){

                        loadMenuItem = new JMenuItem(PluginServices.getText(this, "cargar"));

                }

                return loadMenuItem;

        }

        public JMenuItem getSaveMenuItem() {

                if (saveMenuItem == null){

                        saveMenuItem = new JMenuItem(PluginServices.getText(this, "salvar"));

                }

                return saveMenuItem;

        }

        public JMenuItem getPreferencesMenuItem() {

                if (preferencesMenuItem == null){

                        preferencesMenuItem = new JMenuItem(PluginServices.getText(this, "preferencias"));

                }

                return preferencesMenuItem;

        }

        public FLayer getFLayer() {

                return fLayer;

        }

        public void setFLayer(FLayer layer) {

                fLayer = layer;

                FLyrRasterSE rasterLayer = (FLyrRasterSE)layer;

                BufferFactory dataSource = rasterLayer.getBufferFactory();

                int bands[]=null;        

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

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

                        bands[i]=i;        

                try {

                        grid = new Grid(dataSource, bands);        

                } catch (RasterBufferInvalidException e) {

                        e.printStackTrace();

                }

                view = (View) PluginServices.getMDIManager().getActiveWindow();

                mapControl = view.getMapControl();

                // 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.addMapTool("drawPolygonROI", new Behavior[] {

                                new PolygonBehavior(drawMouseViewListener), new MouseMovementBehavior(sbl)

                        }

                );

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

                                new PolylineBehavior(drawMouseViewListener), new MouseMovementBehavior(sbl)

                        }

                );

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

                                new PointBehavior(drawMouseViewListener), new MouseMovementBehavior(sbl)

                        }

                );

                mapControl.getMapContext().addLayerDrawingListener(listener);

                /*

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

                 */

                clearROIs();

                loadROIs();

        }

        private void clearROIs() {

                try {

                        getTable().removeAllRows();

                } catch (NotInitializeException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                getRoiGraphics().clear();

                getRois().clear();

        }

        /**

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

         *

         */

        private void loadROIs() {

                ArrayList roisArray = ((FLyrRasterSE)fLayer).getRois();

                if (roisArray != null){

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

                        ISymbol symbol = null;

                        FGraphic fGraphic = null;

                        int nPoints, nPolygons, nLines;

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

                                nPoints = 0; nPolygons = 0; nLines = 0;

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

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

                                row[0]= roi.getName();

                                row[4]= roi.getColor();

                                addROI(roi);

                                VectorialROI vectorialROI = (VectorialROI)roi;

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

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

                                        switch (geometry.getGeometryType()) {

                                        case FShape.POINT:

                                                symbol = SymbologyFactory.createDefaultMarkerSymbol();

                                                ((IMarkerSymbol)symbol).setColor(roi.getColor());

                                                nPoints++;

                                                break;

                                        case FShape.POLYGON:

                                                symbol =SymbologyFactory.createDefaultFillSymbol();

                                                ((IFillSymbol)symbol).setFillColor(roi.getColor());

                                                nPolygons++;

                                                break;

                                        case FShape.LINE:

                                                symbol =SymbologyFactory.createDefaultLineSymbol();

                                                ((ILineSymbol)symbol).setLineColor(roi.getColor());

                                                nLines++;

                                                break;

                                        }

                                        fGraphic = new FGraphic(geometry,graphicLayer.addSymbol(symbol));

                                        graphicLayer.addGraphic(fGraphic);

                                        getRoiGraphics(roi.getName()).add(fGraphic);

                                }

                                row[1]= new Integer(nPolygons); row[2]= new Integer (nLines); row[3]= new Integer(nPoints);

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

                        }

                        selectDrawRoiTool();

                        setToolsEnabled(true);

                        getMapControl().drawGraphics();

                }

        }

        public MapControl getMapControl() {

                return mapControl;

        }

        public void selectDrawRoiTool(){

                if (mapControl!=null)

                        if (getPolygonToolButton().isSelected()){

                                Image img = new ImageIcon(getClass().getResource("images/PoligonCursor.png")).getImage();

                                cursor = Toolkit.getDefaultToolkit().createCustomCursor(img, new Point(16, 16), "");

                                mapControl.setTool("drawPolygonROI");

                        }

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

                                Image img = new ImageIcon(getClass().getResource("images/LineCursor.png")).getImage();

                                cursor = Toolkit.getDefaultToolkit().createCustomCursor(img, new Point(16, 16), "");

                                mapControl.setTool("drawLineROI");

                        }

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

                                Image img = new ImageIcon(getClass().getResource("images/PointCursor.png")).getImage();

                                cursor = Toolkit.getDefaultToolkit().createCustomCursor(img, new Point(16, 16), "");

                                mapControl.setTool("drawPointROI");

                        }

        }

        public Grid getGrid() {

                return grid;

        }

        private HashMap getRois() {

                if (rois == null)

                        rois = new HashMap();

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

                }

                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(new ImageIcon(getClass().getResource(pathToImages + "Point.png")));

                }

                return pointToolButton;

        }

        public JToggleButton getLineToolButton() {

                if (lineToolButton==null){

                        lineToolButton = new JToggleButton();

                        lineToolButton.setIcon(new ImageIcon(getClass().getResource(pathToImages + "Line.png")));

                }

                return lineToolButton;

        }

        public JToggleButton getPolygonToolButton() {

                if (polygonToolButton ==null){

                        polygonToolButton = new JToggleButton();

                        polygonToolButton.setIcon(new ImageIcon(getClass().getResource(pathToImages + "Poligon.png")));

                }

                return polygonToolButton;

        }

        public Cursor getToolCursor(){

                return cursor;

        }

        public View getView() {

                return view;

        }

        /**

         * Dibuja en la vista las ROIs que aparecen en la tabla.

         *

         */

        public void drawROIs() {

                Color geometryColor;

                BufferedImage img = getMapControl().getImage();

                Graphics2D gImag = (Graphics2D) img.getGraphics();

                ViewPort vp = getMapControl().getViewPort();

                try {

                        for (int i = 0; i < getTable().getRowCount(); i++){

                                geometryColor = (Color)getTable().getModel().getValueAt(i, 4);

                                VectorialROI vROI = (VectorialROI)getRois().get(getTable().getTable().getJTable().getValueAt(i, 0));

                                for (Iterator iter = vROI.getGeometries().iterator(); iter.hasNext();) {

                                        IGeometry geometry = (IGeometry) iter.next();

                                        ISymbol sym = SymbologyFactory.createDefaultSymbolByShapeType(geometry.getGeometryType(), geometryColor);

                                        geometry.draw(gImag, vp, sym,null);

                                }

                        }

                        getView().repaint();

                } catch (NotInitializeException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

        private HashMap getRoiGraphics() {

                if (roiGraphics == null)

                        roiGraphics = new HashMap();

                return roiGraphics;

        }

        public void addROI(ROI roi) {

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

                getRoiGraphics().put(roi.getName(), new ArrayList());

        }

        public ROI getROI(String roiName){

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

        }

        public void removeROI(String roiName){

                getRois().remove(roiName);

                ArrayList roiGraphics = getRoiGraphics(roiName);

                if (roiGraphics!=null){

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

                                getMapControl().getMapContext().getGraphicsLayer().removeGraphic((FGraphic)getRoiGraphics(roiName).get(i));

                        getMapControl().rePaintDirtyLayers();

                        getRoiGraphics().remove(roiName);

                }

        }

        public ArrayList getRoiGraphics(String roiName){

                return (ArrayList)getRoiGraphics().get(roiName);

        }

        /**

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

                for (Iterator iter = getRoiGraphics().values().iterator(); iter.hasNext();) {

                        ArrayList fGraphics = (ArrayList) iter.next();

                        for (Iterator iterator = fGraphics.iterator(); iterator.hasNext();) {

                                FGraphic fGraphic = (FGraphic) iterator.next();

                                getMapControl().getMapContext().getGraphicsLayer().removeGraphic(fGraphic);

                        }

                }

                if(getMapControl().getMapContext().getGraphicsLayer().getNumGraphics()==0)

                        getMapControl().getMapContext().getGraphicsLayer().clearSymbolsGraphics();

        }

        public ArrayList getROIs() {

                return new ArrayList(getRois().values());

        }

}