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.enhanced.ui;

import java.awt.Color;

import java.awt.Dimension;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.Insets;

import java.awt.event.KeyEvent;

import java.awt.event.KeyListener;

import java.text.NumberFormat;

import javax.swing.JCheckBox;

import javax.swing.JFormattedTextField;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.text.DefaultFormatterFactory;

import javax.swing.text.NumberFormatter;

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

import org.gvsig.fmap.dal.coverage.dataset.Buffer;

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

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.i18n.Messages;

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

import org.gvsig.raster.swing.preview.PreviewBasePanelTab;

import org.gvsig.raster.tools.app.basic.tool.enhanced.graphics.InputHistogram;

import org.gvsig.raster.tools.app.basic.tool.enhanced.graphics.OutputHistogram;

import org.gvsig.tools.locator.LocatorException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Panel que contiene las gr?ficas para la modificaci?n del realce.

 * 

 * 19/02/2008

 * @author Nacho Brodin nachobrodin@gmail.com

 */

public class GraphicsPanel extends JPanel implements PreviewBasePanelTab, KeyListener {

        private static final long   serialVersionUID = 1L;

        private Logger              log              = LoggerFactory.getLogger(GraphicsPanel.class);

        private InputHistogram      input            = null;

        private OutputHistogram     output           = null;

        private JPanel              controls         = null;

        private JLabel              llevels          = null;

        private JFormattedTextField tlevels          = null;

        private JCheckBox           cbRGB            = null;

        private HistogramComputer           hist             = null;

        private BufferHistogram     histRGB          = null;

        private double[]            minList          = null;

        private double[]            maxList          = null;

        private FLyrRaster          lyr              = null;

        /**

         * Crea una instancia del panel GraphicsPanel

         */

        public GraphicsPanel(FLyrRaster lyr) {

                try {

                        this.lyr = lyr;

                        hist = lyr.getDataStore().getHistogramComputer();

                        histRGB = RasterLocator.getManager().getRasterUtils().convertHistogramToRGB(hist.getBufferHistogram());

                        minList = lyr.getDataStore().getStatistics().getMin();

                        maxList = lyr.getDataStore().getStatistics().getMax();

                        if (minList == null)

                                minList = new double[] { 0 };

                        if (maxList == null)

                                maxList = new double[] { 255 };

                } catch (HistogramException e) {

                        hist = null;

                } catch (LocatorException e) {

                } catch (ProcessInterruptedException e) {

                        e.printStackTrace();

                }

                init();

                if (lyr.getDataStore().getDataType()[0] == Buffer.TYPE_BYTE)

                        getRGB().setSelected(true);

                else

                        getRGB().setEnabled(false);

        }

        /**

         * Inicializaci?n de los controles gr?ficos.

         */

        private void init() {

                setLayout(new GridBagLayout());

                setBorder(javax.swing.BorderFactory.createTitledBorder(null, Messages.getText("histograms"), javax.swing.border.TitledBorder.DEFAULT_JUSTIFICATION, javax.swing.border.TitledBorder.DEFAULT_POSITION, null, null));

                GridBagConstraints gridBagConstraints = null;

                JLabel label1 = new JLabel(Messages.getText("input_hist"));

                gridBagConstraints = new GridBagConstraints();

                gridBagConstraints.insets = new Insets(4, 4, 4, 4);

                gridBagConstraints.anchor = java.awt.GridBagConstraints.CENTER;

                label1.setHorizontalAlignment(javax.swing.SwingConstants.CENTER);

                label1.setPreferredSize(new Dimension(150, 14));

                add(label1, gridBagConstraints);

                JLabel label2 = new JLabel(Messages.getText("output_hist"));

                gridBagConstraints = new GridBagConstraints();

                gridBagConstraints.gridx = 2;

                gridBagConstraints.gridy = 0;

                gridBagConstraints.insets = new Insets(4, 4, 4, 4);

                gridBagConstraints.anchor = java.awt.GridBagConstraints.CENTER;

                label2.setHorizontalAlignment(javax.swing.SwingConstants.CENTER);

                label2.setPreferredSize(new Dimension(150, 14));

                add(label2, gridBagConstraints);

                gridBagConstraints = new GridBagConstraints();

                gridBagConstraints.gridx = 0;

                gridBagConstraints.gridy = 1;

                gridBagConstraints.fill = GridBagConstraints.BOTH;

                gridBagConstraints.weightx = 1.0;

                gridBagConstraints.weighty = 1.0;

                gridBagConstraints.insets = new Insets(4, 4, 4, 4);

                add(getInputHistogram(), gridBagConstraints);

                gridBagConstraints = new GridBagConstraints();

                gridBagConstraints.gridx = 1;

                gridBagConstraints.gridy = 1;

                gridBagConstraints.fill = GridBagConstraints.VERTICAL;

                gridBagConstraints.insets = new Insets(4, 4, 4, 4);

                add(getControls(), gridBagConstraints);

                gridBagConstraints = new GridBagConstraints();

                gridBagConstraints.gridx = 2;

                gridBagConstraints.gridy = 1;

                gridBagConstraints.fill = GridBagConstraints.BOTH;

                gridBagConstraints.weightx = 1.0;

                gridBagConstraints.weighty = 1.0;

                gridBagConstraints.insets = new Insets(4, 4, 4, 4);

                add(getOutputHistogram(), gridBagConstraints);                

                setLevelsEnabled(false);

        }

        /**

         * Obtiene el histograma de entrada

         * @return Histogram

         */

        public HistogramComputer getHistogram() {

                return hist;

        }

        /**

         * Obtiene el gr?fico con el histograma de entrada

         * @return InputHistogram

         */

        public InputHistogram getInputHistogram() {

                if (input == null) {

                        try {

                                input = new InputHistogram(hist.getBufferHistogram(), lyr, minList, maxList);

                                input.setHistogram(hist.getBufferHistogram(), minList, maxList);

                        } catch (ProcessInterruptedException e) {

                                log.debug("Proceso de calculo de histograma interrumpido", this, e);

                        } catch (HistogramException e) {

                                log.debug("Excepci?n en el c?lculo del histograma", this, e);

                        }

                }

                return input;

        }

        /**

         * Actualiza el histograma a visualizar por si se ha cambiado de RGB a normal.

         */

        public void updateHistogram() {

                if (getRGB().isSelected()) {

                        double[] min = new double[minList.length];

                        double[] max = new double[maxList.length];

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

                                min[i] = 0.0;

                                max[i] = 255.0;

                        }

                        input.setHistogram(histRGB, min, max);

                } else {

                        try {

                                input.setHistogram(hist.getBufferHistogram(), minList, maxList);

                        } catch (ProcessInterruptedException e) {

                                log.debug("Proceso de calculo de histograma interrumpido", this, e);

                        } catch (HistogramException e) {

                                log.debug("Excepci?n en el c?lculo del histograma", this, e);

                        }

                }

        }

        /**

         * Obtiene el gr?fico con el histograma de salida

         * @return OutputHistogram

         */

        public OutputHistogram getOutputHistogram() {

                if (output == null) {

                        try {

                                output = new OutputHistogram(hist.getBufferHistogram(), lyr, minList, maxList);

                        } catch (ProcessInterruptedException e) {

                                log.debug("Proceso de calculo de histograma interrumpido", this, e);

                        } catch (HistogramException e) {

                                log.debug("Excepci?n en el c?lculo del histograma", this, e);

                        }

                }

                return output;

        }

        /**

         * Obtiene el panel con los controles

         * @return JPanel

         */

        public JPanel getControls() {

                if(controls == null) {

                        controls = new JPanel();

                        controls.setLayout(new GridBagLayout());

                        llevels = new JLabel(Messages.getText("levels"));

                        GridBagConstraints gbc = new GridBagConstraints();

                        gbc.weightx = 1;

                        gbc.insets = new Insets(0, 5, 0, 5);

                        gbc.fill = GridBagConstraints.BOTH;

                        controls.add(llevels, gbc);

                        gbc.gridy = 1;

                        controls.add(getLevels(), gbc);

                        gbc.gridy = 2;

                        controls.add(getRGB(), gbc);

                }

                return controls;

        }

        /**

         * Activa o desactiva el control de niveles de posterizaci?n en level-slice

         * de recorte.

         * @param enabled

         */

        public void setLevelsEnabled(boolean enabled) {

                getLevels().setEnabled(enabled);

                if(enabled)

                        getLevels().setBackground(Color.WHITE);

                else

                        getLevels().setBackground(getControls().getBackground());

        }

        /**

         * Obtiene el n?mero de niveles.

         * @return

         */

        public JFormattedTextField getLevels() {

                if (tlevels == null) {

                        NumberFormat doubleDisplayFormat = NumberFormat.getNumberInstance();

                        doubleDisplayFormat.setMinimumFractionDigits(0);

                        NumberFormat doubleEditFormat = NumberFormat.getNumberInstance();

                        tlevels = new JFormattedTextField(new DefaultFormatterFactory(new NumberFormatter(doubleDisplayFormat), new NumberFormatter(doubleDisplayFormat), new NumberFormatter(doubleEditFormat)));

                        tlevels.addKeyListener(this);

                }

                return tlevels;

        }

        /**

         * Obtiene el checkbox que indica si se va a tratar la imagen como un checkbox

         * @return

         */

        public JCheckBox getRGB() {

                if (cbRGB == null) {

                        cbRGB = new JCheckBox("RGB");

                }

                return cbRGB;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.beans.previewbase.IUserPanelInterface#getPanel()

         */

        public JPanel getPanel() {

                return this;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.beans.previewbase.IUserPanelInterface#getTitle()

         */

        public String getTitle() {

                return "";

        }

        /**

         * Asigna el listener para gestionar el evento de movimiento de gr?ficos

         * @param listener

         */

        public void setListener(EnhancedListener listener) {

                getInputHistogram().setListener(listener);

        }

        /*

         * (non-Javadoc)

         * @see java.awt.event.KeyListener#keyPressed(java.awt.event.KeyEvent)

         */

        public void keyPressed(KeyEvent e) {

                Long lValue = (Long) getLevels().getValue();

                int value = lValue.intValue();

                boolean change = false;

                switch (e.getKeyCode()) {

                        case KeyEvent.VK_UP:

                                value++;

                                change = true;

                                break;

                        case KeyEvent.VK_DOWN:

                                value--;

                                change = true;

                                break;

                }

                if (change) {

                        if (value > 30)

                                value = 30;

                        if (value < 2)

                                value = 2;

                        if (!getLevels().getValue().equals(new Long(value))) {

                                getLevels().setValue(new Long(value));

                                getLevels().postActionEvent();

                        }

                }

        }

        public void keyReleased(KeyEvent e) {}

        public void keyTyped(KeyEvent e) {}

}