Implement statistics module

3 files changed, 53 insertions, 4 deletions

diff --git a/run.py b/run.py
index 9ac0389..c564a0f 100755
--- a/run.py
+++ b/run.py
@@ -15,6 +15,7 @@
 import trace_reader
 import sanitizer
 import gedf_test
+import stats
 import stdout_printer
 
 ###############################################################################
@@ -35,4 +36,5 @@ g6 = [
 stream = trace_reader.trace_reader(g6)
 stream = sanitizer.sanitizer(stream)
 stream = gedf_test.gedf_test(stream)
+stream = stats.stats(stream)
 stdout_printer.stdout_printer(stream)
diff --git a/stats.py b/stats.py
new file mode 100644
index 0000000..6216f44
--- /dev/null
+++ b/stats.py
@@ -0,0 +1,36 @@
+###############################################################################
+# Description
+###############################################################################
+# Compute and produce statistics
+
+
+###############################################################################
+# Public Functions
+###############################################################################
+
+def stats(stream):
+    min_inversion = -1
+    max_inversion = -1
+    sum_inversions = 0
+    num_inversions = 0
+    for record in stream:
+        if record.type_name == 'inversion_end':
+            length = record.job.inversion_end - record.job.inversion_start
+            if length > 0:
+                num_inversions += 1
+                if length > max_inversion:
+                    max_inversion = length
+                if length < min_inversion or min_inversion == -1:
+                    min_inversion = length
+                sum_inversions += length
+        yield record
+    avg_inversion = int(sum_inversions / num_inversions)
+    class Obj(): pass
+    rec = Obj()
+    rec.record_type = "meta"
+    rec.type_name = "stats"
+    rec.num_inversions = num_inversions 
+    rec.min_inversion = min_inversion
+    rec.max_inversion = max_inversion
+    rec.avg_inversion = avg_inversion
+    yield rec
diff --git a/stdout_printer.py b/stdout_printer.py
index 466f734..5bbcead 100644
--- a/stdout_printer.py
+++ b/stdout_printer.py
@@ -12,10 +12,14 @@ def stdout_printer(stream):
     for record in stream:
         if record.record_type == "event":
             _print_event(record)
-        elif record.type_name == 'inversion_start':
+        elif record.record_type == "meta" and record.type_name == "stats":
+            _print_stats(record)
+        elif record.record_type == "error" and record.type_name == 'inversion_start':
             _print_inversion_start(record)
-        elif record.type_name == 'inversion_end':
+        elif record.record_type == "error" and record.type_name == 'inversion_end':
             _print_inversion_end(record)
+        else:
+            continue
         print("")
 
 ###############################################################################
@@ -39,7 +43,7 @@ def _print_inversion_start(record):
     print("On CPU: ",end="")
     for job in record.on_cpu:
         print(str(job) + " ",end="")
-    print("")
+    print("") #newline
 
 def _print_inversion_end(record):
     print("Type: {}".format("Inversion end"))
@@ -55,4 +59,11 @@ def _print_inversion_end(record):
     print("On CPU: ",end="")
     for job in record.on_cpu:
         print(str(job) + " ",end="")
-    print("")
+    print("") #newline
+
+def _print_stats(record):
+    print("Inversion statistics")
+    print("Num inversions: {}".format(record.num_inversions))
+    print("Min inversion: {}".format(record.min_inversion))
+    print("Max inversion: {}".format(record.max_inversion))
+    print("Avg inversion: {}".format(record.avg_inversion))