Add Iterator support to TMF traces

[deliverable/tracecompass.git] / org.eclipse.linuxtools.tmf.core / src / org / eclipse / linuxtools / tmf / core / trace / TmfTrace.java

diff --git a/org.eclipse.linuxtools.tmf.core/src/org/eclipse/linuxtools/tmf/core/trace/TmfTrace.java b/org.eclipse.linuxtools.tmf.core/src/org/eclipse/linuxtools/tmf/core/trace/TmfTrace.java
index 51007e1f4d4221ad76a4dfd723c74f8f9cea61ea..81991d645c5c54466c2a5b4b916f165cbb618cfa 100644 (file)
--- a/org.eclipse.linuxtools.tmf.core/src/org/eclipse/linuxtools/tmf/core/trace/TmfTrace.java
+++ b/org.eclipse.linuxtools.tmf.core/src/org/eclipse/linuxtools/tmf/core/trace/TmfTrace.java
@@ -16,6 +16,7 @@ package org.eclipse.linuxtools.tmf.core.trace;
 import java.io.File;
 import java.util.Collection;
 import java.util.HashMap;
+import java.util.Iterator;
 import java.util.Map;
 
 import org.eclipse.core.resources.IResource;
@@ -628,6 +629,65 @@ public abstract class TmfTrace extends TmfEventProvider implements ITmfTrace {
         return context;
     }
 
+    // ------------------------------------------------------------------------
+    // ITmfTrace - Iterator operations
+    // ------------------------------------------------------------------------
+
+    /* (non-Javadoc)
+     * @see org.eclipse.linuxtools.tmf.core.trace.ITmfTrace#iterator()
+     */
+    /**
+     * @since 2.0
+     */
+    @Override
+    public Iterator<ITmfEvent> iterator() {
+        return new TmfTraceIterator(this);
+    }
+
+    /* (non-Javadoc)
+     * @see org.eclipse.linuxtools.tmf.core.trace.ITmfTrace#iterator(org.eclipse.linuxtools.tmf.core.trace.ITmfLocation)
+     */
+    /**
+     * @since 2.0
+     */
+    @Override
+    public Iterator<ITmfEvent> iterator(ITmfLocation location) {
+        return new TmfTraceIterator(this, location);
+    }
+
+    /* (non-Javadoc)
+     * @see org.eclipse.linuxtools.tmf.core.trace.ITmfTrace#iterator(long)
+     */
+    /**
+     * @since 2.0
+     */
+    @Override
+    public Iterator<ITmfEvent> iterator(long rank) {
+        return new TmfTraceIterator(this, rank);
+    }
+
+    /* (non-Javadoc)
+     * @see org.eclipse.linuxtools.tmf.core.trace.ITmfTrace#iterator(org.eclipse.linuxtools.tmf.core.event.ITmfTimestamp)
+     */
+    /**
+     * @since 2.0
+     */
+    @Override
+    public Iterator<ITmfEvent> iterator(ITmfTimestamp timestamp) {
+        return new TmfTraceIterator(this, timestamp);
+    }
+
+    /* (non-Javadoc)
+     * @see org.eclipse.linuxtools.tmf.core.trace.ITmfTrace#iterator(double)
+     */
+    /**
+     * @since 2.0
+     */
+    @Override
+    public Iterator<ITmfEvent> iterator(double ratio) {
+        return new TmfTraceIterator(this, ratio);
+    }
+
     // ------------------------------------------------------------------------
     // ITmfTrace - Read operations (returning an actual event)
     // ------------------------------------------------------------------------