Application: gvSIG desktop » gvSIG raster

/* 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.raster.tools.app.basic.tool.histogram;

import java.awt.Color;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.beans.PropertyChangeEvent;

import java.beans.PropertyChangeListener;

import java.io.File;

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.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.project.documents.table.TableDocument;

import org.gvsig.app.project.documents.table.TableManager;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

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

import org.gvsig.fmap.dal.coverage.datastruct.BufferHistogram;

import org.gvsig.fmap.dal.coverage.datastruct.HistogramClass;

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

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

import org.gvsig.fmap.dal.coverage.store.props.HistogramComputer;

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

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

import org.gvsig.fmap.dal.feature.EditableFeature;

import org.gvsig.fmap.dal.feature.EditableFeatureType;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.NewFeatureStoreParameters;

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

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

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

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

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

import org.gvsig.raster.tools.app.basic.raster.process.HistogramProcess;

import org.gvsig.raster.tools.app.basic.raster.process.IProcessActions;

import org.gvsig.raster.tools.app.basic.raster.process.StatisticsProcess;

import org.gvsig.raster.tools.app.basic.tool.histogram.ui.HistogramPanel;

import org.gvsig.tools.dataTypes.DataTypes;

/**

 * Listener para eventos del panel de histograma

 *

 * @version 20/03/2007

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

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

 */

public class HistogramPanelListener implements GraphicListener, ActionListener, PropertyChangeListener, IProcessActions {

        private HistogramPanel histogramPanel    = null;

        /**

         * Variable que apunta a uno de los dos histogramas posibles: Real o RGB

         */

        private BufferHistogram      lastHistogram     = null;

        /**

         * Histograma original sin convertir a RGB

         */

        private BufferHistogram      lastHistogramReal = null;

        /**

         * Histograma convertido a rango RGB

         */

        private BufferHistogram      lastHistogramRGB  = 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 FLyrRaster     lyr               = null;

        private Color[]                                                bandsColor     = {

                        Color.red,

                        Color.green,

                        Color.blue,

                        Color.cyan,

                        Color.black,

                        Color.darkGray,

                        Color.gray,

                        Color.magenta,

                        Color.yellow,

                        Color.orange};

        /**

         * Guardamos el histogramable para hacer el histogramProcess

         */

        HistogramComputer histogramable = null;

        /**

         * Constructor. Asigna el panel del histograma

         * @param p Panel

         */

        public HistogramPanelListener(HistogramPanel histogramPanel) {

                this.histogramPanel = histogramPanel;

        }

        /**

         * Obtiene el panel del histograma

         * @return HistogramPanel

         */

        private HistogramPanel getHistogramPanel() {

                return histogramPanel;

        }

        /**

         * Asigna la capa para obtener las fuentes de datos tanto del

         * datasource como de la visualizaci?n.

         * @param lyr Capa

         */

        public void setLayer(FLyrRaster lyr) {

                this.lyr = lyr;

        }

        public void setControlListeners() {

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

        }

        /**

         * Actualizar cuadro de estad?sticas

         */

        private void updateStatistic() {

                if (getLastHistogram() == null) {

                        getHistogramPanel().setStatistic(null);

                        return;

                }

                double first = getHistogramPanel().getGraphicContainer().getX1();

                double end = getHistogramPanel().getGraphicContainer().getX2();

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

        }

        /**

         * Tratamiento de todos los eventos visuales.

         */

        @SuppressWarnings("unchecked")

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

                        ArrayList comboSource = getHistogramPanel().getComboSource();

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

                                String name = (String) ((ArrayList) comboSource.get(i)).get(1);

                                if (name.compareTo(RasterToolsUtil.getText(this, "datos_visualizados")) == 0) {

                                        ((ArrayList) comboSource.get(i)).remove(0);

                                        ((ArrayList) comboSource.get(i)).add(0, ((FLyrRaster) lyr).getRender().getLastRenderBuffer().getHistogramComputer());

                                }

                                if (name.compareTo(RasterToolsUtil.getText(this, "imagen_completa")) == 0) {

                                        ((ArrayList) comboSource.get(i)).remove(0);

                                        ((ArrayList) comboSource.get(i)).add(0, ((FLyrRaster) lyr).getDataStore().getHistogramComputer());

                                }

                        }

                        showHistogram();

                        return;

                }

                // Checkbox de eliminas extremos

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

                        updateGraphic();

                        return;

                }

                // Checkbox de RGB

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

                        selectHistogram();

                        updateStatistic();

                        updateGraphic();

                        return;

                }

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

                //Selecci?n de histograma acumulado y no acumulado

                JComboBox cbt = getHistogramPanel().getComboBoxType();

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

                        getHistogramPanel().getGraphicContainer().getPGraphic().setViewType(getHistogramPanel().getComboBoxType().getSelectedIndex());

                        return;

                }

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

                // Boton Crear Tabla

                JButton table = getHistogramPanel().getBCreateTable();

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

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

                                String fName;

                                JFileChooser chooser = new JFileChooser();

                                chooser.setDialogTitle(RasterToolsUtil.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----------------------

                                        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] = DataTypes.INT;

                                        lengths[0] = 15;

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

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

                                                types[band + 1] = DataTypes.DOUBLE;

                                                lengths[band + 1] = 15;

                                        }

                                        FeatureStore store = null;

                                        try {

                                                DataManager manager = DALLocator.getDataManager();

                                                DataServerExplorerParameters eparams = manager.createServerExplorerParameters("FilesystemExplorer");

                                                eparams.setDynValue("initialpath", "/data");

                                                DataServerExplorer serverExplorer = manager.openServerExplorer("FilesystemExplorer", eparams);

                                                NewFeatureStoreParameters sparams = (NewFeatureStoreParameters)serverExplorer.getAddParameters("DBF");

                                                ((FilesystemStoreParameters) sparams).setFile(file);

                                                EditableFeatureType featureType = sparams.getDefaultFeatureType();

                                                //Creating table structure

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

                                                        featureType.add(names[i], types[i], lengths[i]);

                                                }

                                                manager.newStore("FilesystemExplorer", "DBF", sparams, true);

                                                store = (FeatureStore)manager.openStore(sparams.getDataStoreName(), sparams);

                                                //Loading data

                                                store.edit();

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

                                                        EditableFeature feature = store.createNewFeature();

                                                        feature.set(names[0], j);

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

                                                                feature.set(names[r + 1], histogram[r][j].getValue());

                                                        }

                                                        store.insert(feature);

                                                }

                                                store.finishEditing();

                                        } catch (DataException e1) {

                                                JOptionPane.showMessageDialog(null, getHistogramPanel().getName() + " " + RasterToolsUtil.getText(this,"table_not_create"));

                                        } catch (ValidateDataParametersException e2) {

                                                JOptionPane.showMessageDialog(null, getHistogramPanel().getName() + " " + RasterToolsUtil.getText(this,"table_not_create"));

                                        }

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

                                        ApplicationManager application = ApplicationLocator.getManager(); 

                                        TableDocument tableDocument = (TableDocument)application.getProjectManager().createDocument(TableManager.TYPENAME, 

                                                        "Histogram_"+lyr.getName());

                                        tableDocument.setStore(store);

                                        application.getCurrentProject().add(tableDocument);

                                        tableDocument.getFactory().getMainWindow(tableDocument);

                                }

                }

        }

        /**

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

         */

        private void refreshBands() {

                getHistogramPanel().refreshBands(true);

                if (getLastHistogram() == null)

                        showBands = new boolean[0];

                else

                        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

         */

        @SuppressWarnings("unchecked")

        public void showHistogram() {

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

                        return;

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

                histogramable = (HistogramComputer) ((ArrayList) getHistogramPanel().getComboSource().get(dataSrc)).get(0);

                if (getLastHistogram() == null) {

                        try {

                                if (histogramable != null)

                                        setNewHistogram(histogramable.getBufferHistogram());

                                else

                                        setNewHistogram(null);

                        } catch (HistogramException e) {

                                RasterToolsUtil.messageBoxError("histogram_error", getHistogramPanel(), e);

                                return;

                        } catch (ProcessInterruptedException e) {

                        } 

                        return;

                }

                // Calculo las estadisticas para luego hacer el proceso del histograma.

                // El parametro object es el que se le pasara al siguiente proceso

                // Mirar el metodo end()

                StatisticsProcess statisticsProcess = new StatisticsProcess();

                statisticsProcess.setActions(this);

                statisticsProcess.addParam("layer", lyr);

                statisticsProcess.addParam("force", Boolean.FALSE);

                statisticsProcess.start();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.rastertools.IProcessActions#end(java.lang.Object)

         */

        public void end(Object object) {

                // Si tenemos un histograma en el parametro, es que ya ha finalizado el proceso

                if (object instanceof BufferHistogram) {

                        setNewHistogram((BufferHistogram) object);

                        return;

                }

                // Si no tenemos un histograma, osease, tenemos un layer, 

                // es que aun necesitamos calcularlo

                if (object instanceof FLyrRaster) {

                        if (histogramable != null) {

                                HistogramProcess histogramProcess = new HistogramProcess();

                                histogramProcess.setActions(this);

                                histogramProcess.addParam("histogramable", histogramable);

                                histogramProcess.start();

                        } else {

                                setNewHistogram(null);

                        }

                }

        }

        /**

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

         */

        private void updateGraphic() {

                if (getLastHistogram() == null) {

                        getHistogramPanel().getGraphicContainer().getPGraphic().cleanChart();

                        return;

                }

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

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

                int end = (int) getHistogramPanel().getBoxValueX2();

                //first = 0;

                //end = 100;

                int min = 0;

                int max = histogramClass[0].length - 1;

                first = min + ((first * (max - min))/ 100);

                end = min + ((end * (max - min))/ 100);

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

                if (getHistogramPanel().getCheckBoxDeleteEdges().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 + 1][2];

                String[] bandNames = new String[bandCount];

                int numBand = 0;

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

                        if (!showBands[iBand])

                                continue;

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

                                try {

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

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

                                } catch (ArrayIndexOutOfBoundsException e) {

                                        RasterToolsUtil.messageBoxError("Error al crear el array del histograma. DataType: " + getHistogramPanel().getDataType() + " Posici?n: " + j, this, e);

                                }

                        }

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

        }

        public void selectHistogram() {

                if (getHistogramPanel().getCheckBoxRGB().isSelected())

                        lastHistogram = lastHistogramRGB;

                else

                        lastHistogram = lastHistogramReal;

        }

        /**

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

         * histogramProcess

         * @param histograma nuevo

         */

        public void setNewHistogram(BufferHistogram histogram) {

                getHistogramPanel().panelInizialited = false;

                eventsEnabled = false;

                this.lastHistogramReal = histogram;

                this.lastHistogramRGB = RasterLocator.getManager().getRasterUtils().convertHistogramToRGB(lastHistogramReal);

                selectHistogram();

                refreshBands();

                updateStatistic();

                updateGraphic();

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

                eventsEnabled = true;

                getHistogramPanel().panelInizialited = true;

        }

        /**

         * Obtener ?ltimo histograma

         * @return Histogram

         */

        public BufferHistogram getLastHistogram() {

                return lastHistogram;

        }

        /**

         * Eventos de los BoxValues

         */

        public void actionValueChanged(GraphicEvent e) {

                updateStatistic();

                updateGraphic();

        }

        /**

         *  Cuando se selecciona/deselecciona una banda

         */

        public void propertyChange(PropertyChangeEvent evt) {

                if (!eventsEnabled)

                        return;

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

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

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

                updateStatistic();

                updateGraphic();

        }

        public void interrupted() {}

}