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

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.Iterator;

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

import org.gvsig.gui.beans.incrementabletask.IIncrementable;

import org.gvsig.gui.beans.incrementabletask.IncrementableEvent;

import org.gvsig.gui.beans.incrementabletask.IncrementableListener;

import org.gvsig.gui.beans.incrementabletask.IncrementableTask;

import org.gvsig.raster.buffer.BufferFactory;

import org.gvsig.raster.buffer.RasterBuffer;

import org.gvsig.raster.buffer.RasterBufferInvalidException;

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

import org.gvsig.raster.datastruct.ColorTable;

import org.gvsig.raster.grid.Grid;

import org.gvsig.raster.grid.GridPalette;

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

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

import org.gvsig.raster.grid.filter.bands.ColorTableFilter;

import org.gvsig.raster.grid.filter.bands.ColorTableListManager;

import org.gvsig.raster.grid.filter.enhancement.LinearEnhancementFilter;

import org.gvsig.raster.grid.filter.statistics.TailTrimFilter;

import org.gvsig.raster.grid.roi.ROI;

import org.gvsig.raster.hierarchy.IRasterRendering;

import org.gvsig.raster.process.CancelEvent;

import org.gvsig.raster.process.RasterTask;

import org.gvsig.raster.util.RasterToolsUtil;

import org.gvsig.rastertools.clipping.WriterBufferServer;

import com.iver.andami.PluginServices;

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

import com.iver.cit.gvsig.fmap.MapContext;

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

import com.iver.cit.gvsig.project.documents.view.gui.View;

/**

 * ClassificationGeneraProccess es la clase abstracta base que extienden todos

 * los metodos de clasificaci?n. Recoge procedimientos comunes a todos los metodos:

 * construccion del grid que se va a clasificar, asignacion de la leyenda a la capa

 * resultante o escritura en fichero.

 * Cada procedimiento de clasificaci?n implementar? los m?todos que determinan la

 * clase a la que pertenece un pixel, adem?s del metodo run().

 *@author Alejandro Mu?oz Sanchez (alejandro.munoz@uclm.es)

 *@author Diego Guerrero Sevilla (diego.guerrero@uclm.es)

 *@version 19/10/2007

 * */

public abstract class ClassificationGeneralProcess implements Runnable, IIncrementable, IncrementableListener {

        protected Grid                                         inputGrid                        = null;

        protected IRasterDataSource                raster                                = null;

        protected RasterBuffer                         rasterResult                = null;

        protected MapContext                         mapContext                         = null;

        protected int                                         percent                           = 0;

        protected Thread                                 blinker                                = null;

        protected ArrayList                                rois                                = null;

        protected IncrementableTask                incrementableTask         = null;

        protected WriterBufferServer         writerBufferServer        = null;

        protected String                                 fileNameOutput                 = null;

        protected boolean                                selectedBands[]                = null;

        protected        int                                        numClases                        = 0;

        protected View                                         view                                = null;

        protected RasterTask                        rasterTask                         = new RasterTask(this);

        protected   int[]                                 bandList                        = null;

        /**

        * M?todo que determina la clase de pertenencia del pixel cuyos valores

        * por banda se pasan en un array de bytes.

        * @param  valores del pixel en cada una de las bandas

        * @return clase a la que pertenece el pixel

        */

        public abstract int getPixelClassForTypeByte(byte pixelBandsValues[]);

        /**

        * M?todo que determina la clase de pertenencia del pixel cuyos valores

        * por banda se pasan en un array de shorts.

        * @param  valores del pixel en cada una de las bandas

        * @return clase a la que pertenece el pixel

        */

        public abstract int getPixelClassForTypeShort(short pixelBandsValues[]);

        /**

        * M?todo que determina la clase de pertenencia del pixel cuyos valores

        * por banda se pasan en un array de int.

        * @param  valores del pixel en cada una de las bandas

        * @return clase a la que pertenece el pixel

        */

        public abstract int getPixelClassForTypeInt(int pixelBandsValues[]);

        /**

        * M?todo que determina la clase de pertenencia del pixel cuyos valores

        * por banda se pasan en un array de float.

        * @param  valores del pixel en cada una de las bandas

        * @return clase a la que pertenece el pixel

        */

        public abstract int getPixelClassForTypeFloat(float pixelBandsValues[]);

        /**

        * M?todo que determina la clase de pertenencia del pixel cuyos valores

        * por banda se pasan en un array de double.

        * @param  valores del pixel en cada una de las bandas

        * @return clase a la que pertenece el pixel

        */

        public abstract int getPixelClassForTypeDouble(double pixelBandsValues[]);

        /**

         * Lanzar el Thread del proceso de clasificaci?n.

         */

        public void start() {}

        /**

        * Asigna la leyenda a la capa resultante

        * @param lyr capa a la que se asigna la leyenda

        */

        public  void setLeyend(FLayer lyr) {

                if(lyr instanceof IRasterRendering) {

                        IRasterRendering rendering = (IRasterRendering) lyr;

                        ArrayList colorItems = new ArrayList();

                        ColorItem colorItem = null;

                        int classValue = 0;

                        for (Iterator iter = rois.iterator(); iter.hasNext();) {

                                ROI roi = (ROI) iter.next();

                                colorItem = new ColorItem();

                                colorItem.setColor(roi.getColor());

                                colorItem.setNameClass(roi.getName());

                                colorItem.setValue(classValue);

                                colorItems.add(colorItem);

                                classValue++;

                        }

                        ColorTable colorTable = new ColorTable();

                        colorTable.createPaletteFromColorItems(colorItems, false);

                        RasterFilterList filterList = rendering.getRenderFilterList();

                        RasterFilterListManager manager = new RasterFilterListManager(filterList);

                        ColorTableListManager cManager = (ColorTableListManager) manager.getManagerByClass(ColorTableListManager.class);

                        filterList.remove(ColorTableFilter.class);

                        ((FLyrRasterSE)lyr).setLastLegend(null);

                        filterList.remove(LinearEnhancementFilter.class);

                        filterList.remove(TailTrimFilter.class);

                        GridPalette gridPalette = new GridPalette(colorTable);

                        cManager.addColorTableFilter(gridPalette);

                        ((FLyrRasterSE)lyr).setLastLegend(gridPalette);

                        rendering.setRenderFilterList(filterList);

                }

        }

        /**

        * Establece el grid con las bandas recogidas en bandList.

        */

        public  void setGrid(){

                BufferFactory bufferFactory = new BufferFactory(raster);

                try {

                        bufferFactory.setAreaOfInterest();

                        bufferFactory.setAllDrawableBands();

                        inputGrid = new Grid(bufferFactory,bandList);

                } catch (RasterDriverException e) {

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

                }catch (RasterBufferInvalidException e) {

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

                }catch (InterruptedException e) {

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

                }

        }

        /**

        * Escritura del resultado a disco y carga de la capa en la vista.

        */

        public void writeToFile(){

                try{

                        GeoRasterWriter grw = null;

                        writerBufferServer = new WriterBufferServer(rasterResult);

                        grw = GeoRasterWriter.getWriter(writerBufferServer, fileNameOutput, rasterResult.getBandCount(),raster.getAffineTransform(), rasterResult.getWidth(), rasterResult.getHeight(), rasterResult.getDataType(), GeoRasterWriter.getWriter(fileNameOutput).getParams(), null);

                        grw.dataWrite();

                        grw.setWkt(raster.getWktProjection());

                        grw.writeClose();

                        rasterResult.free();

                        mapContext= view.getModel().getMapContext();

                        mapContext.beginAtomicEvent();

                        FLayer lyr = null;

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

                        if (endIndex < 0)

                                endIndex = fileNameOutput.length();

                        lyr = FLyrRasterSE.createLayer(fileNameOutput.substring(fileNameOutput.lastIndexOf(File.separator) + 1, endIndex),new File(fileNameOutput), null);

                        setLeyend(lyr);

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

                }finally{

                        if (incrementableTask != null)

                                incrementableTask.processFinalize();

                }

        }

        /**

        * @return buffer monobanda con el resultado de la clasificaci?n

        */

        public  RasterBuffer getResult(){

                return rasterResult;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getLabel()

         */

        public String getLabel() {

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

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getLog()

         */

        public String getLog() {

                if (writerBufferServer==null)

                        return PluginServices.getText(this,"obteniendo_imagen")+"...";

                else

                        return PluginServices.getText(this,"escribiendo_resultado")+"...";

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getPercent()

         */

        public int getPercent() {

                if (writerBufferServer==null)

                        return percent;

                else

                        return writerBufferServer.getPercent();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getTitle()

         */

        public String getTitle() {

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

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#actionResumed()

         */

        public void actionResumed(IncrementableEvent e) {

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#actionSuspend()

         */

        public void actionSuspended(IncrementableEvent e) {

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#setIncrementableTask()

         */

        public void setIncrementableTask(IncrementableTask incrementableTask) {

                this.incrementableTask = incrementableTask;

                this.incrementableTask.addIncrementableListener(this);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#actionCanceled()

         */

        public void actionCanceled(IncrementableEvent e) {

                rasterTask.setEvent(new CancelEvent(this));

                if (incrementableTask != null)

                        incrementableTask.processFinalize();

        }

}