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

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

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

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

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

/**

 * Clase base para los filtros de realzado lineal. Lee el m?nimo y m?xmo de la

 * clase Statistic que ser?n calculados por PercentTailTrimFilter o

 * ComputeMinMaxFilter dependiendo de si est? activado el recorte de colas o no.

 * En Statistic tambi?n est?n los segundos valores despu?s del m?nimo y m?ximo

 * que son los que se utilizan con la opci?n eliminar extremos activada. Estos

 * se usaran en vez del m?nimo y m?ximo cuando la variable removeExtrema est? a

 * true.

 *

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

 */

public class LinearEnhancementFilter extends BaseRasterFilter {

        protected double[]                 scale        = new double[3];

        protected double[]                 offset       = new double[3];

        protected Statistics               stats        = null;

        protected double[]                 minBandValue        = null;

        protected double[]                 maxBandValue        = null;

        protected boolean                  removeEnds   = false;

        protected double                   tailTrim     = 0D;

        protected int                      nbands       = 3;

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

        /**

         * Construye un LinearEnhancementFilter

         */

        public LinearEnhancementFilter() {

                setName(names[0]);

        }

        public void pre() throws FilterAddException {

                super.pre();

                stats = (Statistics) params.get("stats");

                removeEnds = ((Boolean) params.get("remove")).booleanValue();

                tailTrim = ((Double) params.get("tailTrim")).doubleValue();

                // En realidad esto no se deberia hacer, pero por logica, un valor del 50%

                // devolveria una capa en blanco, asi que evitamos un resultado supuestamente

                // no esperado

                if ((tailTrim >= 0.5) && (tailTrim < 0.51))

                        tailTrim = 0.51;

                if ((tailTrim > 0.49) && (tailTrim < 0.50))

                        tailTrim = 0.49;

                try {

                        stats.calculate(RasterLibrary.statisticsScale);

                } catch (FileNotOpenException e) {

                        exec = false;

                } catch (RasterDriverException e) {

                        exec = false;

                } catch (ProcessInterruptedException e) {

                        exec = false;

                }

                double[][] tailTrimByBand = (double[][]) stats.getTailTrimValue(tailTrim);

                if ((tailTrim != 0) && (tailTrimByBand != null)) { // Max y Min con recorte de colas

                        scale = new double[tailTrimByBand.length];

                        offset = new double[tailTrimByBand.length];

                        minBandValue = new double[tailTrimByBand.length];

                        maxBandValue = new double[tailTrimByBand.length];

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

                                minBandValue[i] = tailTrimByBand[i][0];

                                maxBandValue[i] = tailTrimByBand[i][1];

                        }

                } else {

                        scale = new double[stats.getMin().length];

                        offset = new double[stats.getMin().length];

                        if (removeEnds) { // Si est? activado eliminar extremos gastamos el 2? m?ximo/m?nimo

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

                                        minBandValue = stats.getSecondMinByteUnsigned();

                                        maxBandValue = stats.getSecondMaxByteUnsigned();

                                } else {

                                        minBandValue = stats.getSecondMin();

                                        maxBandValue = stats.getSecondMax();

                                }

                        } else { // Si no est? activado eliminar extremos

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

                                        minBandValue = stats.getMinByteUnsigned();

                                        maxBandValue = stats.getMaxByteUnsigned();

                                } else {

                                        minBandValue = stats.getMin();

                                        maxBandValue = stats.getMax();

                                }

                        }

                }

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

                        scale[i] = 255D / (maxBandValue[i] - minBandValue[i]);

                        offset[i] = (255D * minBandValue[i]) / (minBandValue[i] - maxBandValue[i]);

                }

                nbands = stats.getBandCount();

                createBufferResult(Buffer.TYPE_BYTE, raster.getBandCount());

        }

        /**

         * Obtiene true si est? activado el flag de eliminar extremos y false si no lo

         * est?

         */

        public Boolean getRemoveEnds() {

                return new Boolean(removeEnds);

        }

        /**

         * Obtiene el porcentaje de recorte de colas aplicado o 0 si no tiene.

         * @return

         */

        public Double getTailTrim(){

                return new Double(tailTrim);

        }

        public int getOutRasterDataType() {

                return Buffer.TYPE_BYTE;

        }

        public String getGroup() {

                return "radiometricos";

        }

        public Params getUIParams(String nameFilter) {

                Params params = new ParamsImpl();

                params.setParam("RemoveEnds",

                                new Boolean(removeEnds),

                                Params.CHECK,

                                null);

                params.setParam("TailTrim",

                                new Double(Math.round(tailTrim * 100.0)),

                                Params.SLIDER,

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

                return params;

        }

        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;

        }

}