| gvSIG desktop 1

/* 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.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.datastruct.serializer.NoDataRmfSerializer;

import org.gvsig.raster.grid.Grid;

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

        /*

         * (non-Javadoc)

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

         */

        public void init() {

                filename = getStringParam("filename");

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

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

        }

        /*

         * (non-Javadoc)

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

         */

        public void process() throws InterruptedException {

                WriterBufferServer writerBufferServer = null;

                IRasterDataSource dsetCopy = null;

                try {

                        insertLineLog(PluginServices.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);

                        rasterFilterList = grid.getFilterList();

                        addSelectedFilters(rasterFilterList, listFilterUsed);

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

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

                        grid.applyFilters();

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

                        buffer = grid.getRasterBuf();

                        writerBufferServer.setBuffer(buffer, -1);

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

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

                        geoRasterWriter.dataWrite();

                        geoRasterWriter.writeClose();

                        // Guardamos en el RMF del fichero el valor NoData

                        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 {

                                RasterUtilities.copyFile(manager.getPath(), manager.getPath() + "~");

                                manager.write(true);

                        } catch (FileNotFoundException e) {

                        } catch (IOException e) {

                        }

                        manager.removeClient(ser.getClass());

                        externalActions.end(filename);

                } 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 (FilterTypeException e) {

                        RasterToolsUtil.messageBoxError(PluginServices.getText(this, "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");

        }

}