perf: some obs were not converted to recordswip-perf

1 files changed, 25 insertions, 28 deletions

diff --git a/unit-trace b/unit-trace
index b4e80f4..9c50488 100755
--- a/unit-trace
+++ b/unit-trace
@@ -94,31 +94,28 @@ if options.gedf is True:
 # import itertools
 # stream1, stream2, stream3 = itertools.tee(stream,3)
 
-def main():
-    # # Call standard out printer
-    # if options.stdout is True:
-    #     from unit_trace import stdout_printer
-    #     stdout_printer.stdout_printer(stream1)
-
-    # # Print G_EDF inversion statistics
-    # if options.num_inversions > -1:
-    #     if options.gedf is not True:
-    #         import sys
-    #         sys.stderr.write("You must enable the G-EDF test module to print" +
-    #             " G-EDF inversion statistics\n")
-    #     else:
-    #         from unit_trace import gedf_inversion_stat_printer
-    #         gedf_inversion_stat_printer.gedf_inversion_stat_printer(stream2,options.num_inversions)
-
-    # Call visualizer
-    if options.visualize is True:
-        from unit_trace import viz
-        try:
-            nsec_time_per_maj = viz.util.parse_time(options.time_per_maj)
-            viz.visualizer.visualizer(stream, nsec_time_per_maj)
-        except ValueError:
-            import sys
-            sys.stderr.write("Time expression should be of the form [TIME] [UNIT]")
-
-import cProfile
-cProfile.run('main()')
+# Call standard out printer
+if options.stdout is True:
+    from unit_trace import stdout_printer
+    stdout_printer.stdout_printer(stream1)
+
+# Print G_EDF inversion statistics
+if options.num_inversions > -1:
+    if options.gedf is not True:
+        import sys
+        sys.stderr.write("You must enable the G-EDF test module to print" +
+            " G-EDF inversion statistics\n")
+    else:
+        from unit_trace import gedf_inversion_stat_printer
+        gedf_inversion_stat_printer.gedf_inversion_stat_printer(stream2,options.num_inversions)
+
+# Call visualizer
+if options.visualize is True:
+    from unit_trace import viz
+    try:
+        nsec_time_per_maj = viz.util.parse_time(options.time_per_maj)
+        viz.visualizer.visualizer(stream, nsec_time_per_maj)
+    except ValueError:
+        import sys
+        sys.stderr.write("Time expression should be of the form [TIME] [UNIT]")
+