Tests: Multi-trace stream intersection test

[babeltrace.git] / bindings / python / examples / sched_switch.py

diff --git a/bindings/python/examples/sched_switch.py b/bindings/python/examples/sched_switch.py
index 8c475565ce7e56981036d20fe3e56d3db9fb38ad..0deb4a5320af4a13f70c17cb3a89df820364b93b 100644 (file)
--- a/bindings/python/examples/sched_switch.py
+++ b/bindings/python/examples/sched_switch.py
@@ -16,6 +16,14 @@
 #
 # The above copyright notice and this permission notice shall be included in
 # all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
 
 # The script takes one optional argument (pid)
 # The script will read events based on pid and
@@ -26,44 +34,46 @@
 # The trace needs PID context (lttng add-context -k -t pid)
 
 import sys
-from babeltrace import *
+import babeltrace.reader
+import babeltrace.common
+
 
 if len(sys.argv) < 2 or len(sys.argv) > 3:
-       raise TypeError("Usage: python sched_switch.py [pid] path/to/trace")
+    raise TypeError("Usage: python sched_switch.py [pid] path/to/trace")
 elif len(sys.argv) == 3:
-       filterPID = True
+    filterPID = True
 else:
-       filterPID = False
+    filterPID = False
 
-traces = TraceCollection()
-ret = traces.add_trace(sys.argv[len(sys.argv)-1], "ctf")
+traces = babeltrace.reader.TraceCollection()
+ret = traces.add_trace(sys.argv[len(sys.argv) - 1], "ctf")
 if ret is None:
-       raise IOError("Error adding trace")
+    raise IOError("Error adding trace")
 
 for event in traces.events:
-       if event.name != "sched_switch":
-               continue
+    if event.name != "sched_switch":
+        continue
 
-       # Getting PID
-       pid = event.field_with_scope("pid", CTFScope.STREAM_EVENT_CONTEXT)
-       if pid is None:
-               print("ERROR: Missing PID info for sched_switch")
-               continue # Next event
+    # Getting PID
+    pid = event.field_with_scope("pid", babeltrace.common.CTFScope.STREAM_EVENT_CONTEXT)
+    if pid is None:
+        print("ERROR: Missing PID info for sched_switch")
+        continue  # Next event
 
-       if filterPID and (pid != long(sys.argv[1])):
-               continue # Next event
+    if filterPID and (pid != long(sys.argv[1])):
+        continue  # Next event
 
-       prev_comm = event["prev_comm"]
-       prev_tid = event["prev_tid"]
-       prev_prio = event["prev_prio"]
-       prev_state = event["prev_state"]
-       next_comm = event["next_comm"]
-       next_tid = event["next_tid"]
-       next_prio = event["next_prio"]
+    prev_comm = event["prev_comm"]
+    prev_tid = event["prev_tid"]
+    prev_prio = event["prev_prio"]
+    prev_state = event["prev_state"]
+    next_comm = event["next_comm"]
+    next_tid = event["next_tid"]
+    next_prio = event["next_prio"]
 
-       # Output
-       print("sched_switch, pid = {}, TS = {}, prev_comm = {},\n\t"
-               "prev_tid = {}, prev_prio = {}, prev_state = {},\n\t"
-               "next_comm = {}, next_tid = {}, next_prio = {}".format(
-               pid, event.timestamp, prev_comm, prev_tid,
-               prev_prio, prev_state, next_comm, next_tid, next_prio))
+    # Output
+    print("sched_switch, pid = {}, TS = {}, prev_comm = {},\n\t"
+          "prev_tid = {}, prev_prio = {}, prev_state = {},\n\t"
+          "next_comm = {}, next_tid = {}, next_prio = {}".format(
+              pid, event.timestamp, prev_comm, prev_tid,
+              prev_prio, prev_state, next_comm, next_tid, next_prio))