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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

import org.gvsig.raster.impl.store.ParamsImpl;

/**

 * Clase base para los filtros de ecualizaci?n de histograma.

 *

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class EqualizationFilter extends BaseRasterFilter {

        protected Statistics               stats              = null;

        protected double[]                 minBandValue              = null;

        protected double[]                 maxBandValue              = null;

        protected int                      nbands             = 3;

        protected int[]                    k                  = null;                   

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

        protected BufferHistogram          histogram          = null;

        protected long                     nPixels            = 0;

        protected int[][]                  resValues          = null;

        protected int                      nElements          = 0;

        protected int[]                    ecualizedBands     = null;

        protected double[][]               aproximationNeg    = null;

        protected int                      nClasses           = RasterLibrary.defaultNumberOfClasses;

        protected long[][]                 lahe               = null;

        protected long[][]                 laheNegative       = null;

        /**

         * Construye un LinearEnhancementFilter

         */

        public EqualizationFilter() {

                setName(names[0]);

        }

        public void pre() throws FilterAddException {

                super.pre();

                stats = (Statistics)getEnv().get("SrcStatistics");

                HistogramComputer hist = (HistogramComputer)getEnv().get("SrcHistogram");

                try {

                        if (!stats.isCalculated()) {

                                try {

                                        stats.calculate(RasterLibrary.statisticsScale);

                                } catch (FileNotOpenException e) {

                                        // No podemos aplicar el filtro

                                        return;

                                } catch (RasterDriverException e) {

                                        // No podemos aplicar el filtro

                                        return;

                                }

                        }

                } catch (ProcessInterruptedException e) {

                        //Si se ha interrumpido no a?adimos el filtro

                }

                try {

                        this.histogram = hist.getBufferHistogram();

                } catch (ProcessInterruptedException e) {

                        exec = false;

                } catch (HistogramException e) {

                        exec = false;

                } 

                ecualizedBands = (int[]) params.get("ecualizedBands");

                if(ecualizedBands == null)

                        ecualizedBands = getColorInterpretation().buildRenderBands();

                height = raster.getHeight();

                width = raster.getWidth();

                nPixels = height * width;

                try {

                        stats.calculate(RasterLibrary.statisticsScale);

                } catch (FileNotOpenException e) {

                        exec = false;

                } catch (RasterDriverException e) {

                        exec = false;

                } catch (ProcessInterruptedException e) {

                        exec = false;

                }

                if(raster.getDataType() == Buffer.TYPE_BYTE) {

                        minBandValue = stats.getMinByteUnsigned();

                        maxBandValue = stats.getMaxByteUnsigned();

                } 

                if((minBandValue == null || maxBandValue == null) || (minBandValue[0] == 0 && maxBandValue[0] == 0)) {

                        minBandValue = new double[raster.getBandCount()];

                        maxBandValue = new double[raster.getBandCount()];

                        if(raster.getDataType() == Buffer.TYPE_BYTE) {

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

                                        minBandValue[i] = 0;

                                        maxBandValue[i] = 255;

                                }

                        } else {

                                minBandValue = stats.getMin();

                                maxBandValue = stats.getMax();

                        }

                }

                BufferHistogram rgbHistogram = null;

                if(raster.getDataType() == Buffer.TYPE_BYTE)

                        rgbHistogram = BufferHistogramImpl.convertHistogramToRGB(histogram);

                else 

                        rgbHistogram = histogram;

                double[][] accumNormalize = BufferHistogramImpl.convertTableToNormalizeAccumulate(rgbHistogram.getTable());

                double[][] accumNormalizeNeg = BufferHistogramImpl.convertTableToNormalizeAccumulate(rgbHistogram.getNegativeTable());

                int value = 255;

                if(raster.getDataType() != RasterBuffer.TYPE_BYTE)

                        value = nClasses;

                lahe = lahe(accumNormalize, value);

                laheNegative = lahe(accumNormalizeNeg, value);

                nElements = (laheNegative[0].length - 1);

                nbands = stats.getBandCount();

                createBufferResult(raster.getDataType(), raster.getBandCount());

        }

        /**

         * M?todo lahe para la ecualizaci?n. Cada posici?n del array resultante tendr? el valor de salida para

         * un valor de entrada dado.

         * @param accumNorm Histograma acumulado

         * @param value Valor m?ximo

         * @return

         */

        private long[][] lahe(double[][] accumNorm, int value) {

                long[][] res = new long[accumNorm.length][accumNorm[0].length];

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

                          for (int j = 0; j < res[i].length; j++) 

                                  res[i][j] = Math.round(accumNorm[i][j] * value);

                return res;

        }

        /**

         * Consulta si la ecualizaci?n est? activa para una banda o no

         * @param band N?mero de banda a consultar si se ha activado la ecualizaci?n

         * @return true si est? activa para esa banda y false si no lo est?.

         */

        protected boolean equalizationActive(int band) {

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

                        if(band == ecualizedBands[i])

                                return true;

                }

                return false;

        }

        public int getOutRasterDataType() {

                return Buffer.TYPE_BYTE;

        }

        public String getGroup() {

                return "realces";

        }

        public Params getUIParams(String nameFilter) {

                return new ParamsImpl();

        }

        public void post() {

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

                raster = null;

        }

        public int getInRasterDataType() {

                return 0;

        }

        public void process(int x, int y) {

        }

        public String[] getNames() {

                return names;

        }

        public boolean isVisible() {

                return true;

        }

}