Application: gvSIG desktop

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

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.List;

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.raster.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.BufferLocator;

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

import org.gvsig.raster.tools.lib.api.RasterClip;

import org.gvsig.raster.tools.lib.api.RasterToolsLocator;

import org.gvsig.raster.tools.lib.api.exceptions.ClipException;

import org.gvsig.raster.tools.lib.api.exceptions.RasterToolException;

import org.gvsig.raster.tools.swing.api.RasterToolsSwingLocator;

import org.gvsig.raster.tools.swing.api.RasterToolsSwingManager;

import org.gvsig.raster.tools.swing.api.clip.ClipPanel;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

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

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

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 ClipExtension extends Extension {

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

    @Override

    public void initialize() {

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

        IconThemeHelper.registerIcon("clip", "downright", this);

        IconThemeHelper.registerIcon("clip", "upleft", this);

        IconThemeHelper.registerIcon("clip", "fullextent-icon", this);

        IconThemeHelper.registerIcon("clip", "viewextent-icon", this);

        IconThemeHelper.registerIcon("clip", "roi", this);

        IconThemeHelper.registerIcon("clip", "round-icon", this);

        IconThemeHelper.registerIcon("clip", "selecttool-icon", this);

        IconThemeHelper.registerIcon("clip", "loadparameters-icon", this);

        IconThemeHelper.registerIcon("clip", "saveparameters-icon", this);

        IconThemeHelper.registerIcon("clip", "up", this);

        IconThemeHelper.registerIcon("clip", "down", this);

    }

    @Override

    public void execute(String actionCommand) {

        if (StringUtils.equalsIgnoreCase(actionCommand, "clip")) {

            IView view = getActiveView();

            MapControl mapControl = view.getMapControl();

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

            if (layers[0] instanceof RasterLayer) {

                RasterLayer layer = (RasterLayer) layers[0];

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

                final RasterToolsSwingManager manager = RasterToolsSwingLocator.getSwingManager();

                final I18nManager i18nManager = ToolsLocator.getI18nManager();

                final ClipPanel panel;

                try {

                    panel = manager.createClipPanel();

                    panel.asJComponent().setPreferredSize(new Dimension(500, 250));

                    panel.set(layer);

                    panel.set(mapControl);

                } catch (RasterToolException e2) {

                    logger.error("Error creating panel", e2);

                    JOptionPane.showMessageDialog(null,

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

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

                        JOptionPane.ERROR_MESSAGE);

                    return;

                }

                final Dialog dialog =

                    winManager.createDialog(panel.asJComponent(), i18nManager.getTranslation("Clip")+": "+layer.getName(), 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:

                            doClip(panel);

                            break;

                        case WindowManager_v2.BUTTON_OK:

                            doClip(panel);

                            panel.asJComponent().setVisible(false);

                            break;

                        case WindowManager_v2.BUTTON_CANCEL:

                            panel.asJComponent().setVisible(false);

                            break;

                        }

                    }

                });

                dialog.show(WindowManager.MODE.WINDOW);

            }

        }

    }

    private void doClip(ClipPanel panel) {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        ThreadSafeDialogsManager dlgManager = ToolsSwingLocator.getThreadSafeDialogsManager();

        RasterClip clip = RasterToolsLocator.getManager().createRasterClip();

        panel.fetch(clip);

        long rasterSize = calculateRasterSize(clip); //Bytes

        if(rasterSize>10000000){

            String message = "_the_resulting_raster_will_have_an_estimated_size_of_XSIZEX_MB_are_you_sure_XquestionX";

            String[] sizes = new String[1];

            sizes[0]= new Long(rasterSize/10000000).toString();

            String title = "Clip";

            int confirm = dlgManager.confirmDialog(

                i18nManager.getTranslation(message, sizes),

                i18nManager.getTranslation(title),

                JOptionPane.YES_NO_OPTION,

                JOptionPane.QUESTION_MESSAGE);

            if(confirm!=JOptionPane.YES_OPTION){

                return;

            }

        }

        try {

            clip.execute(null);

        } catch (ClipException e) {

            logger.warn("Can't create raster clip.", e);

            String message = "_cant_create_raster_clip";

            String title = "_error_creating_clip";

            dlgManager.messageDialog(message, title, JOptionPane.WARNING_MESSAGE);

        }

    }

    private long calculateRasterSize(RasterClip clip) {

        RasterLayer layer = clip.getLayer();

        long columns = clip.getColumns();

        long rows = clip.getRows();

        RasterStore rasterStore = layer.getRasterStore();

        List<Integer> bands = clip.getBands();

        BufferManager bufferManager = BufferLocator.getBufferManager();

        long size = 0;

        for (Integer band : bands) {

            size += rows*columns*bufferManager.getDataTypeSize(rasterStore.getBandDescriptor(band.intValue()).getDataType());

        }

        return size;

    }

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

    }

}