From 1b97e12b1434ee135d7b0c2986e946e8858ba712 Mon Sep 17 00:00:00 2001 From: Lukas Prause Date: Mon, 23 Jan 2023 12:41:49 +0100 Subject: [PATCH] Extends color list for cell plotting. --- plot_single_transmission_timeline.py | 5 ++++- plot_transmission_timeline.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/plot_single_transmission_timeline.py b/plot_single_transmission_timeline.py index b968217..904a024 100755 --- a/plot_single_transmission_timeline.py +++ b/plot_single_transmission_timeline.py @@ -121,9 +121,12 @@ if __name__ == "__main__": color_list = None 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(): 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") p3, = twin2.plot(transmission_df["ack_rtt"].dropna(), color="red", linestyle="dashdot", label="ACK RTT") diff --git a/plot_transmission_timeline.py b/plot_transmission_timeline.py index 1634089..d6d54fa 100755 --- a/plot_transmission_timeline.py +++ b/plot_transmission_timeline.py @@ -205,9 +205,12 @@ if __name__ == "__main__": color_list = None 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(): 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") p2, = twin1.plot(transmission_df["downlink_cqi"], "--", color="green", label="CQI")