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

import java.awt.BorderLayout;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.Insets;

import java.util.ArrayList;

import java.util.Hashtable;

import java.util.List;

import javax.swing.BorderFactory;

import javax.swing.DefaultListModel;

import javax.swing.JCheckBox;

import javax.swing.JPanel;

import javax.swing.border.TitledBorder;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.ui.mdiManager.IWindow;

import org.gvsig.app.project.documents.view.gui.AbstractViewPanel;

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

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

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

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

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

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

import org.gvsig.gui.beans.buttonspanel.ButtonsPanel;

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

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

import org.gvsig.gui.beans.treelist.TreeListContainer;

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

import org.gvsig.raster.mainplugin.beans.createlayer.CreateLayerPanel;

import org.gvsig.raster.swing.RasterSwingLocator;

import org.gvsig.raster.swing.preview.DataSourcePreview;

import org.gvsig.raster.swing.preview.PreviewBasePanelTab;

import org.gvsig.raster.swing.preview.PreviewPanel;

import org.gvsig.raster.swing.preview.PreviewRenderProcess;

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.bean.previewbase.RasterDataSourcePreview;

import org.gvsig.raster.tools.app.basic.tool.filter.FilterListener;

import org.gvsig.raster.util.LayerVisualStatusList;

/**

 * Dialogo para los filtros de raster.

 *

 * @author Nacho Brodin (nachobrodin@gmail.com)

 * @author BorSanZa - Borja S?nchez Zamorano 

 */

public class FilterPanel extends JPanel implements PreviewRenderProcess {

        private static final long serialVersionUID = 7152780112689637266L;

        private PreviewPanel           previewBasePanel    = null;

        private FLyrRaster             layer               = null;

        private FilterListener         filterListener      = null;

        private PropertiesComponent    propertiesComponent = null;

        private LayerVisualStatusList  status              = new LayerVisualStatusList();

        private JCheckBox              jCBShowFilters      = null;

        private String                 viewName            = null;

        private JPanel                 jPanelOptions       = null;

        private FilterMainPanel        filterMain          = null;

        private FilterDialog           filterDialog        = null;

        private CreateLayerPanel       newLayerPanel       = null;

        private boolean                showPreview         = true;        

        /**

         * Constructor

         * @param width Ancho del panel

         * @param height Alto del panel

         */

        public FilterPanel(FLyrRaster layer, FilterDialog filterDialog) {

                this.filterDialog = filterDialog;

                setLayer(layer);

                initialize();

        }

        private void initialize() {

                translate();

                setLayout(new BorderLayout());

                add((Component)getPreviewBasePanel(), BorderLayout.CENTER);

        }

        /**

         * Seccion donde ir?n todas las traducciones invariables del componente

         */

        private void translate() {

                getCBShowFilters().setText(PluginServices.getText(this, "aplicar_vista_previa"));

        }

        /**

         * Obtiene el panel con el histograma

         * @return HistogramPanel

         */

        public PreviewPanel getPreviewBasePanel() {

                if (previewBasePanel == null) {

                        ArrayList<PreviewBasePanelTab> list = new ArrayList<PreviewBasePanelTab>();

                        list.add(getMainPanel());

                        RasterDataSourcePreview source = new RasterDataSourcePreview((FLyrRaster)layer);

                        previewBasePanel = RasterSwingLocator.getSwingManager().createPreviewPanel(

                                        ButtonsPanel.BUTTONS_APPLYCLOSE, 

                                        list, 

                                        null, 

                                        panelOptions(), 

                                        this, 

                                        source);

                        previewBasePanel.getButtonsPanel().addCancel();

                        previewBasePanel.setPreviewSize(new Dimension(230, 215));

                        previewBasePanel.addButtonPressedListener(filterDialog);

                }

                return previewBasePanel;

        }

        /**

         * Devuelve el componente <code>FilterListener</code>, que contendr? el

         * proceso en si del panel

         */

        private FilterListener getFilterListener() {

                if (filterListener == null) {

                        filterListener = new FilterListener(this);

                }

                return filterListener;

        }

        /**

         * Devuelve el panel de Opciones, en caso de no existir, lo crea.

         * @return

         */

        private JPanel panelOptions() {

                if (jPanelOptions == null) {

                        jPanelOptions = new JPanel();

                        GridBagConstraints gridBagConstraints;

                        jPanelOptions.setLayout(new GridBagLayout());

                        gridBagConstraints = new GridBagConstraints();

                        gridBagConstraints.gridx = 0;

                        gridBagConstraints.gridy = 1;

                        gridBagConstraints.anchor = GridBagConstraints.WEST;

                        gridBagConstraints.insets = new Insets(2, 2, 2, 2);

                        jPanelOptions.add(getCBShowFilters(), gridBagConstraints);

                        gridBagConstraints = new GridBagConstraints();

                        gridBagConstraints.gridx = 0;

                        gridBagConstraints.gridy = 4;

                        gridBagConstraints.fill = GridBagConstraints.HORIZONTAL;

                        gridBagConstraints.insets = new Insets(3, 3, 3, 3);

                        jPanelOptions.add(getNewLayerPanel().getJPanel(), gridBagConstraints);

                        gridBagConstraints = new GridBagConstraints();

                        gridBagConstraints.gridx = 0;

                        gridBagConstraints.gridy = 5;

                        gridBagConstraints.weightx = 1.0;

                        gridBagConstraints.weighty = 1.0;

                        jPanelOptions.add(new JPanel(), gridBagConstraints);

                }

                return jPanelOptions;

        }

        public CreateLayerPanel getNewLayerPanel() {

                if (newLayerPanel == null) {

                        newLayerPanel = new CreateLayerPanel(layer);

                         if(layer.isRemote())

                                 newLayerPanel.disableNewLayer();

                }

                return newLayerPanel;

        }

        /**

         * Construye el arbol para el men? de filtros disponibles

         */

        @SuppressWarnings("unchecked")

        private void menuBuild() {

                Hashtable keys = new Hashtable();

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

                int dataType = layer.getDataStore().getDataType()[0];

                if( filterList.getByName("enhanced_stretch") != null ||

                        filterList.getByName("enhanced") != null ||

                        filterList.getByName("colortable") != null)

                        dataType = Buffer.TYPE_BYTE;

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

                int count = 0;

                RasterFilter obj = null;

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

                        Class classFilter = (Class) registeredFilters.get(i);

                        try {

                                obj = (RasterFilter) classFilter.newInstance();

                                initFilter(obj);

                                if (!obj.isVisible())

                                        continue;

                                // Crear grupos de los filtros

                                if (!keys.containsKey(PluginServices.getText(this, new String(obj.getGroup())))) {

                                        keys.put(PluginServices.getText(this, new String(obj.getGroup())), obj);

                                        getMainPanel().getTreeListContainer().addClass(PluginServices.getText(this, new String(obj.getGroup())), count);

                                        count++;

                                }

                                // Crear cada entrada de los filtro

                                String[] names = obj.getNames();

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

                                        obj.setName(names[j]);

                                        getFilterListener().addNewParam(names[j], initFilterParam(obj.getUIParams(names[j])), classFilter);

                                        getMainPanel().getTreeListContainer().addEntry(PluginServices.getText(this, names[j]), PluginServices.getText(this, obj.getGroup()), names[j]);

                                }

                        } catch (InstantiationException e) {

                                RasterToolsUtil.debug("No se ha podido meter el filtro en la lista", this, e);

                        } catch (IllegalAccessException e) {

                                RasterToolsUtil.debug("No se ha podido meter el filtro en la lista", this, e);

                        }

                }

                getMainPanel().getTreeListContainer().getTree().expandRow(0);

        }

        /**

         * Pasamos los parametros basicos a un filtro

         * @param filter

         */

        @SuppressWarnings("unchecked")

        private void initFilter(RasterFilter filter) {

                if (layer != null) {

                        filter.getEnv().put("datastore", layer.getDataStore());

                        filter.getEnv().put("initRaster", layer);

                }

        }

        /**

         * Sirve para a?adir mas parametros genericos por defecto al panel, de momento

         * solo he puesto para activar o desactivar un filtro.

         * @param params

         * @return

         */

        private Params initFilterParam(Params params) {

                params.setParam("enabled", new Boolean(true), Params.CHECK, null);

                Param param = params.getParamById("enabled");

                params.getParams().remove(params.getParams().size() - 1);

                params.getParams().add(0, param);

                return params;

        }

        /**

         * Comprueba que el nombre del filtro esta cargado en las extensiones

         * @param filter

         * @return

         */

        @SuppressWarnings("unchecked")

        private boolean hasThisFilter(String filter) {

                RasterFilterList filterList = RasterLocator.getManager().createEmptyFilterList(Buffer.TYPE_BYTE);

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

                boolean error;

                RasterFilter obj = null;

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

                        Class classFilter = (Class) registeredFilters.get(i);

                        error = true;

                        try {

                                obj = (RasterFilter) classFilter.newInstance();

                                error = false;

                        } catch (InstantiationException e) {

                        } catch (IllegalAccessException e) {

                        }

                        if (!error) {

                                String[] names = obj.getNames();

                                for (int j = 0; j < names.length; j++)

                                        if (filter.equals(names[j]))

                                                return true;

                        }

                }

                return false;

        }

        /**

         * Definir el FLayer del panel, haciendo todas las cargas necesarias despues

         * de especificarlo.

         * @param layer

         */

        private void setLayer(FLyrRaster layer) {

                if (layer == null)

                        return;

                this.layer = layer;

                RasterDataSourcePreview source = new RasterDataSourcePreview(layer);

                getPreviewBasePanel().setLayer(source);

                // Construimos el arbol de filtros

                menuBuild();

                IWindow win = PluginServices.getMDIManager().getActiveWindow();

                AbstractViewPanel view = null;

                if(win instanceof AbstractViewPanel)

                        view = (AbstractViewPanel) win;

                else {

                        RasterToolsUtil.messageBoxInfo("view_not_active", this);

                        return;

                }

                viewName = PluginServices.getMDIManager().getWindowInfo(view).getTitle();

                //Comprobamos que el raster de salida no exceda los 4GB que es el m?ximo 

                //admisible por un TIFF. Si es as? se hace un setCompress para salvar en JPEG2000

                long sizeB = (long)(( layer.getDataStore().getWidth() * layer.getDataStore().getWidth() * layer.getDataStore().getBandCount())); 

                int sizeMB = (int)(sizeB / 1048576);

                if(sizeMB >= 4096) {

                        RasterToolsUtil.messageBoxInfo("file_too_big", this);

                        getNewLayerPanel().setCompress(true);

                }

                RasterFilterList rfl = layer.getRender().getFilterList();

                status.getVisualStatus(layer);

                //Carga el listado de filtros que ya estaban aplicados en la capa al abrir

                int i = 0;

                boolean filters = false;

                while (rfl.get(i) != null) {

                        DefaultListModel list = (DefaultListModel) getMainPanel().getTreeListContainer().getListModel();

                        if (rfl.get(i).isVisible() && hasThisFilter(rfl.get(i).getName())) {

                                list.addElement(PluginServices.getText(this, rfl.get(i).getName()));

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

                                        if (((ParamStruct) getFilterListener().getParamsList().get(j)).getFilterName().equals(rfl.get(i).getName())) {

                                                filters = true;

                                                initFilter(rfl.get(i));

                                                ((ParamStruct) getFilterListener().getParamsList().get(j)).setFilterParam(initFilterParam(rfl.get(i).getUIParams(rfl.get(i).getName())));

                                        }

                                }

                        }

                        i++;

                }

                // Si existen filtros ya en la capa, pondremos los RadioButton para que

                // los cambios se hagan sobre la visualizaci?n

                if (filters)

                        getNewLayerPanel().setOnlyView(true);

                // Selecciona el primer item del arbol

                TreeListContainer lc = getMainPanel().getTreeListContainer();

                lc.getTree().expandRow(1);

                lc.getTree().setSelectionInterval(2, 2);

        }

        public FilterMainPanel getMainPanel() {

                if (filterMain == null) {

                        filterMain = new FilterMainPanel(getFilterListener());

                        JPanel panel = getNewLayerPanel().getFileNamePanel();

                        panel.setBorder(BorderFactory.createTitledBorder(null, PluginServices.getText(this, "nombre_capa"), TitledBorder.DEFAULT_JUSTIFICATION, TitledBorder.DEFAULT_POSITION, null, null));

                        filterMain.getCentralPanel().add(panel, BorderLayout.SOUTH);

                }

                return filterMain;

        }

        /**

         * Volvemos todo a la normalidad cuando se cancela

         */

        public void cancel() {

                if (layer != null)

                        layer.getMapContext().invalidate();

        }

        /**

         * Cuando se aceptan los cambios en el panel ejecutaremos el aceptar del

         * listener

         */

        public void apply() {

                filterListener.accept();

        }

        /**

         * Obtener el PropertiesComponent del panel de filtros, en caso de no existir,

         * lo crea.

         * @return

         */

        public PropertiesComponent getPropertiesComponent() {

                if (propertiesComponent == null) {

                        propertiesComponent = new PropertiesComponent();

                        getMainPanel().getCentralPanel().add(propertiesComponent, BorderLayout.CENTER);

                        propertiesComponent.addStateChangedListener(getFilterListener());

                }

                return propertiesComponent;

        }

        /**

         * Asignar un nuevo componente de PropertiesComponent, eliminando el anterior

         * que hubiera.

         * @param component

         * @param name

         */

        public void setNewPropertiesComponent(PropertiesComponent component, String name) {

                getPropertiesComponent().setVisible(false);

                this.remove(propertiesComponent);

                propertiesComponent = null;

                propertiesComponent = component;

                getMainPanel().getCentralPanel().add(propertiesComponent, BorderLayout.CENTER);

                propertiesComponent.addStateChangedListener(getFilterListener());

                setTitle(name);

                this.validate();

        }

        /**

         * Asigna el t?tulo al panel

         * @param title T?tulo del panel

         */

        public void setTitle(String title) {

                getPropertiesComponent().setBorder(BorderFactory.createTitledBorder(null, PluginServices.getText(this, "filtro_de") + " " + PluginServices.getText(this, title), javax.swing.border.TitledBorder.DEFAULT_JUSTIFICATION, javax.swing.border.TitledBorder.DEFAULT_POSITION, null, null));

        }

        /**

         * Devuelve el estado de los filtros inicial

         * @return VisualStatusLayerStack

         */

        public LayerVisualStatusList getLayerVisualStatus() {

                return status;

        }

        /**

         * Actualizamos la vista previa

         */

        public void refreshPreview() {

                getPreviewBasePanel().refreshPreview();

        }

        /**

         * Devuelve el FLayer asignado al panel

         * @return

         */

        public FLyrRaster getLayer() {

                return layer;

        }

        /**

         * @return the jLabelShowFilters

         */

        public JCheckBox getCBShowFilters() {

                if (jCBShowFilters == null) {

                        jCBShowFilters = new JCheckBox();

                        jCBShowFilters.addActionListener(getFilterListener());

                        jCBShowFilters.setSelected(true);

                }

                return jCBShowFilters;

        }

        /**

         * Obtiene el nombre de la vista

         * @return

         */

        public String getViewName() {

                return viewName;

        }

        /**

         * Especificar el nombre de la nueva capa para el recuadro de texto asign?ndo

         * en cada llamada un nombre consecutivo.

         */

        public void updateNewLayerText() {

                getNewLayerPanel().updateNewLayerText();

        }

        public void process(DataSourcePreview rendering) throws ImageUnavailableException {

                if(!showPreview)

                        throw new ImageUnavailableException(RasterToolsUtil.getText(this, "panel_preview_not_available"));

                getFilterListener().drawImage((FLyrRaster)rendering.getSource());

        }

        /**

         * Obtiene el flag que informa de si se est? mostrando la previsualizaci?n o no.

         * En caso de no mostrarse se lanza una excepci?n ImageUnavailableExcepcion con el 

         * mensaje "La previsualizaci?n no est? disponible para este panel"

         * @return

         */

        public boolean isShowPreview() {

                return showPreview;

        }

        /**

         * Asigna el flag para mostrar u ocultar la preview. En caso de no mostrarse se lanza una 

         * excepci?n ImageUnavailableExcepcion con el mensaje "La previsualizaci?n no est? disponible para

         * este panel"

         * @param showPreview

         */

        public void setShowPreview(boolean showPreview) {

                this.showPreview = showPreview;

        }

}