Changeset f34f95c for benchmark/plot.py

Timestamp:

Apr 5, 2022, 3:48:25 PM (2 years ago)

Author:

Thierry Delisle <tdelisle@…>

Branches:

ADT, ast-experimental, enum, master, pthread-emulation, qualifiedEnum

Children:

11a1240

Parents:

0c3aa67

Message:

Updated plot script to support non-interactive mode

File:

• benchmark/plot.py (modified) (4 diffs)

benchmark/plot.py

@@ -40 +40 @@
 }
 
-def plot(data, x, y):
+def plot(data, x, y, out):
         fig, ax = plt.subplots()
         colors = itertools.cycle(['#0095e3','#006cb4','#69df00','#0aa000','#fb0300','#e30002','#fd8f00','#ff7f00','#8f00d6','#4b009a','#ffff00','#b13f00'])
@@ -67 +67 @@
         ax.yaxis.set_major_formatter( EngFormatter(unit=field_names[y].unit) )
         plt.legend(loc='upper left')
-        plt.show()
+        if out:
+                plt.savefig(out)
+        else:
+                plt.show()
 
 
@@ -75 +78 @@
         parser = parser = argparse.ArgumentParser(description='Python Script to draw R.M.I.T. results')
         parser.add_argument('-f', '--file', nargs='?', type=argparse.FileType('r'), default=sys.stdin)
+        parser.add_argument('-o', '--out', nargs='?', type=str, default=None)
+        parser.add_argument('-y', nargs='?', type=str, default="")
 
         try:
@@ -103 +108 @@
                         fields.add(label)
 
-        print(series)
-        print("fields")
-        for f in fields:
-                print("{}".format(f))
+        if not options.out :
+                print(series)
+                print("fields")
+                for f in fields:
+                        print("{}".format(f))
 
-        plot(data, "Number of processors", "ns per ops")
+        if options.y and options.y in field_names.keys():
+                plot(data, "Number of processors", options.y, options.out)
+        else:
+                if options.y:
+                        print("Could not find key '{}', defaulting to 'ns per ops'".format(options.y))
+                plot(data, "Number of processors", "ns per ops", options.out)