Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2005 IVER T.I. and Generalitat Valenciana.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.raster.regionalpha.filter;

import java.awt.geom.Point2D;

import java.util.ArrayList;

import org.gvsig.raster.buffer.RasterBuffer;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.dataset.MultiRasterDataset;

import org.gvsig.raster.dataset.Params;

import org.gvsig.raster.grid.GridExtent;

import org.gvsig.raster.grid.filter.RasterFilter;

import org.gvsig.raster.grid.roi.ROI;

import org.gvsig.raster.regionalpha.panel.RegionAlphaUI;

import com.iver.cit.gvsig.fmap.layers.FLayer;

/**

 * 

 * @version 15/01/2008

 * @author BorSanZa - Borja S?nchez Zamorano (borja.sanchez@iver.es)

 */

public class RegionAlphaFilter extends RasterFilter {

        public static String[] names           = new String[] { "regionalpha" };

        private RegionAlphaUI  regionAlphaUI   = null;

        private ArrayList      rois            = null;

        private IBuffer        rasterAlpha     = null;

        private int            alphaBandNumber = -1;

        private int            alpha           = 255;

        /* Variables que hacen falta en el process */

        private GridExtent gridExtent;

        private GridExtent windowExtent;

        private MultiRasterDataset dataset;

        /**

         * Constructor

         */

        public RegionAlphaFilter() {

                super();

                setName(names[0]);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.grid.filter.RasterFilter#getGroup()

         */

        public String getGroup() {

                return "radiometricos";

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.grid.filter.RasterFilter#getNames()

         */

        public String[] getNames() {

                return names;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.grid.filter.RasterFilter#getResult(java.lang.String)

         */

        public Object getResult(String name) {

                if (!name.equals("raster"))

                        return null;

                if (!exec)

                        return (Object) this.raster;

                return (Object) this.rasterResult;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.grid.filter.RasterFilter#getUIParams(java.lang.String)

         */

        public Params getUIParams(String nameFilter) {

                Params params = new Params();

                params.setParam("Panel", getRegionAlphaUI(), -1, null);

                params.setParam("FilterName", nameFilter, -1, null);

                params.setParam("Alpha",

                                new Integer(alpha),

                                Params.SLIDER,

                                new String[]{ "0", "255", "0", "1", "25" }); //min, max, valor defecto, intervalo peque?o, intervalo grande;

                return params;

        }

        private RegionAlphaUI getRegionAlphaUI() {

                if (regionAlphaUI == null) {

                        regionAlphaUI = new RegionAlphaUI();

                        FLayer raster = (FLayer) getEnv().get("initRaster");

                        regionAlphaUI.setRois(rois);

                        regionAlphaUI.setLayer(raster);

                }

                return regionAlphaUI;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.grid.filter.RasterFilter#pre()

         */

        public void pre() {

                exec = true;

                raster = rasterResult;

                raster = (RasterBuffer) params.get("raster");

                rois = (ArrayList) params.get("rois");

                if (rois == null)

                        rois = new ArrayList();

                height = raster.getHeight();

                width = raster.getWidth();

                alpha = ((Integer) params.get("alpha")).intValue();

                gridExtent = (GridExtent) environment.get("GridExtent");

                windowExtent = (GridExtent) environment.get("WindowExtent");

                dataset = (MultiRasterDataset) environment.get("MultiRasterDataset");

                Integer b = ((Integer) environment.get("alphaBandNumber"));

                if(b != null)

                        alphaBandNumber = b.intValue();

                if(alphaBandNumber < 0)

                        rasterAlpha = RasterBuffer.getBuffer(IBuffer.TYPE_BYTE, raster.getWidth(), raster.getHeight(), 1, true);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.grid.filter.RasterFilter#post()

         */

        public void post() {

//        En caso de que nadie apunte a raster, se liberara su memoria.

                if(alphaBandNumber < 0)

                        mergeBufferTransparency(rasterAlpha);

                raster = null;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.grid.filter.RasterFilter#process(int, int)

         */

        public void process(int x, int y) {

                if ((windowExtent == null) || (gridExtent == null))

                        return;

                double newX = windowExtent.minX() + ((((double) x) * windowExtent.width()) / ((double) raster.getWidth()));

                double newY = windowExtent.minY() + ((( (double) (raster.getHeight() - (y ))) * windowExtent.height()) / ((double) raster.getHeight()));

                Point2D point2D = dataset.worldToRaster(new Point2D.Double(newX, newY));

                newX = point2D.getX();

                newY = point2D.getY();

                if(alphaBandNumber < 0) {

                        for (int i = 0; i < rois.size(); i++) {

                                if (((ROI) rois.get(i)).isInGrid((int) Math.round(newX), (int) Math.round(newY))) {

                                        rasterAlpha.setElem(y, x, 0, (byte) (255 - alpha));

                                        return;

                                }

                        }

                        rasterAlpha.setElem(y, x, 0, (byte) 255);

                } else {

                        for (int i = 0; i < rois.size(); i++) {

                                if (((ROI) rois.get(i)).isInGrid((int) Math.round(newX), (int) Math.round(newY))) {

                                        raster.setElem(y, x, alphaBandNumber, 

                                                (byte) (((255 - alpha) & 0xff) * (raster.getElemByte(y, x, alphaBandNumber) & 0xff) / 255));

                                        return;

                                }

                        }

                        raster.setElem(y, x, alphaBandNumber, 

                                (byte) ((255 & 0xff) * (raster.getElemByte(y, x, alphaBandNumber) & 0xff) / 255));

                }

        }

        public int getInRasterDataType() {return 0;}

        public int getOutRasterDataType() {return 0;}

}