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

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

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

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 LinearStretchEnhancementFilter extends BaseRasterFilter {

        protected double[][]                      scale              = null;

        protected double[][]                      offset             = null;

        protected Statistics                          stats              = null;

        protected double[]                        minBandValue             = null;

        protected double[]                        maxBandValue             = null;

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

        private boolean                           removeEnds         = false;

        protected DefaultLinearStretchParams      stretchs           = null;

        protected DefaultStretch[]                scaleOffsetList    = null;

        protected int[]                           renderBands        = null; 

        /**

         * Construye un LinearEnhancementFilter

         */

        public LinearStretchEnhancementFilter() {

                setName(names[0]);

        }

        public void pre() throws FilterAddException {

                super.pre();

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

                stretchs = (DefaultLinearStretchParams) params.get("stretchs");

                if (params.get("remove") != null)

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

                if (stretchs == null)

                        return;

                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)

                        stretchs.rgb = false;

                ColorInterpretation outputColorInterp = getColorInterpretation();

                renderBands = outputColorInterp.buildRenderBands();

                stretchs.setMaxMin(stats, renderBands);

                if (removeEnds)

                        stretchs.applyRemoveEndsToStretchs(stats, renderBands);

                stretchs.loadTailTrimValues(stats);

                if (stretchs.hasTailTrim())

                        stretchs.applyTrimToStretchs();

                stretchs.calcLinearScaleAndOffset();

                loadStretchList();

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

        }

        /**

         * La lista de escalas y desplazamientos es un array de 3 elementos en el que

         * cada posici?n es un objeto Stretch con la escala y desplazamiento de la

         * banda que se dibuja en esa posici?n. El objetivo es aplicar a cada banda

         * el m?ximo y m?nimo que le corresponde. Por ejemplo, cuando tenemos una imagen de

         * 3 bandas de tipo short y queremos visualizar en RGB solo la primera banda entonces

         * escaleOffsetList tendr? stretchs.red en las 3 posiciones.

         */

        private void loadStretchList() {

                scaleOffsetList = new DefaultStretch[3];

                scaleOffsetList[0] = stretchs.red;

                scaleOffsetList[1] = stretchs.green;

                scaleOffsetList[2] = stretchs.blue;

        }

        /**

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

         * @return

         */

        public Double getTailTrim() {

                double[] tailTrimList;

                if (stretchs != null)

                        tailTrimList = stretchs.getTailTrimList();

                else

                        tailTrimList = new double[0];

                double median = 0;

                double nValues = tailTrimList.length;

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

                        median += tailTrimList[i];

                return new Double(nValues > 0 ? median / nValues : median);

        }

        public int getOutRasterDataType() {

                return Buffer.TYPE_BYTE;

        }

        /**

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

         * est?

         */

        public Boolean getRemoveEnds() {

                return new Boolean(removeEnds);

        }

        public String getGroup() {

                return "radiometricos";

        }

        public Params getUIParams(String nameFilter) {

                if(stretchs == null)

                        stretchs = (DefaultLinearStretchParams) this.params.get("stretchs");

                Params params = new ParamsImpl();

                params.setParam("TailTrim",

                                new Double(Math.round(getTailTrim().doubleValue() * 100.0)),

                                Params.SLIDER,

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

                params.setParam("StretchInRed",

                                stretchs.red.stretchIn,

                                Params.CHOICE,

                                null);

                params.setParam("StretchInGreen",

                                stretchs.green.stretchIn,

                                Params.CHOICE,

                                null);

                params.setParam("StretchInBlue",

                                stretchs.blue.stretchIn,

                                Params.CHOICE,

                                null);

                params.setParam("StretchOutRed",

                                stretchs.red.stretchOut,

                                Params.CHOICE,

                                null);

                params.setParam("StretchOutGreen",

                                stretchs.green.stretchOut,

                                Params.CHOICE,

                                null);

                params.setParam("StretchOutBlue",

                                stretchs.blue.stretchOut,

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimRedMin",

                                new Double(stretchs.red.tailTrimMin),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimRedMax",

                                new Double(stretchs.red.tailTrimMax),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimGreenMin",

                                new Double(stretchs.green.tailTrimMin),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimGreenMax",

                                new Double(stretchs.green.tailTrimMax),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimBlueMin",

                                new Double(stretchs.blue.tailTrimMin),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimBlueMax",

                                new Double(stretchs.blue.tailTrimMax),

                                Params.CHOICE,

                                null);

                params.setParam("RedMaxValue",

                                new Double(stretchs.red.maxValue),

                                Params.CHOICE,

                                null);

                params.setParam("RedMinValue",

                                new Double(stretchs.red.minValue),

                                Params.CHOICE,

                                null);

                params.setParam("GreenMaxValue",

                                new Double(stretchs.green.maxValue),

                                Params.CHOICE,

                                null);

                params.setParam("GreenMinValue",

                                new Double(stretchs.green.minValue),

                                Params.CHOICE,

                                null);

                params.setParam("BlueMaxValue",

                                new Double(stretchs.blue.maxValue),

                                Params.CHOICE,

                                null);

                params.setParam("BlueMinValue",

                                new Double(stretchs.blue.minValue),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimRedValueMax",

                                new Double(stretchs.red.tailTrimValueMax),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimRedValueMin",

                                new Double(stretchs.red.tailTrimValueMin),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimGreenValueMax",

                                new Double(stretchs.green.tailTrimValueMax),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimGreenValueMin",

                                new Double(stretchs.green.tailTrimValueMin),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimBlueValueMax",

                                new Double(stretchs.blue.tailTrimValueMax),

                                Params.CHOICE,

                                null);

                params.setParam("TailTrimBlueValueMin",

                                new Double(stretchs.blue.tailTrimValueMin),

                                Params.CHOICE,

                                null);

                params.setParam("RedOffset",

                                stretchs.red.offset,

                                Params.CHOICE,

                                null);

                params.setParam("GreenOffset",

                                stretchs.green.offset,

                                Params.CHOICE,

                                null);

                params.setParam("BlueOffset",

                                stretchs.blue.offset,

                                Params.CHOICE,

                                null);

                params.setParam("RedScale",

                                stretchs.red.scale,

                                Params.CHOICE,

                                null);

                params.setParam("GreenScale",

                                stretchs.green.scale,

                                Params.CHOICE,

                                null);

                params.setParam("BlueScale",

                                stretchs.blue.scale,

                                Params.CHOICE,

                                null);

                params.setParam("Remove",

                                new Boolean(removeEnds),

                                Params.CHOICE,

                                null);

                /*if(renderBands == null)

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

                params.setParam("RenderBands",

                                convertArrayToString(renderBands),

                                Params.NONE,

                                null);*/

                params.setParam("RGB",

                                new Boolean(stretchs.rgb),

                                Params.NONE,

                                null);

                return params;

        }

        /**

         * Convierte un array de dobles a una cadena

         * @param values

         * @return

         */

        /*private String convertArrayToString(int[] values) {

                StringBuffer buffer = new StringBuffer();

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

                        buffer.append(values[i]);

                        if (i < (values.length - 1))

                                buffer.append(" ");

                }

                return buffer.toString();

        }*/

        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 false;

        }

        /**

         * @return the stretchs

         */

        public DefaultLinearStretchParams getStretchs() {

                return stretchs;

        }

}