Added support for visualizing arbitrary actions.

(By Jonathan, cherry-picked by Mac)

Conflicts:

	unit_trace/trace_reader.py
	unit_trace/viz/canvas.py
	unit_trace/viz/graph.py
	unit_trace/viz/schedule.py

1 files changed, 10 insertions, 0 deletions

diff --git a/unit_trace/sanitizer.py b/unit_trace/sanitizer.py
index 598379a..fc66170 100644
--- a/unit_trace/sanitizer.py
+++ b/unit_trace/sanitizer.py
@@ -14,6 +14,8 @@ def sanitizer(stream):
     job_2s_released = [] # list of tasks which have released their job 2s
     jobs_switched_to = []
 
+    released = False
+
     for record in stream:
 
         # Ignore records which are not events (e.g. the num_cpus record)
@@ -21,6 +23,13 @@ def sanitizer(stream):
             yield record
             continue
 
+        if record.type_name == 'release':
+            released = released or True
+
+        if record.type_name == 'action' and released:
+            yield record
+            continue
+
         # All records with job < 2 are garbage
         if record.job < 2:
             continue
@@ -50,4 +59,5 @@ def sanitizer(stream):
             if (record.pid,record.job) not in jobs_switched_to:
                 record.job -= 1
 
+
         yield record