Application: gvSIG desktop » gvSIG raster

/* 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.Dimension;

import java.awt.Rectangle;

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.JPanel;

import javax.swing.JSplitPane;

import javax.swing.ListCellRenderer;

import javax.swing.SwingUtilities;

import javax.swing.event.ChangeEvent;

import javax.swing.event.ChangeListener;

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.app.project.documents.view.toolListeners.PanListener;

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

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

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

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.MapContextLocator;

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.fmap.mapcontrol.MapControl;

import org.gvsig.fmap.mapcontrol.MapControlCreationException;

import org.gvsig.fmap.mapcontrol.MapControlLocator;

import org.gvsig.fmap.mapcontrol.tools.Behavior.Behavior;

import org.gvsig.fmap.mapcontrol.tools.Behavior.MoveBehavior;

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.dynobject.DynObject;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.locator.LocatorException;

/**

 * @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 MapControl previewerMapControl;

    private OneLayerPreviewer previewer;

    private boolean initialized;

    private TableModelListener editorTableModelListener;

    private RasterLayer clonedLayerForPreview;

    /**

     *

     */

    public ColorTablePagePanelController() {

        this.initialized = false;

        translate();

        initializeComponents();

    }

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

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

                                colorTable = selector.getSelected();

                                if (((BandItem) cmbBand.getSelectedItem()).getIndex() > -1) {

//                                    ColorTable adaptedColorTable = adaptToTheBand(colorTable);

                                    colorTable = adaptToTheBand(colorTable);

                                    setEditorColorTable();

                                    updatePreviewer();

                                }

                            }

                        });

                    }

                }, "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() {

                                    updatePreviewer();

                                }

                            });

                        }

                    }, "Palette band changed");

                    thread.setDaemon(false);

                    thread.start();

                }

            }

        });

//        editor.addInterpolatedItemListener(new ItemListener() {

//

//            @Override

//            public void itemStateChanged(ItemEvent e) {

//                updatePreviewer();

//            }

//        });

    }

    private OneLayerPreviewer getPreviewer() {

        if(previewer == null){

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

        }

        return this.previewer;

//        if(this.previewerMapControl == null) {

//            try {

//                MapContext mapContext = MapContextLocator.getMapContextManager().createMapContext();

//                this.previewerMapControl = MapControlLocator.getMapControlManager().createJMapControlPanel(mapContext);

//            } catch (MapControlCreationException | LocatorException e) {

//                LOG.warn("Can't create mapControl to preview.", e);

//            }

//            this.previewerMapControl.setPreferredSize(new Dimension(300, 200));

//

//        }

//        return this.previewerMapControl;

    }

    private void initializePreviewer(){

//        MapControl mapControl = getPreviewer();

//        MapContext mapContext = mapControl.getMapContext();

//        mapContext.setProjection(this.rasterLayer.getProjection());

//        mapContext.getLayers().add(this.clonedLayerForPreview);

//        try {

//            mapControl.getViewPort().setEnvelope(this.rasterLayer.getFullEnvelope());

//        } catch (ReadException e) {

//            LOG.warn("Can't get the full envelope of the layer.", e);

//        }

//

        getPreviewer().setLayer(this.clonedLayerForPreview);

        splPanelRight.setTopComponent(previewer.asJComponent());

//        splPanelRight.getTopComponent().setSize(previewer.asJComponent().getPreferredSize());

        // pan

//        PanListener panListener = new PanListener(mapControl);

//        mapControl.addBehavior("pan", new MoveBehavior(panListener,Behavior.BUTTON_LEFT));

//        mapControl.setTool("pan");

    }

    private void updatePreviewer(){

        if (this.initialized) {

//            MapControl mapControl = getPreviewer();

            ColorInterpretation colorInterpretation = this.clonedLayerForPreview.getLegend().getColorInterpretation();

            updateColorInterpretation(colorInterpretation);

//            MapContext mapContext = mapControl.getMapContext();

//            mapContext.invalidate();

//            mapControl.paintImmediately(new Rectangle(mapControl.getSize()));

        }

    }

    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;

    }

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

    }

    /* (non-Javadoc)

     * @see org.gvsig.tools.swing.api.Component#asJComponent()

     */

    @Override

    public JComponent asJComponent() {

        return this;

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.app.mainplugin.colortable.ColorTablePagePanel#fetch(org.gvsig.fmap.mapcontext.layers.FLayer)

     */

    @Override

    public boolean fetch(FLayer layer) {

        RasterLayer rasterLayer = (RasterLayer) layer;

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

        ColorInterpretation colorInterpretation = rasterLegend.getColorInterpretation();

//        updateColorInterpretation(colorInterpretation);

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

        colorInterpretation.setPaletteBand(paletteBand);

        if (paletteBand>=0) {

            try {

                colorInterpretation.setPalette((ColorTable)this.editor.getColorTable().clone());

            } catch (CloneNotSupportedException e) {

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

                LOG.warn(message, e);

                throw new RuntimeException(message, e);

            }

        }

        updateCmbBands(rasterLayer);

        return true;

    }

    /**

     * @param colorInterpretation

     */

    private void updateColorInterpretation(ColorInterpretation colorInterpretation) {

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

        colorInterpretation.setPaletteBand(paletteBand);

        if (paletteBand>=0) {

            colorInterpretation.setPalette(this.editor.getColorTable());

        }

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.app.mainplugin.colortable.ColorTablePagePanel#set(org.gvsig.fmap.mapcontext.layers.FLayer)

     */

    @Override

    public void set(FLayer layer) {

        this.rasterLayer = (RasterLayer) layer;

        try {

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

        } catch (Exception e1) {

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

            return;

        }

        init();

    }

    /**

     *

     */

    private void init() {

        updateCmbBands(this.clonedLayerForPreview);

        RasterLegend legend = null;

        legend = (RasterLegend) this.clonedLayerForPreview.getLegend();

        ColorInterpretation colorInterpretation = legend.getColorInterpretation();

        if (colorInterpretation.isPalette()) {

            splPanel.getLeftComponent().setEnabled(true);

            colorTable = colorInterpretation.getPalette();

        } else {

            splPanel.getLeftComponent().setEnabled(false);

        }

        setEditorColorTable();

        initializePreviewer();

        this.initialized = true;

    }

    /**

     * @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> bandsModel = new DefaultComboBoxModel<BandItem>();

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

            BandDescriptor bandDescriptor = null;

            if(i>=0){

                bandDescriptor = store.getBandDescriptor(i);

            }

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

        }

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

            editor.set(colorTable);

            if (editorTableModelListener == null) {

                editor.addTableModelListener(getEditorTableModelListener());

            }

        }

    }

}