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.datastruct.Stretch;

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

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

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

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

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

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

/**

 * Par?metros necesarios para el filtro de realce por tramos. Contiene

 * informaci?n de cada tramo. Valores m?ximos y m?nimos reales y correspondencia

 * a RGB de cada tramo.

 * 

 * Tambi?n contiene los valores calculados en el preproceso de escala y

 * desplazamiento para cada banda.

 * 

 * @author Nacho Brodin nachobrodin@gmail.com

 */

public class DefaultLinearStretchParams implements LinearStretchParams {

        public DefaultStretch             red               = new DefaultStretch();

        public DefaultStretch             green             = new DefaultStretch();

        public DefaultStretch             blue              = new DefaultStretch();

        public boolean                    rgb               = false;

        public boolean hasTailTrim() {

                return (red.hasTailTrim() ||

                                green.hasTailTrim() ||

                                blue.hasTailTrim());

        }

        public double[] getTailTrimList() {

                return new double[]{

                                red.tailTrimMin, green.tailTrimMin, blue.tailTrimMin, 

                                red.tailTrimMax, green.tailTrimMax,        blue.tailTrimMax};

        }

        public void loadTailTrimValues(Statistics stats) {

                double[][] result;

                result = (double[][]) stats.getTailTrimValue(red.tailTrimMin);

                if (result != null)

                        red.tailTrimValueMin = result[0][0];

                result = (double[][]) stats.getTailTrimValue(red.tailTrimMax);

                if (result != null)

                        red.tailTrimValueMax = result[0][1];

                result = (double[][]) stats.getTailTrimValue(green.tailTrimMin);

                if (result != null && result.length >= 2)

                        green.tailTrimValueMin = result[1][0];

                result = (double[][]) stats.getTailTrimValue(green.tailTrimMax);

                if (result != null && result.length >= 2)

                        green.tailTrimValueMax = result[1][1];

                result = (double[][]) stats.getTailTrimValue(blue.tailTrimMin);

                if (result != null && result.length >= 3)

                        blue.tailTrimValueMin = result[2][0];

                result = (double[][]) stats.getTailTrimValue(blue.tailTrimMax);

                if (result != null && result.length >= 3)

                        blue.tailTrimValueMax = result[2][1];

        }

        public void applyTrimToStretchs() {

                red.applyTrimToStretchs();

                green.applyTrimToStretchs();

                blue.applyTrimToStretchs();

        }

        public void calcLinearScaleAndOffset() {

                red.calcLinearScaleAndOffset();

                green.calcLinearScaleAndOffset();

                blue.calcLinearScaleAndOffset();

        }

        public void setMaxMin(Statistics stats, int[] renderBands) {

                if (renderBands[0] > -1)

                        red.setMaxMin(stats, renderBands[0], rgb);

                if (renderBands[1] > -1)

                        green.setMaxMin(stats, renderBands[1], rgb);

                if (renderBands[2] > -1)

                        blue.setMaxMin(stats, renderBands[2], rgb);

        }

        public void applyRemoveEndsToStretchs(Statistics stats, int[] renderBands) {

                if (renderBands[0] > -1)

                        red.applyRemoveEndsToStretchs(stats, rgb, renderBands[0]);

                if (renderBands[1] > -1)

                        green.applyRemoveEndsToStretchs(stats, rgb, renderBands[1]);

                if (renderBands[2] > -1)

                        blue.applyRemoveEndsToStretchs(stats, rgb, renderBands[2]);

        }

        /**

         * Obtiene un objeto LinearStretchParams para una aplicaci?n de realce lineal estandar, es decir, el rango

         * de datos de salida es 0-255 y con solo un tramo para los datos. 

         * @param nBands N?mero de bandas

         * @param tailTrim Recorte de colas

         * @param stats Estad?sticas

         * @return LinearStretchParams

         * @throws FileNotOpenException

         * @throws RasterDriverException

         */

        public static DefaultLinearStretchParams createStandardParam(int[] renderBands, double tailTrim, Statistics stats, boolean rgb) throws FileNotOpenException, RasterDriverException {

                DefaultLinearStretchParams leParams = new DefaultLinearStretchParams();

                leParams.rgb = rgb;

                try {

                        stats.calculate(RasterLibrary.statisticsScale);

                } catch (ProcessInterruptedException e) {

                        return null;

                }

                if (renderBands[0] >= 0) {

                        if (rgb)

                                leParams.red.stretchIn = new double[] { stats.getMinByteUnsigned()[0], stats.getMaxByteUnsigned()[0] };

                        else

                                leParams.red.stretchIn = new double[] { stats.getMin()[renderBands[0]], stats.getMax()[renderBands[0]] };

                }

                leParams.red.stretchOut = new int[] { 0, 255 };

                leParams.red.tailTrimMin = tailTrim;

                leParams.red.tailTrimMax = tailTrim;

                if (renderBands[1] >= 0) {

                        if (rgb)

                                leParams.green.stretchIn = new double[] { stats.getMinByteUnsigned()[0], stats.getMaxByteUnsigned()[0] };

                        else

                                leParams.green.stretchIn = new double[] { stats.getMin()[renderBands[1]], stats.getMax()[renderBands[1]] };

                }

                leParams.green.stretchOut = new int[] { 0, 255 };

                leParams.green.tailTrimMin = tailTrim;

                leParams.green.tailTrimMax = tailTrim;

                if (renderBands[2] >= 0) {

                        if (rgb)

                                leParams.blue.stretchIn = new double[] { stats.getMinByteUnsigned()[0], stats.getMaxByteUnsigned()[0] };

                        else

                                leParams.blue.stretchIn = new double[] { stats.getMin()[renderBands[2]], stats.getMax()[renderBands[2]] };

                }

                leParams.blue.stretchOut = new int[] { 0, 255 };

                leParams.blue.tailTrimMin = tailTrim;

                leParams.blue.tailTrimMax = tailTrim;

                return leParams;

        }

        public Stretch getRed() {

                return red;

        }

        public void setRed(Stretch red) {

                this.red = (DefaultStretch)red;

        }

        public Stretch getGreen() {

                return green;

        }

        public void setGreen(Stretch green) {

                this.green = (DefaultStretch)green;

        }

        public Stretch getBlue() {

                return blue;

        }

        public void setBlue(Stretch blue) {

                this.blue = (DefaultStretch)blue;

        }

        public boolean isRgb() {

                return rgb;

        }

        public void setRgb(boolean rgb) {

                this.rgb = rgb;

        }

}