Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.multifile.app.panel;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import javax.swing.JOptionPane;

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

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

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

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

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

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

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

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

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

import org.gvsig.i18n.Messages;

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

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

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

import org.gvsig.raster.swing.RasterSwingLibrary;

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

import org.gvsig.tools.locator.LocatorException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Listener for <code>BandSelectorPanel</code>. This interface is implemented by two listeners

 * , the new layer listener and the properties panel listener.

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

 */

public abstract class AbstractBandSelectorListener implements ActionListener, ButtonsPanelListener {

        protected Logger                log              = LoggerFactory.getLogger(AbstractBandSelectorListener.class);

        protected BandSelectorPanel     bandSetupPanel   = null;

        protected boolean               enabled          = true;

        protected String                file             = null;

        protected String                folder           = null;

        /**

         * Constructor

         * @param bs Panel del selector de bandas

         * @param lyr Capa raster

         */

        public AbstractBandSelectorListener(BandSelectorPanel bs) {

                this.bandSetupPanel = bs;

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

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

                bs.getNumBandSelectorCombo().addActionListener(this);

        }

        public void setDestination(String file, String folder) {

                this.file = file;

                this.folder = folder;

        }

        /**

         * Enables or disables the panel action

         */

        public void setEnabledPanelAction(boolean enabled) {

                this.enabled = enabled;

        }

        /**

         * Sets the band position when these are rendered. The position is read from

         * the band table

         */

        public abstract void setNewBandsPositionInRendering();

        public abstract RasterDataStore getResult();

        /**

         * Actions when is applied

         */

        public abstract void apply();

        public abstract void addFileBand();

        public abstract void delFileBand();

        /**

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

        }

        protected JFileChooser createJFileChooser() {

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

                return fileChooser;

        }

        protected boolean checkStoresCompatibility(RasterDataStore mainRasterStore, List<File> fileList) {

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

                        //Checks extents

                        try {

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

                                        JOptionPane.showMessageDialog(null, Messages.getText("extents_no_coincidentes") +  " " + fileList.get(i).getAbsolutePath(), "", JOptionPane.ERROR_MESSAGE);

                                        fileList.remove(i);

                                }

                        } catch (Exception e) {

                                log.debug("Problems check the bounding boxes", e);

                        }

                }

                if(mainRasterStore.isTiled()) {

                        if(!RasterSwingLibrary.messageBoxYesOrNot(Messages.getText("store_tiled"), this)) {

                                return false;

                        }

                }

                return true;

        }

        /**

         * Creates a multifile data provider

         * @param layerName

         * @param path

         * @return

         */

        protected MultiFileProvider createMultiFileProvider(String layerName, String path) {

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

                        if(!path.endsWith(File.separator))

                                path = path + File.separator;

                        File fileURI = new File(path + layerName + ".mff");

                        File rmfURI = new File(path + layerName + ".rmf");

                        int counter = 0;

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

                                fileURI = new File(path + layerName + "_" + counter + ".mff");

                                rmfURI = new File(path + layerName + "_" + counter + ".rmf");

                                counter ++;

                        }

                        if(counter > 0)

                                layerName += "_" + counter;

                        newParamsMultifile.setURI(fileURI.toURI());

                        return (MultiFileProvider)dataManager.createProvider(null, newParamsMultifile);

                } catch (Exception exc) {

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

                }

                return null;

        }

        /**

         * 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

         */

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

        }

        /**

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

         * @param file

         * @return

         * @throws LocatorException

         * @throws NotSupportedExtensionException

         * @throws RasterDriverException

         * @throws CloseException

         */

        protected boolean checkNewFile(RasterDataStore oldStore, String file) throws LocatorException, NotSupportedExtensionException, RasterDriverException, CloseException {

                Rectangle2D extentOrigin = oldStore.getExtent().toRectangle2D();

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

                RasterDataParameters storeParameters = provServ.createParameters(file);

                storeParameters.setSRS(oldStore.getProjection());

                RasterDataStore dataStore = RasterLocator.getManager().getProviderServices().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;

        }

        /**

         * Saves the new layer in disk

         * @param layerName

         * @param file

         * @param uriList

         * @throws IOException

         */

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

        }

}