Application: gvSIG desktop » gvSIG raster

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

 *

 * Copyright (C) 2006 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.raster.tools.app.basic.tool.filter;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.util.ArrayList;

import java.util.EventObject;

import java.util.Hashtable;

import java.util.List;

import javax.swing.ListModel;

import org.gvsig.andami.PluginServices;

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

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

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

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

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

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

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

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

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

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

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

import org.gvsig.gui.beans.propertiespanel.PropertiesComponent;

import org.gvsig.gui.beans.propertiespanel.PropertiesComponentListener;

import org.gvsig.gui.beans.propertiespanel.PropertyStruct;

import org.gvsig.gui.beans.treelist.event.TreeListChangeEvent;

import org.gvsig.gui.beans.treelist.event.TreeListEvent;

import org.gvsig.gui.beans.treelist.listeners.TreeListChangeListener;

import org.gvsig.gui.beans.treelist.listeners.TreeListComponentListener;

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

import org.gvsig.raster.swing.RasterSwingLibrary;

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

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

import org.gvsig.raster.tools.app.basic.raster.process.FilterProcess;

import org.gvsig.raster.tools.app.basic.raster.process.IProcessActions;

import org.gvsig.raster.tools.app.basic.tool.filter.ui.FilterPanel;

import org.gvsig.raster.util.RasterNotLoadException;

/**

 * <code>FilterListener</code> es la clase donde se procesar? gran parte del

 * c?digo que controla el panel para el manejo de un layer en la aplicaci?n de

 * filtros.

 *

 * @author BorSanZa - Borja S?nchez Zamorano 

 */

public class FilterListener implements ActionListener, TreeListComponentListener, TreeListChangeListener, PropertiesComponentListener, FilterUIListener, IProcessActions {

        private FilterPanel               filterPanel  = null;

        private int                       actualParam  = -1;

        private ArrayList<ParamStruct>    paramsList   = new ArrayList<ParamStruct>();

        /**

         * Construye un FilterListener especificando el FilterPanel asociado

         * @param filterDialog

         */

        public FilterListener(FilterPanel filterPanel) {

                this.filterPanel = filterPanel;

        }

        /**

         * Asignamos los valores del PropertiesComponent al Params seleccionado

         */

        @SuppressWarnings("unchecked")

        public void refreshDataProperties() {

                if (actualParam == -1)

                        return;

                ArrayList<PropertyStruct> listValues = getFilterPanel().getPropertiesComponent().getValues();

                Params params = ((ParamStruct) paramsList.get(actualParam)).getFilterParam();

                for (int j = 0; j < listValues.size(); j++) {

                        PropertyStruct ps = (PropertyStruct) listValues.get(j);

                        params.changeParamValue(ps.getKey(), ps.getNewValue());

                }

        }

        /**

         * Obtener la posici?n del Param seleccionado en el ArrayList

         * @param filterName

         * @return

         */

        private int getParamSelected(String filterName) {

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

                        if (((ParamStruct) paramsList.get(i)).getFilterName().equals(filterName))

                                return i;

                }

                return -1;

        }

        /**

         * Cambiar el panel de propiedades central por el nuevo panel, segun el filtro

         * seleccionado que se pasa por par?metro.

         * @param filter

         */

        public void changePanel(String filter) {

                int posParam = getParamSelected(filter);

                refreshDataProperties();

                actualParam = posParam;

                PropertiesComponent propertiesComponent = new PropertiesComponent();

                if (posParam != -1) {

                        Params params = ((ParamStruct) paramsList.get(actualParam)).getFilterParam();

                        if (params != null) {

                                Param paramPanel = params.getParamById("Panel");

                                if(        paramPanel != null && 

                                        paramPanel.getDefaultValue() instanceof RegistrableFilterListener)

                                        ((RegistrableFilterListener)paramPanel.getDefaultValue()).addFilterUIListener(this);

                                RasterToolsUtil.loadPropertiesFromWriterParams(propertiesComponent, params, new String[]{"FilterName"});

                        }

                }

                getFilterPanel().setNewPropertiesComponent(propertiesComponent, filter);

        }

        /**

         * A?adir un nuevo Params a la lista de Params que podemos manejar. Un Params

         * equivale a un filtro cargado. El hecho de trabajar con Params y no con

         * filtros, simplifica totalmente el panel. Sin tener que depender de los

         * filtros nada m?s que para el momento de dibujado o guardado.

         * @param name

         * @param params

         * @param classFilter

         */

        public void addNewParam(String name, Params params, Class<?> classFilter) {

                ParamStruct param = new ParamStruct();

                param.setFilterName(name);

                param.setFilterParam(params);

                param.setFilterClass(classFilter);

                paramsList.add(param);

        }

        public void actionChangeProperties(EventObject e) {

                refreshDataProperties();

                getFilterPanel().refreshPreview();

        }

        public void actionChangeSelection(TreeListChangeEvent e) {

                changePanel(e.getItem());

        }

        public void actionPerformed(ActionEvent e) {

                getFilterPanel().refreshPreview();

        }

        public void elementAdded(TreeListEvent e) {

                getFilterPanel().refreshPreview();

        }

        public void elementMoved(TreeListEvent e) {

                getFilterPanel().refreshPreview();

        }

        public void elementRemoved(TreeListEvent e) {

                getFilterPanel().refreshPreview();

        }

        /**

         * @return the paramsList

         */

        public ArrayList<ParamStruct> getParamsList() {

                return paramsList;

        }

        /**

         * Aqui se seleccionan que filtros se van a aplicar y se devuelven en forma

         * de ArrayList tanto para el dibujado como cuando aceptan o aplican el panel.

         * @param layerRaster

         * @return

         */

        public ArrayList<ParamStruct> applyFilters(FLyrRaster layerRaster) {

                ArrayList<ParamStruct> listFilterUsed = new ArrayList<ParamStruct>();

                int initDataType = layerRaster.getRender().getFilterList().getInitDataType();

                RasterFilterList filterList = RasterLocator.getManager().createEmptyFilterList(initDataType);

                filterList.setEnv(layerRaster.getRender().getFilterList().getEnv());

                filterList.setInitDataType(initDataType);

                List<Class<?>> registeredFilters = filterList.getRegisteredFilterList();

                // Conservamos filtros no visibles ya existentes

                List<RasterFilter> filtersInit =  getFilterPanel().getLayerVisualStatus().getLast().getFilterStatus();

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

                        // Si es visible no hacemos nada

                        if (((RasterFilter) filtersInit.get(i)).isVisible())

                                continue;

                        RasterFilter obj = null;

                        for (int j = 0; j < registeredFilters.size(); j++) {

                                Class<?> classFilter = (Class<?>) registeredFilters.get(j);

                                try {

                                        obj = (RasterFilter) classFilter.newInstance();

                                        if (obj.getName().equals(((RasterFilter) filtersInit.get(i)).getName()))

                                                break;

                                } catch (InstantiationException e) {

                                        RasterSwingLibrary.messageBoxError("error_creando_filtro", this, e);

                                } catch (IllegalAccessException e) {

                                        RasterSwingLibrary.messageBoxError("error_creando_filtro", this, e);

                                }

                        }

                        // Si no encontramos el filtro apropiado, nos olvidamos de el

                        if (obj == null)

                                continue;

                        // Si no es visible tenemos que conservar el filtro

                        try {

                                Params params = (Params) ((RasterFilter) filtersInit.get(i)).getUIParams(((RasterFilter) filtersInit.get(i)).getName()).clone();

                                // A?ado el parametro RenderBands a los parametros del filtro

                                int[] renderBands = layerRaster.getRender().getRenderColorInterpretation().buildRenderBands();

                                String rgb = renderBands[0] + " " + renderBands[1] + " " + renderBands[2];

                                params.setParam("RenderBands", rgb, 0, null);

                                params.setParam("alphaBand", new Integer(layerRaster.getAlphaBandNumber()), 0, null);

                                ParamStruct newParam = new ParamStruct();

                                newParam.setFilterClass(obj.getClass());

                                newParam.setFilterName(((RasterFilter) filtersInit.get(i)).getName());

                                newParam.setFilterParam(params);

                                listFilterUsed.add(newParam);

                        } catch (CloneNotSupportedException e) {

                        }

                }

                // Metemos los filtros seleccionados en el panel

                ListModel list = getFilterPanel().getMainPanel().getTreeListContainer().getListModel();

                for (int i = 0; i < list.getSize(); i++) {

                        Hashtable<?, ?> hastTable = getFilterPanel().getMainPanel().getTreeListContainer().getMap();

                        for (int j = 0; j < paramsList.size(); j++) {

                                boolean active = true;

                                Param param = ((ParamStruct) paramsList.get(j)).getFilterParam().getParamById("enabled");

                                if ((param != null) &&

                                        param.getDefaultValue() instanceof Boolean &&

                                        ((((Boolean)param.getDefaultValue()).booleanValue()) == false))

                                        active = false;

                                if (active) {

                                        if (((ParamStruct) paramsList.get(j)).getFilterName().equals(hastTable.get(list.getElementAt(i)))) {

                                                try {

                                                        Params params = (Params) ((ParamStruct) paramsList.get(j)).getFilterParam().clone();

                                                        // A?ado el parametro RenderBands a los parametros del filtro

                                                        int[] renderBands = layerRaster.getRender().getRenderColorInterpretation().buildRenderBands();

                                                        String rgb = renderBands[0] + " " + renderBands[1] + " " + renderBands[2];

                                                        params.setParam("RenderBands", rgb, 0, null);

                                                        params.setParam("alphaBand", new Integer(layerRaster.getAlphaBandNumber()), 0, null);

                                                        ParamStruct newParam = new ParamStruct();

                                                        newParam.setFilterClass(((ParamStruct) paramsList.get(j)).getFilterClass());

                                                        newParam.setFilterName(((ParamStruct) paramsList.get(j)).getFilterName());

                                                        newParam.setFilterParam(params);

                                                        listFilterUsed.add(newParam);

                                                } catch (CloneNotSupportedException e) {

                                                }

                                        }

                                }

                        }

                }

                return listFilterUsed;

        }

        public void drawImage(FLyrRaster rendering) {

                rendering.getRender().getFilterList().clear();

                if (getFilterPanel().getCBShowFilters().isSelected()) {

                        RasterFilterList filterList = rendering.getRender().getFilterList();

                        ArrayList<ParamStruct> listFilterUsed = applyFilters(rendering);

                        ArrayList<Exception> exc = new ArrayList<Exception>();

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

                                try {

                                        RasterFilterListManager filterManager = filterList.getManagerByFilterClass(((ParamStruct) listFilterUsed.get(i)).getFilterClass());

                                        filterManager.addFilter(((ParamStruct) listFilterUsed.get(i)).getFilterClass(), ((ParamStruct) listFilterUsed.get(i)).getFilterParam());

                                } catch (FilterTypeException e) {

                                        exc.add(e);

                                } catch (FilterManagerException e1) {

                                        exc.add(e1);

                                }

                        }

                        if(exc.size() != 0) {

                                RasterSwingLibrary.messageBoxError(PluginServices.getText(this, "error_adding_filters"), this, exc);

                                exc.clear();

                        }

                }

        }

        /**

         * Que acciones se ejecutaran al haber presionado el bot?n aceptar o aplicar

         */

        public void accept() {

                RasterDataStore raster = ((FLyrRaster) getFilterPanel().getLayer()).getDataStore();

                if (raster == null)

                        return;

                String path = null;

                if (!getFilterPanel().getNewLayerPanel().isOnlyViewSelected()) {

                        path = getFilterPanel().getNewLayerPanel().getFileSelected();

                        if (path == null)

                                return;

                }

                FLyrRaster rendering = (FLyrRaster) getFilterPanel().getLayer();

                // Array para guardar los filtros que se van a usar en forma de ParamStruct

                ArrayList<ParamStruct> listFilterUsed = applyFilters(rendering);

                if (filterPanel.getNewLayerPanel().isOnlyViewSelected()) {

                        try {

                                ParamStruct.addSelectedFilters(rendering.getRender().getFilterList(), listFilterUsed);

                                ((FLyrRaster) getFilterPanel().getLayer()).getRender().setFilterList(rendering.getRender().getFilterList());

                                getFilterPanel().getLayer().getMapContext().invalidate();

                        } catch (FilterTypeException e) {

                                RasterSwingLibrary.messageBoxError(PluginServices.getText(this, "error_adding_filters"), this, e);

                        } catch (FilterManagerException e) {

                                RasterSwingLibrary.messageBoxError(PluginServices.getText(this, "error_adding_filters"), this, e);

                        }

                } else {

                        FilterProcess filterProcess = new FilterProcess();

                        filterProcess.setActions(this);

                        filterProcess.addParam("rendering", rendering);

                        filterProcess.addParam("filename", path);

                        filterProcess.addParam("rasterdatasource", raster);

                        filterProcess.addParam("layer", ((FLyrRaster) getFilterPanel().getLayer()));

                        filterProcess.addParam("listfilterused", listFilterUsed);

                        filterProcess.start();

                        filterProcess = null;

                }

        }

        /**

         * Devuelve el FilterPanel asociado al FilterListener

         * @return

         */

        public FilterPanel getFilterPanel() {

                return filterPanel;

        }

        /**

         * Acciones que se realizan al finalizar de crear los recortes de imagen.

         * Este m?todo es llamado por el thread TailRasterProcess al finalizar.

         */

        public void loadLayerInToc(String fileName) {

                if (!getFilterPanel().getNewLayerPanel().isNewLayerSelected())

                        return;

                if(!new File(fileName).exists())

                        return;

                try {

                        RasterToolsUtil.loadLayer(getFilterPanel().getViewName(), fileName, null);

                } catch (RasterNotLoadException e) {

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

                }

                if(filterPanel != null)

                        filterPanel.updateNewLayerText();

        }

        public void actionValuesCompleted(EventObject e) {

                actionChangeProperties(e);

        }

        public void end(Object param) {

                loadLayerInToc((String) param);

        }

        public void interrupted() {}

}