added figure

figure.py

@@ -0,0 +1,93 @@
+import time
+import re
+import matplotlib.pyplot as plt
+import numpy as np
+
+def timestr_to_timestamp(timestr):
+    return time.mktime(time.strptime(timestr, "%Y-%m-%d %H:%M:%S"))
+
+
+def draw_figure(file_suffix):
+    f_ls = "./log/link_state_log_" + file_suffix
+    f_tr = "./log/transmission_log_" + file_suffix
+
+    # read link_state data
+    timestr, bw, delay, loss = [], [], [], []
+    with open(f_ls, 'r') as f:
+        for line in f:
+            line = line.strip()
+            tstr, b, d, l, c = line.split("\t")
+            timestr.append(tstr)
+            bw.append(int(b))
+            try:
+                delay.append(int(d[:-2]))
+            except:
+                delay.append(150)
+            b = int(l[:-1])
+            loss.append(b if b < 10 else 10)    
+    ls_timestamp = [timestr_to_timestamp(tstr) for tstr in timestr]
+    min_ls_tstamp = min(ls_timestamp)
+    ls_timestamp = [tstamp - min_ls_tstamp for tstamp in ls_timestamp]
+    scaled_loss = [20 * l for l in loss] # scale by 20x
+
+    # read trasmission_log data
+    tr_timestamp, rate = [], []
+    with open(f_tr, 'r') as f:
+        for line in f:
+            line = line.strip()
+            l_tstr, u_tstr, byte_n, r = line.split("\t")
+            l_tstamp = timestr_to_timestamp(l_tstr)
+            u_tstamp = timestr_to_timestamp(u_tstr)
+            tstamp = (l_tstamp + u_tstamp) / 2.
+            tr_timestamp.append(tstamp)
+            r = int(r.split('.')[0])
+            rate.append(r)
+    min_tr_tstamp = min(tr_timestamp)
+    tr_timestamp = [tstamp - min_tr_tstamp for tstamp in tr_timestamp]
+
+    # plot bw, delay, loss and transmit rate
+    suffix = re.sub(r'(\d+)', '', file_suffix)
+    plt.xlabel("time/s") 
+    plt.title("bw, delay, loss and transmit rate for " + suffix.upper())
+    plt.plot(ls_timestamp, bw, label='bandwidth', color="lightgreen", linestyle=":")
+    plt.plot(ls_timestamp, delay, label='delay', color="pink", linestyle=":")
+    plt.plot(ls_timestamp, scaled_loss, label='loss(scaled by 20x)', color="cyan", linestyle=":")
+    plt.plot(tr_timestamp, rate, label='rate', color="red", linestyle="-")
+    plt.legend(loc='upper right')
+    plt.savefig("./figure/" + suffix + ".jpg")
+    plt.close()
+
+
+def draw_rate_cmp_figure(file_suffixes):
+    color_arr = ["lightblue", "lightgreen", "red", "gold"]
+    linestyle_arr = [':', ':', '-', '-']
+
+    for color, linestyle, file_suffix in zip(color_arr, linestyle_arr, file_suffixes):
+        f_tr = "./log/transmission_log_" + file_suffix
+        tr_timestamp, rate = [], []
+        with open(f_tr, 'r') as f:
+            for line in f:
+                line = line.strip()
+                l_tstr, u_tstr, byte_n, r = line.split("\t")
+                l_tstamp = timestr_to_timestamp(l_tstr)
+                u_tstamp = timestr_to_timestamp(u_tstr)
+                tstamp = (l_tstamp + u_tstamp) / 2.
+                tr_timestamp.append(tstamp)
+                r = int(r.split('.')[0])
+                rate.append(r)
+        min_tr_tstamp = min(tr_timestamp)
+        tr_timestamp = [tstamp - min_tr_tstamp for tstamp in tr_timestamp]
+
+        suffix = re.sub(r'(\d+)', '', file_suffix)
+        plt.plot(tr_timestamp, rate, label=suffix+' rate', color=color, linestyle=linestyle)
+
+    plt.legend(loc='upper right')
+    plt.savefig("./figure/cmp.jpg")
+    plt.close()
+
+
+if __name__ == "__main__":
+    file_suffixes = ["reno", "cubic", "bbr1", "mybbr10"]
+    draw_rate_cmp_figure(file_suffixes)
+    for file_suffix in file_suffixes:
+        draw_figure(file_suffix)