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.fmap.mapcontext.raster.swing.impl.bands;

import java.awt.color.ColorSpace;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.ItemEvent;

import java.awt.event.ItemListener;

import java.io.File;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import java.util.Locale;

import javax.swing.JComponent;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

import javax.swing.ListSelectionModel;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import javax.swing.filechooser.FileFilter;

import javax.swing.table.DefaultTableColumnModel;

import javax.swing.table.TableColumn;

import javax.swing.table.TableColumnModel;

import javax.swing.table.TableModel;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

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

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

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

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

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

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

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

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

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

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

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

import org.gvsig.fmap.mapcontext.raster.swing.bands.BandsPanel;

import org.gvsig.raster.lib.buffer.api.BufferLocator;

import org.gvsig.raster.lib.buffer.api.BufferManager;

import org.gvsig.raster.lib.buffer.api.NoData;

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

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

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.threadsafedialogs.ThreadSafeDialogsManager;

/**

 * @author fdiaz

 *

 */

public class BandsPanelController extends BandsPanelView implements BandsPanel, ActionListener, ListSelectionListener {

    /**

     *

     */

    private static final long serialVersionUID = 8255630092178065352L;

    @SuppressWarnings("unused")

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

    private RasterLayer layer;

    private static final String RASTER_FILTER = "All_raster_supported";

    // FLayer layer;

    /**

     * @param layer

     */

    @SuppressWarnings("unchecked")

    public BandsPanelController() {

        DisposeUtils.bind(this);

        translate();

        initializeComponents();

    }

    private void initializeComponents() {

        this.btnAdd.addActionListener(this);

        this.btnRemove.addActionListener(this);

        tblBands.setRowSelectionAllowed(true);

        tblBands.setColumnSelectionAllowed(false);

        tblBands.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        tblBands.getSelectionModel().addListSelectionListener(this);

        BandsTableModel model = new BandsTableModel();

        tblBands.setModel(model);

        tblBands.setColumnModel(new DefaultTableColumnModel());

        TableColumnModel columnModel = tblBands.getColumnModel();

        columnModel.setColumnSelectionAllowed(false);

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

            TableColumn tableColumn = new TableColumn(i);

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

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

            columnModel.addColumn(tableColumn);

        }

        cboColorSpaces.addItem(ColorInterpretation.RGB);

        cboColorSpaces.addItem(ColorInterpretation.CMYK);

        cboColorSpaces.addItem(ColorInterpretation.HSL);

        //FIXME: Uncomment when a operation to convert RGB to/from HSL is implemented.

//        cboColorSpaces.addItem(ColorInterpretation.YCBCR);

        cboColorSpaces.addItemListener(new ItemListener() {

            @Override

            public void itemStateChanged(ItemEvent e) {

                doChangeColorColumnCellEditor();

            }

        });

        // FIXME: esto ser?a lo que habr?a que cambiar si se desea implementar

        // otros espacios de color

//        columnModel.getColumn(0).setCellEditor(new ColorInterpretationCellEditor(BandsTableModel.RGBColorSpace));

        columnModel.getColumn(0).setCellRenderer(new ColorInterpretationCellRenderer());

        columnModel.getColumn(1).setCellRenderer(new NoDataCellRenderer());

        // columnModel.getColumn(1).setCellEditor(new NoDataCellEditor());

        columnModel.getColumn(2).setCellRenderer(new DataTypeCellRenderer());

    }

    private void translate() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

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

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

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

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

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

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

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

    }

    /**

     * @param locale

     *

     */

    public void setLocate(Locale locale) {

        Locale l = super.getLocale();

        if (!l.equals(locale)) {

            translate();

        }

        super.setLocale(locale);

    }

    /**

     * @param layer

     */

    @SuppressWarnings("unchecked")

    public void set(FLayer layer) {

        if(this.layer!=null){

            DisposeUtils.dispose(this.layer);

        }

        this.layer = (RasterLayer) layer;

        DisposeUtils.bind(layer);

        RasterLegend legend = this.layer.getLegend();

        ColorInterpretation colorInterpretation = null;

        if (legend != null) {

            colorInterpretation = legend.getColorInterpretation();

        }

        RasterStore store = this.layer.getRasterStore();

        addBands(store, colorInterpretation);

    }

    /**

     * @param colorInterpretation

     * @param store

     * @param model

     */

    private void addBands(RasterStore store, ColorInterpretation colorInterpretation) {

        String colorSpace = "";

        if (colorInterpretation.hasAnyCMYKBand()) {

            colorSpace = ColorInterpretation.CMYK;

        } else if (colorInterpretation.hasAnyHSLBand()) {

            colorSpace = ColorInterpretation.HSL;

        } else if (colorInterpretation.hasAnyHSLBand()) {

            colorSpace = ColorInterpretation.YCBCR;

        } else {

            colorSpace = ColorInterpretation.RGB;

        }

        cboColorSpaces.setSelectedItem(colorSpace);

        doChangeColorColumnCellEditor();

        BandsTableModel model = ((BandsTableModel) tblBands.getModel());

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

            BandDescriptor bandDescriptor = store.getBandDescriptor(i);

            RasterStoreBand rasterStoreBand = new RasterStoreBand(bandDescriptor.getStore(), bandDescriptor.getBand());

            NoData noData = bandDescriptor.getNoData();

            if (noData != null) {

                rasterStoreBand.setNoDataNumber(noData.getValue());

            }

            rasterStoreBand.setBandName(bandDescriptor.getName());

            rasterStoreBand.setDataType(bandDescriptor.getDataType());

            if (colorInterpretation != null) {

                rasterStoreBand.setBandColorInterpretation(colorInterpretation.get(i));

            } else {

                rasterStoreBand.setBandColorInterpretation(ColorInterpretation.UNDEFINED_BAND);

            }

            model.add(rasterStoreBand);

            DisposeUtils.disposeQuietly(rasterStoreBand);

            model.fireTableDataChanged();

        }

    }

    /**

     * @param layer

     */

    public boolean fetch(FLayer layer) {

        RasterLayer rasterLayer = (RasterLayer) layer;

        RasterStore store = rasterLayer.getRasterStore();

        RasterLegendManager rasterLegendManager = RasterLegendLocator.getRasterLegendManager();

        if (willBeChangedBands(store)) {

            store.clearAdditionalBands();

            BandsTableModel model = (BandsTableModel) tblBands.getModel();

            BufferManager bufferManager = BufferLocator.getBufferManager();

            for (int index = 0; index < model.getRowCount(); index++) {

                RasterStoreBand rasterStoreBand = model.getElementAt(index);

                RasterStore subStore = rasterStoreBand.getStore();

                int band = rasterStoreBand.getBand();

                if (subStore != store) {

                    try {

                        store.addBand(subStore, band);

                    } catch (DataException e) {

                        ThreadSafeDialogsManager dlgManager = ToolsSwingLocator.getThreadSafeDialogsManager();

                        String message = "_cant_add_band_XbandX_from_store_XsourceX_to_store_XtargetX";

                        String title = "_error_adding_bands";

                        dlgManager.messageDialog(message, new String[] { String.valueOf(band), subStore.getName(),

                            store.getName() }, title, JOptionPane.WARNING_MESSAGE);

                        LOG.warn(message, e);

                        return false;

                    }

                }

                BandDescriptor bandDescriptor = store.getBandDescriptor(index);

                NoData noData = bandDescriptor.getNoData();

                Number value = null;

                value = rasterStoreBand.getNoDataNumber();

                if (noData == null) {

                    noData = bufferManager.createNoData(value, null);

                } else {

                    noData.setValue((Number) value);

                }

                bandDescriptor.setNoData(noData);

            }

            rasterLayer.recalculateStatistics();

            List<String> colorInterpretations = getColorInterpretations();

            boolean createNewLegend = true;

            RasterLegend oldLegend = rasterLayer.getLegend();

            if(oldLegend!=null){

                ColorInterpretation oldColorInterpretation = oldLegend.getColorInterpretation();

                if(oldColorInterpretation.length()==colorInterpretations.size()){

                    for (int band=0; band<oldColorInterpretation.length(); band++) {

                        oldColorInterpretation.setColorInterpValue(band, colorInterpretations.get(band));

                    }

                    createNewLegend = false;

                }

            }

            if(createNewLegend) {

                ColorInterpretation colorInterpretation =

                    rasterLegendManager.createColorInterpretation(colorInterpretations.toArray(new String[0]));

                RasterLegend legend = rasterLegendManager.createLegend(colorInterpretation);

                rasterLayer.setLegend(legend);

            }

            rasterLayer.getMapContext().invalidate();

        }

        return true;

    }

    private List<String> getColorInterpretations(){

        List<String> colorInterpretations = new ArrayList();

        BandsTableModel model = (BandsTableModel) tblBands.getModel();

        for (int index = 0; index < model.getRowCount(); index++) {

            RasterStoreBand rasterStoreBand = model.getElementAt(index);

            RasterStore subStore = rasterStoreBand.getStore();

            int band = rasterStoreBand.getBand();

            colorInterpretations.add(rasterStoreBand.getBandColorInterpretation());

        }

        return colorInterpretations;

    }

    private boolean willBeChangedBands(RasterStore store){

        BandsTableModel model = (BandsTableModel) tblBands.getModel();

        if(store.getBands()!=model.getRowCount()){

            return true;

        }

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

        for (int index = 0; index < model.getRowCount(); index++) {

            RasterStoreBand rasterStoreBand = model.getElementAt(index);

            RasterStore subStore = rasterStoreBand.getStore();

            int band = rasterStoreBand.getBand();

            BandDescriptor bandDescriptor = store.getBandDescriptor(index);

            if (bandDescriptor.getStore() != subStore || bandDescriptor.getBand() != band

                || isChangedNoData(bandDescriptor.getNoData(), rasterStoreBand.getNoDataNumber())

                || !(rasterStoreBand.getBandColorInterpretation().equals(colorInterpretation.get(index)))

               ){

                return true;

            }

        }

        return false;

    }

    private boolean isChangedNoData(NoData noData, Number number){

        if(noData!=null){

            return noData.getValue()!=number;

        } else {

            return number!=null;

        }

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    @SuppressWarnings("unchecked")

    @Override

    public void actionPerformed(ActionEvent e) {

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

            BandsTableModel model = (BandsTableModel) tblBands.getModel();

            model.removeElementAt(tblBands.getSelectedRow());

            model.fireTableDataChanged();

        } else if (e.getSource() == btnAdd) {

            DataManager manager = DALLocator.getDataManager();

            JFileChooser fileChooser = createJFileChooser();

            int result = fileChooser.showOpenDialog(this);

            if (result == JFileChooser.APPROVE_OPTION) {

                File[] files = fileChooser.getSelectedFiles();

                for (File file : files) {

                    DataServerExplorerParameters fileExplorerParams;

                    try {

                        fileExplorerParams = manager.createServerExplorerParameters(FilesystemServerExplorer.NAME);

                        FilesystemServerExplorer explorer =

                            (FilesystemServerExplorer) manager.openServerExplorer(FilesystemServerExplorer.NAME,

                                fileExplorerParams);

                        String type = explorer.getProviderName(file);

                        DynObject params = manager.createStoreParameters(type);

                        if (params.getDynClass().getDynField("file") != null) {

                            params.setDynValue("file", file);

                        }

                        // FIXME: ?se le pone la proyecci?n del raster principal

                        // o se busca la que tenga el archivo si la tiene?

                        if (params.getDynClass().getDynField("crs") != null) {

                            params.setDynValue("crs", this.layer.getProjection());

                        }

                        RasterStore store = (RasterStore) manager.openStore(type, params);

                        DisposeUtils.disposeQuietly(explorer);

                        addBands(store, null);

                        DisposeUtils.disposeQuietly(store);

                    } catch (InitializeException | ProviderNotRegisteredException | ValidateDataParametersException e1) {

                        LOG.warn("Can't add bands from the file '" + file.getName() + "'", e1);

                        // e1.printStackTrace();

                    }

                }

            }

        }

    }

    @Override

    public void valueChanged(ListSelectionEvent e) {

        if (e.getSource() == tblBands.getSelectionModel()) {

            if (tblBands.getSelectedRow() >= 0) {

                if (this.layer.getRasterStore() != ((BandsTableModel) tblBands.getModel()).getElementAt(

                    tblBands.getSelectedRow()).getStore()) {

                    btnRemove.setEnabled(true);

                } else {

                    btnRemove.setEnabled(false);

                }

            } else {

                btnRemove.setEnabled(false);

            }

        }

    }

    private JFileChooser createJFileChooser() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        JFileChooser fileChooser = new JFileChooser();

        int mode = FilesystemServerExplorer.MODE_RASTER;

        FilesystemServerExplorer explorer;

        DataManager dm = DALLocator.getDataManager();

        FilesystemServerExplorerParameters param;

        try {

            param =

                (FilesystemServerExplorerParameters) dm.createServerExplorerParameters(FilesystemServerExplorer.NAME);

            explorer = (FilesystemServerExplorer) dm.openServerExplorer(FilesystemServerExplorer.NAME, param);

        } catch (Exception e) {

            throw new RuntimeException(e);

        }

        FilesystemFileFilter filter = explorer.getFilter(mode, i18nManager.getTranslation(RASTER_FILTER));

        fileChooser.setMultiSelectionEnabled(true);

        fileChooser.setAcceptAllFileFilterUsed(false);

        fileChooser.setFileFilter(new MyFileFilter(filter));

        DisposeUtils.dispose(explorer);

        return fileChooser;

    }

    /**

     * @author fdiaz

     *

     */

    public class MyFileFilter extends FileFilter {

        public FilesystemFileFilter filter = null;

        /**

         * @param params

         */

        public MyFileFilter(FilesystemFileFilter params) {

            this.filter = params;

        }

        /**

         * @see javax.swing.filechooser.FileFilter#accept(java.io.File)

         */

        public boolean accept(File f) {

            if (f.isDirectory()) {

                return true;

            }

            return filter.accept(f);

        }

        /**

         * @see javax.swing.filechooser.FileFilter#getDescription()

         */

        public String getDescription() {

            return filter.getDescription();

        }

        /**

         * @return

         */

        public String getName() {

            return filter.getDataStoreProviderName();

        }

    }

    @Override

    public void dispose() {

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

            return;

        }

        DisposeUtils.dispose(this.layer);

        this.layer = null;

        BandsTableModel model = (BandsTableModel) tblBands.getModel();

        for (int index = 0; index < model.getRowCount(); index++) {

            RasterStoreBand rasterStoreBand = model.getElementAt(index);

            DisposeUtils.disposeQuietly(rasterStoreBand);

        }

    }

    private void doChangeColorColumnCellEditor() {

        TableColumnModel columnModel = tblBands.getColumnModel();

        String selectedItem = (String)cboColorSpaces.getSelectedItem();

        if(selectedItem.equalsIgnoreCase(ColorInterpretation.RGB)){

            columnModel.getColumn(0).setCellEditor(new ColorInterpretationCellEditor(BandsTableModel.RGBColorSpace));

        }

        if(selectedItem.equalsIgnoreCase(ColorInterpretation.CMYK)){

            columnModel.getColumn(0).setCellEditor(new ColorInterpretationCellEditor(BandsTableModel.CMYKColorSpace));

        }

        if(selectedItem.equalsIgnoreCase(ColorInterpretation.HSL)){

            columnModel.getColumn(0).setCellEditor(new ColorInterpretationCellEditor(BandsTableModel.HSLColorSpace));

        }

        if(selectedItem.equalsIgnoreCase(ColorInterpretation.YCBCR)){

            columnModel.getColumn(0).setCellEditor(new ColorInterpretationCellEditor(BandsTableModel.YCBCRColorSpace));

        }

        TableModel bandModel = tblBands.getModel();

        for (int i = 0; i < tblBands.getRowCount(); i++) {

            bandModel.setValueAt(ColorInterpretation.UNDEFINED_BAND, i, 0);

        }

    }

}