langspielplatte/measurement-scripts
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Extends color list for cell plotting.

Browse Source
This commit is contained in:
Lukas Prause
2023-01-23 12:41:49 +01:00
parent e1b0cfa32a
commit 1b97e12b14
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
plot_single_transmission_timeline.py
View File

@@ -121,9 +121,12 @@ if __name__ == "__main__":
color_list = None color_list = None
cmap = matplotlib.cm.get_cmap("Set3") cmap = matplotlib.cm.get_cmap("Set3")
unique_cells = transmission_df["cell_color"].unique()
color_list = cmap * (round(len(unique_cells) / len(cmap)) + 1)
for c in transmission_df["cell_color"].unique(): for c in transmission_df["cell_color"].unique():
bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c] bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c]
ax.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[c]) ax.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=color_list[c])
p4, = twin3.plot(transmission_df["snd_cwnd"].dropna(), color="lime", linestyle="dashed", label="cwnd") p4, = twin3.plot(transmission_df["snd_cwnd"].dropna(), color="lime", linestyle="dashed", label="cwnd")
p3, = twin2.plot(transmission_df["ack_rtt"].dropna(), color="red", linestyle="dashdot", label="ACK RTT") p3, = twin2.plot(transmission_df["ack_rtt"].dropna(), color="red", linestyle="dashdot", label="ACK RTT")

5
plot_transmission_timeline.py
View File

@@ -205,9 +205,12 @@ if __name__ == "__main__":
color_list = None color_list = None
cmap = matplotlib.cm.get_cmap("Set3") cmap = matplotlib.cm.get_cmap("Set3")
unique_cells = transmission_df["cell_color"].unique()
color_list = cmap * (round(len(unique_cells) / len(cmap)) + 1)
for c in transmission_df["cell_color"].unique(): for c in transmission_df["cell_color"].unique():
bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c] bounds = transmission_df[["index", "cell_color"]].groupby("cell_color").agg(["min", "max"]).loc[c]
ax.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=cmap.colors[c]) ax.axvspan(bounds.min(), bounds.max(), alpha=0.3, color=color_list[c])
p1, = ax.plot(transmission_df["goodput_rolling"], "-", color="blue", label="goodput") p1, = ax.plot(transmission_df["goodput_rolling"], "-", color="blue", label="goodput")
p2, = twin1.plot(transmission_df["downlink_cqi"], "--", color="green", label="CQI") p2, = twin1.plot(transmission_df["downlink_cqi"], "--", color="green", label="CQI")