Application: gvSIG desktop

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

*

* Copyright (C) 2005 IVER T.I. 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.

*/

package org.gvsig.rastertools.histogram;

import java.awt.Color;

import java.awt.Component;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.beans.PropertyChangeEvent;

import java.beans.PropertyChangeListener;

import java.io.File;

import java.io.IOException;

import java.io.RandomAccessFile;

import java.nio.channels.FileChannel;

import java.nio.channels.WritableByteChannel;

import java.util.ArrayList;

import javax.swing.JButton;

import javax.swing.JComboBox;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

import javax.swing.table.DefaultTableModel;

import org.gvsig.gui.beans.graphic.GraphicEvent;

import org.gvsig.gui.beans.graphic.GraphicListener;

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

import org.gvsig.raster.util.Histogram;

import org.gvsig.raster.util.HistogramClass;

import org.gvsig.raster.util.HistogramException;

import org.gvsig.raster.util.IHistogramable;

import org.gvsig.rastertools.histogram.ui.HistogramPanel;

import com.hardcode.driverManager.DriverLoadException;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.hardcode.gdbms.engine.data.DataSource;

import com.hardcode.gdbms.engine.data.DataSourceFactory;

import com.hardcode.gdbms.engine.data.NoSuchTableException;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

import com.iver.andami.PluginServices;

import com.iver.cit.gvsig.ProjectExtension;

import com.iver.cit.gvsig.fmap.drivers.shp.DbaseFileHeaderNIO;

import com.iver.cit.gvsig.fmap.drivers.shp.DbaseFileWriterNIO;

import com.iver.cit.gvsig.fmap.edition.EditableAdapter;

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

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

import com.iver.cit.gvsig.project.ProjectFactory;

import com.iver.cit.gvsig.project.documents.table.ProjectTable;

import com.iver.cit.gvsig.project.documents.table.gui.Table;

/**

 * Listener para eventos del panel de histograma

 *

 * @version 20/03/2007

 * @author Nacho Brodin (brodin_ign@gva.es)

 * @author Borja Sanchez Zamorano (borja.sanchez@iver.es)

 */

public class HistogramPanelListener implements GraphicListener, ActionListener, PropertyChangeListener {

        private HistogramPanel                                histogramPanel = null;

        private Histogram                                        lastHistogram = null;

        public boolean                                                eventsEnabled = false;

        /**

         * Bandas que se est?n mostrando en el gr?fico. Se inicializa con las 3 bandas

         * RGB de la visualizaci?n. Este array puede tener m?s elementos ya que si las 

         * bandas no son de visualizaci?n (bandas de la imagen en disco) tendr? un elemento

         * por cada una. 

         */

        private boolean[]                                        showBands = null;

  private Color[]                                                bandsColor = {

                  Color.red,

                  Color.green,

                  Color.blue,

                  Color.cyan,

                  Color.black,

                  Color.darkGray,

                  Color.gray,

                  Color.magenta,

                  Color.yellow,

                  Color.orange};

        public HistogramPanelListener(HistogramPanel p){

                histogramPanel = p;

        }

        public HistogramPanel getHistogramPanel() {

                return histogramPanel;

        }

        public void setControlListeners(){

                getHistogramPanel().getGraphicContainer().addValueChangedListener(this);

        }

        /**

         * Actualizar cuadro de estad?sticas

         */

        private void updateStatistic() {

                int first = (int) getHistogramPanel().getBoxesValues(true)[1];

                int end = (int) getHistogramPanel().getBoxesValues(true)[0];

                getHistogramPanel().setStatistic(getLastHistogram().getBasicStats((int) first, (int) end, showBands));

        }

        /**

         * Tratamiento de todos los eventos visuales.

         */

        public void actionPerformed(ActionEvent e) {

                if (!eventsEnabled) return;

                // Boton de desmarcar todas las bandas

                if (e.getSource() == getHistogramPanel().getButtonClean()) {

                        getHistogramPanel().refreshBands(false);

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

                                showBands[i] = false;

                        updateStatistic();

                        updateGraphic();

                        return;

                }

                // Boton de marcar todas las bandas

                if (e.getSource() == getHistogramPanel().getButtonShowAll()) {

                        getHistogramPanel().refreshBands(true);

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

                                showBands[i] = true;

                        updateStatistic();

                        updateGraphic();

                        return;

                }

                //--------------------------------------

                //Selecci?n de fuente de datos del histograma

                JComboBox cbo = getHistogramPanel().getComboBoxSource();

                if (e.getSource() == cbo) {

                        showHistogram();

                        return;

                }

                // Checkbox de eliminas extremos

                if (e.getSource() == getHistogramPanel().getCBDeleteEdges()) {

                        updateGraphic();

                        return;

                }

                //--------------------------------------

                //Selecci?n de histograma acumulado y no acumulado

                JComboBox cbt = getHistogramPanel().getComboBoxType();

                if (e.getSource() == cbt) {

                        updateStatistic();

                        updateGraphic();

                        return;

                }

                //--------------------------------------

                // Boton Crear Tabla

                JButton table = getHistogramPanel().getBCreateTable();

                if (e.getSource() == table) {

                        try {

//                        -------Mostrar un fileChooser------------------

                                String fName;

                                JFileChooser chooser = new JFileChooser();

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

                                int returnVal = chooser.showOpenDialog(getHistogramPanel());

                                if (returnVal == JFileChooser.APPROVE_OPTION) {

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

                                        if (!fName.endsWith(".dbf"))

                                                fName += ".dbf";

                                        //-------------Crear el dbf----------------------

                                        DbaseFileWriterNIO dbfWrite = null;

                                        DbaseFileHeaderNIO myHeader;

                                        Value[] record;

                                        HistogramClass[][] histogram = getLastHistogram().getHistogram();

                                        int numBands = histogram.length;

                                        int numRecors = histogram[0].length;

                                        File file = new File(fName);

                                        String names[] = new String[numBands+1];

                                        int types[] = new int [numBands+1];

                                        int lengths[] = new int [numBands+1];

                                        names[0]="Value";

                                        types[0]=4;

                                        lengths[0]=15;

                                        for (int band = 0; band < numBands; band++){

                                                names[band+1]="Band"+band;

                                                types[band+1]=4;

                                                lengths[band+1]=15;

                                        }

                                        myHeader = DbaseFileHeaderNIO.createDbaseHeader(names,types,lengths);

                                        myHeader.setNumRecords(numRecors);

                                        dbfWrite = new DbaseFileWriterNIO(myHeader, (FileChannel) getWriteChannel(file.getPath()));

                                        record = new Value[numBands+1];

                                        for (int j = 0; j < numRecors; j++) {

                                                record[0] = ValueFactory.createValue(j);

                                                for (int r = 0; r < numBands; r++) {

                                                        record[r+1] = ValueFactory.createValue(histogram[r][j].getValue());

                                                }

                                                dbfWrite.write(record);

                                        }

                                        dbfWrite.close();

                                        //------------A?adir el dbf al proyecto--------------

                                        ProjectExtension ext = (ProjectExtension) PluginServices.getExtension(ProjectExtension.class);

                                        String name = file.getName();

                                        LayerFactory.getDataSourceFactory().addFileDataSource("gdbms dbf driver", name, fName);

                                        DataSource dataSource;

                                        dataSource = LayerFactory.getDataSourceFactory().createRandomDataSource(name, DataSourceFactory.AUTOMATIC_OPENING);

                                        SelectableDataSource sds = new SelectableDataSource(dataSource);

                                        EditableAdapter auxea=new EditableAdapter();

                                        auxea.setOriginalDataSource(sds);

                                        ProjectTable projectTable = ProjectFactory.createTable(name, auxea);

                                        //ext.getProject().addTable(projectTable);

                                        ext.getProject().addDocument(projectTable);

                                        Table t = new Table();

                                        t.setModel(projectTable);

                                        //projectTable.setAndamiWindow(t);

                                        PluginServices.getMDIManager().addWindow(t);

                                }

                        } catch (IOException e1) {

                                JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),getHistogramPanel().getName() + " " + PluginServices.getText(this,"table_not_create"));

                        } catch (DriverLoadException e1) {

                                JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),getHistogramPanel().getName() + " " + PluginServices.getText(this,"table_not_create"));

                        } catch (NoSuchTableException e1) {

                                JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),getHistogramPanel().getName() + " " + PluginServices.getText(this,"table_not_create"));

                        } catch (ReadDriverException e1) {

                                JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),getHistogramPanel().getName() + " " + PluginServices.getText(this,"table_not_create"));

                        }

                }

        }

        /**

         * Actualizar la variable de las bandas visibles y su componente visual.

         */

        private void refreshBands() {

                getHistogramPanel().refreshBands(true);

                showBands = new boolean[getLastHistogram().getNumBands()];

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

                        showBands[i] = true;

        }

        /**

         * Lanza los dos threads para procesar el histograma y visualizar la

         * ventana de incremento

         */

        public void showHistogram() {

                if (getHistogramPanel().getComboBoxSource().getSelectedIndex() < 0) 

                        return;

                int dataSrc = getHistogramPanel().getComboBoxSource().getSelectedIndex();

                IHistogramable iaux = (IHistogramable) ((ArrayList) getHistogramPanel().getComboSource().get(dataSrc)).get(0);

                if (lastHistogram == null) {

                        try {

                                setNewHistogram(iaux.getHistogram());

                        } catch (HistogramException e) {

                                e.printStackTrace();

                        }

                        return;

                }

                HistogramProcess histogramProcess = new HistogramProcess(iaux, this);

                IncrementableTask incrementableTask = new IncrementableTask(histogramProcess);

                histogramProcess.setIncrementableTask(incrementableTask);

                incrementableTask.showWindow();

                iaux.resetPercent();

                histogramProcess.start();

                incrementableTask.start();

        }

        /**

         * A?ade o elimina una banda de la visualizaci?n. Si la banda se est? visualizando

         * se elimina y si no entonces se muestra

         * @param band banda a visualizar o borrar del gr?fico

         */

        public void addOrRemoveGraphicBand(int band){

                if (band > showBands.length) return;

                showBands[band] = !showBands[band];

        }

        /**

         * Actualiza la grafica con los datos que ya teniamos del histograma.

         */

        private void updateGraphic() {

                if (getLastHistogram() == null) return;

                HistogramClass[][] histogramClass = getLastHistogram().getHistogramByType(Histogram.getType(getHistogramPanel().getComboBoxType().getSelectedIndex()));

                if (histogramClass == null) return;

                double[][][] datos = new double[histogramClass.length][histogramClass[0].length][2];

                for (int iBand=0; iBand < histogramClass.length; iBand++) {

                        for (int i=0; i<histogramClass[iBand].length; i++) {

                                datos[iBand][i][0] = histogramClass[iBand][i].getMin();

                                datos[iBand][i][1] = histogramClass[iBand][i].getValue();

                        }

                }

                // Definimos el principio y final de la grafica, sirve para descartar valores.

                int first = (int) getHistogramPanel().getBoxesValues(false)[1];

                int end = (int) getHistogramPanel().getBoxesValues(false)[0];

                // Si hay que eliminar los limites, quitamos el ultimo y primer valor de la grafica

                if (getHistogramPanel().getCBDeleteEdges().isSelected()) {

                        if ((first + 1) <= end)

                                first++;

                        if ((end - 1) >= first)

                                end--;

                }

                int bandCount = 0;

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

                        if (showBands[i]) bandCount++;

                }

                double[][][] newHistogram = new double[bandCount][end - first][2];

                String[] bandNames = new String[bandCount];

                bandCount = datos.length;

                int numBand = 0;

                for (int iBand = 0; iBand < datos.length; iBand++) {

                        if (!showBands[iBand]) continue;

                        for (int j=first; j<end; j++) {

                                newHistogram[numBand][j-first][0] = datos[iBand][j][0];

                                newHistogram[numBand][j-first][1] = datos[iBand][j][1];

                        }

                        bandNames[numBand] = (String) ((DefaultTableModel) getHistogramPanel().getJTableBands().getModel()).getValueAt(iBand, 1);

                        getHistogramPanel().getGraphicContainer().setBandColor(numBand, bandsColor[iBand % bandsColor.length]);

                        numBand++;

                }

                getHistogramPanel().getGraphicContainer().getPGraphic().setNewChart(newHistogram, bandNames);

        }

        /**

         * Definir el nuevo histograma, metodo pu?blico para ser invocado desde

         * histogramProcess

         * @param histograma nuevo

         */

        public void setNewHistogram(Histogram value) {

                getHistogramPanel().panelInizialited = false;

                eventsEnabled = false;

                lastHistogram = value;

                refreshBands();

                updateGraphic();

                updateStatistic();

                // Activo la ejecucion de los eventos porque seguro que ya tenemos un histograma

                eventsEnabled = true;

                getHistogramPanel().panelInizialited = true;

        }

        /**

         * Obtener ?ltimo histograma

         * @return Histogram

         */

        public Histogram getLastHistogram() {

                return lastHistogram;

        }

        /**

         * Eventos de los BoxValues

         */

        public void actionValueChanged(GraphicEvent e) {

                updateStatistic();

        }

        /**

         * Proceso para guardar una estadistica en un fichero.

         * @param path

         * @return

         * @throws IOException

         */

        private WritableByteChannel getWriteChannel(String path) throws IOException {

                WritableByteChannel channel;

                File f = new File(path);

                if (!f.exists()) {

                        System.out.println("Creando fichero " + f.getAbsolutePath());

                        if (!f.createNewFile()) {

                                throw new IOException("Cannot create file " + f);

                        }

                }

                RandomAccessFile raf = new RandomAccessFile(f, "rw");

                channel = raf.getChannel();

                return channel;

        }

        /**

         *  Cuando se selecciona/deselecciona una banda

         */

        public void propertyChange(PropertyChangeEvent evt) {

                if (!eventsEnabled) return;

                int countRow = ((DefaultTableModel) histogramPanel.getJTableBands().getModel()).getRowCount();

                for (int i=0; i<countRow; i++) {

                        showBands[i] = ((Boolean) ((DefaultTableModel) histogramPanel.getJTableBands().getModel()).getValueAt(i, 0)).booleanValue();

                }

                updateGraphic();

                updateStatistic();

        }

}