tmf: Split StateHistorySystem into two interfaces

[deliverable/tracecompass.git] / org.eclipse.linuxtools.lttng2.kernel.core / src / org / eclipse / linuxtools / internal / lttng2 / kernel / core / stateprovider / CtfKernelHandler.java

diff --git a/org.eclipse.linuxtools.lttng2.kernel.core/src/org/eclipse/linuxtools/internal/lttng2/kernel/core/stateprovider/CtfKernelHandler.java b/org.eclipse.linuxtools.lttng2.kernel.core/src/org/eclipse/linuxtools/internal/lttng2/kernel/core/stateprovider/CtfKernelHandler.java
index 7065cb62aded70fa7f3c7073788fce713e210d95..8f965f6df24cc03ff853710fa1944b19b35e5033 100644 (file)
--- a/org.eclipse.linuxtools.lttng2.kernel.core/src/org/eclipse/linuxtools/internal/lttng2/kernel/core/stateprovider/CtfKernelHandler.java
+++ b/org.eclipse.linuxtools.lttng2.kernel.core/src/org/eclipse/linuxtools/internal/lttng2/kernel/core/stateprovider/CtfKernelHandler.java
@@ -21,9 +21,8 @@ import org.eclipse.linuxtools.lttng2.kernel.core.trace.LttngStrings;
 import org.eclipse.linuxtools.tmf.core.ctfadaptor.CtfTmfEvent;
 import org.eclipse.linuxtools.tmf.core.event.ITmfEventField;
 import org.eclipse.linuxtools.tmf.core.statesystem.AttributeNotFoundException;
-import org.eclipse.linuxtools.tmf.core.statesystem.StateHistorySystem;
-import org.eclipse.linuxtools.tmf.core.statesystem.StateSystem;
 import org.eclipse.linuxtools.tmf.core.statesystem.TimeRangeException;
+import org.eclipse.linuxtools.tmf.core.statesystem.helpers.IStateSystemBuilder;
 import org.eclipse.linuxtools.tmf.core.statevalue.ITmfStateValue;
 import org.eclipse.linuxtools.tmf.core.statevalue.StateValueTypeException;
 import org.eclipse.linuxtools.tmf.core.statevalue.TmfStateValue;
@@ -37,7 +36,7 @@ import org.eclipse.linuxtools.tmf.core.statevalue.TmfStateValue;
 class CtfKernelHandler implements Runnable {
 
     private final BlockingQueue<CtfTmfEvent> inQueue;
-    private StateSystem ss;
+    private IStateSystemBuilder ss;
 
     private CtfTmfEvent currentEvent;
 
@@ -60,14 +59,10 @@ class CtfKernelHandler implements Runnable {
         knownEventNames = fillEventNames();
     }
 
-    void assignStateSystem(StateSystem targetSS) {
+    void assignStateSystem(IStateSystemBuilder targetSS) {
         this.ss = targetSS;
     }
 
-    StateSystem getStateSystem() {
-        return ss;
-    }
-
     @Override
     public void run() {
         if (ss == null) {
@@ -94,16 +89,14 @@ class CtfKernelHandler implements Runnable {
 
     private void closeStateSystem() {
         /* Close the History system, if there is one */
-        if (ss.getClass() == StateHistorySystem.class) {
-            try {
-                ((StateHistorySystem) ss).closeHistory(currentEvent.getTimestamp().getValue());
-            } catch (TimeRangeException e) {
-                /*
-                 * Since we're using currentEvent.getTimestamp, this shouldn't
-                 * cause any problem
-                 */
-                e.printStackTrace();
-            }
+        try {
+            ss.closeHistory(currentEvent.getTimestamp().getValue());
+        } catch (TimeRangeException e) {
+            /*
+             * Since we're using currentEvent.getTimestamp, this shouldn't
+             * cause any problem
+             */
+            e.printStackTrace();
         }
     }