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

import java.awt.event.ActionEvent;

import java.awt.geom.Point2D;

import java.io.File;

import java.net.URI;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.io.FilenameUtils;

import org.gvsig.andami.PluginServices;

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

import org.gvsig.fmap.dal.DALLocator;

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

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

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.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.swing.JFileChooser;

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

import org.gvsig.i18n.Messages;

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

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

import org.gvsig.raster.mainplugin.RasterMainPluginUtils;

import org.gvsig.raster.mainplugin.config.Configuration;

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

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

import org.gvsig.raster.swing.RasterSwingLibrary;

import org.gvsig.raster.swing.basepanel.ButtonsPanelEvent;

/**

 * 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 BandSelectorPropertiesListener extends AbstractBandSelectorListener {

        private JFileChooser          fileChooser    = null;

        private FLyrRaster            fLayer         = null;

        private List<File>            fileList       = null;

        /**

         * Constructor

         * @param bs Panel del selector de bandas

         */

        public BandSelectorPropertiesListener(BandSelectorPanel bs) {

                super(bs);

        }

        /**

         * Constructor

         * @param lyr Capa raster

         */

        public void init(FLyrRaster lyr) {

                fLayer = lyr;

        }

        public RasterDataStore getResult() {

                return fLayer.getDataStore();

        }

        /**

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

                super.actionPerformed(e);

                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 numBandToRed = bandSetupPanel.getColorInterpretationByColorBandBand(RasterDataStore.RED_BAND);

                        int numBandToGreen = bandSetupPanel.getColorInterpretationByColorBandBand(RasterDataStore.GREEN_BAND);

                        int numBandToBlue = bandSetupPanel.getColorInterpretationByColorBandBand(RasterDataStore.BLUE_BAND);

                        int numBandToAlpha = bandSetupPanel.getColorInterpretationByColorBandBand(RasterDataStore.ALPHA_BAND);

                        if (!isCorrectAssignedBand(numBandToRed, numBandToGreen, numBandToBlue, numBandToAlpha)) {

                                RasterSwingLibrary.messageBoxError(Messages.getText("combinacion_no_asignable"), bandSetupPanel);

                                return;

                        }

                        RasterSwingLibrary.messageBoxYesOrNot(Messages.getText("color_interpretation_continue"), this);

                        RasterDataStore dataSource = fLayer.getDataStore();

                        if(dataSource == null) {

                                RasterSwingLibrary.messageBoxError(Messages.getText("error_carga_capa"), bandSetupPanel);

                                return;

                        }

                        //ColorInterpretation ci = dataSource.getColorInterpretation();

                        try {

                                String[] bands = new String[bandSetupPanel.getARGBTable().getRowCount()];

                                ColorInterpretation ci = RasterLocator.getManager().getDataStructFactory().createColorInterpretation(bands);

                                // Combinaci?n GRAY

                                if ((numBandToRed == numBandToGreen) && (numBandToRed == numBandToBlue) && (numBandToRed >= 0)) {

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

                                                ci.setColorInterpValue(iBand, ColorInterpretation.UNDEF_BAND);

                                        }

                                        ci.setColorInterpValue(0, ColorInterpretation.GRAY_BAND);

                                        ci.setColorInterpValue(numBandToAlpha, 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();

                                dataSource.setColorInterpretation(ci);

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

                        } catch (RmfSerializerException exc) {

                                RasterSwingLibrary.messageBoxError(Messages.getText("error_salvando_rmf"), bandSetupPanel, exc);

                        } catch (NotInitializeException exc) {

                                RasterSwingLibrary.messageBoxError(Messages.getText("table_not_initialize"), bandSetupPanel, exc);

                        }

                }

                boolean autoRefreshView = Configuration.getValue("general_auto_preview", Boolean.TRUE).booleanValue();

                if (!autoRefreshView)

                        return;

                bandSetupPanel.onlyApply();

        }

        /**

         * A?ade una banda al raster

         */

        public void addFileBand() {

                fileChooser = createJFileChooser();

                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

                                URI[] uris = dataStore.getURIByProvider();

                                boolean exists = false;

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

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

                                                RasterSwingLibrary.messageBoxError( Messages.getText("fichero_existe") + ": " + files[i].getAbsolutePath(), bandSetupPanel);

                                                exists = true;

                                                break;

                                        }

                                }

                                if(!exists)

                                        fileList.add(files[i]);

                        }

                        if(!checkStoresCompatibility(fLayer.getDataStore(), fileList))

                                return;

                        if(dataStore.isMultiFile()) {

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

                                        try {

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

                                        } catch (InvalidSourceException e) {

                                                RasterSwingLibrary.messageBoxError(Messages.getText("addband_error"), bandSetupPanel, e);

                                        }

                                }

                                dataStore.setProvider(dataStore.getProvider());

                                //It shows the files and bands in the panel

                                try {

                                        bandSetupPanel.addFiles(dataStore);

                                } catch (NotInitializeException e) {

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

                                RasterMainPluginUtils.addWindow(dialog);

                        }

                }

        }

        /**

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

         * seleccionado ninguna banda no hace nada.

         *

         */

        public 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(new File(pathName));

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

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

                                bandSetupPanel.removeFile(file);

                        }

                }

                setNewBandsPositionInRendering();

        }

        /**

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

                File firstFile = new File(paramFirstFile.getURI());

        String path = FilenameUtils.getFullPath(firstFile.getAbsolutePath());

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

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

                try {

                        MultiFileProvider provMultifile = createMultiFileProvider(layerName, path);

                        MultiFileDataParameters newParamsMultifile = (MultiFileDataParameters)provMultifile.getDataParameters();

                        newParamsMultifile.addProvider(dataStore);

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

                                RasterSwingLibrary.messageBoxError(Messages.getText("table_not_initialize"), this, ex);

                        }

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

                        uriList.add(firstFile);

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

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

                        }

                        saveMultiFileLayer(layerName, path, uriList);

                } catch (Exception exc) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("addband_error"), bandSetupPanel, exc);

                }

        }

        /**

         * Acciones a ejecutar cuando se aplica

         */

        public void apply() {

                if (enabled)

                        setNewBandsPositionInRendering();

        }

        public void setNewBandsPositionInRendering() {

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

                        fLayer.getRender().setRenderColorInterpretation(bandSetupPanel.getSelectedColorInterpretation());

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

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

                        if(gt != null)

                                gt.setTransparencyBand(alphaBand);

                        fLayer.getMapContext().invalidate();

                }

        }

}