Application: gvSIG desktop

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

         *

         * Copyright (C) 2006 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 Iba?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.remotesensing.classification.gui;

import java.awt.BorderLayout;

import java.awt.Dimension;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.Insets;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.FocusEvent;

import java.awt.event.FocusListener;

import java.io.File;

import javax.swing.BorderFactory;

import javax.swing.ButtonGroup;

import javax.swing.ComboBoxModel;

import javax.swing.DefaultComboBoxModel;

import javax.swing.JButton;

import javax.swing.JFileChooser;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.JRadioButton;

import javax.swing.JScrollPane;

import javax.swing.JTabbedPane;

import javax.swing.JTable;

import javax.swing.JTextField;

import javax.swing.SwingConstants;

import javax.swing.border.CompoundBorder;

import javax.swing.border.EmptyBorder;

import javax.swing.border.TitledBorder;

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

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

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

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

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

import org.gvsig.gui.beans.incrementabletask.IncrementableTask;

import org.gvsig.raster.dataset.GeoRasterWriter;

import org.gvsig.raster.dataset.IRasterDataSource;

import org.gvsig.rastertools.RasterModule;

import org.gvsig.rastertools.roi.ui.ROIsTablePanel;

import org.gvsig.remotesensing.classification.ClassificationGeneralProcess;

import org.gvsig.remotesensing.classification.ClassificationMaximumLikelihoodProcess;

import org.gvsig.remotesensing.classification.ClassificationMinimumDistanceProcess;

import org.gvsig.remotesensing.principalcomponents.gui.BandTableFormat;

import com.iver.andami.PluginServices;

import com.iver.andami.Utilities;

import com.iver.andami.ui.mdiManager.IWindow;

import com.iver.andami.ui.mdiManager.WindowInfo;

import com.iver.cit.gvsig.addlayer.fileopen.FileOpenWizard;

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

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

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

import com.iver.utiles.swing.JComboBox;

/**

 * Clase que implementa la interfaz del proceso de clasificacion de una imagen.

 * 

 * */

public class ClassificationPanel extends DefaultButtonsPanel implements IWindow, ButtonsPanelListener,ActionListener,FocusListener{

        private static final long                 serialVersionUID = 1L;

        private View                                         view = null;

        private ClassificationPanel         classificationPanel= null;

        private int                                         width=400;

        private int                                         height=550;

        private MapContext                                 m_MapContext=null;

        private FLayers                                 layers=null;

        private int nMethods                        = 2;

        private JRadioButton                         rButtonFile=null;

        private JRadioButton                         rButtonMemory =null;

        private JTextField                                 jTextNombreCapa=null;

        private JPanel                                         panelTab1=null;

        private JPanel                                         panelTab2=null;

        private JScrollPane                                scrollBandas=null;

        private JScrollPane                                scrollTable=null;

        private JComboBox                                comboLayers= null;

        private JComboBox                                comboMethods= null;

        private ROIsTablePanel                         roisTablePanel= null;

        private BandTableFormat                 mModeloTabla=null;

        private JTable                                         jTableBandas=null;

        private JPanel                                         nombreCapa=null;

        private int[]                                         bands= null;

        public WindowInfo getWindowInfo(){

                WindowInfo m_viewinfo = new WindowInfo(WindowInfo.MODELESSDIALOG | WindowInfo.RESIZABLE | WindowInfo.MAXIMIZABLE);

                m_viewinfo.setWidth(width);

                m_viewinfo.setHeight(height);

                m_viewinfo.setTitle(PluginServices.getText(this,"classification"));

                m_viewinfo.setX(300);

                return m_viewinfo;

        }

        public ClassificationPanel(View view) {

                this.view = view;

                m_MapContext = view.getModel().getMapContext();

                layers = m_MapContext.getLayers();

                this.setPreferredSize(new Dimension(width, height));

                this.setSize(width, height);

                this.setLayout(new BorderLayout(1, 1));

                this.addButtonPressedListener(this);

                Inicializar();

        }

        public ClassificationPanel getClasificatorPanel() {

                if (classificationPanel==null){

                        classificationPanel = new ClassificationPanel(view);

                        classificationPanel.addButtonPressedListener(this);

                }

                return classificationPanel;

        }

        private void Inicializar(){

                BorderLayout bd=new BorderLayout();

                this.setLayout(bd);

                TitledBorder topBorder = BorderFactory.createTitledBorder(" ");

                topBorder.setTitlePosition(TitledBorder.TOP);

                this.setBorder(new CompoundBorder(topBorder,new EmptyBorder(0,5,5,5)));

                JTabbedPane tabbedPane = new JTabbedPane();

                 tabbedPane.addTab(PluginServices.getText(this, "operacion"), getPanelTab1());

                tabbedPane.addTab(PluginServices.getText(this, "opciones"), getPanelTab2());

                this.setLayout(new BorderLayout());

                this.add(tabbedPane, BorderLayout.CENTER);

                getBandas();

        }

         private JPanel getPanelTab1(){         

                if (panelTab1==null){

                        panelTab1 = new JPanel();        

                        panelTab1.setLayout(new GridBagLayout());

                        // Panel superior opcion Selecion de imagen y bandas

                        JPanel selectionImagenPanel= new JPanel();

                        TitledBorder topBorder = BorderFactory.createTitledBorder(" Imagen");

                        topBorder.setTitlePosition(TitledBorder.TOP);

                        JPanel aux= new JPanel();

                        aux.setPreferredSize(new Dimension(300,110));

                        BorderLayout bd=new BorderLayout();

                        aux.setLayout(bd);

                        aux.add(getComboCapas(),BorderLayout.NORTH);

                        aux.add(getScrollBands(),BorderLayout.SOUTH);

                        selectionImagenPanel.add(aux);

                        selectionImagenPanel.setBorder(topBorder);

                        GridBagConstraints c = new GridBagConstraints();

                        c.fill = GridBagConstraints.BOTH;

                        c.insets=new Insets(2,2,2,2);

                        c.weightx = 1;

                        c.weighty = 0.1;

                        c.gridx = 0;

                        c.gridy = 0;

                        panelTab1.add(selectionImagenPanel, c);

                        // Panel seleccion metodo

                        JPanel methodPanel= new JPanel();

                        TitledBorder topBorderMethod = BorderFactory.createTitledBorder("Metodo");

                        topBorder.setTitlePosition(TitledBorder.TOP);

                        methodPanel.setBorder(topBorderMethod);

                        methodPanel.add(new JLabel(PluginServices.getText(this, "metodo")));

                        methodPanel.add(getComboMethods());

                        c.fill = GridBagConstraints.BOTH;

                        c.insets=new Insets(2,2,2,2);

                        c.weightx = 1;

                        c.weighty = 0.1;

                        c.gridx = 0;

                        c.gridy = 1;

                        panelTab1.add(methodPanel, c);

                        // Insercion del panel de las clases

                        c.fill = GridBagConstraints.BOTH;

                        c.insets=new Insets(2,2,2,2);

                        c.weightx = 1;

                        c.weighty = 1;

                        c.gridx = 0;

                        c.gridy = 2; 

                        panelTab1.add(getRoisTablePanel(), c);         

                 }

                 return panelTab1;

         }

         private JPanel getPanelTab2(){         

                if (panelTab2==null){

                        panelTab2 = new JPanel();        

                        panelTab2.setLayout(new GridBagLayout());

                        JPanel parameterPanel= new JPanel();

                        TitledBorder topBorder = BorderFactory.createTitledBorder(PluginServices.getText(this, "parametros"));

                        topBorder.setTitlePosition(TitledBorder.TOP);

                        JPanel aux= new JPanel();

                        aux.setPreferredSize(new Dimension(300,110));

                        BorderLayout bd=new BorderLayout();

                        aux.setLayout(bd);

                        aux.add(new JButton(),BorderLayout.NORTH);

                        parameterPanel.add(aux);

                        parameterPanel.setBorder(topBorder);

                        GridBagConstraints c = new GridBagConstraints();

                        c.fill = GridBagConstraints.BOTH;

                        c.insets=new Insets(2,2,2,2);

                        c.weightx = 1;

                        c.weighty = 1;

                        c.gridx = 0;

                        c.gridy = 0;

                        panelTab2.add(parameterPanel, c);

                        // Insercion del panel de las clases

                        c.fill = GridBagConstraints.BOTH;

                        c.insets=new Insets(2,2,2,2);

                        c.weightx = 0.05;

                        c.weighty = 0.05;

                        c.gridx = 0;

                        c.gridy = 1; 

                        panelTab2.add(getNombreCapa(), c);

                }

                return panelTab2;

         }

         // Scroll con la tabla de clases que pueden intervenir en el proceso de clasificacion

         public JScrollPane getScrollTable() {

                if(scrollTable==null){

                        scrollTable=new JScrollPane();

                        TitledBorder topBorder = BorderFactory.createTitledBorder((PluginServices.getText(this,"clases")));

                    topBorder.setTitlePosition(TitledBorder.TOP);

                    scrollTable.setBorder(new CompoundBorder(topBorder,new EmptyBorder(2,2,2,2)));                

                        scrollTable.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_NEVER);

                        scrollTable.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);        

                }

                return scrollTable;

        }

         public JTextField getJTextNombreCapa() {

                if (jTextNombreCapa==null){

                        jTextNombreCapa=new JTextField(10);

                        jTextNombreCapa.setText("NewLayer_" + RasterModule.layerCount);

                        RasterModule.layerCount++;

                        jTextNombreCapa.addFocusListener(this);

                        }

                return jTextNombreCapa;

        }

        /**

         * Especificar el nombre de la nueva capa para el recuadro de texto asign?ndo

         * en cada llamada un nombre consecutivo.

         */

        public void setNewLayerText() {

                getJTextNombreCapa().setText("NewLayer_" + RasterModule.layerCount);

                RasterModule.layerCount++;

        }

        public JRadioButton getRadioFile(){

                if (rButtonFile == null){

                        rButtonFile = new JRadioButton(PluginServices.getText(this,"a_fichero"));

                        rButtonFile.addActionListener(this);

                }

                return rButtonFile;

        }

        public JRadioButton getRadioMemory(){

                if (rButtonMemory == null){

                        rButtonMemory = new JRadioButton(PluginServices.getText(this,"a_memoria"),true);

                        rButtonMemory.addActionListener(this);

                }

                return rButtonMemory;

        }

         public JComboBox getComboCapas() {

                if (comboLayers==null){                

                        ComboBoxModel jComboBoxLayersModel = new 

                DefaultComboBoxModel(getLayerNames());

                        comboLayers = new JComboBox();

                        comboLayers.setModel(jComboBoxLayersModel);

                        comboLayers.setEnabled(true);

                        comboLayers.addActionListener(this);        

                        comboLayers.setSelectedIndex(0);

                }                                

                return comboLayers;

        }

        public void close(){        

                try {

                        getRoisTablePanel().clearRoiGraphics();

                        getRoisTablePanel().getMapControl().rePaintDirtyLayers();

                        getRoisTablePanel().setPreviousTool();

                        PluginServices.getMDIManager().closeWindow(this);

                } catch (ArrayIndexOutOfBoundsException ex) {

                        // Si la ventana no se puede eliminar no hacemos nada

                }

        }

        /**

         *           @return array de string con los nombre de las capas raster

         */

        private String[] getLayerNames() {                

                int i, capasraster=0; 

                for (i = 0; i < layers.getLayersCount(); i++) {                      

                        if (layers.getLayer(i)instanceof  FLyrRasterSE)

                                capasraster++;     

                }

                // Solo se toman las capas raster. 

                String[] sNames = new String[capasraster];

                capasraster=0;

                for (i = 0; i < layers.getLayersCount(); i++) {                      

                        if (layers.getLayer(i)instanceof FLyrRasterSE){

                                sNames[capasraster] =layers.getLayer(i).getName();

                                capasraster++;

                        }

                }    

                return sNames;

        }

        public String getFileSelected() {

                String path = "";

                if (getRadioFile().isSelected()) {        

                        JFileChooser chooser = new JFileChooser(FileOpenWizard.getLastPath());

                        chooser.setDialogTitle(PluginServices.getText(this, "seleccionar_fichero"));

                        //A?adimos las extensiones que hayan sido registradas en el driver

                        String[] extList = GeoRasterWriter.getDriversExtensions();

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

                                chooser.addChoosableFileFilter(new WriterFilter(extList[i]));

                        if (chooser.showOpenDialog(this) != JFileChooser.APPROVE_OPTION)

                                return null;

                        String fName = chooser.getSelectedFile().toString();                                

                        String ext = ((WriterFilter)chooser.getFileFilter()).getDescription();

                        ext = ext.toLowerCase().substring(ext.lastIndexOf(".") + 1, ext.length());

                        if ((fName != null) && !fName.equals(""))

                                if (!fName.endsWith("." + ext))        

                                        fName = fName + "." + ext;

                        FileOpenWizard.setLastPath(chooser.getSelectedFile().getPath().substring(0, chooser.getSelectedFile().getPath().lastIndexOf(File.separator)));

                                path = fName;

                } else {

                        path = Utilities.createTempDirectory() + File.separator +getJTextNombreCapa().getText() + ".tif";

                        setNewLayerText();        

                }

                return path;        

        }

        public void accept(){

                // Imagen que va a ser clasificada

                IRasterDataSource rasterLayer= ((FLyrRasterSE)layers.getLayer((String) getComboCapas().getSelectedItem())).getDataSource();

                ClassificationGeneralProcess process=null;

                if(getComboMethods().getSelectedIndex()==0)

                        process = new ClassificationMaximumLikelihoodProcess(rasterLayer,getRoisTablePanel().getROIs(),view,getFileSelected());

                if(getComboMethods().getSelectedIndex()==1)

                        process =  new ClassificationMinimumDistanceProcess(rasterLayer,getRoisTablePanel().getROIs(),view,getFileSelected());

                IncrementableTask incrementableTask = new IncrementableTask(process);        

                process.setIncrementableTask(incrementableTask);

                incrementableTask.showWindow();

                process.start();

                incrementableTask.start();        

        }                

        public void actionButtonPressed(ButtonsPanelEvent e) {

                        // Bot?n de Aceptar

                if (e.getButton() == ButtonsPanel.BUTTON_ACCEPT) {

                        accept();

                        close();

                }

                // Bot?n de Aplicar

                if (e.getButton() == ButtonsPanel.BUTTON_APPLY) {

                        accept();

                }

                // Bot?n de Cerrar

                if (e.getButton() == ButtonsPanel.BUTTON_CANCEL) {

                        close();

                }                        

         }

        public void actionPerformed(ActionEvent arg0) {

                if(arg0.getSource()==getComboCapas()){

                        getBandas();                

                }

        }

        public void focusGained(FocusEvent arg0) {

        }

        public void focusLost(FocusEvent arg0) {

        }

        private JTable getTableBands(){

                if (jTableBandas==null){        

                        jTableBandas=new JTable(getModeloTabla());

                }

                return jTableBandas;

        }

        public BandTableFormat getModeloTabla() {         

                if(mModeloTabla==null){

                        mModeloTabla=new BandTableFormat(); 

                } 

                return mModeloTabla;        

        }

        public JScrollPane getScrollBands() {        

                if (scrollBandas==null){

                        scrollBandas=new JScrollPane();                                

                }        

                getTableBands().getColumn("").setPreferredWidth(30);

                scrollBandas.setPreferredSize(new Dimension(250,80));

                scrollBandas.setViewportView(getTableBands());

                scrollBandas.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_NEVER);

                return scrollBandas;        

        }

        public ROIsTablePanel getRoisTablePanel() {

                if (roisTablePanel==null){        

                        roisTablePanel= new ROIsTablePanel();

                }

                return roisTablePanel;

        }

        public JComboBox getComboMethods() {        

                if (comboMethods==null){

                        comboMethods= new JComboBox();

                        String[] sNames = new String[nMethods];

                        sNames[0]= new String(PluginServices.getText(this, "metodo_maxima_probabilidad"));

                        sNames[1]= new String(PluginServices.getText(this, "metodo_minima_diatancia"));

                        ComboBoxModel jComboBoxLayersModel = new  DefaultComboBoxModel(sNames);

                        comboMethods.setModel(jComboBoxLayersModel);

                        comboMethods.setEnabled(true);

                        comboMethods.addActionListener(this);                

                }                

                return comboMethods;

        }

        // Toma las Bandas de la imagen seleccionada en el combo Imagen

        public void getBandas(){

                // Antes de insertar se borra lo anterior

                getModeloTabla().LimpiarLista();

                FLyrRasterSE rasterLayer= (FLyrRasterSE)layers.getLayer((String)getComboCapas().getSelectedItem());

                String bandas[]=new String[rasterLayer.getBandCount()];        

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

                if (layers.getLayer((String)getComboCapas().getSelectedItem()) instanceof  FLyrRasterSE ){                

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

                                        String s=(PluginServices.getText(this,"banda"))+(j+1);

                                        bandas[j]=s;

                                        bands[j]=j;

                                }        

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

                                getModeloTabla().addRow(bandas[i]); 

                        getTableBands().updateUI();

                        // PROVISIONAL

                        getRoisTablePanel().setFLayer((FLyrRasterSE)layers.getLayer((String)getComboCapas().getSelectedItem()));

                }        

        }

        /**Panel que incluye el nombre de la capa y las opciones de almacenamieto del grid generado*/

        public JPanel getNombreCapa() {

                if (nombreCapa==null){ 

                        nombreCapa=new JPanel();

                        nombreCapa.setPreferredSize(new Dimension(400,130));

                        TitledBorder topBorder = BorderFactory.createTitledBorder(PluginServices.getText(this, "salida"));

                        topBorder.setTitlePosition(TitledBorder.TOP);

                        nombreCapa.setBorder(topBorder);

                        GridBagConstraints gridBagConstraints;

                        JPanel radioPanel = new JPanel();

                        radioPanel.setLayout(new GridBagLayout());

                        radioPanel.setBorder(BorderFactory.createTitledBorder(""));

                        ButtonGroup buttonGroup = new ButtonGroup();

                        buttonGroup.add(getRadioMemory());

                        gridBagConstraints = new java.awt.GridBagConstraints();

                        gridBagConstraints.gridx = 0;

                        gridBagConstraints.gridy = 1;

                        gridBagConstraints.anchor = GridBagConstraints.WEST;

                        gridBagConstraints.insets = new java.awt.Insets(3, 3, 3, 3);

                        radioPanel.add(getRadioMemory(),gridBagConstraints);

                        buttonGroup.add(getRadioFile());

                        gridBagConstraints = new java.awt.GridBagConstraints();

                        gridBagConstraints.gridx = 0;

                        gridBagConstraints.gridy = 2;

                        gridBagConstraints.anchor = GridBagConstraints.WEST;

                        gridBagConstraints.insets = new java.awt.Insets(3, 3, 3, 3);

                        radioPanel.add(getRadioFile(),gridBagConstraints);

                        //Establece la separacion entre los elementos

                        nombreCapa.setLayout(new GridBagLayout());

                        gridBagConstraints = new java.awt.GridBagConstraints();

                        gridBagConstraints.gridx = 0;

                        gridBagConstraints.gridy = 0;

                        gridBagConstraints.anchor = GridBagConstraints.WEST;

                        gridBagConstraints.insets = new java.awt.Insets(3, 3, 3, 3);

                        nombreCapa.add(new JLabel(("Nombre:"),SwingConstants.RIGHT ),gridBagConstraints);

                        gridBagConstraints = new java.awt.GridBagConstraints();

                        gridBagConstraints.gridx = 1;

                        gridBagConstraints.gridy = 0;

                        gridBagConstraints.anchor = GridBagConstraints.WEST;

                        gridBagConstraints.insets = new java.awt.Insets(3, 3, 3, 3);

                        nombreCapa.add(getJTextNombreCapa(),gridBagConstraints);

                        gridBagConstraints = new java.awt.GridBagConstraints();

                        gridBagConstraints.gridx = 2;

                        gridBagConstraints.gridy = 0;

                        gridBagConstraints.anchor = GridBagConstraints.WEST;

                        gridBagConstraints.insets = new java.awt.Insets(3, 3, 3, 3);

                        nombreCapa.add(radioPanel,gridBagConstraints);        

                }

                return nombreCapa;

        }

}

/**

 * @author Nacho Brodin <brodin_ign@gva.es>

 * Filtro para el selector de formatos de escritura

 */

class WriterFilter extends javax.swing.filechooser.FileFilter {

        private String                                filter;

        public WriterFilter(String fil) {

                this.filter = fil;

        }

        public boolean accept(File f) {

                return f.isDirectory() || f.getName().toLowerCase().endsWith("." + filter);

        }

        public String getDescription() {

                return "." + filter;

        }

}