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.app.mainplugin.colortable;

import java.awt.Component;

import java.awt.event.ComponentAdapter;

import java.awt.event.ComponentEvent;

import java.awt.event.ItemEvent;

import java.awt.event.ItemListener;

import java.io.File;

import java.util.List;

import java.util.Locale;

import javax.swing.DefaultComboBoxModel;

import javax.swing.JComponent;

import javax.swing.JLabel;

import javax.swing.JList;

import javax.swing.JSplitPane;

import javax.swing.ListCellRenderer;

import javax.swing.SwingUtilities;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import javax.swing.event.TableModelEvent;

import javax.swing.event.TableModelListener;

import org.apache.commons.lang3.tuple.Pair;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.PluginsLocator;

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

import org.gvsig.fmap.dal.raster.BandDescriptor;

import org.gvsig.fmap.dal.raster.RasterSet;

import org.gvsig.fmap.dal.raster.RasterStore;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.raster.api.RasterLayer;

import org.gvsig.fmap.mapcontext.raster.swing.MapContextRasterSwingLocator;

import org.gvsig.fmap.mapcontext.raster.swing.previewer.OneLayerPreviewer;

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

import org.gvsig.raster.lib.legend.api.RasterLegend;

import org.gvsig.raster.lib.legend.api.RasterLegendLocator;

import org.gvsig.raster.lib.legend.api.colorinterpretation.ColorInterpretation;

import org.gvsig.raster.lib.legend.api.colortable.ColorTable;

import org.gvsig.raster.lib.legend.api.colortable.colortableclass.ColorTableClass;

import org.gvsig.raster.swing.legend.RasterSwingLegendLocator;

import org.gvsig.raster.swing.legend.colortable.editor.ColorTableEditorPanel;

import org.gvsig.raster.swing.legend.colortable.selector.ColorTableSelectorPanel;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.i18n.I18nManager;

/**

 * @author fdiaz

 *

 */

public class ColorTablePagePanelController extends ColorTablePagePanelView implements ColorTablePagePanel {

    /**

     *

     */

    private static final long serialVersionUID = 798930218545360652L;

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

    private ColorTableSelectorPanel selector;

    private ColorTableEditorPanel editor;

    private ColorTable colorTable;

    private JSplitPane splPanelRight = new JSplitPane(JSplitPane.VERTICAL_SPLIT);

    private RasterLayer rasterLayer;

    private OneLayerPreviewer previewer;

    private boolean initialized;

    private TableModelListener editorTableModelListener;

    private RasterLayer clonedLayerForPreview;

    private boolean isChanged;

    /**

     *

     */

    public ColorTablePagePanelController() {

        this.initialized = false;

        DisposeUtils.bind(this);

        initializeComponents();

        translate();

    }

    private void initializeComponents() {

        selector = RasterSwingLegendLocator.getSwingManager().createColorTableSelectorPanel();

        PluginServices rasterPlugin = PluginsLocator.getManager().getPlugin("org.gvsig.raster.app.mainplugin");

        DynObject properties = rasterPlugin.getPluginProperties();

        File defaultColorTableLibraryPath = (File) properties.getDynValue("defaultColorTableLibraryPath");

        List<Pair<File, ColorTable>> colorTables = RasterLegendLocator.getRasterLegendManager().getColorTables(defaultColorTableLibraryPath);

        selector.set(colorTables);

        selector.setEditable(false);

        splPanelRight.setTopComponent(getPreviewer().asJComponent());

        splPanelRight.setBottomComponent(selector.asJComponent());

        splPanel.setRightComponent(splPanelRight);

        editor = RasterSwingLegendLocator.getSwingManager().createColorTableEditorPanel();

        splPanel.setLeftComponent(editor.asJComponent());

        editor.addChkInterpolatedItemListener(new ItemListener() {

            @Override

            public void itemStateChanged(ItemEvent e) {

                Thread thread = new Thread(new Runnable() {

                    @Override

                    public void run() {

                        SwingUtilities.invokeLater(new Runnable() {

                            @Override

                            public void run() {

                                updatePreviewer();

                            }

                        });

                    }

                }, "Check por interpolate changed");

                thread.setDaemon(false);

                thread.start();

            }

        });

        selector.addListSelectionListener(new ListSelectionListener() {

            @Override

            public void valueChanged(ListSelectionEvent e) {

                Thread thread = new Thread(new Runnable() {

                    @Override

                    public void run() {

                        SwingUtilities.invokeLater(new Runnable() {

                            @Override

                            public void run() {

                                updateColorTable();

                            }

                        });

                    }

                }, "Selector changed");

                thread.setDaemon(false);

                thread.start();

            }

        });

        splPanel.setDividerLocation(0.75);

        cmbBand.setRenderer(new ComboBandsCellRenderer());

        cmbBand.addItemListener(new ItemListener() {

            @Override

            public void itemStateChanged(ItemEvent e) {

                if(e.getStateChange()==ItemEvent.SELECTED){

                    Thread thread = new Thread(new Runnable() {

                        @Override

                        public void run() {

                            SwingUtilities.invokeLater(new Runnable() {

                                @Override

                                public void run() {

                                    updateColorTable();

                                }

                            });

                        }

                    }, "Palette band changed");

                    thread.setDaemon(false);

                    thread.start();

                }

            }

        });

    }

    private void translate() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

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

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

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

    }

    /**

     * @param locale

     *

     */

    public void setLocate(Locale locale) {

        Locale l = super.getLocale();

        if (!l.equals(locale)) {

            translate();

        }

        super.setLocale(locale);

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    private OneLayerPreviewer getPreviewer() {

        if(previewer == null){

            previewer = MapContextRasterSwingLocator.getSwingManager().createOneLayerPreviewer();

        }

        return this.previewer;

    }

    private void initializePreviewer(){

        getPreviewer().setLayer(this.clonedLayerForPreview);

        splPanelRight.setTopComponent(previewer.asJComponent());

    }

    /**

     *

     */

    private void updateColorTable() {

        if (this.initialized) {

            ColorTable selectedColorTable = selector.getSelected();

            if (selectedColorTable != null) {

                colorTable = selectedColorTable;

            } else {

                if (colorTable == null) {

                    selector.select(0);

                    selectedColorTable = selector.getSelected();

                    colorTable = selectedColorTable;

                }

            }

            if (((BandItem) cmbBand.getSelectedItem()).getIndex() > -1 && selectedColorTable != null) {

                colorTable = adaptToTheBand(selectedColorTable);

                setEditorColorTable();

            }

            updatePreviewer();

        }

    }

    private void updatePreviewer() {

        if (this.initialized) {

            RasterLegend legend = getClonedLayerForPreview().getLegend();

            ColorInterpretation colorInterpretation = legend.getColorInterpretation();

            updateColorInterpretation(colorInterpretation);

            isChanged = true;

        }

    }

    protected ColorTable adaptToTheBand(ColorTable colorTable) {

        ColorTable cloned = RasterLegendLocator.getRasterLegendManager().createColorTable();

        cloned.copyFrom(colorTable);

        List<ColorTableClass> classes = cloned.getClasses();

        double minClassValue = Double.POSITIVE_INFINITY;

        double maxClassValue = Double.NEGATIVE_INFINITY;

        for (ColorTableClass colorTableClass : classes) {

            double value = colorTableClass.getValue();

            if(value < minClassValue){

                minClassValue = value;

            }

            if(value > maxClassValue){

                maxClassValue = value;

            }

        }

        Statistics statistics = this.rasterLayer.getStatistics(null);

        int bandIndex = ((BandItem)cmbBand.getSelectedItem()).getIndex();

        double minBandValue = statistics.getMin()[bandIndex];

        double maxBandValue = statistics.getMax()[bandIndex];

        double ratio = (maxBandValue-minBandValue)/(maxClassValue-minClassValue);

        for (ColorTableClass colorTableClass : classes) {

            double value = colorTableClass.getValue();

            double adaptedValue = ((value-minClassValue)*ratio)+minBandValue;

            colorTableClass.setValue(adaptedValue);

        }

        return cloned;

    }

    @Override

    public boolean fetch(FLayer layer) {

        if(isChanged){

            RasterLayer rasterLayer = (RasterLayer) layer;

            RasterLegend rasterLegend = (RasterLegend) rasterLayer.getLegend();

            ColorInterpretation colorInterpretation = rasterLegend.getColorInterpretation();

            updateColorInterpretation(colorInterpretation);

            updateCmbBands(rasterLayer);

        }

        return true;

    }

    /**

     * @param colorInterpretation

     */

    private void updateColorInterpretation(ColorInterpretation colorInterpretation) {

        int paletteBand = ((BandItem)cmbBand.getSelectedItem()).getIndex();

        colorInterpretation.setPaletteBand(paletteBand);

        ColorTable colorTable2 = this.editor.getColorTable();

        if(colorTable2!=null){

            try {

                colorInterpretation.setPalette((ColorTable)colorTable2.clone());

            } catch (CloneNotSupportedException e) {

                String message = "Can't clone ColorTable.";

                LOG.warn(message, e);

                throw new RuntimeException(message, e);

            }

        }

    }

    @Override

    public void set(FLayer layer) {

        if(this.rasterLayer != null) {

            DisposeUtils.dispose(this.rasterLayer);

        }

        this.rasterLayer = (RasterLayer) layer;

        DisposeUtils.bind(this.rasterLayer);

        init();

    }

    /**

    *

    */

   private void init() {

       updateCmbBands(getClonedLayerForPreview());

       RasterLegend legend = null;

       legend = getClonedLayerForPreview().getLegend();

       ColorInterpretation colorInterpretation = legend.getColorInterpretation();

       if (colorInterpretation.isPalette()) {

           splPanel.getLeftComponent().setEnabled(true);

           colorTable = colorInterpretation.getPalette();

       } else {

           splPanel.getLeftComponent().setEnabled(false);

       }

       setEditorColorTable();

       isChanged = false;

       initializePreviewer();

       initialized = true;

   }

    private RasterLayer getClonedLayerForPreview() {

        if(this.clonedLayerForPreview==null){

            try {

                this.clonedLayerForPreview = (RasterLayer) this.rasterLayer.cloneLayer();

            } catch (Exception e1) {

                LOG.warn("Can't clone the raster layer to preview.", e1);

            }

        }

        return this.clonedLayerForPreview;

    }

    /**

     * @return

     */

    private TableModelListener getEditorTableModelListener() {

        if (editorTableModelListener == null) {

            editorTableModelListener = new TableModelListener() {

                @Override

                public void tableChanged(TableModelEvent e) {

                    updatePreviewer();

                }

            };

        }

        return editorTableModelListener;

    }

    private void updateCmbBands(RasterLayer layer){

        RasterStore store = layer.getRasterStore();

        DefaultComboBoxModel<BandItem> currentBandsModel = (DefaultComboBoxModel<BandItem>) cmbBand.getModel();

        DefaultComboBoxModel<BandItem> bandsModel = new DefaultComboBoxModel<BandItem>();

        boolean equals = true;

        for (int i = -1; i < store.getBands(); i++) {

            BandDescriptor bandDescriptor = null;

            if(i>=0){

                bandDescriptor = store.getBandDescriptor(i);

                if(currentBandsModel==null || currentBandsModel.getSize()<=i+1 || currentBandsModel.getElementAt(i+1) == null ||!bandDescriptor.equals(currentBandsModel.getElementAt(i+1).getDescriptor())){

                    equals = false;

                }

            }

            bandsModel.addElement(new BandItem(i,bandDescriptor));

        }

        if (!equals) {

            cmbBand.setModel(bandsModel);

            ColorInterpretation colorInterpretation = layer.getLegend().getColorInterpretation();

            if (colorInterpretation.isPalette()) {

                cmbBand.setSelectedIndex(colorInterpretation.getPaletteBand() + 1);

            } else {

                cmbBand.setSelectedIndex(0);

            }

        }

    }

    private static class BandItem{

        private int index;

        private BandDescriptor descriptor;

        public BandItem(int index, BandDescriptor descriptor) {

            this.setIndex(index);

            this.setDescriptor(descriptor);

        }

        /**

         * @return the index

         */

        public int getIndex() {

            return index;

        }

        /**

         * @param index the index to set

         */

        public void setIndex(int index) {

            this.index = index;

        }

        /**

         * @return the descriptor

         */

        public BandDescriptor getDescriptor() {

            return descriptor;

        }

        /**

         * @param descriptor the descriptor to set

         */

        public void setDescriptor(BandDescriptor descriptor) {

            this.descriptor = descriptor;

        }

    }

    private static class ComboBandsCellRenderer extends javax.swing.JLabel implements ListCellRenderer<BandItem> {

        /**

         *

         */

        private static final long serialVersionUID = 7763209775607437499L;

        @Override

        public Component getListCellRendererComponent(JList<? extends BandItem> list, BandItem value, int index,

            boolean isSelected, boolean cellHasFocus) {

            I18nManager i18nManager = ToolsLocator.getI18nManager();

            if (value == null || value.getIndex() < 0) {

                setHorizontalAlignment(JLabel.CENTER);

                setText("\u2015");

            } else {

                setHorizontalAlignment(JLabel.LEFT);

                setText(i18nManager.getTranslation("_band") + ": " + (value.getIndex()) + " - '"

                    + value.getDescriptor().getName() + "'");

            }

            return this;

        }

    }

    @Override

    public void setVisible(boolean aFlag) {

        super.setVisible(aFlag);

        splPanel.setDividerLocation(0.60);

        splPanelRight.setDividerLocation(0.40);

    }

    /**

     *

     */

    private void setEditorColorTable() {

        if (colorTable != null) {

            if(editor.getColorTable()!=null){

                editor.removeTableModelListener(getEditorTableModelListener());

            }

            editor.set(colorTable);

            if (editorTableModelListener == null) {

                editor.addTableModelListener(getEditorTableModelListener());

            }

        }

    }

    @Override

    public void dispose() {

        if(initialized) {

            if(!ToolsLocator.getDisposableManager().release(this)){

                return;

            }

            DisposeUtils.dispose(this.rasterLayer);

            this.rasterLayer = null;

            DisposeUtils.dispose(this.previewer);

            this.previewer = null;

            DisposeUtils.disposeQuietly(this.clonedLayerForPreview);

            this.clonedLayerForPreview = null;

        }

    }

}