Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.mosaic.app;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.andami.PluginServices;

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

import org.gvsig.app.project.documents.view.gui.AbstractViewPanel;

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

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

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

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

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

import org.gvsig.fmap.mapcontext.exceptions.LoadLayerException;

import org.gvsig.i18n.Messages;

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

import org.gvsig.raster.impl.provider.RasterProvider;

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

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

import org.gvsig.raster.mosaic.swing.main.AddFilesEvent;

import org.gvsig.raster.mosaic.swing.main.AddFilesListener;

import org.gvsig.raster.mosaic.swing.main.AddFilesPanel;

import org.gvsig.raster.swing.RasterSwingLibrary;

import org.gvsig.tools.locator.LocatorException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class MainDialogActions implements AddFilesListener {

        private Logger                          log                  = LoggerFactory.getLogger(MainDialogActions.class);

        private String                          file                 = null;

        private String                          folder               = null;

        private AddFilesPanel                   addFilesPanel        = null;

        private AbstractViewPanel               selectedView         = null;

        public MainDialogActions(AddFilesPanel panel, String file, String folder) {

                this.file = file;

                this.folder = folder;

                this.addFilesPanel = panel;

        }

        public void saveLayer() {

                RasterDataStore mainRasterStore = addFilesPanel.getResult(false);

                try {

                        saveMosaicFileLayer(mainRasterStore);

                } catch (IOException e) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("error_saving_mosaic"), addFilesPanel, e);

                }

        }

        /**

         * Saves the new layer in disk

         * @param layerName

         * @param file

         * @param uriList

         * @throws IOException

         */

        private String saveMosaicFileLayer(RasterDataStore mainRasterStore) throws IOException {

                MosaicDataParameters params = null; 

                if(mainRasterStore.getParameters() instanceof MosaicDataParameters)        {        

                                params = (MosaicDataParameters)mainRasterStore.getParameters();

                }

                if(mainRasterStore.getParameters() instanceof TileDataParameters) {

                        TileDataParameters tileparams = (TileDataParameters)mainRasterStore.getParameters();

                        params = (MosaicDataParameters)tileparams.getDataParameters();

                }

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

                List<RasterProvider> providers = params.getProviders();

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

                        uriList.add(new File(providers.get(i).getURI()));

                }

                try {

                        return MosaicRasterFormat.saveMosaicRasterFormat(file, folder, uriList);

                } catch (IOException e) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("error_salvando_mrf"), addFilesPanel, e);

                }

                return null;

        }

        public void loadLayer() {

                RasterDataStore mainRasterStore = addFilesPanel.getResult(true);

                if(mainRasterStore == null) {

                        return;

                }

                try {

                        loadLayerInView(mainRasterStore);

                } catch (LoadLayerException e) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("error_loading_layer"), addFilesPanel, e);

                }

        }

        private void loadLayerInView(RasterDataStore mainRasterStore) throws LoadLayerException {

                MapContextManager mcm = MapContextLocator.getMapContextManager();

                FLyrRaster lyr = (FLyrRaster) mcm.createLayer(file, mainRasterStore);

                lyr.reload();

                getView().getMapControl().getMapContext().beginAtomicEvent();

                getView().getMapControl().getMapContext().getLayers().addLayer(lyr);

                getView().getMapControl().getMapContext().invalidate();

                getView().getMapControl().getMapContext().endAtomicEvent();

        }

        private AbstractViewPanel getView() {

                if(selectedView == null) {

                        IWindow[] wList = PluginServices.getMDIManager().getAllWindows();

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

                                if(wList[i] instanceof AbstractViewPanel)

                                        selectedView = (AbstractViewPanel)wList[i];

                        }

                }

                return selectedView;

        }

        public void actionAddFiles(AddFilesEvent e) {

                File[] fList = e.getFileList();

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

                        try {

                                RasterDataStore store = RasterLocator.getManager().getProviderServices().open(fList[i].getAbsolutePath());

                                if(store != null)

                                        addFilesPanel.addDstFile(

                                                        store, 

                                                        getLayerNameFromFileName(store.getFullName()), 

                                                        store.getBandCount(), 

                                                        store.getDataType()[0],

                                                        store.getCellSize(),

                                                        store.getNoDataValue());

                        } catch (RasterDriverException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        } catch (InitializeException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        } catch (ProviderNotRegisteredException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        } catch (LocatorException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        } catch (NotSupportedExtensionException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        }

                }

        }

        public void actionAddFolder(AddFilesEvent e) {

                File[] folderList = e.getFileList();

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

                        try {

                                RasterDataStore store = RasterLocator.getManager().getProviderServices().open(folderList[i].getAbsolutePath());

                                if(store != null)

                                        addFilesPanel.addDstFile(

                                                        store, 

                                                        getLayerNameFromFileName(store.getFullName()), 

                                                        store.getBandCount(), 

                                                        store.getDataType()[0],

                                                        store.getCellSize(),

                                                        store.getNoDataValue());

                        } catch (RasterDriverException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        } catch (InitializeException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        } catch (ProviderNotRegisteredException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        } catch (LocatorException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        } catch (NotSupportedExtensionException e1) {

                                log.debug("Error loading a RasterDataStore from the list", e1);

                        }

                }

        }

        private String getLayerNameFromFileName(String fullName) {

                if(fullName.contains(File.separator)) {

                        int index = fullName.lastIndexOf(File.separator);

                        fullName = fullName.substring(index + 1);

                }

                /*if(fullName.contains(".")) {

                        int index = fullName.lastIndexOf(".");

                        fullName = fullName.substring(0, index);

                }*/

                return fullName;

        }

}