Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.app.mainplugin.colortable;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.util.Iterator;

import java.util.List;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.StringUtils;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

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

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.selector.ColorTableSelectorPanel;

import org.gvsig.tools.ToolsLocator;

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.windowmanager.Dialog;

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

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

/**

 *

 * @author fdiaz

 *

 */

public class ColorTablesLibraryExtension extends Extension {

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

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

    @Override

    public void initialize() {

        IconThemeHelper.registerIcon("action", "color-tables-library-icon", this);

        IconThemeHelper.registerIcon("colortable", "new-color-table-icon", this);

        IconThemeHelper.registerIcon("colortable", "edit-color-table-icon", this);

        IconThemeHelper.registerIcon("colortable", "import-color-table-icon", this);

        IconThemeHelper.registerIcon("colortable", "export-color-table-icon", this);

        IconThemeHelper.registerIcon("colortable", "remove-color-table-icon", this);

        IconThemeHelper.registerIcon("editcolortable","add-color-table-class-icon", this);

        IconThemeHelper.registerIcon("editcolortable","remove-color-table-class-icon", this);

        IconThemeHelper.registerIcon("editcolortable","remove-all-color-table-class-icon", this);

    }

    @Override

    public void execute(String actionCommand) {

        ApplicationManager applicationManager = ApplicationLocator.getManager();

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        if (StringUtils.equalsIgnoreCase(actionCommand, "color-tables-library")) {

            //Mostrar una ventana con el controller

            RasterSwingLegendManager rasterSwingLegendManager = RasterSwingLegendLocator.getSwingManager();

            final ColorTableSelectorPanel panel = rasterSwingLegendManager.createColorTableSelectorPanel();

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

            DynObject properties = rasterPlugin.getPluginProperties();

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

            previousColorTables = RasterLegendLocator.getRasterLegendManager().getColorTables(defaultColorTableLibraryPath);

//            List<Triple<File, ColorTable, Modification>> modifiableColorTables = new ArrayList<Triple<File, ColorTable, Modification>>(previousColorTables.size());

//            for (Iterator iterator = previousColorTables.iterator(); iterator.hasNext();) {

//                Pair<File, ColorTable> pair = (Pair<File, ColorTable>) iterator.next();

//                ImmutableTriple<File, ColorTable, Modification> triple = new ImmutableTriple<File, ColorTable, Modification>(pair.getKey(), pair.getValue(), Modification.NONE);

//                modifiableColorTables.add(triple);

//            }

            panel.set(previousColorTables);

            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-tables-library"),

                    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:

                        doApply(panel);

                        break;

                    case WindowManager_v2.BUTTON_OK:

                        if (doApply(panel)) {

                            panel.asJComponent().setVisible(false);

                        }

                        break;

                    case WindowManager_v2.BUTTON_CANCEL:

                        panel.asJComponent().setVisible(false);

                        break;

                    }

                }

            });

            dialog.show(WindowManager.MODE.WINDOW);

        }

    }

    protected boolean doApply(ColorTableSelectorPanel panel) {

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        List<Pair<File, ColorTable>> removedColorTables = panel.getRemovedColorTables();

        for (Pair<File, ColorTable> pair : removedColorTables) {

            File file = pair.getLeft();

            file.delete();

        }

        List<Pair<File, ColorTable>> modifiedColorTables = panel.getModifiedColorTables();

        for (Pair<File, ColorTable> pair : modifiedColorTables) {

            File oldFile = null;

            File file = pair.getLeft();

            ColorTable colorTable = pair.getRight();

            if (!FilenameUtils.getBaseName(file.getName()).equalsIgnoreCase(colorTable.getName())) {

                oldFile = file;

                file = new File(file.getParent(), colorTable.getName()+ "." + FilenameUtils.getExtension(file.getAbsolutePath()));

            }

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

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

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

                if (colorTableIOFactory.accept(file)) {

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

                    }

                }

            }

            if(oldFile!=null){

                oldFile.delete();

            }

        }

        List<ColorTable> addedColorTables = panel.getAddedColorTables();

        for (ColorTable colorTable : addedColorTables) {

            colorTable.getName();

            PluginServices rasterPlugin =

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

            DynObject properties = rasterPlugin.getPluginProperties();

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

            // FIXME: De momento creo tablas de color en formato de gvSIG

            // (ver de coger la extension en lugar de ponerla fija)

            File file = new File(defaultColorTableLibraryPath, colorTable.getName() + ".xml");

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

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

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

                if (colorTableIOFactory.accept(file)) {

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

                    }

                }

            }

        }

        return true;

    }

    @Override

    public boolean isEnabled() {

        return true;

    }

    @Override

    public boolean isVisible() {

        return true;

    }

}