Application: gvSIG desktop

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

         *

         * Copyright (C) 2006 Instituto de Desarrollo Regional 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.

         *

         * For more information, contact:

         *

         *  Generalitat Valenciana

         *   Conselleria d'Infraestructures i Transport

         *   Av. Blasco Iba?ez, 50

         *   46010 VALENCIA

         *   SPAIN

         *

         *      +34 963862235

         *   gvsig@gva.es

         *      www.gvsig.gva.es

         *

         *    or

         *

         *   Instituto de Desarrollo Regional (Universidad de Castilla La-Mancha)

         *   Campus Universitario s/n

         *   02071 Alabacete

         *   Spain

         *

         *   +34 967 599 200

         */

package org.gvsig.remotesensing.mosaic.process;

import java.io.File;

import java.io.IOException;

import org.gvsig.fmap.raster.layers.FLyrRasterSE;

import org.gvsig.raster.RasterProcess;

import org.gvsig.raster.buffer.BufferFactory;

import org.gvsig.raster.buffer.RasterBuffer;

import org.gvsig.raster.buffer.RasterBufferInvalidException;

import org.gvsig.raster.buffer.WriterBufferServer;

import org.gvsig.raster.dataset.GeoRasterWriter;

import org.gvsig.raster.dataset.IRasterDataSource;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.io.RasterDriverException;

import org.gvsig.raster.datastruct.Histogram;

import org.gvsig.raster.datastruct.HistogramException;

import org.gvsig.raster.grid.Grid;

import org.gvsig.raster.grid.GridException;

import org.gvsig.raster.grid.OutOfGridException;

import org.gvsig.raster.util.RasterToolsUtil;

import com.iver.andami.PluginServices;

import com.iver.cit.gvsig.exceptions.layers.LoadLayerException;

import com.iver.cit.gvsig.fmap.layers.FLayer;

/** 

 * Clase que implementa el proceso de Histogram Matching para una imagen.

 * Para el proceso son necesarios dos raster, el raster fuente, que se pretende hacer corresponder

 * con el histograma del raster de referencia y el raster maestro o de referencia.

 * 

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

 * @version 30/4/2008

 * */

public class HistogramMatchingProcess extends RasterProcess {

        private FLyrRasterSE rasterSource                         = null;

        private FLyrRasterSE rasterReference                = null;

        private Grid sourceGrid                                                = null;

        private Grid gridResult                                                = null;

        // PROVISIONAL -para cargar capa en vista-

        //protected MapContext mapContext                         = null;

        private Histogram histogramSource                        = null;

        private Histogram histogramReference                = null;

        private double [][] acumulateR                                = null;

        private double [][] acumulateS                                 = null;

//        Indicador de progreso

        private int                     percent            = 0;

        protected WriterBufferServer         writerBufferServer        = null;

        /**

         *  Se inicializan los par?metros. 

         *  

         *  rasterSource raster al que se aplica el proceso de Histogram Matching

         *  rasterReference raster que se toma como maestro para hacer la correspondencia

         * */

        public void init() {

                try {

                        rasterSource = (FLyrRasterSE) ((FLyrRasterSE)getParam("rasterSource")).cloneLayer();

                        rasterReference = (FLyrRasterSE) ((FLyrRasterSE)getParam("rasterReference")).cloneLayer();

                } catch (Exception e1) {

                        e1.printStackTrace();

                }

                // PROVISIONAL

        //        mapContext = (MapContext)getParam("contexto");

                try {

                        histogramSource= rasterSource.getBufferFactory().getDataSource().getHistogram();

                        histogramReference= rasterReference.getBufferFactory().getDataSource().getHistogram();

                        IRasterDataSource dsetCopy = null; 

                        dsetCopy = rasterSource.getDataSource().newDataset();

                        BufferFactory bufferFactory = new BufferFactory(dsetCopy);

                        // Inicializacion del grid de datos y el grid resultante

                        if (!RasterBuffer.loadInMemory(dsetCopy))

                                bufferFactory.setReadOnly(true);        

                        sourceGrid = new Grid(bufferFactory,rasterSource.getRenderBands());                

                        gridResult = new Grid(sourceGrid.getGridExtent(), sourceGrid.getGridExtent(),sourceGrid.getDataType(), new int[]{0,1,2});

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

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

                }catch (RasterBufferInvalidException e) {

                        RasterToolsUtil.messageBoxError("buffer_incorrecto", this, e);

                } catch (HistogramException e) {

                        return;

                } catch (InterruptedException e) {

                        Thread.currentThread().interrupt();        

                }

        }

        /** 

         * Proceso de correspondecia de Histogramas 

         * */

        public void process() throws InterruptedException {

                byte tableAsign[][]= new byte[3][256];

                int progress=0;

                int index=0;

                double inValue=0;

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

                for (int band=0; band < rasterSource.getRenderBands().length; band++){

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

                                inValue = acumulateS[band][i];

                                byte value=(byte) (searchInHistogramReference(inValue,band)-128);

                                tableAsign[band][i]= value;

                        }

                }

                try {

                // Construccion de la nueva imagen en base a la tabla estimada

                        for (int band=0; band < rasterSource.getRenderBands().length; band++){

                                gridResult.setBandToOperate(band);

                                sourceGrid.setBandToOperate(band);

                                for(int i=0; i< sourceGrid.getNY(); i++){

                                        progress++;

                                        for(int j=0;j<sourceGrid.getNX();j++){                                                                                

                                                byte data= (byte) ((sourceGrid.getCellValueAsByte(j,i)));

                                                index=data+128;

                                                gridResult.setCellValue(j,i,(byte)((tableAsign[band][index])));

                                                percent=(int)( progress*100/(gridResult.getLayerNY()*rasterSource.getRenderBands().length));        

                                        }        

                                }        

                        }

                } catch (GridException e) {

                        RasterToolsUtil.messageBoxError(PluginServices.getText(this, "grid_error"), this, e);

                } catch (OutOfGridException e) {

                        RasterToolsUtil.messageBoxError(PluginServices.getText(this, "bad_access_grid"), this, e);

                }        

                // PROVISIONAL

                writeToFile();

        }

        /**

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

        }

        /**

         *  Escritura en fichero y carga de la capa en la vista.

         *  Metodo Provisional.

         * 

         * */

        public void writeToFile(){

                try{

                        String filename= "histogramMatching2.img";

                        GeoRasterWriter grw = null;

                        writerBufferServer = new WriterBufferServer(gridResult.getRasterBuf());

                        grw = GeoRasterWriter.getWriter(writerBufferServer, filename, gridResult.getBandCount(),rasterSource.getAffineTransform(), gridResult.getRasterBuf().getWidth(), gridResult.getRasterBuf().getHeight(), gridResult.getRasterBuf().getDataType(), GeoRasterWriter.getWriter(filename).getParams(), null);

                        grw.dataWrite();

                        grw.setWkt(rasterSource.getWktProjection());

                        grw.writeClose();

                        //mapContext.beginAtomicEvent();

                        FLayer lyr = null;

                        int endIndex = filename.lastIndexOf(".");

                        if (endIndex < 0)

                                endIndex = filename.length();

                        lyr = FLyrRasterSE.createLayer(

                                        filename.substring(filename.lastIndexOf(File.separator) + 1, endIndex),

                                        filename,

                                        rasterSource.getProjection()

                                        );

                /*        mapContext.getLayers().addLayer(lyr);

                        //mapContext.endAtomicEvent();

                        mapContext.invalidate();*/

                } catch (NotSupportedExtensionException e) {

                        RasterToolsUtil.messageBoxError(PluginServices.getText(this, "error_writer_notsupportedextension"), this, e);

                } catch (RasterDriverException e) {

                        RasterToolsUtil.messageBoxError(PluginServices.getText(this, "error_writer"), this, e);        

                } catch (IOException e) {

                        RasterToolsUtil.messageBoxError(PluginServices.getText(this, "error_writer"), this, e);

                }catch (LoadLayerException e) {

                        RasterToolsUtil.messageBoxError("error_cargar_capa", this, e);

                }catch (InterruptedException e) {

                        Thread.currentThread().interrupt();

                }         

        }

        /**

         * @return title

         * */

        public String getTitle() {

                return PluginServices.getText(this,"histogram_matching");

        }

        /**

         * @return indicador de progreso 

         * */

        public int getPercent() {

                if(writerBufferServer==null)

                        return percent;

                else

                        return writerBufferServer.getPercent();

        }

}