Extends color list for cell plotting.

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")

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")