Application: gvSIG desktop » gvSIG raster

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

import java.awt.Graphics2D;

import java.awt.image.BufferedImage;

import java.io.IOException;

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

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.MapContextException;

import org.gvsig.fmap.mapcontext.ViewPort;

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

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

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

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

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

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.raster.tools.swing.api.RasterToolsSwingLocator;

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

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.locator.LocatorException;

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

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

/**

 *

 * @author fdiaz

 *

 */

public class ExportViewToImageExtension extends Extension {

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

    private MapContext mapContext;

    @Override

    public void initialize() {

        IconThemeHelper.registerIcon("action", "view-export-image", this);

    }

    @Override

    public void execute(String actionCommand) {

        if (StringUtils.equalsIgnoreCase(actionCommand, "export-view-to-image")) {

            IView view = getActiveView();

            MapControl mapControl = view.getMapControl();

            mapContext = mapControl.getMapContext();

            ViewPort viewPort = mapControl.getViewPort();

            ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

            BufferedImage image = toolsSwingManager.createBufferedImage(viewPort.getImageWidth(), viewPort.getImageHeight(), BufferedImage.TYPE_3BYTE_BGR);

            Graphics2D g = image.createGraphics();

            try {

                mapContext.draw(image, g, null, mapContext.getScaleView());

            } catch (ReadException | MapContextException e2) {

                logger.warn("Can't draw mapContext.", e2);

            }

            Envelope envelope = viewPort.getAdjustedEnvelope();

            BufferManager bufferManager = BufferLocator.getBufferManager();

            Buffer buffer = null;

            try {

                buffer = bufferManager.createBufferFromAwtRaster(image.getRaster(), viewPort.getProjection(), envelope);

                RasterSwingBufferManager rasterSwingBufferManager = RasterSwingBufferLocator.getSwingManager();

                SaveBufferDialog saveBufferDialog = rasterSwingBufferManager.createSaveBufferDialog(buffer);

                saveBufferDialog.showDialog(null);

            } catch (IOException | BufferException e1) {

                logger.warn("Can't create buffer from view's image.", e1);

                return;

            } finally {

                DisposeUtils.disposeQuietly(buffer);

            }

        }

    }

    @Override

    public boolean isEnabled() {

        IView view = getActiveView();

        if (view != null) {

            ViewDocument viewDocument = view.getViewDocument();

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

            for (int i = 0; i < layers.getLayersCount(); i++) {

                FLayer layer = layers.getLayer(i);

                if (layer != null && layer.isAvailable()) {

                    return true;

                }

            }

        }

        return false;

    }

    @Override

    public boolean isVisible() {

        IView view = getActiveView();

        return (view != null);

    }

    private IView getActiveView() {

        ApplicationManager application = ApplicationLocator.getManager();

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

        return view;

    }

}