Added argument to omit line plots

examples/interference/visualizer.py

@@ -6,6 +6,11 @@
 import glob
 import os
 import sys
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument("--noline", action="store_true", help="Omit line plot generation")
+args = parser.parse_args()
 
 file_list = []
 script_dir = os.path.dirname(__file__)
@@ -95,22 +100,25 @@
     plt.savefig(if_phy + "_" + payload_size + "_" + sir + ".png")
     plt.close()
 
-styles = ['s-', 'o-', '^-', 'x-', 'd-', 'h-']
-# NOTE loops over each (if_phy, payload_size, offset) tuple in line_info
-for if_phy, payload_size, offset in line_info:
-    title = "Interference: " + if_phy + ", Payload: " + payload_size + ", Offset: " + offset
-    # NOTE sort (alfabetically) by sir
-    line_info[if_phy, payload_size, offset] = dict(sorted(line_info[if_phy, payload_size, offset].items(), key=lambda x: x[0].lower()))
-    # NOTE first transpose the dataframe (i.e., reflect the dataframe over its main 
-    # diagonal by writing rows as columns and vice-versa) and plot it as a line plot
-    df = pd.DataFrame(line_info[if_phy, payload_size, offset]).T
-    ax = df.plot(kind='line', figsize=(10,7), style=styles, xticks=range(len(list(df.index.values))))
-    ax.legend(loc='lower right')
-    ax.set_xlabel('SIR between TX and IF')
-    ax.set_ylabel('Packet Reception Rate [%]')
-    ax.set_ylim([0, 120])
-    ax.set_yticks([0,20,40,60,80,100])
-    ax.set_xlim([-0.05,len(list(df.index.values)) - 0.95])
-    plt.title(title)
-    plt.savefig(if_phy + "_" + payload_size + "_" + offset + ".png")
-    plt.close()
+if not args.noline:
+    styles = ['s-', 'o-', '^-', 'x-', 'd-', 'h-']
+    # NOTE loops over each (if_phy, payload_size, offset) tuple in line_info
+    for if_phy, payload_size, offset in line_info:
+        title = "Interference: " + if_phy + ", Payload: " + payload_size + ", Offset: " + offset
+        # NOTE sort (alfabetically) by sir
+        line_info[if_phy, payload_size, offset] = dict(sorted(line_info[if_phy, payload_size, offset].items(), key=lambda x: x[0].lower()))
+        # NOTE first transpose the dataframe (i.e., reflect the dataframe over its main 
+        # diagonal by writing rows as columns and vice-versa) and plot it as a line plot
+        df = pd.DataFrame(line_info[if_phy, payload_size, offset]).T
+        ax = df.plot(kind='line', figsize=(10,7), style=styles, xticks=range(len(list(df.index.values))))
+        ax.legend(loc='lower right')
+        ax.set_xlabel('SIR between TX and IF')
+        ax.set_ylabel('Packet Reception Rate [%]')
+        ax.set_ylim([0, 120])
+        ax.set_yticks([0,20,40,60,80,100])
+        ax.set_xlim([-0.05,len(list(df.index.values)) - 0.95])
+        plt.title(title)
+        plt.savefig(if_phy + "_" + payload_size + "_" + offset + ".png")
+        plt.close()
+else:
+    pass