Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2006 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.raster.grid.filter.enhancement;

import org.gvsig.raster.buffer.RasterBuffer;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.dataset.Params;

import org.gvsig.raster.dataset.properties.DatasetListStatistics;

import org.gvsig.raster.grid.filter.RasterFilter;

/**

 * 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 RasterFilter {

        public static String                genericName = "enhanced";

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

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

        protected DatasetListStatistics        stats = null; 

        protected double[]                         minBandValue = null;

        protected double[]                         maxBandValue = null;

        protected boolean                         removeEnds = false;

        protected double                                 tailTrim = 0D;

        protected IBuffer                                rasterResult = null;

        protected int                                         nbands = 3;

        protected int[]                                        renderBands = null;

        /**

         * Construye un LinearEnhancementFilter

         */

  public LinearEnhancementFilter() {

                super();

                super.fName = genericName;

        }

  /*

         * (non-Javadoc)

         * 

         * @see org.cresques.io.raster.IRasterFilter#pre()

         */

  public void pre() {

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

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

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

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

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

                height = raster.getHeight();

                width = raster.getWidth();

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

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

                        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

                                minBandValue = stats.getSecondMin();

                                maxBandValue = stats.getSecondMax();

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

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

                rasterResult = RasterBuffer.getBuffer(IBuffer.TYPE_BYTE, raster.getWidth(), raster.getHeight(), raster.getBandCount(), true);

        }

  /**

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

  }

  /*

   *  (non-Javadoc)

   * @see org.gvsig.fmap.grid.filter.IRasterFilter#getOutRasterDataType()

   */

  public int getOutRasterDataType() {

                return IBuffer.TYPE_BYTE;

        }

        public Object getResult(String name) {

                return (Object) this.rasterResult;

        }

        /*

         * (non-Javadoc)

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

         */

        public String getName() {

                return genericName;

        }

        /*

         * (non-Javadoc)

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

         */

        public String getGroup() {

                return "basics";

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.grid.filter.IRasterFilter#getUIParams()

         */

        public Params getUIParams() {

                Params params = new Params();

                params.setParam("RemoveEnds",

                                removeEnds + "",

                                Params.CHECK,

                                null);

                params.setParam("TailTrim",

                                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;

        }

        /*

         * (non-Javadoc)

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

         */

        public void post() {

                if (nbands < rasterResult.getBandCount()) {

                        for (int band = nbands; band < rasterResult.getBandCount(); band++) {

                                if (renderBands != null && band < renderBands.length) {

                                        for (int col = 0; col < rasterResult.getWidth(); col++)

                                                for (int row = 0; row < rasterResult.getHeight(); row++)

                                                        rasterResult.setElem(row, col, band, rasterResult.getElemByte(row, col, renderBands[band]));

                                }

                        }

                }

        }

        /*

         * (non-Javadoc)

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

         */

        public int getInRasterDataType() {

                return 0;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.grid.filter.RasterFilter#process(int, int)

         */

        public void process(int x, int y) {

        }

        public Object clone() throws CloneNotSupportedException {

    Object obj = null;

                obj = super.clone();

                return obj;

        }

}