Improve stats.py to show longest n inversions

2 files changed, 36 insertions, 1 deletions

diff --git a/unit_trace/stats.py b/unit_trace/stats.py
index 17c5fc9..4462ef1 100644
--- a/unit_trace/stats.py
+++ b/unit_trace/stats.py
@@ -8,11 +8,16 @@
 # Public Functions
 ###############################################################################
 
-def stats(stream):
+def stats(stream,num):
+
+    # State
     min_inversion = -1
     max_inversion = -1
     sum_inversions = 0
     num_inversions = 0
+    longest_inversions = []
+
+    # Iterate over records, updating state
     for record in stream:
         if record.type_name == 'inversion_end':
             length = record.job.inversion_end - record.job.inversion_start
@@ -23,11 +28,29 @@ def stats(stream):
                 if length < min_inversion or min_inversion == -1:
                     min_inversion = length
                 sum_inversions += length
+                if len(longest_inversions) == num:
+                    si = longest_inversions[0]
+                    if length > (si.job.inversion_end -
+                        si.job.inversion_start):
+                        longest_inversions.append(record)
+                        longest_inversions = _sort_longest_inversions(
+                            longest_inversions)
+                        del longest_inversions[0]
+                else:
+                    longest_inversions.append(record)
+                    longest_inversions = _sort_longest_inversions(
+                        longest_inversions)
+
         yield record
+
+    # We've seen all records.
+    # Further update state
     if num_inversions > 0:
         avg_inversion = int(sum_inversions / num_inversions)
     else:
         avg_inversion = 0
+
+    # Produce a meta type record with all our information.
     class Obj(object): pass
     rec = Obj()
     rec.record_type = "meta"
@@ -36,4 +59,13 @@ def stats(stream):
     rec.min_inversion = min_inversion
     rec.max_inversion = max_inversion
     rec.avg_inversion = avg_inversion
+    longest_inversions = _sort_longest_inversions(longest_inversions)
+    rec.longest_inversions = longest_inversions
     yield rec
+
+def _sort_longest_inversions(longest_inversions):
+    """ Sort longest inversions"""
+    def sortkey(x):
+        return x.job.inversion_end - x.job.inversion_start
+    longest_inversions.sort(key=sortkey)
+    return longest_inversions
diff --git a/unit_trace/stdout_printer.py b/unit_trace/stdout_printer.py
index a56acce..dd1e2bd 100644
--- a/unit_trace/stdout_printer.py
+++ b/unit_trace/stdout_printer.py
@@ -75,3 +75,6 @@ def _print_stats(record):
     print "Min inversion: %d" % (record.min_inversion)
     print "Max inversion: %d" % (record.max_inversion)
     print "Avg inversion: %d" % (record.avg_inversion)
+    for inv in record.longest_inversions:
+        print ""
+        _print_inversion_end(inv)