Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.tasseledcap.app;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

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

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.gui.beans.buttonspanel.ButtonsPanel;

import org.gvsig.gui.beans.defaultbuttonspanel.DefaultButtonsPanel;

import org.gvsig.i18n.Messages;

import org.gvsig.raster.algorithm.BasicAPISwingPanel;

import org.gvsig.raster.algorithm.process.DataProcess;

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

import org.gvsig.raster.swing.RasterSwingLibrary;

import org.gvsig.raster.tasseledcap.app.toolbox.ToolboxTasseledCapLoader;

import org.gvsig.raster.tasseledcap.swing.TasseledCapSwingLocator;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.extensionpoint.ExtensionPoint;

import org.gvsig.tools.extensionpoint.ExtensionPointManager;

public class TasseledCapExtension extends Extension implements ActionListener {

    private BasicAPISwingPanel   mainPanel              = null;

    private MainWindow           windowMainPanel        = null;

    private FLyrRaster           lyr                    = null;

    public void initialize() {

            // Adds an entry to the TOC's floating menu to those layers defined in this extension

                //ExtensionPoint exPoint = ToolsLocator.getExtensionPointManager().add("My_TocActions");

                //exPoint.append("MyMenuEntry", "", new PrincipalComponentsTocMenuEntry());

        // Adds a new tab to the "add layer" dialog

            //AddLayer.addWizard(PrincipalComponentsWizard.class);

            // Adds a new button in the raster bar

            // TODO: Remove for providers

                ExtensionPointManager extensionPoints = ToolsLocator.getExtensionPointManager();

                ExtensionPoint point = extensionPoints.add("GenericToolBarMenu");

                point.append("TasseledCap", "", TasseledCapTocMenuEntry.getSingleton());

                TasseledCapTocMenuEntry.setExtension(this);

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

                        TasseledCapExtension.class.getClassLoader(),

                        TasseledCapExtension.class.getClass().getName());

            initilizeIcons();

    }

        public void postInitialize() {

                new ToolboxTasseledCapLoader().registerTool();

        }

    public void execute(String actionCommand) {

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

                    lyr = getSelectedLayer();

                    if(lyr == null) {

                            RasterSwingLibrary.messageBoxError(Messages.getText("layer_not_found"), null);

                            return;

                    }

                        String defaultROIEPSG = getFirstMapCtrl().getProjection().getAbrev();

                    mainPanel = TasseledCapSwingLocator.getSwingManager().createTasseledCapMainPanel(

                                    lyr.getDataStore(),

                                    lyr.getName(),

                                    lyr.getDataStore().getBandCount(),

                                    defaultROIEPSG);

                    windowMainPanel = new MainWindow(mainPanel.getComponent(),

                                    Messages.getText("tasseled_cap"),

                                    400, 420,

                                    this);

                    PluginServices.getMDIManager().addCentredWindow(windowMainPanel);

                    mainPanel.initialize();

            }

    }

        /**

         * Gets the Mapcontrol for the first view found.

         * @return

         */

    private MapControl getFirstMapCtrl() {

            IWindow[] windowList = PluginServices.getMDIManager().getOrderedWindows();

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

                    if(windowList[i] instanceof IView) {

                            return ((IView)windowList[i]).getMapControl();

                    }

            }

            return null;

    }

    /**

     * Gets the selected and valid layer

     * @return

         */

    private FLyrRaster getSelectedLayer() {

        MapControl firstMapCtrl = getFirstMapCtrl();

        if (firstMapCtrl != null) {

            FLayer[] lyrList =

                getFirstMapCtrl().getMapContext().getLayers().getActives();

            if (lyrList.length <= 1) {

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

                    if (lyrList[i] instanceof FLyrRaster) {

                        return (FLyrRaster) lyrList[i];

                    }

                }

            }

        }

        return null;

    }

    public boolean isEnabled() {

      FLyrRaster selectedLayer = getSelectedLayer();

      if (selectedLayer == null) {

        return false;

      }

      if (!(selectedLayer instanceof ILayerState))

        return false;

      if (!((ILayerState) selectedLayer).isOpen())

        return false;

      if((selectedLayer).getRender().getFilterList().get("colortable") != null)

        return false;

      if(selectedLayer.getDataStore().getBandCount() < 4)

        return false;

      return true;

    }

    public boolean isVisible() {

      FLyrRaster selectedLayer = getSelectedLayer();

      if (selectedLayer == null)

        return false;

      if (!(selectedLayer instanceof IRasterLayerActions))

        return false;

      return ((IRasterLayerActions) selectedLayer).isActionEnabled(IRasterLayerActions.ENHANCED);

    }

    void initilizeIcons(){

                IconThemeHelper.registerIcon(null, "tasseled-cab", this);

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

                IconThemeHelper.registerIcon(null, "gvsig-icon16x16", this);

    }

        public void actionPerformed(ActionEvent e) {

                //ACCEPT MAIN PANEL

                if( e.getSource() == ((DefaultButtonsPanel)windowMainPanel).getButtonsPanel().getButton(ButtonsPanel.BUTTON_APPLY) ||

                        e.getSource() == ((DefaultButtonsPanel)windowMainPanel).getButtonsPanel().getButton(ButtonsPanel.BUTTON_ACCEPT)) {

                        DataProcess task = null;

                        try {

                                task = (DataProcess)mainPanel.getResult();

                        } catch (Exception e1) {

                                RasterSwingLibrary.messageBoxError(Messages.getText("problems_creating_process"), null);

                                return;

                        }

                        task.setActions(new ProcessEndActions(windowMainPanel, lyr));

                        task.init();

                        task.start();

                }

        }

        public void interrupted() {

        }

        public void updateProgress(int current, int total) {

        }

}