Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.roimask.app;

import java.awt.Graphics2D;

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

import java.io.File;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import org.gvsig.fmap.dal.coverage.RasterLocator;

import org.gvsig.fmap.dal.coverage.dataset.Buffer;

import org.gvsig.fmap.dal.coverage.datastruct.Extent;

import org.gvsig.fmap.dal.coverage.datastruct.NoData;

import org.gvsig.fmap.dal.coverage.exception.FilterTypeException;

import org.gvsig.fmap.dal.coverage.exception.ProcessInterruptedException;

import org.gvsig.fmap.dal.coverage.exception.ROIException;

import org.gvsig.fmap.dal.coverage.grid.RasterFilter;

import org.gvsig.fmap.dal.coverage.grid.RasterFilterList;

import org.gvsig.fmap.dal.coverage.grid.render.ImageDrawer;

import org.gvsig.fmap.dal.coverage.store.props.ColorInterpretation;

import org.gvsig.fmap.dal.coverage.store.props.Transparency;

import org.gvsig.gui.beans.imagenavigator.ImageUnavailableException;

import org.gvsig.raster.algorithm.RasterBaseAlgorithmLibrary;

import org.gvsig.raster.algorithm.process.DataProcess;

import org.gvsig.raster.algorithm.process.ProcessException;

import org.gvsig.raster.algorithm.process.ProcessParamsManagement;

import org.gvsig.raster.fmap.layers.FLyrRaster;

import org.gvsig.raster.roi.ROI;

import org.gvsig.raster.roimask.algorithm.ROIMaskProcess;

import org.gvsig.raster.roimask.app.ROIMaskExtension.LayerElement;

import org.gvsig.raster.roimask.swing.ROIMaskData;

import org.gvsig.raster.swing.preview.DataSourcePreview;

import org.gvsig.tools.task.Cancellable;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@SuppressWarnings("deprecation")

public class ROIMaskPreview implements DataSourcePreview {

        private FLyrRaster              inputLyr          = null;

        private Extent                  bbox              = null;

        private ROIMaskData             data              = null;

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

        public void setData(ROIMaskData data) {

                this.data = data;

                loadLayers();

        }

        private void loadLayers() {

                if(data == null)

                        return;

                Object obj = data.getInputLayer();

                if(obj != null)

                        inputLyr = (FLyrRaster)((LayerElement)data.getInputLayer()).getLayer();

                if(inputLyr != null)

                        bbox = inputLyr.getFullRasterExtent();                

        }

        public void closePreviewLayer() {

        }

        public double getCellSize() {

                if(inputLyr != null)

                        return inputLyr.getDataStore().getCellSize();

                return 1;

        }

        public DataSourcePreview cloneDataSourcePreview() {

                return this;

        }

        public void popStatus() {

        }

        public void pushStatus() {

        }

        public void draw(BufferedImage image, Graphics2D g, double[] coords, int w, int h, Cancellable cancel, double scale) throws ImageUnavailableException {

                //Posici?n de inicio dibujado dentro del Graphics

                int x = 0;

                int y = 0;

                //Coordenadas de la petici?n

                double iniX = coords[0];

                double iniY = coords[1];

                double endX = coords[2];

                double endY = coords[3];

                //Se ajustan las coordenadas de petici?n, ancho y alto en coordenadas p?xel, 

                //as? como la coordenada de inicio de dibujado 

                if(bbox.getULX() > coords[0]) {

                        iniX = bbox.getULX();

                        x = (int)(((iniX - coords[0]) * w) / Math.abs(coords[2] - coords[0]));

                }

                if(bbox.getULY() < coords[1]) {

                        iniY = bbox.getULY();

                        y = (int)(((coords[1] - iniY) * h) / Math.abs(coords[3] - coords[1]));

                }

                if(bbox.getLRX() < coords[2]) 

                        endX = bbox.getLRX();

                if(bbox.getLRY() > coords[3]) 

                        endY = bbox.getLRY();

                if(iniX != coords[0] || endX != coords[2]) {

                        w = (int)((Math.abs(endX - iniX) * w) / Math.abs(coords[2] - coords[0]));

                }

                if(iniY != coords[1] || endY != coords[3]) {

                        h = (int)((Math.abs(iniY - endY) * h) / Math.abs(coords[3] - coords[1]));

                }

                Extent ext = RasterLocator.getManager().getDataStructFactory().createExtent(iniX, iniY, endX, endY);

                BufferedImage img = throwProcess(ext, w, h);

                g.drawImage(img, x, y, null);

        }

        /**

         * Throws the process sequentially. Since the preview is small,

         * it should not taken a lot of time

         * is small

         * @param ext

         * @param w

         * @param h

         * @return

         */

        private BufferedImage throwProcess(Extent ext, int w, int h) {

                loadLayers();

                DataProcess task;

                try {

                        task = RasterBaseAlgorithmLibrary.getManager().createRasterTask("ROIMaskProcess");

                        task.addParam(ROIMaskProcess.RASTER_STORE1, inputLyr.getDataStore());

                        task.addParam(ProcessParamsManagement.PREVIEW, true);

                        task.addParam(ProcessParamsManagement.WINDOW, ext);

                        task.addParam(ProcessParamsManagement.OUTPUT_WIDTH, w);

                        task.addParam(ProcessParamsManagement.OUTPUT_HEIGHT, h);

                        task.addParam(ROIMaskProcess.ROIS, getROIsSelected());

                        task.addParam(ROIMaskProcess.ALPHA, data.getAlpha());

                        task.addParam(ROIMaskProcess.INVERSE, data.isInverse());

                        task.addParam(ROIMaskProcess.NODATA, getNoDataSelected());

                        task.addParam(ROIMaskProcess.EXPORT, false);

                } catch (ProcessException e1) {

                        logger.debug("Error creating task", e1);

                        return null;

                } catch (ROIException e) {

                        logger.debug("Error creating task", e);

                        return null;

                }

                task.init();

                try {

                        task.process();

                } catch (ProcessInterruptedException e1) {

                        return null;

                } catch (ProcessException e) {

                        logger.debug("Error in process");

                        return null;

                }

                HashMap<String, Object> map = (HashMap<String, Object>)task.getResult();

                Buffer buf = (Buffer)map.get(ROIMaskProcess.BUFFER);

                buf.setDataExtent(ext.toRectangle2D());

                ColorInterpretation ci = inputLyr.getDataStore().getColorInterpretation();

                Transparency transparency = null;

                if(ci.isRGB() || ci.isRGBA()) {

                        Integer alphaBandNumber = (Integer)map.get(ROIMaskProcess.ALPHA_BAND);

                        transparency = RasterLocator.getManager().getDataStructFactory().createTransparency(buf.getBandCount());

                        transparency.setTransparencyBand(alphaBandNumber);

                        transparency.activeTransparency();

                }

                ImageDrawer drawer = RasterLocator.getManager().createImageDrawerService();

                drawer.setBuffer(buf);

                drawer.setLastTransparency(transparency);

                try {

                        applyFiltersToPreview(drawer);

                        return (BufferedImage)drawer.drawBufferOverImageObject();

                } catch (ProcessInterruptedException e) {

                        return null;

                } 

    }

        /**

         * Adds filters to the preview. If the original render has a color table

         * , a generic enhanced will be added.

         * @param drawer

         */

        private void applyFiltersToPreview(ImageDrawer drawer) {

                RasterFilterList filterList = inputLyr.getRender().getFilterList();

                List<RasterFilter> list = new ArrayList<RasterFilter>();

                if(filterList.lenght() > 0) {

                        for (int i = 0; i < filterList.lenght(); i++) {

                                RasterFilter rf = filterList.get(i);

                                if(rf.getName().equals("colortable")) {

                                        drawer.addEnhanced(inputLyr.getDataStore().getStatistics(), true, 0.2);

                                        return;

                                }

                                list.add(rf);

                        }

                }

                try {

                        drawer.addFilters(inputLyr.getDataStore().getStatistics(), list);

                } catch (FilterTypeException e) {

                        logger.debug("Tirando sin filtros", e);

                }

        }

        private NoData getNoDataSelected() {

                NoData nodata = RasterLocator.getManager().getDataStructFactory().createDefaultNoData(

                                inputLyr.getDataStore().getBandCount(), inputLyr.getDataStore().getDataType()[0]);

                nodata.setValue(data.getValueNoData());

                return nodata;

        }

        private ROI[] getROIsSelected() throws ROIException {

                List<ROI> roi = inputLyr.getRois();

                boolean[] roisSelected = data.getROIsSelected();

                List<ROI> roisToDraw = new ArrayList<ROI>(); 

                for (int i = 0; i < roisSelected.length; i++) {

                        if(roisSelected[i])

                                roisToDraw.add(roi.get(i));

                }

                return roisToDraw.toArray(new ROI[0]);

        }

        public Object getSource() {

                return null;

        }

        public Rectangle2D getBBoxRectangle() {

                if(inputLyr != null)

                        return inputLyr.getFullRasterExtent().toRectangle2D();

                return new Rectangle2D.Double();

        }

        public void addParams(DataProcess task, String path, String fileName) throws ROIException {

                if(!fileName.matches("([*]\\.[???])")) {

                        fileName += ".tif";

                }

                task.addParam(ROIMaskProcess.PATH, path + File.separator + fileName);

                task.addParam(ROIMaskProcess.RASTER_STORE1, inputLyr.getDataStore());

                task.addParam(ROIMaskProcess.ROIS, getROIsSelected());

                task.addParam(ROIMaskProcess.ALPHA, data.getAlpha());

                task.addParam(ROIMaskProcess.INVERSE, data.isInverse());

                task.addParam(ROIMaskProcess.NODATA, getNoDataSelected());

                task.addParam(ROIMaskProcess.EXPORT, true);

        }

}