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

import java.awt.Color;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.util.List;

import javax.swing.JFileChooser;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import javax.swing.event.TableModelEvent;

import javax.swing.event.TableModelListener;

import javax.swing.filechooser.FileFilter;

import javax.swing.table.DefaultTableModel;

import org.cresques.cts.IProjection;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.project.documents.Document;

import org.gvsig.app.project.documents.view.BaseViewDocument;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.coverage.RasterLocator;

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

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

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

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

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

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

import org.gvsig.gui.beans.buttonspanel.ButtonsPanelEvent;

import org.gvsig.gui.beans.buttonspanel.ButtonsPanelListener;

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

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

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

import org.gvsig.raster.roi.ROI;

import org.gvsig.raster.roi.ROIReader;

import org.gvsig.raster.roi.ROIWriter;

import org.gvsig.raster.swing.RasterSwingLibrary;

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

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

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

import org.gvsig.raster.tools.app.basic.tool.roi.ROIDataModel.LayerElement;

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

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

/**

 * Listener para el panel ROIsTablePanel

 *

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

 *

 */

public class ROIsTablePanelListener implements ButtonsPanelListener, ActionListener, ListSelectionListener, TableModelListener {

        private ROIPanel       panel              = null;

        private ROIDataModel   dataModel          = null;

        private String                    roiSelectedName    = "";

        private ROIWindow      window             = null;

        private File           lastDirectory      = null;

        private String         defaultROIFileName = null;

        public ROIsTablePanelListener(

                        ROIPanel panel, 

                        ROIDataModel dataModel,

                        ROIWindow window) {

                this.panel = panel;

                this.dataModel = dataModel;

                this.window = window;

        }

        public void setDefaultROIFileName(String defaultROIFileName) {

                this.defaultROIFileName = defaultROIFileName;

        }

        public void setPathToROI(File ROI) {

                this.lastDirectory = ROI;

        }

        public void actionButtonPressed(ButtonsPanelEvent e) {

        }

        public void actionPerformed(ActionEvent e) {

                if (e.getSource() == panel.getButton(ROIPanel.NEW_BUTTON)) {

                        String roiName;

                        try {

                                roiName = "ROI-" + String.valueOf(System.currentTimeMillis());

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

                                row[0]= roiName;

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

                                ROI roi = RasterLocator.getManager().createVectorialROI(dataModel.getRasterDataStore());

                                roi.setName(roiName);

                                roi.setColor((Color)row[4]);

                                dataModel.addROI(roi);

                                panel.getTable().setSelectedIndex(panel.getTable().getRowCount()-1);

                                dataModel.selectDrawRoiTool();

                                panel.setToolsEnabled(true);

                        } catch (NotInitializeException e1) {

                                RasterToolsUtil.messageBoxError("error_tabla_rois", panel, e1);

                                window.close();

                        }

                } else if (e.getSource() == panel.getButton(ROIPanel.DELETE_BUTTON)) {

                        try {

                                if (panel.getTable().getSelectedRows().length > 0) {

                                        String className = (String)panel.getTable().getModel().getValueAt(panel.getTable().getSelectedRow(),0);

                                        dataModel.removeROI(className);

                                        int selectedRow = panel.getTable().getSelectedRow();

                                        ((DefaultTableModel)panel.getTable().getModel()).removeRow(selectedRow);

                                        selectedRow = selectedRow >= panel.getTable().getRowCount() ? panel.getTable().getRowCount() - 1 : selectedRow; 

                                        panel.getTable().setSelectedIndex(selectedRow);

                                }

                        } catch (NotInitializeException e1) {

                                RasterToolsUtil.messageBoxError("error_tabla_rois", panel, e1);

                                window.close();

                        }

                } else if (e.getSource() == panel.getButton(ROIPanel.POINT_BUTTON)) {

                        panel.getButton(ROIPanel.POINT_BUTTON).setSelected(true);

                        panel.getButton(ROIPanel.LINE_BUTTON).setSelected(false);

                        panel.getButton(ROIPanel.POLYGON_BUTTON).setSelected(false);

                        dataModel.selectDrawRoiTool();

                } else if (e.getSource() == panel.getButton(ROIPanel.LINE_BUTTON)) {

                        panel.getButton(ROIPanel.LINE_BUTTON).setSelected(true);

                        panel.getButton(ROIPanel.POINT_BUTTON).setSelected(false);

                        panel.getButton(ROIPanel.POLYGON_BUTTON).setSelected(false);

                        dataModel.selectDrawRoiTool();

                } else if (e.getSource() == panel.getButton(ROIPanel.POLYGON_BUTTON)) {

                        panel.getButton(ROIPanel.POLYGON_BUTTON).setSelected(true);

                        panel.getButton(ROIPanel.LINE_BUTTON).setSelected(false);

                        panel.getButton(ROIPanel.POINT_BUTTON).setSelected(false);

                        dataModel.selectDrawRoiTool();

                } else if (e.getSource() == panel.getButton(ROIPanel.EXPORT_BUTTON)) {

                        showChooserDialog(ROIPanel.EXPORT_BUTTON);

                } else if (e.getSource() == panel.getButton(ROIPanel.IMPORT_BUTTON)) {

                        showChooserDialog(ROIPanel.IMPORT_BUTTON);

                } else if (e.getSource() == panel.getButton(ROIPanel.REMOVE_ROI_FILE_BUTTON)) {

                        if(RasterToolsUtil.messageBoxYesOrNot("remove_roi_file_link", null)) {

                                dataModel.clearROIs();

                                removeLinkBetweenFileROIAndLayer();

                        }

                } else if (e.getSource() == panel.getButton(ROIPanel.ADDFLYR_BUTTON)) {

                        Object obj = panel.getLayerList(ROIPanel.FLYRS_COMBO).getSelectedItem();

                        if(obj instanceof LayerElement) {

                                FLyrVect lyrVect = ((LayerElement)obj).getLayer();

                                ROIReader reader = null;

                                try {

                                        reader = RasterLocator.getManager().createVectorialROIReader(

                                                        lyrVect.getFeatureStore(), 

                                                        dataModel.getRasterDataStore(), 

                                                        getProjectionFromLayer());

                                        List<ROI> rois = reader.read(dataModel.getROIs());

                                        dataModel.clearROIs();

                                        dataModel.setROIs(rois);

                                        DataStoreParameters params = lyrVect.getDataStore().getParameters();

                                        if(params instanceof FilesystemStoreParameters)

                                                dataModel.setROIFile(((FilesystemStoreParameters)params).getFile());

                                } catch (ROIException e1) {

                                        RasterSwingLibrary.messageBoxError("error_file_not_valid", panel, e1);

                                } catch (GridException e1) {

                                        RasterSwingLibrary.messageBoxError("error_file_not_valid", panel, e1);

                                }

                        }

                }

        }

        private void removeLinkBetweenFileROIAndLayer() {

                try {

                        ((FLyrRaster)dataModel.getFLayer()).setROIsFiles(null);

                } catch (RmfSerializerException e) {

                        RasterSwingLibrary.messageBoxError("error_saving_rois_rmf", panel, e);

                }

        }

        private void showChooserDialog(int action) {

                JFileChooser fileChooser = new JFileChooser();

                fileChooser.removeChoosableFileFilter(fileChooser.getAcceptAllFileFilter());

                fileChooser.addChoosableFileFilter(new ShpFileFilter());

                fileChooser.setSelectedFile(new File(defaultROIFileName));

                File oldROIFile = dataModel.getROIFile();

                if(oldROIFile != null) {

                        fileChooser.setCurrentDirectory(oldROIFile);

                        fileChooser.setSelectedFile(oldROIFile);

                } else

                        fileChooser.setCurrentDirectory(lastDirectory);

                if(action == ROIPanel.IMPORT_BUTTON) {

                        if (fileChooser.showOpenDialog(panel.getComponent()) == JFileChooser.APPROVE_OPTION) {

                                File file = fileChooser.getSelectedFile();

                                loadROIFromFile(file);

                        }

                }

                if(action == ROIPanel.EXPORT_BUTTON) {

                        if (fileChooser.showSaveDialog(panel.getComponent()) == JFileChooser.APPROVE_OPTION) {

                                File file = fileChooser.getSelectedFile();

                                saveROIToFile(file);

                        }

                }

        }

        private void saveROIToFile(File file) {

                lastDirectory = new File(file.getParent());

                if(file.exists()) {

                        if(!RasterToolsUtil.messageBoxYesOrNot("raster_error_file_exists", null))

                                return;

                }

                ROIWriter writer = RasterLocator.getManager().createROIWriter(

                                file.getPath(), 

                                getProjectionFromLayer());

                writer.write(dataModel.getROIs());

                dataModel.setROIFile(file);

        }

        private void loadROIFromFile(File file) {

                lastDirectory = new File(file.getParent());

                try {

                        ROIReader reader = RasterLocator.getManager().createROIReader( 

                                        file.getPath(), 

                                        dataModel.getRasterDataStore(), 

                                        getProjectionFromLayer());

                        dataModel.clearROIs();

                        List<ROI> rois = reader.read(dataModel.getROIs());

                        dataModel.setROIFile(file);

                        dataModel.setROIs(rois);

                } catch (FileNotExistsException e1) {

                        RasterSwingLibrary.messageBoxError("error_file_not_found", panel, e1);

                } catch (ROIException e1) {

                        RasterSwingLibrary.messageBoxError("problems_loading_rois", panel, e1);

                } catch (GridException e1) {

                        RasterSwingLibrary.messageBoxError("problems_loading_rois", panel, e1);

                } 

        }

        /**

         * Gets the projection from the layer or from the view if the 

         * layer does not have

         * @return

         */

        private IProjection getProjectionFromLayer() {

                IProjection proj = dataModel.getFLayer().getProjection();

                if(proj == null) {

                        Document doc = ApplicationLocator.getManager().getActiveDocument();

                        if(doc instanceof BaseViewDocument)

                                proj = ((BaseViewDocument)doc).getProjection();

                }

                return proj;

        }

        public void valueChanged(ListSelectionEvent e) {

                try {

                        if (panel.getTable().getSelectedRows().length <= 0) {

                                panel.setToolsEnabled(false);

                                dataModel.setPreviousTool();

                        } else {

                                panel.setToolsEnabled(true);

                                dataModel.selectDrawRoiTool();

                                int selectedRow = panel.getTable().getSelectedRow();

                                roiSelectedName = (String)panel.getTable().getTable().getJTable().getValueAt(selectedRow,0);

                        }

                } catch (NotInitializeException e1) {

                        RasterToolsUtil.messageBoxError("error_rois_table", panel, e1);

                }

        }

        @SuppressWarnings("deprecation")

        public void tableChanged(TableModelEvent e) {

                //Cambio de color en una ROI:

                try {

                        if(e.getColumn() == 4) {

                                String roiName = (String)panel.getTable().getTable().getJTable().getValueAt(e.getFirstRow(),0);

                                if(dataModel.getMapControl() != null) {

                                        ROIFeatureManager roiFeatureManager = dataModel.getRoiFeatureManager();

                                        Color color = (Color) panel.getTable().getTable().getJTable().getValueAt(e.getFirstRow(), 4);

                                        roiFeatureManager.changeColor(dataModel.getGraphicLayer(), roiName, color);

                                        dataModel.getROI(roiName).setColor(color);

                                        dataModel.getMapControl().drawGraphics();

                                        dataModel.getMapControl().getMapContext().invalidate();

                                }

                        } else if(e.getColumn() == 0) {

                                String newName = (String)panel.getTable().getTable().getJTable().getValueAt(e.getFirstRow(),0);

                                dataModel.changeRoiName(roiSelectedName, newName);

                                roiSelectedName = newName;

                        }

                } catch (DataException e1) {

                        RasterToolsUtil.messageBoxError(

                                        "error_getting_roi_features", panel, e1);

                }

        }

}

/**

 * Filtro para el selector de expresiones matem?ticas.

 * @author Alejandro Mu?oz (alejandro.munoz@uclm.es)

 *

 */

class ShpFileFilter extends FileFilter {

        final static String exp = "shp";

        public boolean accept(File f) {

                if (f.isDirectory()) {

                        return true;

                }

                String s = f.getName();

                int i = s.lastIndexOf('.');

                if (i > 0 &&  i < s.length() - 1) {

                        String extension = s.substring(i+1).toLowerCase();

                        if (exp.equals(extension)){

                                return true;

                        } else {

                                return false;

                        }

                }

                return false;

        }

        public String getDescription() {

                return "Shape file .shp";

        }

}