Application: gvSIG desktop » gvSIG raster

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

import javax.swing.JOptionPane;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.project.documents.view.ViewDocument;

import org.gvsig.app.project.documents.view.gui.IView;

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

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

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

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

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

import org.gvsig.raster.lib.buffer.api.exceptions.BufferOperationException;

import org.gvsig.raster.lib.buffer.api.operations.OperationList;

import org.gvsig.raster.swing.buffer.RasterSwingBufferLocator;

import org.gvsig.raster.swing.buffer.RasterSwingBufferManager;

import org.gvsig.raster.swing.buffer.save.SaveBufferDialog;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

/**

 *

 * @author fdiaz

 *

 */

public class SaveRasterLayerExtension extends Extension {

    private static final Logger logger = LoggerFactory.getLogger(SaveRasterLayerExtension.class);

    @Override

    public void initialize() {

        IconThemeHelper.registerIcon("action", "layer-saveas", this);

    }

    @Override

    public void execute(String actionCommand) {

        if (StringUtils.equalsIgnoreCase(actionCommand, "save-raster-layer")) {

            I18nManager i18nManager = ToolsLocator.getI18nManager();

            IView view = getActiveView();

            MapControl mapControl = view.getMapControl();

            FLayer[] layers = mapControl.getMapContext().getLayers().getActives();

            if (layers[0] instanceof RasterLayer) {

                final RasterLayer layer = (RasterLayer) layers[0];

                final RasterStore store = layer.getRasterStore();

                Buffer buffer;

                try {

                    buffer = store.getRasterSet();

                } catch (DataException e) {

                    logger.error("Error creating raster store's buffer", e);

                    JOptionPane.showMessageDialog(null,

                        new StringBuilder().append(i18nManager.getTranslation("error_creating_buffer")).append(":")

                            .append(e.getLocalizedMessage()), i18nManager.getTranslation("error"),

                        JOptionPane.ERROR_MESSAGE);

                    return;

                }

                OperationList filters = layer.getLegend().getFilters();

                if(buffer != null){

                    Buffer filteredBuffer = null;

                    try {

                        filteredBuffer = filters.execute(buffer);

                    } catch (BufferOperationException e) {

                        logger.error("Error applying the filters", e);

                        JOptionPane.showMessageDialog(null,

                            new StringBuilder().append(i18nManager.getTranslation("error_applying_buffer")).append(":")

                                .append(e.getLocalizedMessage()), i18nManager.getTranslation("error"),

                            JOptionPane.ERROR_MESSAGE);

                        return;

                    }

                    RasterSwingBufferManager rasterSwingBufferManager = RasterSwingBufferLocator.getSwingManager();

                    SaveBufferDialog saveBufferDialog = rasterSwingBufferManager.createSaveBufferDialog();

                    saveBufferDialog.setBuffer(filteredBuffer);

                    saveBufferDialog.showDialog(null);

                    return;

                }

            }

        }

    }

    @Override

    public boolean isEnabled() {

        IView view = getActiveView();

        if (view != null) {

            ViewDocument viewDocument = view.getViewDocument();

            FLayer[] layers = viewDocument.getMapContext().getLayers().getActives();

            if (layers.length == 1 && layers[0] instanceof RasterLayer) {

                return true;

            }

        }

        return false;

    }

    @Override

    public boolean isVisible() {

        IView view = getActiveView();

        if (view != null) {

            ViewDocument viewDocument = view.getViewDocument();

            FLayers layers = viewDocument.getMapContext().getLayers();

            for (FLayer fLayer : layers) {

                if (fLayer instanceof RasterLayer) {

                    return true;

                }

            }

        }

        return false;

    }

    private IView getActiveView() {

        ApplicationManager application = ApplicationLocator.getManager();

        IView view = (IView) application.getActiveComponent(ViewDocument.class);

        return view;

    }

}