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.tools.app.basic.raster.process;

import java.io.IOException;

import java.util.ArrayList;

import javax.swing.SwingUtilities;

import org.gvsig.andami.PluginServices;

import org.gvsig.fmap.dal.coverage.RasterLocator;

import org.gvsig.fmap.dal.coverage.RasterManager;

import org.gvsig.fmap.dal.coverage.dataset.Buffer;

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.coverage.grid.Grid;

import org.gvsig.fmap.dal.coverage.grid.RasterFilterList;

import org.gvsig.fmap.dal.coverage.grid.RasterFilterListManager;

import org.gvsig.fmap.dal.coverage.store.DataServerWriter;

import org.gvsig.fmap.dal.coverage.store.RasterDataStore;

import org.gvsig.fmap.dal.coverage.store.RasterQuery;

import org.gvsig.fmap.dal.coverage.store.RasterWriter;

import org.gvsig.fmap.dal.coverage.store.props.ColorInterpretation;

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

import org.gvsig.raster.tools.app.basic.RasterToolsUtil;

import org.gvsig.raster.tools.app.basic.raster.bean.previewbase.ParamStruct;

/**

 * Clase donde se hara todo el proceso de aplicar una lista de filtros 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 RasterDataStore          rasterDataSource = null;

        private ArrayList<ParamStruct>   listFilterUsed   = null;

        private RasterFilterList         rasterFilterList = null;

        private RasterWriter             geoRasterWriter  = null;

        private FLyrRaster               lyr              = null;

        private RasterManager            rManager         = RasterLocator.getManager();

        /*

         * (non-Javadoc)

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

         */

        @SuppressWarnings("unchecked")

        public void init() {

                filename = getStringParam("filename");

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

                lyr = (FLyrRaster)getParam("layer");

                rasterDataSource = lyr.getDataStore();

        }

        /**

         * Devuelve una interpretacion de color. Es usado cuando no conseguimos un

         * rendering y no podemos saber que interpretacion l?gica ten?a antes.

         * @param geoRasterWriter

         */

        private ColorInterpretation getColorIntepretation(Buffer buffer, Grid grid) {

                ColorInterpretation colorInterpretation = null;

                do {

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

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

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

                                break;

                        }

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

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

                                colorInterpretation = rasterDataSource.getColorInterpretation();

                                break;

                        }

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

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

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

                                        colorInterp[i] = ColorInterpretation.GRAY_BAND;

                                        continue;

                                }

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

                        }

                        colorInterpretation = rManager.createColorInterpretation(colorInterp);

                } while (false);

                return colorInterpretation;

        }

        /*

         * (non-Javadoc)

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

         */

        public void process() throws ProcessInterruptedException {

                DataServerWriter writerBufferServer = null;

                if(lyr != null)

                        lyr.setReadingData(Thread.currentThread().toString());

                RasterDataStore dstoreCopy = null;

                RasterQuery query = null;

                Buffer buffer = null;

                try {

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

                        //Creaci?n del BufferFactory

                        dstoreCopy = rasterDataSource.newDataStore();

                        query = rManager.createQuery();

                        //if (!RasterBuffer.loadInMemory(dsetCopy))

                        query.setReadOnly(true);

                        //Asignaci?n de bandas

                        int[] renderBands = lyr.getRender().getRenderBands();

                        if (renderBands != null) {

                                // Si es gris, se reduce a una sola banda

                                if ((renderBands.length == 3) && (renderBands[0] == renderBands[1]) && (renderBands[1] == renderBands[2])) 

                                        renderBands = new int[] { renderBands[0] };

                                query.setDrawableBands(renderBands);

                        } else

                                query.setAllDrawableBands();

                        query.setAreaOfInterest();

                        Buffer buff = dstoreCopy.query(query);

                        if(buff.isReadOnlyBuffer()) 

                                buff.addDrawableBands(renderBands);

                        Grid grid = rManager.createGrid(buff, dstoreCopy, true);

                        //Obtenemos la lista de filtros

                        rasterFilterList = grid.getFilterList();

                        //Aplicamos los filtros usados en el panel

                        addSelectedFilters(rasterFilterList, listFilterUsed);

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

                        grid.setNoDataValue(rasterDataSource.getNoDataValue());

                        grid.applyFilters();

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

                        buffer = grid.getRasterBuf();

                        writerBufferServer = rManager.createDataServerWriter();

                        writerBufferServer.setBuffer(buffer, -1);

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

                        //Calculo de la interpretaci?n de color

                        ColorInterpretation colorInterpretation = null;

                        if(lyr.existColorTable() || buffer.getBandCount() > 1)

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

                        else if(buffer.getBandCount() == 2) {

                                renderBands = lyr.getRender().getRenderBands();

                                String[] ci = new String[renderBands.length];

                                for (int i = 0; i < renderBands.length; i++) {

                                        switch (renderBands[i]) {

                                        case 0:ci[i] = ColorInterpretation.RED_BAND; break;

                                        case 1:ci[i] = ColorInterpretation.GREEN_BAND; break;

                                        case 2:ci[i] = ColorInterpretation.BLUE_BAND; break;

                                        default: ci[i] = ColorInterpretation.UNDEF_BAND; 

                                        }

                                }

                                colorInterpretation = rManager.createColorInterpretation(ci);

                        } else

                                colorInterpretation = rManager.createColorInterpretation(new String[] { ColorInterpretation.GRAY_BAND });

                        //Si la imagen original ten?a una banda de transparencia se asignar? esta. Si los filtros generados

                        //crean una banda se mezclar? con la original. 

                        int nbands = buffer.getBandCount();

                        if( rasterDataSource.getTransparency() != null &&

                                        rasterDataSource.getTransparency().getAlphaBandNumber() >= 0) {

                                query.setDrawableBands(new int[]{rasterDataSource.getTransparency().getAlphaBandNumber()});

                                query.setAreaOfInterest();

                                Buffer alpha = dstoreCopy.query(query);

                                if(grid.getFilterList().getAlphaBand() != null)

                                        alpha = rManager.getColorConversion().mergeTransparencyBuffers(alpha, grid.getFilterList().getAlphaBand());

                                writerBufferServer.setAlphaBuffer(alpha);

                                //Asignamos la interpretaci?n de color de la banda alpha                                

                                ColorInterpretation alphaI = rManager.createColorInterpretation(new String[] { ColorInterpretation.ALPHA_BAND });

                                colorInterpretation.addColorInterpretation(alphaI);

                                nbands ++;

                        } else if(grid.getFilterList().getAlphaBand() != null) {

                                writerBufferServer.setAlphaBuffer(grid.getFilterList().getAlphaBand());

                                //Asignamos la interpretaci?n de color de la banda alpha

                                ColorInterpretation alphaI = rManager.createColorInterpretation(new String[] { ColorInterpretation.ALPHA_BAND });

                                colorInterpretation.addColorInterpretation(alphaI);

                                nbands ++;

                        }

                        geoRasterWriter = rManager.createWriter(writerBufferServer, 

                                        filename,

                                        nbands, 

                                        rasterDataSource.getAffineTransform(), 

                                        buffer.getWidth(), 

                                        buffer.getHeight(), 

                                        buffer.getDataType(), 

                                        rManager.createWriter(filename).getParams(), 

                                        null);

                        if (lyr.getRender() == null)

                                colorInterpretation = getColorIntepretation(buffer, grid);

                        geoRasterWriter.setColorBandsInterpretation(colorInterpretation.getValues());

                        geoRasterWriter.dataWrite();

                        geoRasterWriter.writeClose();

                        geoRasterWriter = null;

                        // Guardamos en el RMF el valor NoData

                        //Si el del buffer tiene valor es porque alg?n filtro le ha salvado valor nodata por lo que se pone ese

                        //sino se pone el de la imagen original

                        NoData nodataOrigin = (NoData)rasterDataSource.getNoDataValue();

                        NoData nodataBuffer =  (NoData)buffer.getNoDataValue();

                        if(nodataBuffer != null && nodataBuffer.isDefined()) {

                                nodataBuffer.setFileName(filename);

                                nodataBuffer.save();

                        } else if(nodataOrigin != null && nodataOrigin.isDefined()) {

                                nodataOrigin.setFileName(filename);

                                nodataOrigin.save();

                        }

                        SwingUtilities.invokeLater(new Runnable() {

                                public void run() {

                                        if (externalActions != null)

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

                } catch (InvalidSetViewException e) {

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

                } catch (FilterManagerException e) {

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

                } finally {

                        rasterDataSource = null;

                        if(buffer != null)

                                buffer.free();

                        if(lyr != null)

                                lyr.setReadingData(null);

                }

        }

        /**

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

         * de ArrayList

         * @param filterList

         * @param listFilterUsed

         * @throws FilterTypeException 

         * @throws FilterTypeException 

         */

        public void addSelectedFilters(RasterFilterList filterList, ArrayList<ParamStruct> listFilterUsed) throws FilterManagerException, FilterTypeException {

                filterList.clear();

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

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

                        RasterFilterListManager filterManager = filterList.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");

        }

}