Adds cooloring for cell ids.

This commit is contained in:
2023-01-13 10:24:42 +01:00
parent c4ea007aa2
commit e924df98a5

View File

@@ -5,6 +5,7 @@ from argparse import ArgumentParser
from math import ceil from math import ceil
from time import sleep from time import sleep
import matplotlib
import pandas as pd import pandas as pd
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from mpl_toolkits import axisartist from mpl_toolkits import axisartist
@@ -171,11 +172,26 @@ if __name__ == "__main__":
left_index=True, left_index=True,
) )
cc_algo = transmission_df["congestion_control"].iloc[0]
cc_algo = cc_algo.upper()
transmission_direction = transmission_df["direction"].iloc[0]
# transmission timeline
scaley = 1.5 scaley = 1.5
scalex = 1.0 scalex = 1.0
plt.figure(figsize=[6.4 * scaley, 4.8 * scalex]) plt.figure(figsize=[6.4 * scaley, 4.8 * scalex])
plt.title("{} with {}".format(transmission_direction, cc_algo))
host = host_subplot(111, axes_class=axisartist.Axes) host = host_subplot(111, axes_class=axisartist.Axes)
cmap = matplotlib.cm.get_cmap("Set3")
for c in transmission_df["cellID"].unique():
bounds = transmission_df[["cellID"]].groupby("color").agg(["min", "max"]).loc[c]
host.axvspan(bounds.min(), bounds.max() + 1, alpha=0.3, color=cmap.colors[c])
plt.subplots_adjust() plt.subplots_adjust()
# additional y axes # additional y axes
@@ -191,20 +207,17 @@ if __name__ == "__main__":
par12.axis["right"].toggle(all=True) par12.axis["right"].toggle(all=True)
# par13.axis["right"].toggle(all=True) # par13.axis["right"].toggle(all=True)
host.plot(transmission_df["goodput"], "-", color="blue", label="goodput" ) host.plot(transmission_df["goodput"], "-", color="blue", label="goodput")
host.set_xlabel("datetime") host.set_xlabel("datetime")
host.set_ylabel("goodput [Mbps]") host.set_ylabel("goodput [Mbps]")
#host.set_ylim([0, 13])
#host.set_yscale("log")
#host.set_yscale("log")
#host.set_yscale("log")
#host.set_yscale("log")
par11.plot(transmission_df["downlink_cqi"], "--", color="green", label="CQI") par11.plot(transmission_df["downlink_cqi"], "--", color="green", label="CQI")
par11.set_ylabel("CQI") par11.set_ylabel("CQI")
par11.set_ylim([0, 15]) par11.set_ylim([0, 15])
par12.plot() par12.plot(transmission_df["ach_rtt"], "-.", color="red", label="ACK RTT")
par12.set_ylabel("ACK RTT [s]")
par12.set_ylim([0, 5])
if args.save: if args.save:
plt.savefig("{}timeline_plot.pdf".format(args.save)) plt.savefig("{}timeline_plot.pdf".format(args.save))
@@ -218,7 +231,8 @@ if __name__ == "__main__":
plot_cdf(transmission_df, "goodput") plot_cdf(transmission_df, "goodput")
plt.xlabel("goodput [mbps]") plt.xlabel("goodput [mbps]")
plt.ylabel("CDF") plt.ylabel("CDF")
plt.legend(["BBR"]) plt.legend([cc_algo])
plt.title("{} with {}".format(transmission_direction, cc_algo))
if args.save: if args.save:
plt.savefig("{}{}_cdf_plot.pdf".format(args.save, "goodput")) plt.savefig("{}{}_cdf_plot.pdf".format(args.save, "goodput"))
@@ -232,7 +246,8 @@ if __name__ == "__main__":
plot_cdf(transmission_df, "ack_rtt") plot_cdf(transmission_df, "ack_rtt")
plt.xlabel("ACK RTT [s]") plt.xlabel("ACK RTT [s]")
plt.ylabel("CDF") plt.ylabel("CDF")
plt.legend(["BBR"]) plt.legend([cc_algo])
plt.title("{} with {}".format(transmission_direction, cc_algo))
if args.save: if args.save:
plt.savefig("{}{}_cdf_plot.pdf".format(args.save, "ack_rtt")) plt.savefig("{}{}_cdf_plot.pdf".format(args.save, "ack_rtt"))