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.swing.legend.impl.colortable.selector;

import java.awt.Color;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.ComponentListener;

import java.awt.event.FocusListener;

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

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import javax.swing.table.DefaultTableColumnModel;

import javax.swing.table.TableColumn;

import javax.swing.table.TableColumnModel;

import org.apache.commons.io.FilenameUtils;

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

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

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

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

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

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

import org.gvsig.raster.lib.legend.api.exceptions.ColorTableIOException;

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

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

import org.gvsig.raster.swing.legend.colortable.create.CreateColorTablePanel;

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

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

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

import org.gvsig.raster.swing.legend.impl.colortable.editor.ColorTableEditorPanelController;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

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

import org.gvsig.tools.swing.api.windowmanager.Dialog;

import org.gvsig.tools.swing.api.windowmanager.WindowManager;

import org.gvsig.tools.swing.api.windowmanager.WindowManager_v2;

import org.gvsig.tools.swing.icontheme.IconTheme;

/**

 * @author fdiaz

 *

 */

public class ColorTableSelectorPanelController extends ColorTableSelectorPanelView implements ColorTableSelectorPanel {

    /**

     *

     */

    private static final long serialVersionUID = -2427828721121167549L;

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

    private List<Pair<File,ColorTable>> colorTables;

    private List<ColorTable> addedColorTables;

    private List<Pair<File,ColorTable>> modifiedColorTables;

    private List<Pair<File,ColorTable>> removedColorTables;

    private boolean editable;

    public ColorTableSelectorPanelController() {

        translate();

        this.editable = true;

        initializeComponents();

    }

    /* (non-Javadoc)

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

     */

    @Override

    public JComponent asJComponent() {

        return this;

    }

    private void initializeComponents() {

        IconTheme iconTheme = ToolsSwingLocator.getIconThemeManager().getCurrent();

        btnNew.setIcon(iconTheme.get("new-color-table-icon"));

        btnEdit.setIcon(iconTheme.get("edit-color-table-icon"));

        btnExport.setIcon(iconTheme.get("export-color-table-icon"));

        btnImport.setIcon(iconTheme.get("import-color-table-icon"));

        btnRemove.setIcon(iconTheme.get("remove-color-table-icon"));

        btnNew.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doNewColorTable();

            }

        });

        btnEdit.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doEditColorTable();

            }

        });

        btnExport.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doExportColorTable();

            }

        });

        btnImport.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doImportColorTable();

            }

        });

        btnRemove.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doRemoveColorTable();

            }

        });

        tblColorTables.setRowSelectionAllowed(true);

        tblColorTables.setColumnSelectionAllowed(false);

        tblColorTables.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        tblColorTables.getSelectionModel().addListSelectionListener(new ListSelectionListener() {

            @Override

            public void valueChanged(ListSelectionEvent e) {

                updateComponents();

            }

        });

        updateComponents();

    }

    protected void doEditColorTable() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        int index = this.tblColorTables.getSelectedRow();

        ColorTable sourceColorTable = ((ColorTablesTableModel)this.tblColorTables.getModel()).getColorTable(index).getValue();

        //Le pasamos una copiamos de la tabla de color al editor para que no la modifique autom?ticamente

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

        colorTable.copyFrom(colorTable);

        //FIXME: ?Al manager?

        final ColorTableEditorPanelController panel = new ColorTableEditorPanelController();

        panel.set(colorTable);

        panel.asJComponent().setPreferredSize(new Dimension(600, 400));

        final WindowManager_v2 winManager = (WindowManager_v2) ToolsSwingLocator.getWindowManager();

        final Dialog dialog =

            winManager.createDialog(panel.asJComponent(), i18nManager.getTranslation("color-table-edit"),

                null, WindowManager_v2.BUTTONS_APPLY_OK_CANCEL);

        dialog.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                switch (dialog.getAction()) {

                case WindowManager_v2.BUTTON_APPLY:

                    doApplyEditColorTable(panel);

                    break;

                case WindowManager_v2.BUTTON_OK:

                    if (doApplyEditColorTable(panel)) {

                        panel.asJComponent().setVisible(false);

                    }

                    break;

                case WindowManager_v2.BUTTON_CANCEL:

                    panel.asJComponent().setVisible(false);

                    break;

                }

            }

        });

        dialog.show(WindowManager.MODE.DIALOG);

    }

    protected boolean doApplyEditColorTable(ColorTableEditorPanel panel) {

        int selectedRow = tblColorTables.getSelectedRow();

        ColorTable colorTableEdited = panel.getColorTable();

        if(panel.isModified()){

            ColorTablesTableModel colorTablesTableModel = (ColorTablesTableModel)this.tblColorTables.getModel();

            colorTablesTableModel.setColorTable(selectedRow, colorTableEdited);

            if(!modifiedColorTables.contains(colorTablesTableModel.getColorTable(selectedRow))){

                modifiedColorTables.add(colorTablesTableModel.getColorTable(selectedRow));

            }

            tblColorTables.setRowSelectionInterval(selectedRow, selectedRow);

        }

        return true;

    }

    protected void doRemoveColorTable() {

        int row = tblColorTables.getSelectedRow();

        ColorTablesTableModel colorTablesTableModel = (ColorTablesTableModel)tblColorTables.getModel();

        Pair<File, ColorTable> pair = colorTablesTableModel.getColorTable(row);

        removedColorTables.add(pair);

        colorTablesTableModel.removeColorTable(tblColorTables.getSelectedRow());

    }

    protected void doImportColorTable() {

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        JFileChooser fileChooser = new JFileChooser();

        List<ColorTableIOFactory> factories = legendManager.getColorTableIOFactories();

        for (Iterator iterator = factories.iterator(); iterator.hasNext();) {

            ColorTableIOFactory colorTableIOFactory = (ColorTableIOFactory) iterator.next();

            fileChooser.addChoosableFileFilter(new MyFileFilter(colorTableIOFactory));

        }

        fileChooser.setMultiSelectionEnabled(true);

        fileChooser.setAcceptAllFileFilterUsed(false);

        int result = fileChooser.showOpenDialog(this);

        if (result == JFileChooser.APPROVE_OPTION) {

            File[] files = fileChooser.getSelectedFiles();

            for (File file : files) {

                for (Iterator iterator = factories.iterator(); iterator.hasNext();) {

                    ColorTableIOFactory colorTableIOFactory = (ColorTableIOFactory) iterator.next();

                    if(colorTableIOFactory.accept(file)){

                        ColorTableIO colorTableIO = colorTableIOFactory.create();

                        ColorTable colorTable;

                        try {

                            colorTable = colorTableIO.read(file);

                        } catch (ColorTableIOException e) {

                            LOG.warn("Can't read color table '"+file.getName()+"'.", e);

                            //FIXME: mostrar aviso al usuario

                            throw new RuntimeException("Can't read color table '"+file.getName()+"'.", e);

                        }

                        addedColorTables.add(colorTable);

                        ((ColorTablesTableModel)tblColorTables.getModel()).addColorTable(new MutablePair<File, ColorTable>(null,colorTable));

                        continue;

                    }

                }

            }

        }

    }

    protected void doExportColorTable() {

        ColorTable colorTable =

            ((ColorTablesTableModel) tblColorTables.getModel()).getColorTable(tblColorTables.getSelectedRow())

                .getValue();

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        JFileChooser fileChooser = new JFileChooser();

        List<ColorTableIOFactory> factories = legendManager.getColorTableIOFactories();

        for (Iterator iterator = factories.iterator(); iterator.hasNext();) {

            ColorTableIOFactory colorTableIOFactory = (ColorTableIOFactory) iterator.next();

            fileChooser.addChoosableFileFilter(new MyFileFilter(colorTableIOFactory));

        }

        fileChooser.setMultiSelectionEnabled(false);

        fileChooser.setAcceptAllFileFilterUsed(false);

        int result = fileChooser.showSaveDialog(this);

        if (result == JFileChooser.APPROVE_OPTION) {

            File file = fileChooser.getSelectedFile();

            MyFileFilter fileFilter = (MyFileFilter) fileChooser.getFileFilter();

            File parent = file.getParentFile();

            String filename = file.getName();

            ColorTableIOFactory colorTableIOFactory = fileFilter.getColorTableIOFactory();

            if (FilenameUtils.getExtension(filename).isEmpty()) {

                file = new File(parent, filename + "." + colorTableIOFactory.getDefaultExtension());

            }

            ;

            try {

                colorTableIOFactory.create().write(colorTable, file);

            } catch (ColorTableIOException e) {

                LOG.warn("Can't write color table '" + file.getName() + "'.", e);

                // FIXME: mostrar aviso al usuario

                throw new RuntimeException("Can't write color table '" + file.getName() + "'.", e);

            }

        }

    }

    protected void doNewColorTable() {

        RasterSwingLegendManager rasterSwingLegendManager = RasterSwingLegendLocator.getSwingManager();

        final CreateColorTablePanel panel = rasterSwingLegendManager.createCreateColorTablePanel();

        panel.setFromColor(new Color(0,0,0,255));

        panel.setToColor(new Color(255,255,255,255));

        panel.setMinimum(0);

        panel.setMaximum(255);

        panel.setIntervals(1);

        panel.setMode(MakeColorTable.MODE_NUMBER_OF_INTERVALS);

        panel.setName("");

        panel.asJComponent().setPreferredSize(new Dimension(400, 300));

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        final WindowManager_v2 winManager = (WindowManager_v2) ToolsSwingLocator.getWindowManager();

        final Dialog dialog =

            winManager.createDialog(panel.asJComponent(), i18nManager.getTranslation("color-tables-library"),

                null, WindowManager_v2.BUTTONS_OK_CANCEL);

        dialog.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                switch (dialog.getAction()) {

                case WindowManager_v2.BUTTON_OK:

                    if (doMakeColorTable(panel)) {

                        panel.asJComponent().setVisible(false);

                    }

                    break;

                case WindowManager_v2.BUTTON_CANCEL:

                    panel.asJComponent().setVisible(false);

                    break;

                }

            }

        });

        dialog.show(WindowManager.MODE.WINDOW);

    }

    protected boolean doMakeColorTable(CreateColorTablePanel panel) {

        MakeColorTable makeColorTable = RasterLegendLocator.getRasterLegendManager().createMakeColorTable();

        panel.fetch(makeColorTable);

        ColorTable colorTable = makeColorTable.getColorTable();

        ((ColorTablesTableModel)tblColorTables.getModel()).addColorTable(new MutablePair<File, ColorTable>(null, colorTable));

        addedColorTables.add(colorTable);

        return true;

    }

    private void translate() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

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

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

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

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

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.tools.swing.api.colortable.ColorTableSelector#fetch(org.gvsig.raster.lib.legend.api.ColorTable)

     */

    @Override

    public ColorTable getSelected() {

        int row = tblColorTables.getSelectedRow();

        return (ColorTable)((ColorTablesTableModel)tblColorTables.getModel()).getValueAt(row, 0);

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.tools.swing.api.colortable.ColorTableSelector#set(java.util.List)

     */

    @Override

    public void set(List<Pair<File,ColorTable>> colorTables) {

        if(colorTables!=null){

            this.colorTables = colorTables;

        } else {

            this.colorTables = new ArrayList<Pair<File,ColorTable>>();

        }

        this.modifiedColorTables = new ArrayList<Pair<File,ColorTable>>();

        this.addedColorTables = new ArrayList<ColorTable>();

        this.removedColorTables = new ArrayList<Pair<File,ColorTable>>();

        this.init();

    }

    private void init() {

        ColorTablesTableModel tableModel = new DefaultColorTablesTableModel(this.colorTables);

        tblColorTables.setModel(tableModel);

        tblColorTables.setColumnModel(new DefaultTableColumnModel());

        TableColumnModel columnModel = tblColorTables.getColumnModel();

        columnModel.setColumnSelectionAllowed(false);

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

            TableColumn tableColumn = new TableColumn(i);

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

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

            if(i==DefaultColorTablesTableModel.COLUMN_PREVIEW){

                tableColumn.setCellRenderer(new PreviewerColorTableCellRenderer());

            }

            columnModel.addColumn(tableColumn);

        }

    }

    private void updateComponents() {

        boolean selected = tblColorTables.getSelectedRowCount() >= 1;

        btnExport.setEnabled(selected);

        btnRemove.setEnabled(selected && this.editable);

        btnEdit.setEnabled(selected && this.editable);

        btnEdit.setVisible(this.editable);

        btnImport.setVisible(this.editable);

        btnNew.setVisible(this.editable);

        btnRemove.setVisible(this.editable);

        btnExport.setVisible(this.editable);

    }

    @Override

    public List<Pair<File, ColorTable>> getModifiedColorTables() {

        return modifiedColorTables;

    }

    @Override

    public List<Pair<File, ColorTable>> getRemovedColorTables() {

        return removedColorTables;

    }

    @Override

    public List<ColorTable> getAddedColorTables() {

        return addedColorTables;

    }

    @Override

    public void setEditable(boolean editable) {

        this.editable = editable;

        updateComponents();

    }

    /**

     * @author fdiaz

     *

     */

    private static class MyFileFilter extends javax.swing.filechooser.FileFilter {

        private ColorTableIOFactory factory;

        /**

         */

        public MyFileFilter(ColorTableIOFactory factory) {

            this.factory = factory;

        }

        @Override

        public boolean accept(File f) {

            if (f.isDirectory()) {

                return true;

            }

            return this.getColorTableIOFactory().accept(f);

        }

        @Override

        public String getDescription() {

            return this.getColorTableIOFactory().getName();

        }

        /**

         * @return the fileFilter

         */

        public ColorTableIOFactory getColorTableIOFactory() {

            return this.factory;

        }

    }

    @Override

    public synchronized void addFocusListener(FocusListener l) {

        // TODO Auto-generated method stub

        super.addFocusListener(l);

    }

    @Override

    public void addListSelectionListener(ListSelectionListener listener) {

        tblColorTables.getSelectionModel().addListSelectionListener(listener);

    }

}