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

import java.awt.image.DataBuffer;

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

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

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

import org.gvsig.raster.roi.ROI;

/**

 * Filtro que aplica una m?scara con la lista de ROIS. Los p?xeles dentro

 * de la ROI se ponen al valor de la imagen de origen. Los p?xeles fuera

 * de la ROI se ponen a NoData. Esta clase es la gestiona los datos para los

 * raster de tipo short.

 * 

 * @author Nacho Brodin nachobrodin@gmail.com

 */

public class MaskShortFilter extends MaskFilter {

        private short nodata     = RasterLibrary.defaultShortNoDataValue;

        public void pre() throws FilterAddException {

                super.pre();

                noData = (NoData)params.get("nodata");

                if(noData != null && noData.isDefined())

                        nodata = noData.getValue().shortValue();

        }

        public void process(int x, int y) {

                if(bufferExtent == null) {

                        for (int j = 0; j < raster.getBandCount(); j++) 

                                rasterResult.setElem(y, x, j, raster.getElemShort(y, x, j));

                        return;

                }

                double wcX = bufferExtent.minX() + ((((double) x) * bufferExtent.width()) / ((double) raster.getWidth()));

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

                if(inverse) {

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

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

                                        for (int j = 0; j < numberOfBandsToProcess(); j++) 

                                                rasterResult.setElem(y, x, j, (short)nodata);

                                        return;

                                }

                        }

                        for (int j = 0; j < numberOfBandsToProcess(); j++) 

                                rasterResult.setElem(y, x, j, raster.getElemShort(y, x, j));

                        return;

                }

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

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

                                for (int j = 0; j < numberOfBandsToProcess(); j++) 

                                        rasterResult.setElem(y, x, j, raster.getElemShort(y, x, j));

                                return;

                        }

                }

                for (int j = 0; j < numberOfBandsToProcess(); j++) 

                        rasterResult.setElem(y, x, j, (short)nodata);

                return;

        }

        public int getInRasterDataType() {

                return DataBuffer.TYPE_SHORT;

        }

        public int getOutRasterDataType() { 

                return DataBuffer.TYPE_SHORT;

        }

}