Add support for intervals

Can now filter by earliest record of interest and
latest record of interest, using -e and -l.

3 files changed, 54 insertions, 0 deletions

diff --git a/unit-trace.py b/unit-trace.py
index 5d772e8..93ab169 100755
--- a/unit-trace.py
+++ b/unit-trace.py
@@ -21,6 +21,8 @@ from unit_trace import visualizer
 from unit_trace import progress
 from unit_trace import skipper
 from unit_trace import maxer
+from unit_trace import earliest
+from unit_trace import latest
 
 # Get trace files from command line arguments
 from optparse import OptionParser
@@ -42,6 +44,10 @@ parser.add_option("-v", "--visual", action="store_true", dest="visualize",
     default=False, help="Use visualizer")
 parser.add_option("-c", "--clean", action="store_true", dest="clean",
     default=False, help="Use sanitizer to clean garbage records")
+parser.add_option("-e", "--earliest", default=0, type=int, dest="earliest",
+    help="Earliest timestamp of interst")
+parser.add_option("-l", "--latest", default=0, type=int, dest="latest",
+    help="Latest timestamp of interst")
 (options, traces) = parser.parse_args()
 traces = list(traces)
 if len(traces) < 1:
@@ -63,6 +69,14 @@ if options.skipnum > 0:
 if options.maxnum > 0:
     stream = maxer.maxer(stream, options.maxnum)
 
+# Enfore earliest timestamp
+if options.earliest > 0:
+    stream = earliest.earliest(stream,options.earliest)
+
+# Enfore latest timestamp
+if options.latest > 0:
+    stream = latest.latest(stream,options.latest)
+
 # Filter out garbage events
 if options.clean is True:
     stream = sanitizer.sanitizer(stream)
diff --git a/unit_trace/earliest.py b/unit_trace/earliest.py
new file mode 100644
index 0000000..eaf0cf8
--- /dev/null
+++ b/unit_trace/earliest.py
@@ -0,0 +1,21 @@
+###############################################################################
+# Description
+###############################################################################
+
+# Enforce earliest record
+
+###############################################################################
+# Public functions
+###############################################################################
+
+def earliest(stream, earliest):
+    for record in stream:
+        if record.record_type=="event":
+            if record.when < earliest:
+                pass
+            else:
+                yield record
+                break
+        yield record
+    for record in stream:
+        yield record
diff --git a/unit_trace/latest.py b/unit_trace/latest.py
new file mode 100644
index 0000000..4abd3a2
--- /dev/null
+++ b/unit_trace/latest.py
@@ -0,0 +1,19 @@
+###############################################################################
+# Description
+###############################################################################
+
+# Enforce latest record
+
+###############################################################################
+# Public functions
+###############################################################################
+
+def latest(stream, latest):
+    for record in stream:
+        if record.record_type=="event":
+            if record.when > latest:
+                break
+            else:
+                yield record
+        else:
+            yield record