Created maxer and skipper

Maxer: Allow a maximum number of records to be parsed
Skipper: Skip a given number of records at the beginning

2 files changed, 42 insertions, 0 deletions

diff --git a/unit_trace/maxer.py b/unit_trace/maxer.py
new file mode 100644
index 0000000..086c6bc
--- /dev/null
+++ b/unit_trace/maxer.py
@@ -0,0 +1,20 @@
+###############################################################################
+# Description
+###############################################################################
+
+# Parse at most the given number of records
+
+###############################################################################
+# Public functions
+###############################################################################
+
+def maxer(stream, number):
+    for record in stream:
+        if record.record_type=="event":
+            if number > 0:
+                number -= 1
+                yield record
+            else:
+                break
+        else:
+            yield record
diff --git a/unit_trace/skipper.py b/unit_trace/skipper.py
new file mode 100644
index 0000000..4a3bd46
--- /dev/null
+++ b/unit_trace/skipper.py
@@ -0,0 +1,22 @@
+###############################################################################
+# Description
+###############################################################################
+
+# Skip over the given number of records.
+
+###############################################################################
+# Public functions
+###############################################################################
+
+def skipper(stream, number):
+    for record in stream:
+        if record.record_type=="event":
+            if number > 0:
+                number -= 1
+            else:
+                yield record
+                break
+        else:
+            yield record
+    for record in stream:
+       yield record