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.impl.grid.filter.convolution;

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

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

import org.gvsig.raster.impl.DefaultRasterManager;

import org.gvsig.raster.impl.buffer.RasterBuffer;

import org.gvsig.raster.impl.datastruct.Kernel;

/**

 * Filtro de convolucion para Buffer de tipo Byte

 * @author Alejandro Mu?oz        <alejandro.munoz@uclm.es>

 * @author Diego Guerrero Sevilla  <diego.guerrero@uclm.es>

 * */

public class ConvolutionByteFilter extends ConvolutionFilter {

        public ConvolutionByteFilter() {

                super();

        }

        /**

         * @param Kernel a aplicar. En caso de que no se trate de un kernel definido en ConvolutionFilter, se puede pasar como

         * parametro el kernel se pretende aplicar.

         * **/

        public ConvolutionByteFilter(Kernel k) {

                super();

                super.kernel = k;

        }

        public void pre() throws FilterAddException {

                super.pre();

                rasterResult = DefaultRasterManager.getInstance().createBuffer(Buffer.TYPE_BYTE, raster.getWidth(), raster.getHeight(), raster.getBandCount(), true);

        }

        /** Aplicacion del filtro para el pixel de la posicion line, col */

        public void process(int col, int line) {

                int lado = kernel.getLado();

                int semiLado = (lado - 1) >> 1;

                double ventana[][] = new double[lado][lado];

                double resultConvolution = 0;

                for (int band = 0; band < numberOfBandsToProcess(); band++) {

                        if ((col - semiLado >= 0) && (line - semiLado >= 0) && (col + semiLado < width) && (line + semiLado < height)) {

                                for (int j = -semiLado; j <= semiLado; j++)

                                        for (int i = -semiLado; i <= semiLado; i++)

                                                ventana[i + semiLado][j + semiLado] = raster.getElemByte(line + j, col + i, band) & 0xff;

                                Kernel Kventana = new Kernel(ventana);

                                resultConvolution = kernel.convolution(Kventana);

                                if (resultConvolution > 255)

                                        resultConvolution = 255;

                                else

                                        if (resultConvolution < 0)

                                                resultConvolution = 0;

                                rasterResult.setElem(line, col, band, (byte) resultConvolution);

                        } else

                                rasterResult.setElem(line, col, band, (byte) raster.getElemByte(line, col, band));

                }

                writeAlphaBand(line, col);

        }

        /**

         * @return  tipo de dato del buffer de entrada

         * */

        public int getInRasterDataType() {

                return RasterBuffer.TYPE_BYTE;

        }

        /**

         * @return tipo de dato del buffer de salida

         * */

        public int getOutRasterDataType() {

                return RasterBuffer.TYPE_BYTE;

        }

}