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;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import java.util.Locale;

import javax.swing.JComponent;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.plugins.Extension;

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

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

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

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

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

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

import org.gvsig.i18n.Messages;

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

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

import org.gvsig.raster.multifile.app.panel.BandSelectorPanel;

import org.gvsig.raster.multifile.app.panel.MainWindow;

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

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

import org.gvsig.raster.swing.RasterSwingLibrary;

import org.gvsig.raster.swing.RasterSwingLocator;

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

import org.gvsig.raster.swing.newlayer.CreateNewLayerPanel;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.extensionpoint.ExtensionPoint;

import org.gvsig.tools.extensionpoint.ExtensionPointManager;

/**

 * Extension for adding grid netcdf support to gvSIG.

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

 */

public class MultifileClientExtension extends Extension implements ActionListener {

        private CreateNewLayerPanel             newLayerPanel        = null;

        private MainWindow                      layerNamewindow      = null;

        private MainWindow                      bandSelectorwindow   = null;

        private BandSelectorPanel               panel                = null; 

        private AbstractViewPanel               selectedView         = null;

        private String                          file                 = null;

        private String                          folder               = null;

        public void initialize() {

                ExtensionPointManager extensionPoints = ToolsLocator.getExtensionPointManager();

                ExtensionPoint point = extensionPoints.get("AplicationPreferences");

                point = extensionPoints.get("RasterSEPropertiesDialog");

                point.append("Bandas", "", BandSelectorPanel.class);

                point = extensionPoints.add("GenericToolBarMenu");

                point.append("Multifile", "", MultiFileCreatorTocMenuEntry.getSingleton());

                MultiFileCreatorTocMenuEntry.setExtension(this);

                if (!Messages.hasLocales()) {

                        Messages.addLocale(Locale.getDefault());

                }

                Messages.addResourceFamily("org.gvsig.raster.multifile.app.i18n.text",

                                MultifileClientExtension.class.getClassLoader(),

                                MultifileClientExtension.class.getClass().getName());

                initilizeIcons();

        }

        public void execute(String actionCommand) {

                if (actionCommand.compareTo("MultifileCreator") == 0) {

                        layerNamewindow = new MainWindow((JComponent)getNewLayerPanel(), 

                                        Messages.getText("bands"), 300, 150, this);

                        PluginServices.getMDIManager().addCentredWindow(layerNamewindow);

            }

        }

        private void initilizeIcons() {

                IconThemeHelper.registerIcon(null, "multifile-icon", this);

    }

        public boolean isEnabled() {

                return true;

        }

        public boolean isVisible() {

                return false;

        }

        public CreateNewLayerPanel getNewLayerPanel() {

                if(newLayerPanel == null) {

                        newLayerPanel = RasterSwingLocator.getSwingManager().createNewLayerPanel();

                }

                return newLayerPanel;

        }

        public void actionPerformed(ActionEvent e) {

                //Accept the window with the layer name

                if(e.getSource() == layerNamewindow.getButtonsPanel().getButton(IButtonsPanel.BUTTON_ACCEPT)) {

                        file = getNewLayerPanel().getFileSelected();

                        folder = getNewLayerPanel().getDirectorySelected();

                        panel = new BandSelectorPanel(BandSelectorPanel.TYPE_DIALOG);

                        panel.getListener().setDestination(file, folder);

                        bandSelectorwindow = new MainWindow(panel, Messages.getText("bands"), 500, 400, this, true);

                        PluginServices.getMDIManager().addCentredWindow(bandSelectorwindow);

                }

                if(e.getSource() == bandSelectorwindow.getButtonsPanel().getButton(IButtonsPanel.BUTTON_USR1)) {

                        loadLayer();

                }

                if(e.getSource() == bandSelectorwindow.getButtonsPanel().getButton(IButtonsPanel.BUTTON_USR2)) {

                }

        }

        private void saveMff(RasterDataStore mainRasterStore) {

                MultiFileDataParameters params = (MultiFileDataParameters)mainRasterStore.getParameters();

                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 {

                        MultiFileFormat.saveMultiFileFormat(file, folder, uriList);

                } catch (IOException e) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("error_salvando_mff"), panel, e);

                }

        }

        private void saveMetadata() {

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

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

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

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

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

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

                        return;

                }

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

                                new String[]{ColorInterpretation.ALPHA_BAND, ColorInterpretation.RED_BAND, ColorInterpretation.GREEN_BAND, ColorInterpretation.BLUE_BAND});

                try {

                        // Combinaci?n GRAY

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

                                for (int iBand = 0; iBand < panel.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 < panel.getARGBTable().getRowCount(); iBand++)

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

                        }

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

                } catch (RmfSerializerException exc) {

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

                } catch (NotInitializeException exc) {

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

                }

        }

        public void loadLayerInView(RasterDataStore mainRasterStore) throws LoadLayerException {

                MapContextManager mcm = MapContextLocator.getMapContextManager();

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

                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 loadLayer() {

                RasterDataStore mainRasterStore = panel.getResult();

                if(mainRasterStore == null) {

                        return;

                }

                saveMff(mainRasterStore);

                saveMetadata();

                try {

                        loadLayerInView(mainRasterStore);

                } catch (LoadLayerException e) {

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

                }

        }

        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;

        }

}