Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 */

package org.gvsig.raster.tools.app.basic.tool.filter.regionalpha;

import java.util.ArrayList;

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

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

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

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

import org.gvsig.fmap.dal.coverage.grid.filter.BaseRasterFilter;

import org.gvsig.fmap.dal.coverage.store.RasterDataStore;

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

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

import org.gvsig.raster.roi.ROI;

/**

 * 

 * @version 15/01/2008

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

 */

public class RegionAlphaFilter extends BaseRasterFilter {

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

        private RegionAlphaUI      regionAlphaUI   = null;

        private ArrayList<ROI>     rois            = null;

        private Buffer             rasterAlpha     = null;

        private int                alpha           = 255;

        protected boolean          inverse         = false;

        /* Variables que hacen falta en el process */

        private GridExtent         gridExtent      = null;

        private GridExtent         windowExtent    = null;

        private RasterDataStore    dataset         = null;

        private double             cellsize        = 0D;

        /**

         * Constructor

         */

        public RegionAlphaFilter() {

                super();

                setName(names[0]);

        }

        /*

         * (non-Javadoc)

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

         */

        public String getGroup() {

                return "mascaras";

        }

        /*

         * (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("alphaBand"))

                        if(rois == null || rois.size() == 0)

                                return null;

                        else

                                return rasterAlpha;

                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 = RasterLocator.getManager().createParams(

                                "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);

                        if(raster instanceof FLyrRaster)

                                regionAlphaUI.setLayer((FLyrRaster)raster);

                }

                return regionAlphaUI;

        }

        /*

         * (non-Javadoc)

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

         */

        @SuppressWarnings("unchecked")

        public void pre() {

                exec = true;

                raster = rasterResult;

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

                Boolean inverseBoolean = (Boolean) params.get("inverse");

                if (inverseBoolean != null)

                        inverse = inverseBoolean.booleanValue();

                rois = (ArrayList<ROI>) params.get("rois");

                if (rois == null)

                        rois = new ArrayList<ROI>();

                height = raster.getHeight();

                width = raster.getWidth();

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

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

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

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

                rasterAlpha = RasterLocator.getManager().createBuffer(Buffer.TYPE_BYTE, raster.getWidth(), raster.getHeight(), 1, true);

                cellsize = dataset.getCellSize();

        }

        /*

         * (non-Javadoc)

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

         */

        public void post() {

        }

        /*

         * (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 wcX = windowExtent.minX() + ((((double) x) * windowExtent.width()) / ((double) raster.getWidth()));

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

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

                        if (((ROI) rois.get(i)).isInside(wcX, wcY, cellsize, cellsize)) {

                                if (inverse)

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

                                else

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

                                return;

                        }

                }

                if (inverse)

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

                else

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

        }

        public int getInRasterDataType() {return 0;}

        public int getOutRasterDataType() {return 0;}

}