control: Add enhanced support for loading of sessions

[deliverable/tracecompass.git] / lttng / org.eclipse.tracecompass.lttng2.control.ui / src / org / eclipse / tracecompass / internal / lttng2 / control / ui / views / dialogs / TraceControlDialogFactory.java

diff --git a/lttng/org.eclipse.tracecompass.lttng2.control.ui/src/org/eclipse/tracecompass/internal/lttng2/control/ui/views/dialogs/TraceControlDialogFactory.java b/lttng/org.eclipse.tracecompass.lttng2.control.ui/src/org/eclipse/tracecompass/internal/lttng2/control/ui/views/dialogs/TraceControlDialogFactory.java
index 65073548c5a3f09aa3812bf32dc2ea93744068b6..7f1f5b84936a06467a014c0f1b0ae1f375caae68 100644 (file)
--- a/lttng/org.eclipse.tracecompass.lttng2.control.ui/src/org/eclipse/tracecompass/internal/lttng2/control/ui/views/dialogs/TraceControlDialogFactory.java
+++ b/lttng/org.eclipse.tracecompass.lttng2.control.ui/src/org/eclipse/tracecompass/internal/lttng2/control/ui/views/dialogs/TraceControlDialogFactory.java
@@ -53,6 +53,11 @@ public final class TraceControlDialogFactory {
      */
     private ISelectCommandScriptDialog fCommandScriptDialog;
 
+    /**
+     * The command script selection dialog.
+     */
+    private ILoadDialog fLoadDialog;
+
     /**
      * The enable events dialog.
      */
@@ -153,6 +158,25 @@ public final class TraceControlDialogFactory {
         return fCommandScriptDialog;
     }
 
+    /**
+     * @return command script selection dialog implementation
+     */
+    public ILoadDialog getLoadDialog() {
+        if (fLoadDialog == null) {
+            fLoadDialog = new LoadDialog(PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell());
+        }
+        return fLoadDialog;
+    }
+
+    /**
+     * Sets a load dialog implementation
+     * @param loadDialog
+     *            a load dialog implementation
+     */
+    public void setLoadDialog(ILoadDialog loadDialog) {
+        fLoadDialog = loadDialog;
+    }
+
     /**
      * Sets a create session dialog implementation.
      * @param createSessionDialog - a create session implementation.