Application: gvSIG desktop

/* 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.rastertools.filter;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.Graphics2D;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.geom.AffineTransform;

import java.awt.geom.Dimension2D;

import java.awt.geom.Point2D;

import java.io.File;

import java.util.ArrayList;

import java.util.EventObject;

import java.util.Hashtable;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

import javax.swing.ListModel;

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

import org.gvsig.gui.beans.imagenavigator.IClientImageNavigator;

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

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

import org.gvsig.raster.dataset.IRasterDataSource;

import org.gvsig.raster.dataset.InvalidSetViewException;

import org.gvsig.raster.dataset.Params;

import org.gvsig.raster.dataset.Params.Param;

import org.gvsig.raster.datastruct.Extent;

import org.gvsig.raster.datastruct.ViewPortData;

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

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

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

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

import org.gvsig.raster.grid.render.Rendering;

import org.gvsig.rastertools.RasterModule;

import org.gvsig.rastertools.filter.ui.FilterPanel;

import com.iver.andami.PluginServices;

import com.iver.andami.Utilities;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.ui.mdiManager.IWindow;

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

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

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

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

/**

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

 *

 * @version 24/05/2007

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

 */

public class FilterListener implements ActionListener, IClientImageNavigator, TreeListComponentListener, TreeListChangeListener, PropertiesComponentListener {

        FilterPanel                                filterPanel        = null;

        public int                                actualParam        = -1;

        public ArrayList        paramsList        = new ArrayList();

        public class ParamStruct {

                String filterName = null;

                Params filterParam = null;

                Class filterClass = null;

                /**

                 * @return the filterName

                 */

                public String getFilterName() {

                        return filterName;

                }

                /**

                 * @param filterName the filterName to set

                 */

                public void setFilterName(String filterName) {

                        this.filterName = filterName;

                }

                /**

                 * @return the filterParam

                 */

                public Params getFilterParam() {

                        return filterParam;

                }

                /**

                 * @param filterParam the filterParam to set

                 */

                public void setFilterParam(Params filterParam) {

                        this.filterParam = filterParam;

                }

                /**

                 * @return the filterClass

                 */

                public Class getFilterClass() {

                        return filterClass;

                }

                /**

                 * @param filterClass the filterClass to set

                 */

                public void setFilterClass(Class filterClass) {

                        this.filterClass = filterClass;

                }

        }

        /**

         * Construye un FilterListener especificando el FilterPanel asociado

         * @param fp

         */

        public FilterListener(FilterPanel fp) {

                filterPanel = fp;

        }

        /**

         * Asignamos los valores del PropertiesComponent al Params seleccionado

         */

        public void RefreshDataProperties() {

                if (actualParam == -1) return;

                ArrayList 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().toString());

                }

        }

        /**

         * 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) {

                                for (int i = 0; i < params.getNumParams(); i++) {

                                        Param p = params.getParam(i);

                                        String name = p.id;

                                        String key = p.id;

                                        if (name.equals("FilterName")) continue;

                                        Object[] types = null;

                                        int selectedValue = 0;

                                        switch (params.getParam(i).type) {

                                                case Params.CHECK:

                                                        propertiesComponent.addValue(name, key, new Boolean((String) p.defaultValue), types);

                                                        break;

                                                case Params.CHOICE:

                                                        ArrayList list = new ArrayList();

                                                        for (int j = 0; j < p.list.length; j++) {

                                                                list.add(p.list[j]);

                                                                if (((String) p.defaultValue).compareTo(p.list[j]) == 0)

                                                                        selectedValue = j;

                                                        }

                                                        types = new Object[] { new Integer(PropertiesComponent.TYPE_COMBO), list };

                                                        propertiesComponent.addValue(name, key, new Integer(selectedValue), types);

                                                        break;

                                                case Params.SLIDER:

                                                        types = new Object[]{new Integer(PropertiesComponent.TYPE_SLIDER), new Integer(p.list[0]), new Integer(p.list[1])};

                                                        propertiesComponent.addValue(name, key, new Integer((int) Math.round(new Double((String) p.defaultValue).doubleValue())), types);

                                                        break;

                                                default:

                                                        propertiesComponent.addValue(params.getParam(i).id, params.getParam(i).id, params.getParam(i).defaultValue, null);

                                                        break;

                                        }

                                }

                        }

                }

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

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.propertiespanel.PropertiesComponentListener#actionChangeProperties(java.util.EventObject)

         */

        public void actionChangeProperties(EventObject e) {

                RefreshDataProperties();

                getFilterPanel().refreshPreview();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.treelist.listeners.TreeListChangeListener#actionChangeSelection(org.gvsig.gui.beans.treelist.event.TreeListChangeEvent)

         */

        public void actionChangeSelection(TreeListChangeEvent e) {

                changePanel(e.getItem());

        }

        /*

         * (non-Javadoc)

         * @see java.awt.event.ActionListener#actionPerformed(java.awt.event.ActionEvent)

         */

        public void actionPerformed(ActionEvent e) {

                getFilterPanel().refreshPreview();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.treelist.listeners.TreeListComponentListener#elementAdded(org.gvsig.gui.beans.treelist.event.TreeListEvent)

         */

        public void elementAdded(TreeListEvent e) {

                getFilterPanel().refreshPreview();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.treelist.listeners.TreeListComponentListener#elementMoved(org.gvsig.gui.beans.treelist.event.TreeListEvent)

         */

        public void elementMoved(TreeListEvent e) {

                getFilterPanel().refreshPreview();

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.treelist.listeners.TreeListComponentListener#elementRemoved(org.gvsig.gui.beans.treelist.event.TreeListEvent)

         */

        public void elementRemoved(TreeListEvent e) {

                getFilterPanel().refreshPreview();

        }

        /**

         * @return the paramsList

         */

        public ArrayList getParamsList() {

                return paramsList;

        }

        /**

         * Devuelve la ruta del fichero donde se va a guardar, en caso de guardarse

         * en memoria, calcula el nombre sin preguntar y devuelve la ruta.

         * @return

         */

        public String getFileSelected() {

                String path = "";

                if (getFilterPanel().getRadioFileGenerate().isSelected()) {

                        JFileChooser chooser = new JFileChooser(RasterModule.getPath());

                        chooser.setDialogTitle(PluginServices.getText(this, "seleccionar_fichero"));

                        //A?adimos las extensiones que hayan sido registradas en el driver

                        String[] extList = GeoRasterWriter.getDriversExtensions();

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

                                chooser.addChoosableFileFilter(new WriterFilter(extList[i]));

                        if (chooser.showOpenDialog(getFilterPanel()) != JFileChooser.APPROVE_OPTION)

                                return null;

                        String fName = chooser.getSelectedFile().toString();

                        String ext = ((WriterFilter)chooser.getFileFilter()).getDescription();

                        ext = ext.toLowerCase().substring(ext.lastIndexOf(".") + 1, ext.length());

                        if ((fName != null) && !fName.equals(""))

                                if (!fName.endsWith("." + ext))

                                        fName = fName + "." + ext;

                        RasterModule.setPath(chooser.getSelectedFile().getPath().substring(0, chooser.getSelectedFile().getPath().lastIndexOf(File.separator)));

                        path = fName;

                } else {

                        path = Utilities.createTempDirectory() + File.separator + getFilterPanel().getFilenameTextField().getText() + ".tif";

                        filterPanel.setNewLayerText();

                }

                return path;

        }

        /**

         * 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 rendering

         * @return

         */

        public ArrayList applyFilters(Rendering rendering) {

                ArrayList listFilterUsed = new ArrayList();

                RasterFilterList filterList = new RasterFilterList();

                RasterFilterListManager stackManager = new RasterFilterListManager(filterList);

                // Conservamos filtros no visibles ya existentes

                ArrayList filtersInit = getFilterPanel().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 < stackManager.getRasterFilterList().size(); j++) {

                                Class classFilter = (Class) stackManager.getRasterFilterList().get(j);

                                try {

                                        obj = (RasterFilter) classFilter.newInstance();

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

                                                break;

                                } catch (InstantiationException e) {

                                        NotificationManager.addError("Error instanciando un filtro de la lista", e);

                                } catch (IllegalAccessException e) {

                                        NotificationManager.addError("Error instanciando un filtro de la lista", 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

                                String rgb = rendering.getRenderBands()[0] + " " + rendering.getRenderBands()[1] + " " + rendering.getRenderBands()[2];

                                params.setParam("RenderBands", rgb, 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().getTreeListContainer().getListModel();

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

                        Hashtable hastTable = getFilterPanel().getTreeListContainer().getMap();

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

                                boolean active = true;

                                Param param = ((ParamStruct) paramsList.get(j)).getFilterParam().getParamById(PluginServices.getText(this, "enabled"));

                                if ((param != null) && ((new Boolean((String) param.defaultValue).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

                                                        String rgb = rendering.getRenderBands()[0] + " " + rendering.getRenderBands()[1] + " " + rendering.getRenderBands()[2];

                                                        params.setParam("RenderBands", rgb, 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;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.imagenavigator.IClientImageNavigator#drawImage(java.awt.Graphics2D, double, double, double, double, double, int, int)

         */

        public void drawImage(Graphics2D g, double x1, double y1, double x2, double y2, double zoom, int width, int height) {

                if (getFilterPanel().getLayer() == null)

                        return;

                Rendering rendering = ((FLyrRasterSE) getFilterPanel().getLayer()).getRender();

                if ((rendering == null) || ((x2 - x1) == 0.0) || ((y2 - y1) == 0.0))

                        return;

                rendering.getFilterList().pushStatus();

                rendering.getFilterList().clear();

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

                        RasterFilterList filterList = rendering.getFilterList();

                        RasterFilterListManager stackManager = new RasterFilterListManager(filterList);

                        ArrayList listFilterUsed = applyFilters(rendering);

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

                                IRasterFilterListManager filterManager = stackManager.getManagerByFilterClass(((ParamStruct) listFilterUsed.get(i)).getFilterClass());

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

                        }

                }

                ViewPort vp = new ViewPort(null);

                Dimension2D dimension = new Dimension(width, height);

                Extent extent = new Extent(new Point2D.Double(x1, y1), new Point2D.Double(x2, y2), new Point2D.Double(x2, y1), new Point2D.Double(x1, y2));

                //Extent extent = new Extent(x1, y1, x2, y2);

                ViewPortData vp2 = new ViewPortData(vp.getProjection(), extent, dimension);

                vp2.setMat(new AffineTransform(zoom, 0.0, 0.0, zoom, -x1*zoom, -y1*zoom));

                Extent extent2 = ((FLyrRasterSE) getFilterPanel().getLayer()).getBufferFactory().getExtent();

                AffineTransform trans = g.getTransform();

                // Calcular cuanto sobresale la imagen y rectificar ese desplazamiento

                if (y1 > extent2.maxY()) {

                        g.translate(0.0, (-(extent2.maxY() - y1) * zoom)*2.0);

                }

                try {

                        rendering.draw(g, vp2);

                } catch (ArrayIndexOutOfBoundsException e) {

                        NotificationManager.addError("Error en el renderizado de la mini-imagen del panel de tabla de color.", e);

                } catch (InvalidSetViewException e) {

                        NotificationManager.addError("Error asignando la vista en el renderizado de la mini-imagen del panel de tabla de color.", e);

                } catch (InterruptedException e) {

                        NotificationManager.addWarning("Dibujado de la preview detenido por el usuario.", e);

                }

                g.setTransform(trans);

                rendering.getFilterList().popStatus();

        }

        /**

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

         */

        public void accept() {

                IRasterDataSource raster = ((FLyrRasterSE) getFilterPanel().getLayer()).getDataSource();

                if (raster == null)

                        return;

                String path = null;

                if (!getFilterPanel().getRadioOnlyView().isSelected()) {

                        path = getFileSelected();

                        if (path == null)

                                return;

                }

                Rendering rendering = ((FLyrRasterSE) getFilterPanel().getLayer()).getRender();

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

                ArrayList listFilterUsed = applyFilters(rendering);

                if (filterPanel.getRadioOnlyView().isSelected()) {

                        FilterProcess.addSelectedFilters(rendering.getFilterList(), listFilterUsed);

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

                } else {

                        FilterProcess filterProcess = new FilterProcess(this, path, raster, listFilterUsed);

                        IncrementableTask incrementableTask = new IncrementableTask(filterProcess);

                        filterProcess.setIncrementableTask(incrementableTask);

                        incrementableTask.showWindow();

                        filterProcess.start();

                        incrementableTask.start();

                }

        }

        /**

         * 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().getRadioNewLayer().isSelected())

                        return;

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

                        return;

                // seleccionamos la vista de gvSIG

                com.iver.cit.gvsig.project.documents.view.gui.View theView = null;

                try {

                        IWindow[] allViews = PluginServices.getMDIManager().getAllWindows();

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

                                if (allViews[i] instanceof com.iver.cit.gvsig.project.documents.view.gui.View

                                                && PluginServices.getMDIManager().getWindowInfo((View) allViews[i])

                                                                .getTitle().equals(getFilterPanel().getViewName()))

                                        theView = (com.iver.cit.gvsig.project.documents.view.gui.View) allViews[i];

                        }

                        if (theView == null)

                                return;

                } catch (ClassCastException ex) {

                        // logger.error(PluginServices.getText(this,"cant_get_view "), ex);

                        return;

                }

                // Cargamos las capas

                theView.getMapControl().getMapContext().beginAtomicEvent();

                //Driver driver = LayerFactory.getDM().getDriver("gvSIG Raster Driver");

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

                if (endIndex < 0)

                        endIndex = fileName.length();

                FLayer lyr = null;

                try {

                        lyr = FLyrRasterSE.createLayer(fileName.substring(

                                        fileName.lastIndexOf(File.separator) + 1, endIndex),

                                        new File(fileName), theView.getMapControl().getProjection());

                } catch (LoadLayerException e) {

                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),

                                        PluginServices.getText(this, "error_cargar_capa"));

                }

                theView.getMapControl().getMapContext().getLayers().addLayer(lyr);

                theView.getMapControl().getMapContext().endAtomicEvent();

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

        }

        /*public void loadLayerInToc(IBuffer b, Extent e) {

                // seleccionamos la vista de gvSIG

                com.iver.cit.gvsig.project.documents.view.gui.View theView = null;

                try {

                        IWindow[] allViews = PluginServices.getMDIManager().getAllWindows();

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

                                if (allViews[i] instanceof com.iver.cit.gvsig.project.documents.view.gui.View

                                                && PluginServices.getMDIManager().getWindowInfo((View) allViews[i])

                                                                .getTitle().equals(getFilterPanel().getViewName()))

                                        theView = (com.iver.cit.gvsig.project.documents.view.gui.View) allViews[i];

                        }

                        if (theView == null)

                                return;

                } catch (ClassCastException ex) {

                        // logger.error(PluginServices.getText(this,"cant_get_view "), ex);

                        return;

                }

                // Cargamos las capas

                theView.getMapControl().getMapContext().beginAtomicEvent();

                //Driver driver = LayerFactory.getDM().getDriver("gvSIG Raster Driver");

                MemoryRasterDriverParam d = new MemoryRasterDriverParam(b, e);

                FLayer lyr = null;

                try {

                        lyr = FLyrRasterSE.createLayer("capa", d, theView.getMapControl().getProjection());

                } catch (LoadLayerException e1) {

                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),

                                        PluginServices.getText(this, "error_cargar_capa"));

                }

                theView.getMapControl().getMapContext().getLayers().addLayer(lyr);

                theView.getMapControl().getMapContext().endAtomicEvent();

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

        }*/

}

/**

 * @author Nacho Brodin <brodin_ign@gva.es>

 * Filtro para el selector de formatos de escritura

 */

class WriterFilter extends javax.swing.filechooser.FileFilter {

        private String                                filter;

        public WriterFilter(String fil) {

                this.filter = fil;

        }

        public boolean accept(File f) {

                return f.isDirectory() || f.getName().toLowerCase().endsWith("." + filter);

        }

        public String getDescription() {

                return "." + filter;

        }

}