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

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

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

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

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

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

import org.gvsig.raster.impl.DefaultRasterManager;

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

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

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

/**

 * Filtro para la aplicaci?n de HistogramMatching a un raster.

 * 

 * @author aMu?oz (alejandro.munoz@uclm.es)

 * @version 27-5-2008

 * */

public class HistogramMatchingFilter  extends BaseRasterFilter {

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

        public BufferHistogram           histogramReference  = null;

        private BufferHistogram          histogramSource     = null;

        private double [][]              acumulateS          = null;

        private double [][]              acumulateR          = null;

        public String                    fileNameOutput      = null;

        public byte                      tableAsign[][]      = null;

        public int                       numbands            = 0;

        /**

         * Constructor 

         **/

        public HistogramMatchingFilter() {

                        super();

        }

        public String getGroup() {

                return "HistogramMatch";

        }

        public int getInRasterDataType() {

                return 0;

        }

        public String[] getNames() {

                return names;                        

        }

        public int getOutRasterDataType() {

                return RasterBuffer.TYPE_BYTE;

        }

        public Params getUIParams(String nameFilter) {

                Params params = new ParamsImpl();

                return params;

        }

        public boolean isVisible() {

                return false;

        }

        /**

         * Acciones antes de la ejecuci?n del filtro

         * */

        public void pre() {        

                // Carga de los parametros del filtro y obtencion de histogramas

                try {

                        loadParam();

                } catch (HistogramException e) {

                        exec = false;

                } catch (ProcessInterruptedException e) {

                        exec = false;

                }

                // Funcion que calcula la correspondencia entre los histogramas, El resultado es tableAsign completa

                // para cada clase en el histograma fuente asigan su clase en el histograma corregido.

                double inValue=0;

                tableAsign= new byte[numbands][256];

                // C?lculo de la tabla de asignaciones entre los histogramas

                for (int band=0; band <numbands; band++){

                        for(int i= 0; i< tableAsign[0].length; i++){

                                inValue = acumulateS[band][i];

                                int value= (searchInHistogramReference(inValue,band));

                                tableAsign[band][i]= (byte)value;

                        }

                }

        }

        /**

         * Acciones posteriores a la ejecuci?n del filtro

         * */

        public void post() {                

        }

        /**

         *  Se recogen los parametros necesarios para la aplicacion del filtro. 

         *         histogramReference histograma de referencia.

         *        fileNameOutput 

         * @throws HistogramException 

         *  

         */

        protected void loadParam() throws HistogramException, ProcessInterruptedException {        

                // Raster

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

                height = raster.getHeight();

                width = raster.getWidth();

                histogramSource = raster.getHistogramComputer().getBufferHistogram();

                histogramSource = BufferHistogramImpl.convertHistogramToRGB(histogramSource);

                // COMPROBACION HISTOGRAMA RGB

                if (!histogramSource.isInRangeRGB())

                        return;

                // Histograma de referencia

                histogramReference = (BufferHistogram)params.get("histogramReference");

                histogramReference = BufferHistogramImpl.convertHistogramToRGB(histogramReference);

                //numbands = (int)params.get("numbads");

                numbands = ((Integer)params.get("numbands")).intValue();

                // Histogramas acumilados y normalizados 

                acumulateS = BufferHistogramImpl.convertTableToNormalizeAccumulate(histogramSource.getTable());

                acumulateR = BufferHistogramImpl.convertTableToNormalizeAccumulate(histogramReference.getTable());

                rasterResult = DefaultRasterManager.getInstance().createBuffer(RasterBuffer.TYPE_BYTE,raster.getWidth(),raster.getHeight(),numbands,true);

                fileNameOutput = (String)params.get("fileNameOutput");

        }

        /**

         * M?todo que realiza la correspondencia entre las clases de los histogramas

         **/

        public int searchInHistogramReference(double value,int band) {

                int i = 0;

                while(value > acumulateR[band][i]) {

                        i++;

                        if(i == 255) 

                                return (int)255;

                }

                if(i == 0)

                        return 0;

                if((acumulateR[band][i]-value) < (acumulateR[band][i-1]-value))

                        return (int)i;

                else 

                        return (int)(i-1);

        }

        public void process(int x, int y) {

        }

}