| gvSIG desktop 1

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

 *

 * Copyright (C) 2007 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.saveas;

import java.awt.Component;

import java.beans.PropertyChangeEvent;

import java.beans.PropertyChangeListener;

import java.io.File;

import java.util.ArrayList;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

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

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

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

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

import org.gvsig.raster.buffer.BufferInterpolation;

import org.gvsig.raster.dataset.GeoRasterWriter;

import org.gvsig.raster.dataset.IRasterDataSource;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.Params;

import org.gvsig.raster.dataset.RasterDriverException;

import org.gvsig.raster.hierarchy.IRasterDataset;

import org.gvsig.raster.hierarchy.IRasterGeoOperations;

import org.gvsig.raster.hierarchy.IRasterOperations;

import org.gvsig.rastertools.RasterModule;

import org.gvsig.rastertools.RasterToolsUtil;

import org.gvsig.rastertools.cutting.CuttingProcess;

import org.gvsig.rastertools.cutting.WriterBufferServer;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

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

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

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

import com.iver.cit.gvsig.project.documents.view.toc.AbstractTocContextMenuAction;

import com.iver.cit.gvsig.project.documents.view.toc.ITocItem;

/**

 * <p>

 * Entrada del TOC que corresponde con la opci?n "Salvar Como" de raster. Esta se apoya

 * en el proceso de recorte de raster para salvar datos.

 * </p>

 * <p>

 * Cuando se abre el dialogo de "Salvar como" se cargan las extensiones soportadas en el

 * selector. Cada tipo de datos de la fuente soporta unas extensiones de escritura u otras.

 * Por ejemplo, si la capa de entrada  es FLOAT no podremos escribir a JPG2000 porque no

 * lo soporta, tendriamos que convertila primero a RGB.

 * </p>

 * <p>

 * Cambiando el tipo de extensi?n en el selector cambian el panel de propiedades asociado.

 * El fichero de salidad se salvar? con las propiedades ajustadas.

 * </P>

 * @version 30/05/2007

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

 *

 */

public class SaveAsTocMenuEntry extends AbstractTocContextMenuAction implements PropertyChangeListener {

        private JFileChooser chooser = null;

        private PropertiesComponent panelProperty = null;

        /*

         * (non-Javadoc)

         * @see com.iver.cit.gvsig.project.documents.contextMenu.AbstractContextMenuAction#getGroup()

         */

        public String getGroup() {

                return "Save";

        }

        /*

         * (non-Javadoc)

         * @see com.iver.cit.gvsig.project.documents.contextMenu.AbstractContextMenuAction#getGroupOrder()

         */

        public int getGroupOrder() {

                return 50;

        }

        /*

         * (non-Javadoc)

         * @see com.iver.cit.gvsig.project.documents.contextMenu.AbstractContextMenuAction#getOrder()

         */

        public int getOrder() {

                return 1;

        }

        /*

         * (non-Javadoc)

         * @see com.iver.cit.gvsig.project.documents.IContextMenuAction#getText()

         */

        public String getText() {

                return PluginServices.getText(this, "saveas");

        }

        /*

         * (non-Javadoc)

         * @see com.iver.cit.gvsig.project.documents.view.toc.AbstractTocContextMenuAction#isEnabled(com.iver.cit.gvsig.project.documents.view.toc.ITocItem, com.iver.cit.gvsig.fmap.layers.FLayer[])

         */

        public boolean isEnabled(ITocItem item, FLayer[] selectedItems) {

                return selectedItems.length == 1;

        }

        /*

         * (non-Javadoc)

         * @see com.iver.cit.gvsig.project.documents.view.toc.AbstractTocContextMenuAction#isVisible(com.iver.cit.gvsig.project.documents.view.toc.ITocItem, com.iver.cit.gvsig.fmap.layers.FLayer[])

         */

        public boolean isVisible(ITocItem item, FLayer[] selectedItems) {

                if (isTocItemBranch(item))

                        return( getNodeLayer(item) instanceof FLyrRasterSE && 

                                    ((FLyrRasterSE)getNodeLayer(item)).isActionEnabled(IRasterLayerActions.SAVEAS));

                return false;

        }

        /*

         * (non-Javadoc)

         * @see com.iver.cit.gvsig.project.documents.view.toc.AbstractTocContextMenuAction#execute(com.iver.cit.gvsig.project.documents.view.toc.ITocItem, com.iver.cit.gvsig.fmap.layers.FLayer[])

         */

        public void execute(ITocItem item, FLayer[] selectedItems) {

                FLayer fLayer = null;

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

                if (selectedItems.length != 1)

                        return;

                fLayer = selectedItems[0];

                if (!(fLayer instanceof IRasterOperations))

                        return;

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

                chooser.addPropertyChangeListener(this);

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

                // Cargamos las extensiones en la lista

                ArrayList extList = new ArrayList();

                try {

                        extList = GeoRasterWriter.getExtensionsSupported(((IRasterGeoOperations) fLayer).getDataType()[0], ((IRasterGeoOperations) fLayer).getBandCount());

                } catch (RasterDriverException e2) {

                        RasterToolsUtil.messageBoxError("extensiones_soportadas_error", chooser);

                        NotificationManager.addError("Error al obtener la lista de extensiones soportadas. No se puede exportar la capa", e2);

                }

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

                        chooser.addChoosableFileFilter(new WriterFilter((String) extList.get(i)));

                // Cargamos el panel de propiedades en el selector

                panelProperty = loadPanelProperties((String) extList.get(extList.size() - 1));

                chooser.setAccessory(panelProperty);

                if (w instanceof View) {

                        if (chooser.showOpenDialog(((View) w).getComponent(0)) == JFileChooser.APPROVE_OPTION) {

                                // Creaci?n de par?metros

                                String tit = PluginServices.getMDIManager().getWindowInfo(w).getTitle();

                                WriterBufferServer dataWriter = new WriterBufferServer();

                                int[] dValues = new int[] { 0, (int) ((IRasterOperations) fLayer).getPxHeight(), (int) ((IRasterOperations) fLayer).getPxWidth(), 0 };

                                IRasterDataSource dataset = (IRasterDataSource)((IRasterDataset) fLayer).getDataSource();

                                int[] drawableBands = new int[((IRasterOperations) fLayer).getBandCount()];

                                for (int i = 0; i < ((IRasterOperations) fLayer).getBandCount(); i++)

                                        drawableBands[i] = i;

                                RasterModule.setPath(chooser.getCurrentDirectory().toString());

                                String file = chooser.getSelectedFile().toString() + chooser.getFileFilter().getDescription();

                                if (!RasterToolsUtil.canWrite(chooser.getCurrentDirectory().toString(), this))

                                        return;

                                Params params = null;

                                try {

                                        params = GeoRasterWriter.getWriter(file).getParams();

                                } catch (NotSupportedExtensionException e1) {

                                        RasterToolsUtil.messageBoxError("no_driver_escritura", this);

                                        NotificationManager.addError(PluginServices.getText(this, "no_driver_escritura"), e1);

                                } catch (RasterDriverException e1) {

                                        RasterToolsUtil.messageBoxError("no_driver_escritura", this);

                                        NotificationManager.addError(PluginServices.getText(this, "no_driver_escritura"), e1);

                                }

                                // Lanzamiento del proceso de guardado

                                CuttingProcess cuttingProcess = new CuttingProcess(false, tit, dValues, file, dataWriter, dataset, dataset.getAffineTransform(), drawableBands, false, BufferInterpolation.INTERPOLATION_NearestNeighbour);

                                if (params != null)

                                        RasterToolsUtil.loadWriterParamsFromPropertiesPanel(panelProperty, params);

                                cuttingProcess.setParams(params);

                                cuttingProcess.setResolution((int) ((IRasterOperations) fLayer).getPxWidth(), (int) ((IRasterOperations) fLayer).getPxHeight());

                                IncrementableTask incrementableTask = new IncrementableTask(dataWriter);

                                cuttingProcess.setIncrementableTask(incrementableTask);

                                incrementableTask.showWindow();

                                cuttingProcess.start();

                                incrementableTask.start();

                        }

                }

        }

        /**

         * Evento que se produce al cambiar el driver de escritura. Esto

         * sustituye el panel properties por el del nuevo driver seleccionado

         */

        public void propertyChange(PropertyChangeEvent evt) {

                if(evt.getNewValue() instanceof WriterFilter) {

                        String ext = ((WriterFilter)evt.getNewValue()).getDescription();

                        panelProperty = loadPanelProperties(ext);

                        chooser.setAccessory(panelProperty);

                        chooser.revalidate();

                }

        }

        /**

         * Obtiene el objeto PropertiesComponent para una extensi?n concreta de fichero

         * sobre el que se va a escribir.

         * @param file Fichero raster sobre el que se escribir?

         * @return PropertiesComponent

         */

        private PropertiesComponent loadPanelProperties(String file) {

                PropertiesComponent panelProperty = new PropertiesComponent();

                Params params = null;

                try {

                        params = GeoRasterWriter.getWriter(file).getParams();

                } catch (NotSupportedExtensionException e1) {

                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(), PluginServices.getText(this, "no_driver_escritura"));

                        return null;

                } catch (RasterDriverException e1) {

                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(), PluginServices.getText(this, "no_driver_escritura"));

                        return null;

                }

                RasterToolsUtil.loadPropertiesFromWriterParams(panelProperty, params);

                return panelProperty;

        }

        /**

         * @author Nacho Brodin (nachobrodin@gmail.com) 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;

                        }

        }

}