Application: gvSIG desktop

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

*

* Copyright (C) 2005 IVER T.I. 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.

*/

package org.gvsig.rastertools.filter;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.util.ArrayList;

import org.gvsig.raster.Configuration;

import org.gvsig.raster.buffer.BufferFactory;

import org.gvsig.raster.buffer.RasterBuffer;

import org.gvsig.raster.dataset.GeoRasterWriter;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.dataset.IRasterDataSource;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

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

import org.gvsig.raster.dataset.io.rmf.RmfBlocksManager;

import org.gvsig.raster.dataset.properties.DatasetColorInterpretation;

import org.gvsig.raster.datastruct.serializer.NoDataRmfSerializer;

import org.gvsig.raster.grid.Grid;

import org.gvsig.raster.grid.GridTransparency;

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

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

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

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

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

import org.gvsig.raster.hierarchy.IRasterRendering;

import org.gvsig.raster.util.RasterToolsUtil;

import org.gvsig.raster.util.RasterUtilities;

import org.gvsig.rastertools.RasterProcess;

import org.gvsig.rastertools.clipping.WriterBufferServer;

import org.gvsig.rastertools.filter.FilterListener.ParamStruct;

import com.iver.andami.PluginServices;

/**

 * Clase donde se hara todo el proceso de aplicar un filtro a una capa. Muestra

 * una ventana de dialogo de incremento informativa.

 *

 * @version 24/05/2007

 * @author BorSanZa - Borja S?nchez Zamorano (borja.sanchez@iver.es)

 */

public class FilterProcess extends RasterProcess {

        private String            filename         = "";

        private IRasterDataSource rasterDataSource = null;

        private ArrayList         listFilterUsed   = null;

        private RasterFilterList  rasterFilterList = null;

        private GeoRasterWriter   geoRasterWriter  = null;

        private IRasterRendering  rendering        = null;

        /*

         * (non-Javadoc)

         * @see org.gvsig.rastertools.RasterProcess#init()

         */

        public void init() {

                rendering = (IRasterRendering) getParam("rendering");

                filename = getStringParam("filename");

                rasterDataSource = (IRasterDataSource) getParam("rasterdatasource");

                listFilterUsed = (ArrayList) getParam("listfilterused");

        }

        /**

         * 

         * @param geoRasterWriter

         */

        private DatasetColorInterpretation getColorIntepretation(GeoRasterWriter geoRasterWriter, IBuffer buffer, Grid grid) {

                DatasetColorInterpretation colorInterpretation = null;

                do {

                        // Si tiene una tabla de color asignamos las tres bandas

                        if (grid.getFilterList().isActive("colortable")) {

                                colorInterpretation = new DatasetColorInterpretation(new String[] { DatasetColorInterpretation.RED_BAND, DatasetColorInterpretation.GREEN_BAND, DatasetColorInterpretation.BLUE_BAND });

                                break;

                        }

                        // Si el numero de bandas coincide asignamos la misma interpretacion que tenia antes

                        if (buffer.getBandCount() == rasterDataSource.getBandCount()) {

                                colorInterpretation = rendering.getRender().getColorInterpretation();

                                break;

                        }

                        String[] colorInterp = new String[rasterDataSource.getColorInterpretation(0).getValues().length];

                        for (int i = 0; i<rasterDataSource.getColorInterpretation(0).getValues().length; i++) {

                                if (rasterDataSource.getColorInterpretation(0).getValues()[i].equals(DatasetColorInterpretation.UNDEF_BAND)) {

                                        colorInterp[i] = DatasetColorInterpretation.GRAY_BAND;

                                        continue;

                                }

                                colorInterp[i] = rasterDataSource.getColorInterpretation(0).getValues()[i];

                        }

                        colorInterpretation = new DatasetColorInterpretation(colorInterp);

                } while (false);

                return colorInterpretation;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.rastertools.RasterProcess#process()

         */

        public void process() throws InterruptedException {

                WriterBufferServer writerBufferServer = null;

                IRasterDataSource dsetCopy = null;

                try {

                        insertLineLog(RasterToolsUtil.getText(this, "leyendo_raster"));

                        dsetCopy = rasterDataSource.newDataset();

                        BufferFactory bufferFactory = new BufferFactory(dsetCopy);

                        if (!RasterBuffer.loadInMemory(dsetCopy))

                                bufferFactory.setReadOnly(true);

                        bufferFactory.setAllDrawableBands();

                        IBuffer buffer = null;

                        writerBufferServer = new WriterBufferServer();

                        bufferFactory.setAreaOfInterest();

                        Grid grid = new Grid(bufferFactory, true);

                        // Obtengo la lista de filtros

                        rasterFilterList = grid.getFilterList();

                        // Le aplico los filtros usados en el panel

                        addSelectedFilters(rasterFilterList, listFilterUsed);

                        // Quito el filtro NoData porque vamos a guardar su valor en un RMF

                        rasterFilterList.remove(NoDataFilter.class);

                        insertLineLog(RasterToolsUtil.getText(this, "aplicando_filtros"));

                        grid.setNoDataValue(rasterDataSource.getNoDataValue(0));

                        grid.applyFilters();

                        GridTransparency transparency = (GridTransparency) grid.getFilterList().getEnv().get("Transparency");

                        insertLineLog(RasterToolsUtil.getText(this, "guardando_capa"));

                        buffer = grid.getRasterBuf();

                        writerBufferServer.setBuffer(buffer, -1);

                        // TODO: FUNCIONALIDAD: Poner los getWriter con la proyecci?n del fichero fuente

                        if (transparency.isTransparencyActive()) {

                                // Asigno el buffer alpha

                                writerBufferServer.setAlphaBuffer(transparency.getAlphaBand());

                                // Creo un GeoRasterWriter con una banda mas para el buffer alpha

                                geoRasterWriter = GeoRasterWriter.getWriter(writerBufferServer, filename, buffer.getBandCount() + 1, rasterDataSource.getAffineTransform(), buffer.getWidth(), buffer.getHeight(), buffer.getDataType(), GeoRasterWriter.getWriter(filename).getParams(), null);

                        } else {

                                // Creo un GeoRasterWriter

                                geoRasterWriter = GeoRasterWriter.getWriter(writerBufferServer, filename, buffer.getBandCount(), rasterDataSource.getAffineTransform(), buffer.getWidth(), buffer.getHeight(), buffer.getDataType(), GeoRasterWriter.getWriter(filename).getParams(), null);

                        }

                        DatasetColorInterpretation colorInterpretation = getColorIntepretation(geoRasterWriter, buffer, grid);

                        if (transparency.isTransparencyActive()) {

                                DatasetColorInterpretation alphaI;

                                alphaI = new DatasetColorInterpretation(new String[] { DatasetColorInterpretation.ALPHA_BAND });

                                colorInterpretation.addColorInterpretation(alphaI);

                        }

                        geoRasterWriter.setColorBandsInterpretation(colorInterpretation.getValues());

                        geoRasterWriter.dataWrite();

                        geoRasterWriter.writeClose();

                        // Guardamos en el RMF del fichero el valor NoData

                        if (Configuration.getValue("nodata_transparency_enabled", Boolean.FALSE).booleanValue()) {

                                String filenameRMF = RasterUtilities.getNameWithoutExtension(filename) + ".rmf";

                                RmfBlocksManager manager = dsetCopy.getDataset(0)[0].getRmfBlocksManager();

                                NoDataRmfSerializer ser;

                                ser = new NoDataRmfSerializer(Double.valueOf(rasterDataSource.getNoDataValue(0)));

                                manager.setPath(filenameRMF);

                                if (!manager.checkRmf())

                                        return;

                                manager.addClient(ser);

                                try {

                                        manager.write(true);

                                } catch (FileNotFoundException e) {

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

                                } catch (IOException e) {

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

                                }

                                manager.removeClient(ser.getClass());

                        }

                        externalActions.end(filename);

                } catch (NotSupportedExtensionException e) {

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

                } catch (RasterDriverException e) {

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

                } catch (IOException e) {

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

                } catch (FilterTypeException e) {

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

                }

        }

        /**

         * Sustituye la lista de filtros de filterList por la que le pasamos en forma

         * de ArrayList

         * @param filterList

         * @param listFilterUsed

         * @throws FilterTypeException 

         */

        public static void addSelectedFilters(RasterFilterList filterList, ArrayList listFilterUsed) throws FilterTypeException {

                filterList.clear();

                RasterFilterListManager stackManager = new RasterFilterListManager(filterList);

                for (int i = 0; i < listFilterUsed.size(); i++) {

                        ParamStruct aux = (ParamStruct) listFilterUsed.get(i);

                        IRasterFilterListManager filterManager = stackManager.getManagerByFilterClass(aux.getFilterClass());

                        filterManager.addFilter(aux.getFilterClass(), aux.getFilterParam());

                }

                filterList.resetPercent();

        }

        /*

         * (non-Javadoc)

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

         */

        public int getPercent() {

                if (rasterFilterList == null)

                        return 0;

                if (rasterFilterList.getPercent() < 100)

                        return rasterFilterList.getPercent();

                if (geoRasterWriter == null)

                        return 0;

                return geoRasterWriter.getPercent();

        }

        /*

         * (non-Javadoc)

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

         */

        public String getTitle() {

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

        }

}