Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.principalcomponents.app;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.HashMap;

import javax.swing.JPanel;

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.algorithm.process.IProcessActions;

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

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

import org.gvsig.raster.principalcomponents.algorithm.PCStatsDataStructure;

import org.gvsig.raster.principalcomponents.algorithm.PrincipalComponentsAlgorithmLibrary;

import org.gvsig.raster.principalcomponents.app.toolbox.ToolboxPCLoader;

import org.gvsig.raster.principalcomponents.swing.PrincipalComponentsSwingLibrary;

import org.gvsig.raster.principalcomponents.swing.PrincipalComponentsSwingLocator;

import org.gvsig.raster.swing.RasterSwingLibrary;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.extensionpoint.ExtensionPoint;

import org.gvsig.tools.extensionpoint.ExtensionPointManager;

public class PrincipalComponentsExtension extends Extension implements ActionListener, IProcessActions {

    private BasicAPISwingPanel   mainPanel              = null;

    private BasicAPISwingPanel   componentsPanel        = null;

    private MainWindow           windowMainPanel        = null;

    private MainWindow           windowComponentsPanel  = null;

    private MainWindow           statisticsPanel        = null;

    private FLyrRaster           lyr                    = null;

    private PCStatsDataStructure stats                  = 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("PrincipalComponents", "", PrincipalComponentsTocMenuEntry.getSingleton());

                PrincipalComponentsTocMenuEntry.setExtension(this);

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

                        PrincipalComponentsExtension.class.getClassLoader(),

                        PrincipalComponentsExtension.class.getClass().getName());

            initilizeIcons();

    }

        public void postInitialize() {

                new ToolboxPCLoader().registerTool();

        }

    public void execute(String actionCommand) {

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

                    lyr = getSelectedLayer();

                    if(lyr == null) {

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

                            return;

                    }

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

                    mainPanel = PrincipalComponentsSwingLocator.getSwingManager().createPCAMainPanel(

                                    lyr.getDataStore(),

                                    lyr.getName(),

                                    lyr.getDataStore().getBandCount(),

                                    defaultROIEPSG);

                    windowMainPanel = new MainWindow(mainPanel.getComponent(),

                                    Messages.getText("principal_components"),

                                    400, 350,

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

                FLayer[] lyrList = getFirstMapCtrl().getMapContext().getLayers().getActives();

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

                        if(lyrList[i] instanceof FLyrRaster)

                                return (FLyrRaster)lyrList[i];

                }

                return null;

        }

    public boolean isEnabled() {

        return false;

    }

    public boolean isVisible() {

        return false;

    }

    void initilizeIcons(){

                IconThemeHelper.registerIcon(null, "principal-components", 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(this);

                        task.init();

                        task.start();

                }

                //GENERATE STATISTICS BUTTON

                if( componentsPanel != null &&

                        PrincipalComponentsSwingLibrary.GENERATE_STATIST == componentsPanel.getComponentIDByObject(e.getSource())) {

                        try {

                                Object[] result = (Object[])componentsPanel.getResult();

                                JPanel statsPanel = PrincipalComponentsSwingLocator.getSwingManager().createStatisticsPanel(

                                                stats,

                                                (Boolean)result[1],

                                                (Boolean)result[2],

                                                (Boolean)result[3]);

                                statisticsPanel = new MainWindow(statsPanel,

                                            Messages.getText("stats"),

                                            800, 500,

                                            this,

                                            true);

                            PluginServices.getMDIManager().addWindow(statisticsPanel);

                        } catch (Exception e1) {

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

                                return;

                        }

                }

                //ACCEPT COMPONENTS PANEL

                if( windowComponentsPanel != null &&

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

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

                        DataProcess task = null;

                        try {

                                Object[] result = (Object[])componentsPanel.getResult();

                                task = (DataProcess)result[0];

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

                                task.init();

                                task.start();

                        } catch (Exception e1) {

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

                                return;

                        }

                }

        }

        @SuppressWarnings("unchecked")

        public void end(Object param) {

                HashMap<String, Object> params = (HashMap<String, Object>)param;

                String processName = (String)params.get(ProcessParamsManagement.PROCESS_NAME);

                //END STATISTICS PROCESS (after accept main dialog)

                if (processName.equals(PrincipalComponentsAlgorithmLibrary.PC_STATS_PROCESS_LABEL)) {

                        PluginServices.getMDIManager().closeWindow(windowMainPanel);

                        stats = (PCStatsDataStructure)params.get("STATS_RESULT");

                        String fileName = (String)params.get("PATH");

                        boolean[] bands = (boolean[])params.get("BANDS");

                        String rois = (String)params.get("ROI_EPSG");

                        componentsPanel = PrincipalComponentsSwingLocator.getSwingManager().createPCAComponentsListPanel(

                                        lyr.getDataStore(),

                                        stats,

                                        fileName,

                                        bands,

                                        rois);

                        componentsPanel.addListener(this);

                        windowComponentsPanel = new MainWindow(componentsPanel.getComponent(),

                                    Messages.getText("principal_components"),

                                    400, 350,

                                    this);

                    PluginServices.getMDIManager().addCentredWindow(windowComponentsPanel);

                    componentsPanel.initialize();

                }

        }

        public void interrupted() {

        }

        public void updateProgress(int current, int total) {

        }

}