Re-structure LTTng sub-project as per the Linux Tools guidelines

[deliverable/tracecompass.git] / org.eclipse.linuxtools.lttng.ui / src / org / eclipse / linuxtools / lttng / ui / tracecontrol / wizards / ConfigureTraceWizard.java

/*******************************************************************************\r
 * Copyright (c) 2011 Ericsson\r
 * \r
 * All rights reserved. This program and the accompanying materials are\r
 * made available under the terms of the Eclipse Public License v1.0 which\r
 * accompanies this distribution, and is available at\r
 * http://www.eclipse.org/legal/epl-v10.html\r
 * \r
 * Contributors:\r
 *   Bernd Hufmann - Initial API and implementation\r
 *   \r
 *******************************************************************************/\r
package org.eclipse.linuxtools.lttng.ui.tracecontrol.wizards;\r
\r
import java.util.Iterator;\r
import java.util.concurrent.TimeUnit;\r
\r
import org.eclipse.jface.viewers.IStructuredSelection;\r
import org.eclipse.jface.wizard.Wizard;\r
import org.eclipse.linuxtools.lttng.core.tracecontrol.model.TraceResource;\r
import org.eclipse.linuxtools.lttng.core.tracecontrol.model.config.TraceChannel;\r
import org.eclipse.linuxtools.lttng.core.tracecontrol.model.config.TraceChannels;\r
import org.eclipse.linuxtools.lttng.core.tracecontrol.model.config.TraceConfig;\r
import org.eclipse.linuxtools.lttng.core.tracecontrol.service.ILttControllerService;\r
import org.eclipse.linuxtools.lttng.ui.LTTngUiPlugin;\r
import org.eclipse.linuxtools.lttng.ui.tracecontrol.TraceControlConstants;\r
import org.eclipse.linuxtools.lttng.ui.tracecontrol.subsystems.TraceSubSystem;\r
import org.eclipse.rse.services.clientserver.messages.SystemMessageException;\r
import org.eclipse.rse.ui.SystemBasePlugin;\r
import org.eclipse.tm.tcf.protocol.IToken;\r
import org.eclipse.tm.tcf.util.TCFTask;\r
import org.eclipse.ui.INewWizard;\r
import org.eclipse.ui.IWorkbench;\r
\r
/**\r
 * <b><u>ConfigureTraceWizard</u></b>\r
 * <p>\r
 *  Wizard implementation to configure a trace.\r
 * </p>\r
 */\r
public class ConfigureTraceWizard extends Wizard implements INewWizard {\r
\r
    // ------------------------------------------------------------------------\r
    // Attributes\r
    // ------------------------------------------------------------------------\r
\r
    ITraceChannelConfigurationPage channelConfigPage;\r
    TraceConfigurationPage traceConfigPage;\r
\r
    private TraceResource fSelectedTrace = null;\r
\r
    // ------------------------------------------------------------------------\r
    // Constructors\r
    // ------------------------------------------------------------------------\r
\r
    public ConfigureTraceWizard() {\r
        super();\r
    }\r
\r
    // ------------------------------------------------------------------------\r
    // Operations\r
    // ------------------------------------------------------------------------\r
 \r
    /*\r
     * (non-Javadoc)\r
     * @see org.eclipse.jface.wizard.Wizard#performFinish()\r
     */\r
    @Override\r
    public boolean performFinish() {\r
        return true;\r
    }\r
\r
    /*\r
     * (non-Javadoc)\r
     * @see org.eclipse.jface.wizard.Wizard#addPages()\r
     */\r
    @Override\r
	public void addPages() {\r
        traceConfigPage = new TraceConfigurationPage(this);\r
        addPage(traceConfigPage);\r
\r
        if (fSelectedTrace.isUst()) {\r
            // User space trace\r
            TraceChannels channels = null;\r
            if ((fSelectedTrace.getTraceConfig()) != null && fSelectedTrace.getTraceConfig().getTraceChannels() != null) {\r
                channels = fSelectedTrace.getTraceConfig().getTraceChannels().clone(); \r
            }\r
            else {\r
                channels = new TraceChannels();\r
                channels.put(TraceChannel.UST_TRACE_CHANNEL_NAME, new TraceChannel(TraceChannel.UST_TRACE_CHANNEL_NAME));\r
            }\r
            channelConfigPage = new UstTraceChannelConfigurationPage(channels);\r
            addPage(channelConfigPage);\r
        } else {\r
            // Kernel trace \r
            TraceChannels channels = null;\r
            if ((fSelectedTrace.getTraceConfig()) != null && (fSelectedTrace.getTraceConfig().getTraceChannels() != null)) {\r
                channels = fSelectedTrace.getTraceConfig().getTraceChannels().clone(); \r
            }\r
            else {\r
                String[] channelNames = new String[0];\r
                try {\r
                    final ILttControllerService service = ((TraceSubSystem)fSelectedTrace.getSubSystem()).getControllerService();\r
                    channelNames = new TCFTask<String[]>() {\r
                        @Override\r
                        public void run() {\r
                            // Get targets using Lttng controller service proxy\r
                            service.getChannels(fSelectedTrace.getParent().getParent().getName(), fSelectedTrace.getParent().getName(), fSelectedTrace.getName(), new ILttControllerService.DoneGetChannels() {\r
\r
                                @Override\r
                                public void doneGetChannels(IToken token, Exception error, String[] str) {\r
                                    if (error != null) {\r
                                        // Notify with error\r
                                        error(error);\r
                                        return;\r
                                    }\r
                                    // Notify with active trace list\r
                                    done(str);\r
                                }\r
                            });\r
                        }}.get(TraceControlConstants.DEFAULT_TCF_TASK_TIMEOUT, TimeUnit.SECONDS);\r
                } catch (Exception e) {\r
                    if (e instanceof SystemMessageException) {\r
                        SystemBasePlugin.logError("Trace Configuration", e); //$NON-NLS-1$;\r
                    }\r
                    else {\r
                        SystemBasePlugin.logError("TraceSubSystem", new SystemMessageException(LTTngUiPlugin.getDefault().getMessage(e))); //$NON-NLS-1$\r
                    }\r
                }\r
\r
                channels = new TraceChannels();\r
                channels.putAll(channelNames);\r
            }\r
\r
            channelConfigPage = new KernelTraceChannelConfigurationPage(channels, fSelectedTrace.getTraceState());\r
            addPage(channelConfigPage);\r
        }\r
\r
        getShell().setImage(LTTngUiPlugin.getDefault().getImage(LTTngUiPlugin.ICON_ID_CONFIG_TRACE));\r
    }\r
\r
    /*\r
     * (non-Javadoc)\r
     * @see org.eclipse.ui.IWorkbenchWizard#init(org.eclipse.ui.IWorkbench, org.eclipse.jface.viewers.IStructuredSelection)\r
     */\r
    @Override\r
    @SuppressWarnings("unchecked")\r
	public void init(IWorkbench workbench, IStructuredSelection selection) {\r
\r
        fSelectedTrace = null;\r
\r
        // store the selected targets to be used when running\r
        Iterator<IStructuredSelection> theSet = selection.iterator();\r
        while (theSet.hasNext()) {\r
            Object obj = theSet.next();\r
            if (obj instanceof TraceResource) {\r
                fSelectedTrace = (TraceResource)obj;\r
                break; // only one is allowed\r
            }\r
        }\r
    }\r
\r
    /**\r
     * Gets the relevant selected trace that will be configured.\r
     * \r
     * @return selected trace.\r
     */\r
    public TraceResource getSelectedTrace() {\r
        return fSelectedTrace;\r
    }\r
    \r
    /**\r
     * Gets the new trace configuration of the trace.\r
     *  \r
     * @return trace configuration\r
     */\r
    public TraceConfig getTraceConfig() {\r
        TraceConfig config = traceConfigPage.getTraceConfig();\r
        config.setTraceChannels(channelConfigPage.getTraceChannels());\r
        return config;\r
    }\r
}\r