Added "side effect remover" function. This is currently used to remove

switch_aways immediately following completions. Now, the switch_aways +
completions == switch_tos, except that job1 does an additional
switch_away and because the last side-effect switch-away is /before/ the
completion and thus not removed. In the future, it will be good to
also make sure the switch_away being removed is on the same CPU

1 files changed, 9 insertions, 0 deletions

diff --git a/sta.py b/sta.py
index 152faeb..34ed2f1 100755
--- a/sta.py
+++ b/sta.py
@@ -88,6 +88,15 @@ class Trace:
         """Slice the trace iterator"""
         self.iter = list(self.iter)[start:end]
 
+    def remove_side_effects(self, type1, type2):
+        """Remove records of type2 immediately following records of type1"""
+        self.iter = list(self.iter)
+        i = 0
+        while i < len(self.iter) - 1:
+            if self.iter[i]['type'] == type1 and self.iter[i+1]['type']==type2:
+                del self.iter[i+1]
+            i += 1
+
     def print_records(self):
         """Prints all records in the trace"""
         for record in self.iter: