Application: gvSIG desktop

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2017 gvSIG Association

 *

 * 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.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.swing.buffer.impl.histogram;

import java.awt.Color;

import java.awt.Container;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.ItemListener;

import java.util.List;

import java.util.Locale;

import javax.swing.JComponent;

import javax.swing.JScrollPane;

import javax.swing.JTable;

import javax.swing.JViewport;

import javax.swing.ListSelectionModel;

import javax.swing.SwingUtilities;

import javax.swing.event.TableModelEvent;

import javax.swing.event.TableModelListener;

import javax.swing.table.DefaultTableColumnModel;

import javax.swing.table.TableColumn;

import javax.swing.table.TableColumnModel;

import org.jfree.data.xy.XYSeries;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.raster.lib.buffer.api.statistics.HistogramBand;

import org.gvsig.raster.lib.buffer.api.statistics.Statistics;

import org.gvsig.raster.swing.buffer.RasterSwingBufferLocator;

import org.gvsig.raster.swing.buffer.RasterSwingBufferManager;

import org.gvsig.raster.swing.buffer.SelectableBandsTableModel;

import org.gvsig.raster.swing.buffer.exceptions.RasterCreatingPanelException;

import org.gvsig.raster.swing.buffer.histogram.HistogramPanel;

import org.gvsig.raster.swing.buffer.impl.histogram.graphic.GraphicContainer;

import org.gvsig.raster.swing.buffer.impl.histogram.graphic.GraphicEvent;

import org.gvsig.raster.swing.buffer.impl.histogram.graphic.GraphicListener;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

/**

 * @author fdiaz

 *

 */

public class HistogramPanelController extends HistogramPanelView implements HistogramPanel {

    /**

     *

     */

    private static final long serialVersionUID = 3098738174477175358L;

    /**

     *

     */

    private static final Logger LOG = LoggerFactory.getLogger(HistogramPanelController.class);

    private static String[] sources = { "_full", "_displayed_data" };

    private static String[] types = { "_normal", "_accumulated", "_logarithmic" };

    private static Color[] defaultColors = new Color[] {

        Color.red,

        Color.green,

        Color.blue,

        Color.darkGray,

        Color.cyan,

        Color.magenta,

        Color.yellow,

        Color.black,

        Color.orange };

    private Statistics statistics;

    private SelectableBandsTableModel tableModel;

    private Statistics fullStatistics;

    private Statistics visibleStatistics;

    private boolean enabledVisibleStatistics;

    private GraphicContainer graphicContainer;

    private Color[] colors;

    boolean initialized;

    /**

     * @throws RasterCreatingPanelException

     */

    public HistogramPanelController() throws RasterCreatingPanelException {

        this.initialized = false;

        translate();

        initializeComponents();

    }

    public void set(Statistics statistics) throws RasterCreatingPanelException{

        this.fullStatistics = statistics;

        this.statistics = statistics;

        if(this.tableModel != null){

            init();

        }

    }

    public void set(SelectableBandsTableModel tableModel) throws RasterCreatingPanelException{

        this.tableModel = tableModel;

        if(this.statistics != null){

            init();

        }

    }

    public void setVisibleStatistics(Statistics visibleStatistics) {

        this.visibleStatistics = visibleStatistics;

        if(this.initialized){

            updateComponents();

        }

    }

    @Override

    public void setBandColors(Color[] colors) {

        this.colors = colors;

        if(this.initialized){

            updateComponents();

        }

    }

    private void init() throws RasterCreatingPanelException{

        RasterSwingBufferManager swingManager = RasterSwingBufferLocator.getSwingManager();

        swingManager.makeASelectableBandsTable(tblBands, this.tableModel);

        ((SelectableBandsTableModel)tblBands.getModel()).addTableModelListener(new TableModelListener() {

            @Override

            public void tableChanged(TableModelEvent e) {

                updateComponents();

            }

        });

        updateComponents();

        this.initialized = true;

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    private void initializeComponents() {

        cmbSource.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                updateComponents();

            }

        });

        cmbType.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                updateComponents();

            }

        });

        btnAllBands.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doSelectAllBands();

            }

        });

        btnClearBands.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doUnselectAllBands();

            }

        });

        btnExportTable.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doExportTable();

            }

        });

        cmbSource.setSelectedIndex(0);

        cmbSource.setEditable(false);

        tblBands.setRowSelectionAllowed(true);

        tblBands.setColumnSelectionAllowed(false);

        tblBands.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        tblStatistics.setRowSelectionAllowed(false);

        tblBands.setColumnSelectionAllowed(false);

        chkRemoveTails.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doRemoveTails();

            }

        });

        chkShowStatistics.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doShowStatistics();

            }

        });

        pnlHistogram.setLayout(new GridBagLayout());

        GridBagConstraints c = new GridBagConstraints();

        c.weightx = 1.0;

        c.weighty = 1.0;

        c.fill = GridBagConstraints.BOTH;

        c.gridx = 0;

        c.gridy = 0;

        this.graphicContainer = getGraphicContainer();

        this.graphicContainer.addValueChangedListener(new GraphicListener() {

            @Override

            public void actionValueChanged(GraphicEvent e) {

                updateComponents();

            }

        });

        pnlHistogram.add(getGraphicContainer(), c);

    }

    /**

     * This method initializes jPanel

     *

     * @return javax.swing.JPanel

     */

    public GraphicContainer getGraphicContainer() {

        if (this.graphicContainer == null) {

            this.graphicContainer = new GraphicContainer(true);

        }

        return this.graphicContainer;

    }

    protected void doShowStatistics() {

            boolean showStatistics = chkShowStatistics.isSelected();

            JScrollPane scrollPane = getScrollPane(tblStatistics);

            if(scrollPane!=null){

                scrollPane.setVisible(showStatistics);

            }

            tblStatistics.setVisible(showStatistics);

            tblStatistics.getTableHeader().setVisible(showStatistics);

            btnExportTable.setVisible(showStatistics);

            chkRemoveTails.setVisible(showStatistics);

    }

    private JScrollPane getScrollPane(JTable table) {

        JScrollPane scrollPane = null;

        final Container parent = SwingUtilities.getUnwrappedParent(table);

        if (parent instanceof JViewport) {

            final Container container = parent.getParent();

            if (container instanceof JScrollPane) {

                scrollPane = (JScrollPane) container;

            }

        }

        return scrollPane;

    }

    protected void doRemoveTails() {

        updateComponents();

    }

    protected void doExportTable() {

        // TODO Auto-generated method stub

    }

    protected void doUnselectAllBands() {

        ((SelectableBandsTableModel)tblBands.getModel()).unselectAllBands();

    }

    protected void doSelectAllBands() {

        ((SelectableBandsTableModel)tblBands.getModel()).selectAllBands();

    }

    private void translate() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        lblSource.setText(i18nManager.getTranslation(lblSource.getText()));

        lblSource.setToolTipText(i18nManager.getTranslation(lblSource.getToolTipText()));

        lblType.setText(i18nManager.getTranslation(lblType.getText()));

        lblType.setToolTipText(i18nManager.getTranslation(lblType.getToolTipText()));

        lblBands.setText(i18nManager.getTranslation(lblBands.getText()));

        lblBands.setToolTipText(i18nManager.getTranslation(lblBands.getToolTipText()));

        cmbSource.removeAllItems();

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

            cmbSource.addItem(i18nManager.getTranslation(sources[i]));

        }

        cmbType.removeAllItems();

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

            cmbType.addItem(i18nManager.getTranslation(types[i]));

        }

        btnAllBands.setText(i18nManager.getTranslation(btnAllBands.getText()));

        btnAllBands.setToolTipText(i18nManager.getTranslation(btnAllBands.getToolTipText()));

        btnClearBands.setText(i18nManager.getTranslation(btnClearBands.getText()));

        btnClearBands.setToolTipText(i18nManager.getTranslation(btnClearBands.getToolTipText()));

        btnExportTable.setText(i18nManager.getTranslation(btnExportTable.getText()));

        btnExportTable.setToolTipText(i18nManager.getTranslation(btnExportTable.getToolTipText()));

        chkRemoveTails.setText(i18nManager.getTranslation(chkRemoveTails.getText()));

        chkRemoveTails.setToolTipText(i18nManager.getTranslation(chkRemoveTails.getToolTipText()));

        chkShowStatistics.setText(i18nManager.getTranslation(chkShowStatistics.getText()));

        chkShowStatistics.setToolTipText(i18nManager.getTranslation(chkShowStatistics.getToolTipText()));

    }

    /**

     * @param locale

     *

     */

    public void setLocate(Locale locale) {

        Locale l = super.getLocale();

        if (!l.equals(locale)) {

            translate();

        }

        super.setLocale(locale);

    }

    private void updateComponents() {

        if (this.fullStatistics == null || this.tableModel == null) {

            return;

        }

        if(this.enabledVisibleStatistics){

            this.cmbSource.setEnabled(true);

        } else {

            this.cmbSource.setSelectedIndex(0);

            this.cmbSource.setEnabled(false);

        }

        if (cmbSource.getSelectedIndex() == 0) {

            this.statistics = this.fullStatistics;

        } else {

            this.statistics = this.visibleStatistics;

        }

        updateStatisticsTable();

        updateHistogramGraphic();

    }

    /**

     *

     */

    private void updateStatisticsTable() {

        List<Integer> selectedBands = ((SelectableBandsTableModel) tblBands.getModel()).getSelectedBands();

        StatisticsTableModel statisticsModel = new StatisticsTableModel(this.statistics, selectedBands);

        tblStatistics.setModel(statisticsModel);

        tblStatistics.setColumnModel(new DefaultTableColumnModel());

        TableColumnModel statisticsColumnModel = tblStatistics.getColumnModel();

        statisticsColumnModel.setColumnSelectionAllowed(false);

        for (int i = 0; i < statisticsModel.getColumnCount(); i++) {

            TableColumn tableColumn = new TableColumn(i);

            tableColumn.setIdentifier(statisticsModel.getColumnName(i));

            tableColumn.setHeaderValue(statisticsModel.getColumnName(i));

            if (i > StatisticsTableModel.COLUMN_LENGTH) {

                tableColumn.setCellRenderer(new PerhapsNotValidValueCellRenderer());

            }

            statisticsColumnModel.addColumn(tableColumn);

        }

    }

    private void updateHistogramGraphic() {

        if(this.statistics==null){

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

            return;

        }

        List<Integer> selectedBands = ((SelectableBandsTableModel)tblBands.getModel()).getSelectedBands();

        Color[] bandsColor = new Color[selectedBands.size()];

        String[] bandNames = new String[selectedBands.size()];

        int index = 0;

        for (Integer selectedBand : selectedBands) {

            if( colors == null || colors.length!=tblBands.getModel().getRowCount()){

                bandsColor[index] = defaultColors[index % defaultColors.length];

            } else {

                bandsColor[index] = colors[selectedBand];

            }

            bandNames[index] = selectedBand.toString();

            getGraphicContainer().setBandColor(index, bandsColor[index]);

            index++;

        }

        getGraphicContainer().getPGraphic().setViewType(cmbType.getSelectedIndex());

        HistogramBand[] histogramBands = this.statistics.getHistogram();

        XYSeries[] datos = new XYSeries[selectedBands.size()];

        for (int iBand = 0; iBand < selectedBands.size(); iBand++) {

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

            int first = (int) getGraphicContainer().getX1(); //.getX1(); //(int) getHistogramPanel().getBoxValueX1();

            int end = (int) getGraphicContainer().getX2(); //(int) getHistogramPanel().getBoxValueX2();

            HistogramBand histogramBand = histogramBands[selectedBands.get(iBand)];

            int min = 0;

            int max = histogramBand.getNumClasses() - 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 (chkRemoveTails.isSelected()) {

                if ((first + 1) <= end)

                    first++;

                if ((end - 1) >= first)

                    end--;

            }

            datos[iBand] = new XYSeries(bandNames[iBand]);

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

                datos[iBand].add(histogramBand.getIntervalMin(i), histogramBand.getValueCount(i));

            }

        }

        getGraphicContainer().getPGraphic().setNewChart(datos);

    }

    @Override

    public void setEnabledVisibleStatistics(boolean enabledVisibleStatistics) {

        this.enabledVisibleStatistics = enabledVisibleStatistics;

        if(this.initialized){

            updateComponents();

        }

    }

    @Override

    public boolean isEnabledVisibleStatistics() {

        return this.enabledVisibleStatistics;

    }

    public void addChangeItemSourceListener(ItemListener listener){

        if(this.initialized){

            cmbSource.addItemListener(listener);

        }

    }

    public void removeChangeItemSourceListener(ItemListener listener){

        if(this.initialized){

            cmbSource.addItemListener(listener);

        }

    }

}