sched: cope with SysTrace's sched_switch event

For trace-cmd traces, we use the raw trace to parse sched_switch.
That's not possible for SysTrace traces, we don't have access to the raw
data.  Luckily SysTrace doesn't mangle the data as much as trace-cmd's
plugin, so a simple string replace is enough to get rid of a stray
"==>".

1 files changed, 16 insertions, 3 deletions

diff --git a/trappy/sched.py b/trappy/sched.py
index ca00268..693be99 100644
--- a/trappy/sched.py
+++ b/trappy/sched.py
@@ -98,11 +98,24 @@ SchedWakeupNew = register_dynamic_ftrace("SchedWakeupNew", "sched_wakeup_new:",
                                          "sched", parse_raw=True)
 """Register SchedWakeupNew Event"""
 
-SchedSwitch = register_dynamic_ftrace("SchedSwitch", "sched_switch", "sched",
-                                      parse_raw=True)
-"""Register SchedSwitch Event"""
 # pylint: enable=invalid-name
 
+class SchedSwitch(Base):
+    """Parse sched_switch"""
+
+    unique_word = "sched_switch:"
+
+    def __init__(self):
+        super(SchedSwitch, self).__init__(parse_raw=True)
+
+    def create_dataframe(self):
+        self.data_array = [line.replace(" ==> ", " ", 1)
+                           for line in self.data_array]
+
+        super(SchedSwitch, self).create_dataframe()
+
+register_ftrace_parser(SchedSwitch, "sched")
+
 class SchedCpuFrequency(Base):
     """Corresponds to Linux kernel trace event power/cpu_frequency"""