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.multifile.panel;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import javax.swing.JOptionPane;

import org.gvsig.andami.PluginServices;

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

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.coverage.store.parameter.RasterDataParameters;

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

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

import org.gvsig.fmap.dal.coverage.util.ProviderServices;

import org.gvsig.fmap.dal.exception.CloseException;

import org.gvsig.fmap.dal.raster.spi.CoverageStoreProvider;

import org.gvsig.fmap.dal.serverexplorer.filesystem.swing.FilesystemExplorerWizardPanel;

import org.gvsig.fmap.dal.spi.DataManagerProviderServices;

import org.gvsig.fmap.dal.spi.DataStoreProvider;

import org.gvsig.fmap.dal.spi.DataStoreProviderServices;

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

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

import org.gvsig.gui.beans.swing.JFileChooser;

import org.gvsig.gui.beans.table.exceptions.NotInitializeException;

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

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

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

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

import org.gvsig.raster.tools.app.basic.raster.gui.wizard.DriverFileFilter;

import org.gvsig.raster.tools.multifile.io.MultiFileDataParameters;

import org.gvsig.raster.tools.multifile.io.MultiFileFormat;

import org.gvsig.raster.tools.multifile.io.MultiFileProvider;

import org.gvsig.tools.locator.LocatorException;

/**

 * Clase que maneja los eventos del panel BandSetupPanel. Gestiona el a?adir o

 * eliminar ficheros de la lista y contiene las acciones a realizar cuando en

 * panel registrable se pulsa aceptar, aplicar o cancelar.

 *

 * @author Nacho Brodin (brodin_ign@gva.es)

 */

public class BandSelectorListener implements ActionListener, ButtonsPanelListener {

        private BandSelectorPanel     bandSetupPanel = null;

        private JFileChooser          fileChooser    = null;

        private FLyrRaster            fLayer         = null;

        private boolean               enabled        = true;

        private ArrayList<File>       fileList       = null;

        /**

         * Constructor

         * @param bs Panel del selector de bandas

         * @param lyr Capa raster

         */

        public BandSelectorListener(BandSelectorPanel bs) {

                this.bandSetupPanel = bs;

                bs.getFileList().getJButtonAdd().addActionListener(this);

                bs.getFileList().getJButtonRemove().addActionListener(this);

                bs.getNumBandSelectorCombo().addActionListener(this);

        }

        /**

         * Comprobar si la asignacion de color es correcta para las 4 bandas. No puede

         * existir una banda con distintas interpretaciones de color. Se comprueban dos

         * casos, asignaciones en escala de grises o en RGB.

         * @param r

         * @param g

         * @param b

         * @param a

         * @return

         */

        private boolean isCorrectAssignedBand(int r, int g, int b, int a) {

                // Si es gris es correcta la asignacion

                if ((r == g) && (r == b) && (r >= 0)) {

                        // Si el alpha esta asignado a la misma banda es incorrecto

                        if (r == a)

                                return false;

                        // En caso contrario es correcto

                        return true;

                }

                // Si dos bandas coinciden, se dice que no es correcta la asignacion

                int list[] = { r, g, b, a };

                for (int i = 0; i <= 3; i++)

                        for (int j = 0; j <= 3; j++)

                                if ((i != j) && (list[i] == list[j]) && (list[i] > -1))

                                        return false;

                return true;

        }

        /**

         * Constructor

         * @param bs Panel del selector de bandas

         * @param lyr Capa raster

         */

        public void init(FLyrRaster lyr) {

                fLayer = lyr;

        }

        /**

         * Listener para la gesti?n de los botones de a?adir, eliminar fichero y

         * el combo de selecci?n de bandas.

         */

        public void actionPerformed(ActionEvent e) {

                if (e.getSource().equals(bandSetupPanel.getFileList().getJButtonAdd()))

                        addFileBand();

                if (e.getSource().equals(bandSetupPanel.getFileList().getJButtonRemove()))

                        delFileBand();

                if (e.getSource().equals(bandSetupPanel.getNumBandSelectorCombo())) {

                        String vBands = (String) bandSetupPanel.getNumBandSelectorCombo().getSelectedItem();

                        if (vBands != null) {

                                if (vBands.compareTo("3") == 0)

                                        bandSetupPanel.resetMode(3);

                                if (vBands.compareTo("2") == 0)

                                        bandSetupPanel.resetMode(2);

                                if (vBands.compareTo("1") == 0)

                                        bandSetupPanel.resetMode(1);

                        }

                }

                if (e.getSource().equals(bandSetupPanel.getSaveButton())) {

                        int rBand = bandSetupPanel.getAssignedBand(RasterDataStore.RED_BAND);

                        int gBand = bandSetupPanel.getAssignedBand(RasterDataStore.GREEN_BAND);

                        int bBand = bandSetupPanel.getAssignedBand(RasterDataStore.BLUE_BAND);

                        int aBand = bandSetupPanel.getAssignedBand(RasterDataStore.ALPHA_BAND);

                        if (!isCorrectAssignedBand(rBand, gBand, bBand, aBand)) {

                                RasterToolsUtil.messageBoxError("combinacion_no_asignable", bandSetupPanel);

                                return;

                        }

                        RasterToolsUtil.messageBoxYesOrNot("color_interpretation_continue", this);

                        RasterDataStore dataSource = fLayer.getDataStore();

                        if(dataSource == null) {

                                RasterToolsUtil.messageBoxError("error_carga_capa", bandSetupPanel);

                                return;

                        }

                        ColorInterpretation ci = dataSource.getColorInterpretation();

                        try {

                                // Combinaci?n GRAY

                                if ((rBand == gBand) && (rBand == bBand) && (rBand >= 0)) {

                                        for (int iBand = 0; iBand < bandSetupPanel.getARGBTable().getRowCount(); iBand++) {

                                                ci.setColorInterpValue(iBand, ColorInterpretation.UNDEF_BAND);

                                        }

                                        ci.setColorInterpValue(rBand, ColorInterpretation.GRAY_BAND);

                                        ci.setColorInterpValue(aBand, ColorInterpretation.ALPHA_BAND);

                                } else {

                                        // Combinaci?n RGB

                                        for (int iBand = 0; iBand < bandSetupPanel.getARGBTable().getRowCount(); iBand++)

                                                ci.setColorInterpValue(iBand, bandSetupPanel.getColorInterpretationByBand(iBand));

                                }

                                String fileName = fLayer.getDataStore().getName();

                                RasterLocator.getManager().getProviderServices().saveObjectToRmfFile(fileName, ci);

                        } catch (RmfSerializerException exc) {

                                RasterToolsUtil.messageBoxError("error_salvando_rmf", bandSetupPanel, exc);

                        } catch (NotInitializeException exc) {

                                RasterToolsUtil.messageBoxError("table_not_initialize", bandSetupPanel, exc);

                        }

                }

                if (!RasterExtension.autoRefreshView)

                        return;

                bandSetupPanel.onlyApply();

        }

        /**

         * Checks if the new file is compatible with the old one

         * @param file

         * @return

         * @throws LocatorException

         * @throws NotSupportedExtensionException

         * @throws RasterDriverException

         * @throws CloseException

         */

        private boolean checkNewFile(String file) throws LocatorException, NotSupportedExtensionException, RasterDriverException, CloseException {

                Rectangle2D extentOrigin = fLayer.getFullRasterExtent().toRectangle2D();

                ProviderServices provServ = RasterLocator.getManager().getProviderServices();

                RasterDataParameters storeParameters = provServ.createParameters(file);

                storeParameters.setSRS(fLayer.getDataStore().getProjection());

                RasterDataStore dataStore = RasterLocator.getManager().open(storeParameters);

                Extent extentNewFile = dataStore.getExtent();

                // Comprobamos que el extent y tama?o del fichero a?adido sea igual al

                // fichero original. Si no es as? no abrimos la capa y mostramos un aviso

                double widthNewFile = (extentNewFile.getMax().getX() - extentNewFile.getMin().getX());

                double heightNewFile = (extentNewFile.getMax().getY() - extentNewFile.getMin().getY());

                if ((widthNewFile - extentOrigin.getWidth()) > 1.0 || (widthNewFile - extentOrigin.getWidth()) < -1.0 || (heightNewFile - extentOrigin.getHeight()) > 1.0

                                || (heightNewFile - extentOrigin.getHeight()) < -1.0) {

                        return false;

                }

                if ((extentNewFile.getMax().getX() - extentNewFile.getMin().getX()) != extentOrigin.getWidth()

                                || (extentNewFile.getMax().getY() - extentNewFile.getMin().getY()) != extentOrigin.getHeight()) {

                        return false;

                }

                dataStore.close();

                return true;

        }

        /**

         * A?ade una banda al raster

         * @param e

         */

        private void addFileBand() {

                // String[] driverNames = null;

                // Creaci?n del dialogo para selecci?n de ficheros

                fileChooser = new JFileChooser(

                                FilesystemExplorerWizardPanel.OPEN_LAYER_FILE_CHOOSER_ID,

                                JFileChooser.getLastPath(FilesystemExplorerWizardPanel.OPEN_LAYER_FILE_CHOOSER_ID, null));

                fileChooser.setMultiSelectionEnabled(true);

                fileChooser.setAcceptAllFileFilterUsed(false);

                fileChooser.addChoosableFileFilter(new DriverFileFilter());

                int result = fileChooser.showOpenDialog(bandSetupPanel);

                if (result == JFileChooser.APPROVE_OPTION) {

                        RasterDataStore dataStore = fLayer.getDataStore();

                        File[] files = fileChooser.getSelectedFiles();

                        JFileChooser.setLastPath(FilesystemExplorerWizardPanel.OPEN_LAYER_FILE_CHOOSER_ID, files[0]);

                        fileList = new ArrayList<File>();

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

                                //Checks that the file does not exist

                                String[] uris = dataStore.getURIByProvider();

                                boolean exists = false;

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

                                        if (uris[j].endsWith(files[i].getName())) {

                                                RasterToolsUtil.messageBoxError( PluginServices.getText( this, "fichero_existe") + ": " + files[i].getAbsolutePath(), bandSetupPanel);

                                                exists = true;

                                                break;

                                        }

                                }

                                if(!exists)

                                        fileList.add(files[i]);

                        }

                        for (int i = fileList.size() - 1; i >= 0; i--) {

                                //Checks extents

                                try {

                                        if(!checkNewFile(fileList.get(i).getAbsolutePath())) {

                                                JOptionPane.showMessageDialog(null, PluginServices.getText(this, "extents_no_coincidentes") +  " " + files[i].getAbsolutePath(), "", JOptionPane.ERROR_MESSAGE);

                                                fileList.remove(i);

                                        }

                                } catch (Exception e) {

                                        e.printStackTrace();

                                }

                        }

                        if(dataStore.isMultiFile()) {

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

                                        try {

                                                dataStore.addFile(fileList.get(i).getAbsolutePath());

                                        } catch (InvalidSourceException e) {

                                                RasterToolsUtil.messageBoxError("addband_error", bandSetupPanel, e);

                                        }

                                }

                                dataStore.setProvider(dataStore.getProvider());

                                //It shows the files and bands in the panel

                                try {

                                        bandSetupPanel.addFiles(dataStore);

                                } catch (NotInitializeException e) {

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

                                }

                        } else {

                                //New layer name

                                WindowInfo wi = PluginServices.getMDIManager().getActiveWindow().getWindowInfo();

                                LayerNameDialog dialog = new LayerNameDialog(new Point2D.Double(wi.getX(), wi.getY()), 300, 80, this);

                                RasterToolsUtil.addWindow(dialog);

                        }

                }

        }

        /**

         * Catchs the events from LayerNameDialog to get the name of the new layer

         */

        public void actionButtonPressed(ButtonsPanelEvent e) {

                String layerName = (String)e.getSource();

                RasterDataStore dataStore = fLayer.getDataStore();

                RasterDataParameters paramFirstFile = (RasterDataParameters)dataStore.getParameters();//(RasterDataParameters)((RasterProvider)dataStore.getProvider()).getDataParameters();

                ProviderServices provServ = RasterLocator.getManager().getProviderServices();

                DataManagerProviderServices dataManager = (DataManagerProviderServices)DALLocator.getDataManager();

                try {

                        //It creates the new Multifile provider

                        DataServerExplorerParameters explParams = dataManager.createServerExplorerParameters(MultiFileProvider.NAME);

                        DataServerExplorer explorer = dataManager.openServerExplorer(MultiFileProvider.NAME, explParams);

                        MultiFileDataParameters newParamsMultifile = (MultiFileDataParameters)dataManager.createStoreParameters(explorer.getProviderName());

                        String path = paramFirstFile.getURI().substring(0, paramFirstFile.getURI().lastIndexOf(File.separator) + 1);

                        String fileURI = path + layerName + ".mff";

                        String rmfURI = path + layerName + ".rmf";

                        int counter = 0;

                        while(new File(fileURI).exists() || new File(rmfURI).exists()) {

                                fileURI = path + layerName + "_" + counter + ".mff";

                                rmfURI = path + layerName + "_" + counter + ".rmf";

                                counter ++;

                        }

                        if(counter > 0)

                                layerName += "_" + counter;

                        newParamsMultifile.setURI(fileURI);

                        newParamsMultifile.addProvider(dataStore);

                        DataStoreProvider provMultifile = dataManager.createProvider((DataStoreProviderServices)dataStore, newParamsMultifile);

                        //And now it creates and adds the new ones

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

                                ArrayList<RasterDataParameters> storeParametersList = provServ.createParametersList(fileList.get(i).getAbsolutePath());

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

                                        DataStoreProvider newFileProv = dataManager.createProvider((DataStoreProviderServices)dataStore, storeParametersList.get(j));

                                        newParamsMultifile.addProviderNotTiled(newFileProv);

                                }

                        }

                        ((DefaultFLyrRaster)fLayer).setName(layerName);

                        //Assigns the MultifileProvider to the store

                        dataStore.setProvider((CoverageStoreProvider)provMultifile);

                        //It shows the files and bands in the panel

                        try {

                                bandSetupPanel.addFiles(dataStore);

                        } catch (NotInitializeException ex) {

                                RasterToolsUtil.messageBoxError("table_not_initialize", this, ex);

                        }

                        ArrayList<File> uriList = new ArrayList<File>();

                        uriList.add(new File(paramFirstFile.getURI()));

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

                                uriList.add(fileList.get(i));

                        }

                        saveMultiFileLayer(layerName, paramFirstFile.getURI(), uriList);

                } catch (Exception exc) {

                        RasterToolsUtil.messageBoxError("addband_error", bandSetupPanel, exc);

                }

        }

        /**

         * Saves the new layer in disk

         * @param layerName

         * @param file

         * @param uriList

         * @throws IOException

         */

        private String saveMultiFileLayer(String layerName, String file, ArrayList<File> uriList) throws IOException {

                String path = file.substring(0, file.lastIndexOf(File.separator) + 1);

                path = path + layerName + ".mff";

                /*File filePath = new File(path);

                if(filePath.exists()) {

                        RasterToolsUtil.messageBoxInfo("file_exists_rename", bandSetupPanel);

                        filePath.renameTo(new File(path + "~"));

                }*/

                MultiFileFormat format = new MultiFileFormat();

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

                        format.addFile(uriList.get(i));

                }

                format.setName(layerName);

                format.write(path);

                return path;

        }

        /**

         * Elimina una banda del raster. Si queda solo un fichero o no se ha

         * seleccionado ninguna banda no hace nada.

         *

         * @param e

         */

        private void delFileBand() {

                Object[] objects = bandSetupPanel.getFileList().getJList().getSelectedValues();

                RasterDataStore dataStore = fLayer.getDataStore();

                for (int i = objects.length - 1; i >= 0; i--) {

                        if (bandSetupPanel.getFileList().getNFiles() > 1) {

                                String pathName = objects[i].toString();

                                dataStore.removeFile(pathName);

                                String file = pathName.substring(pathName.lastIndexOf(File.separator) + 1);

                                file = file.substring(file.lastIndexOf("\\") + 1);

                                bandSetupPanel.removeFile(file);

                        }

                }

                setNewBandsPositionInRendering();

        }

        /**

         * Acciones a ejecutar cuando se aplica

         */

        public void apply() {

                if (enabled)

                        setNewBandsPositionInRendering();

        }

        /**

         * Asigna la posici?n de las bandas en el rederizado basandose en la selecci?n

         * hecho en la tabla de bandas.

         */

        public void setNewBandsPositionInRendering() {

                if (fLayer != null && fLayer.getRender() != null) {

                        fLayer.getRender().setRenderBands(new int[]{bandSetupPanel.getAssignedBand(RasterDataStore.RED_BAND),

                                        bandSetupPanel.getAssignedBand(RasterDataStore.GREEN_BAND),

                                        bandSetupPanel.getAssignedBand(RasterDataStore.BLUE_BAND)});

                        int alphaBand = bandSetupPanel.getAssignedBand(RasterDataStore.ALPHA_BAND);

                        // Ultima transparencia aplicada en el renderizador

                        Transparency gt = fLayer.getRender().getLastTransparency();

                        if(gt != null)

                                gt.setTransparencyBand(alphaBand);

                        // Transparencia del dataset

                        /*if(fLayer.getDataStore() != null) {

                                Transparency t = fLayer.getDataStore().getTransparency();

                                if(t != null)

                                        t.setTransparencyBand(alphaBand);

                        }*/

                        fLayer.getMapContext().invalidate();

                }

        }

        /**

         * Activa o desactiva la acci?n del panel

         * @param enabled true para activa y false para desactivar.

         */

        public void setEnabledPanelAction(boolean enabled) {

                this.enabled = enabled;

        }

}