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.mosaic.app.gui;

import java.awt.Component;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.geom.Point2D;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import javax.swing.event.TableModelEvent;

import javax.swing.event.TableModelListener;

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

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

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemServerExplorer;

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

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.raster.fmap.layers.DefaultFLyrRaster;

import org.gvsig.raster.mosaic.io.MosaicDataParameters;

import org.gvsig.raster.mosaic.io.MosaicProvider;

import org.gvsig.raster.mosaic.io.MosaicRasterFormat;

import org.gvsig.raster.mosaic.io.MosaicRasterFormatSerializer;

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

/**

 * Class for listen to events in mosaic GUI

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

 */

public class MosaicPropertiesListener implements TableModelListener, ActionListener, ButtonsPanelListener {

        private MosaicPropertiesPanel                panel                  = null;

        private String                               layerName              = null;

        public MosaicPropertiesListener(MosaicPropertiesPanel panel) {

                this.panel = panel;

                panel.getAddButton().addActionListener(this);

                panel.getCreateButton().addActionListener(this);

                panel.getSeeListCheck().addActionListener(this);

                panel.getDeleteButton().addActionListener(this);

                panel.getModel().addTableModelListener(this);

        }

        /**

         * Events in the table

         */

        public void tableChanged(TableModelEvent e) {

                if(e.getColumn() == 0) {

                        MosaicDataParameters params = (MosaicDataParameters)panel.getfLayer().getDataStore().getParameters();

                        Object[][] data = panel.getModel().getDataWithoutPath();

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

                                boolean b = ((Boolean)data[i][0]).booleanValue();

                                params.getProviders().get(i).getDataParameters().setVisible(b);

                        }

                }

        }

        public void accept() {

                apply();

        }

        @SuppressWarnings("deprecation")

        public void apply() {

                try {

                        RasterDataStore dataStore = panel.getfLayer().getDataStore();

                        RasterDataParameters paramFirstFile = (RasterDataParameters)dataStore.getParameters();

                        boolean create = !(paramFirstFile instanceof MosaicDataParameters);

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

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

                        String path = basePath + layerName + ".mrf";

                        MosaicDataParameters newParamsMosaic = null;

                        //Gets the mosaic provider parameters if we are creating a new mosaic

                        if(create) {

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

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

                                newParamsMosaic = (MosaicDataParameters)dataManager.createStoreParameters(explorer.getProviderName());

                                newParamsMosaic.setURI(path);

                                //Temporal hasta que est? el cuadro de configuraci?n. Se pone el tama?o de pixel de la imagen maestra

                                newParamsMosaic.setPixelSize(dataStore.getCellSize());

                        } else

                                newParamsMosaic = (MosaicDataParameters)paramFirstFile;

                        Object[][] data = panel.getModel().getDataWithPath();

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

                        DataServerExplorerParameters param = (DataServerExplorerParameters) dataManager.createServerExplorerParameters(FilesystemServerExplorer.NAME);

                        FilesystemServerExplorer fsexplorer = (FilesystemServerExplorer) dataManager.createServerExplorer(param);

                        //Iterates all files and builds a new provider for each one

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

                                String file = (String)data[i][1];

                                uriList.add(new File(file));

                                if(!newParamsMosaic.existsProvider(file)) {

                                        File f = new File(file);

                                        fsexplorer.setCurrentPath(new File(basePath));

                                        ArrayList<String> provName = fsexplorer.getProviderNameList(f);

                                        if(provName.size() > 0) {

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

                                                        DataStoreParameters newparams = fsexplorer.createStoreParameters(f, provName.get(j));

                                                        ((FilesystemStoreParameters)newparams).setFile(f); 

                                                        DataStoreProvider prov = dataManager.createProvider((DataStoreProviderServices)dataStore, newparams);

                                                        if(newParamsMosaic.addNotTiledProvider(prov))

                                                                break;

                                                }

                                        }

                                }

                        }

                        if(create) {

                                MosaicProvider provMosaic = (MosaicProvider)dataManager.createProvider((DataStoreProviderServices)dataStore, newParamsMosaic);

                                ((DefaultFLyrRaster)panel.getfLayer()).setName(layerName);

                                //Assigns the MultifileProvider to the store

                                dataStore.setProvider(provMosaic);

                        }

                        //((DefaultFLyrRaster)panel.getfLayer()).getRender().setLastTransparency(null);

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

                        panel.setCreateStatus();

                } catch (Exception exc) {

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

                }

        }

        /**

         * Saves the new layer in disk

         * @param layerName

         * @param file

         * @param uriList

         * @throws IOException

         */

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

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

                path = path + layerName + ".mrf";

                MosaicRasterFormat format = new MosaicRasterFormat();

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

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

                }

                format.setName(layerName);

                MosaicRasterFormatSerializer.write(format, path);

                return path;

        }

        public void cancel() {

        }

        public void selected() {

        }

        /*

         * (non-Javadoc)

         * @see java.awt.event.ActionListener#actionPerformed(java.awt.event.ActionEvent)

         */

        public void actionPerformed(ActionEvent e) {

                if(e.getSource() == panel.getCreateButton()) {

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

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

                        RasterToolsUtil.addWindow(dialog);

                        return;

                }

                if(e.getSource() == panel.getAddButton()) {

                        if(panel.getAddFilesRadio().isSelected()) {

                                selectFiles();

                        } else {

                                selectDirectory();

                        }

                }

                if(e.getSource() == panel.getDeleteButton()) {

                        int[] index = panel.getFileList().getSelectedRows();

                        panel.getModel().delete(index);

                }

                if(e.getSource() == panel.getSeeListCheck()) {

                        panel.getModel().setSeeFlag(panel.getSeeListCheck().isSelected());

                        panel.getFileList().repaint();

                }

                panel.setCreateStatus();

        }

        /**

         * Selects a list of files

         */

        private void selectFiles() {

                JFileChooser chooser = new JFileChooser("DATA_INPUT_LISTENER", JFileChooser.getLastPath("DATA_INPUT_LISTENER", null));

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

                chooser.setAcceptAllFileFilterUsed(true);

                chooser.setMultiSelectionEnabled(true);

                int returnVal = chooser.showOpenDialog((Component)panel);

                if (returnVal == JFileChooser.APPROVE_OPTION) {

                        File[] fileList = chooser.getSelectedFiles();

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

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

                                if(!existsFile(fileList[i]))

                                        fileArray.add(fileList[i]);

                        }

                        addData(fileArray);

                }

        }

        /**

         * Returns true if exists the file in the table

         * @param file

         * @return

         */

        private boolean existsFile(File file) {

                for (int i = 0; i < panel.getModel().getRowCount(); i++) {

                        String path = (String)panel.getModel().getDataWithPath()[i][1];

                        if(path.compareTo(file.getAbsolutePath()) == 0)

                                return true;

                }

                return false;

        }

        /**

         * Selects the files of a directory

         */

        private void selectDirectory() {

                JFileChooser chooser = new JFileChooser("DATA_INPUT_LISTENER", JFileChooser.getLastPath("DATA_INPUT_LISTENER", null));

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

                chooser.setAcceptAllFileFilterUsed(true);

                chooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);

                int returnVal = chooser.showOpenDialog((Component)panel);

                if (returnVal == JFileChooser.APPROVE_OPTION) {

                        File dir = chooser.getSelectedFile();

                        if(dir.isDirectory()) {

                                String[] files = dir.list();

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

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

                                        File f = new File(dir + File.separator + files[j]);

                                        if(!existsFile(f)) {

                                                fileArray.add(f);

                                        }

                                }

                                addData(fileArray);

                        }

                }

        }

        /**

         * Adds data to the array of data

         * @param fileList

         * @param data

         * @return

         */

        public void addData(ArrayList<File> fileArray) {

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

                        if(fileArray.get(i).isDirectory())

                                fileArray.remove(i);

                }

                Object[][] dataOld = panel.getModel().getDataWithPath();

                Object[][] dataWithoutOld = panel.getModel().getDataWithoutPath();

                Object[][] newdata = null;

                Object[][] newdataWithout = null;

                if(dataOld == null) {

                        newdata = new Object[fileArray.size()][2];

                        newdataWithout = new Object[fileArray.size()][2];

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

                                if(!fileArray.get(i).isDirectory()) {

                                        newdata[i][0] = newdataWithout[i][0] = new Boolean(true);

                                        newdata[i][1] = fileArray.get(i).getAbsolutePath();

                                        newdataWithout[i][1] = fileArray.get(i).getName();

                                }

                        }

                } else {

                        newdata = new Object[dataOld.length + fileArray.size()][2];

                        newdataWithout = new Object[dataWithoutOld.length + fileArray.size()][2];

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

                                newdata[i][0] = newdataWithout[i][0] = dataOld[i][0];

                                newdata[i][1] = dataOld[i][1];

                                newdataWithout[i][1] = dataWithoutOld[i][1];

                        }

                        for (int i = dataOld.length; i < dataOld.length + fileArray.size(); i++) {

                                newdata[i][0] = newdataWithout[i][0] = new Boolean(true);

                                if(!fileArray.get(i - dataOld.length).isDirectory()) {

                                        newdata[i][1] = fileArray.get(i - dataOld.length).getAbsolutePath();

                                        newdataWithout[i][1] = fileArray.get(i - dataOld.length).getName();

                                }

                        }

                }

                panel.getModel().setData(newdata, newdataWithout);

                panel.getFileList().repaint();

        }

        public void actionButtonPressed(ButtonsPanelEvent e) {

                //After introduce the name of the new layer

                layerName = (String)e.getSource();

                panel.getModel().clearModel();

                if(panel.getfLayer() != null) {

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

                        f.add(new File(panel.getfLayer().getDataStore().getName()));

                        addData(f);

                }

                if(panel.getAddFilesRadio().isSelected()) {

                        selectFiles();

                } else {

                        selectDirectory();

                }

        }

}