Adds srtts to csv.

calc_bandwidth_goodput_csv.py

@@ -0,0 +1,198 @@
+#!/usr/bin/env python3
+import math
+import multiprocessing
+import os
+from argparse import ArgumentParser
+
+import matplotlib
+import numpy as np
+import pandas as pd
+import matplotlib.pyplot as plt
+
+import seaborn as sns
+
+sns.set()
+#sns.set(font_scale=1.5)
+
+tex_fonts = {
+    "pgf.texsystem": "lualatex",
+    # "legend.fontsize": "x-large",
+    # "figure.figsize": (15, 5),
+    "axes.labelsize": 15,  # "small",
+    # "axes.titlesize": "x-large",
+    "xtick.labelsize": 15,  # "small",
+    "ytick.labelsize": 15,  # "small",
+    "legend.fontsize": 15,
+    "axes.formatter.use_mathtext": True,
+    "mathtext.fontset": "dejavusans",
+}
+
+
+# plt.rcParams.update(tex_fonts)
+
+
+def convert_cellid(value):
+    if isinstance(value, str):
+        try:
+            r = int(value.split(" ")[-1].replace("(", "").replace(")", ""))
+            return r
+        except Exception as e:
+            return -1
+    else:
+        return int(-1)
+
+
+if __name__ == "__main__":
+    parser = ArgumentParser()
+    parser.add_argument("-s", "--serial_file", required=True, help="Serial csv file.")
+    parser.add_argument(
+        "-p", "--pcap_csv_folder", required=True, help="PCAP csv folder."
+    )
+    parser.add_argument("--save", required=True, help="Location to save pdf file.")
+    parser.add_argument(
+        "-i",
+        "--interval",
+        default=10,
+        type=int,
+        help="Time interval for rolling window.",
+    )
+
+    args = parser.parse_args()
+
+    pcap_csv_list = list()
+    for filename in os.listdir(args.pcap_csv_folder):
+        if filename.endswith(".csv") and "tcp" in filename:
+            pcap_csv_list.append(filename)
+
+    counter = 1
+    if len(pcap_csv_list) == 0:
+        print("No CSV files found.")
+
+    pcap_csv_list.sort(key=lambda x: int(x.split("_")[-1].replace(".csv", "")))
+
+    concat_frame = None
+
+    for csv in pcap_csv_list:
+        print(
+            "\rProcessing {} out of {} CSVs.\t({}%)\t".format(
+                counter, len(pcap_csv_list), math.floor(counter / len(pcap_csv_list))
+            )
+        )
+
+        # try:
+        transmission_df = pd.read_csv(
+            "{}{}".format(args.pcap_csv_folder, csv),
+            dtype=dict(is_retranmission=bool, is_dup_ack=bool),
+        )
+
+        transmission_df["datetime"] = pd.to_datetime(
+            transmission_df["datetime"]
+        ) - pd.Timedelta(hours=1)
+        transmission_df = transmission_df.set_index("datetime")
+        transmission_df.index = pd.to_datetime(transmission_df.index)
+        transmission_df = transmission_df.sort_index()
+
+        # srtt to [s]
+        transmission_df["srtt"] = transmission_df["srtt"].apply(lambda x: x / 10 ** 6)
+
+        # key for columns and level for index
+        transmission_df["goodput"] = (
+            transmission_df["payload_size"]
+            .groupby(pd.Grouper(level="datetime", freq="{}s".format(args.interval)))
+            .transform("sum")
+        )
+        transmission_df["goodput"] = transmission_df["goodput"].apply(
+            lambda x: ((x * 8) / args.interval) / 10 ** 6
+        )
+
+        transmission_df["goodput_rolling"] = (
+            transmission_df["payload_size"].rolling("{}s".format(args.interval)).sum()
+        )
+        transmission_df["goodput_rolling"] = transmission_df["goodput_rolling"].apply(
+            lambda x: ((x * 8) / args.interval) / 10 ** 6
+        )
+
+        # set meta values and remove all not needed columns
+        cc_algo = transmission_df["congestion_control"].iloc[0]
+        cc_algo = cc_algo.upper()
+        transmission_direction = transmission_df["direction"].iloc[0]
+
+        # transmission_df = transmission_df.filter(["goodput", "datetime", "ack_rtt", "goodput_rolling", "snd_cwnd"])
+
+        # read serial csv
+        serial_df = pd.read_csv(
+            args.serial_file, converters={"Cell_ID": convert_cellid}
+        )
+        serial_df["datetime"] = pd.to_datetime(serial_df["datetime"]) - pd.Timedelta(
+            hours=1
+        )
+        serial_df = serial_df.set_index("datetime")
+        serial_df.index = pd.to_datetime(serial_df.index)
+        serial_df.sort_index()
+
+        # print(serial_df["Cell_ID"])
+
+        # serial_df["Cell_ID"] = serial_df["Cell_ID"].apply(
+        #    lambda x: int(x.split(" ")[-1].replace("(", "").replace(")", "")))
+
+        transmission_df = pd.merge_asof(
+            transmission_df,
+            serial_df,
+            tolerance=pd.Timedelta("1s"),
+            right_index=True,
+            left_index=True,
+        )
+
+        #transmission_df.index = transmission_df["arrival_time"]
+
+        # replace 0 in RSRQ with Nan
+        transmission_df["NR5G_RSRQ_(dB)"] = transmission_df["NR5G_RSRQ_(dB)"].replace(
+            0, np.NaN
+        )
+        transmission_df["RSRQ_(dB)"] = transmission_df["RSRQ_(dB)"].replace(0, np.NaN)
+
+        # filter active state
+        for i in range(1, 5):
+            transmission_df["LTE_SCC{}_effective_bw".format(i)] = transmission_df[
+                "LTE_SCC{}_bw".format(i)
+            ]
+
+            mask = transmission_df["LTE_SCC{}_state".format(i)].isin(["ACTIVE"])
+            transmission_df["LTE_SCC{}_effective_bw".format(i)] = transmission_df[
+                "LTE_SCC{}_effective_bw".format(i)
+            ].where(mask, other=0)
+
+        # filter if sc is usesd for uplink
+        for i in range(1, 5):
+            mask = transmission_df["LTE_SCC{}_UL_Configured".format(i)].isin([False])
+            transmission_df["LTE_SCC{}_effective_bw".format(i)] = transmission_df[
+                "LTE_SCC{}_effective_bw".format(i)
+            ].where(mask, other=0)
+
+        # sum all effective bandwidth for 5G and 4G
+        transmission_df["SCC1_NR5G_effective_bw"] = transmission_df[
+            "SCC1_NR5G_bw"
+        ].fillna(0)
+
+        transmission_df["lte_effective_bw_sum"] = (
+                transmission_df["LTE_SCC1_effective_bw"].fillna(0)
+                + transmission_df["LTE_SCC2_effective_bw"].fillna(0)
+                + transmission_df["LTE_SCC3_effective_bw"].fillna(0)
+                + transmission_df["LTE_SCC4_effective_bw"].fillna(0)
+                + transmission_df["LTE_bw"].fillna(0))
+        transmission_df["nr_effective_bw_sum"] = transmission_df["SCC1_NR5G_effective_bw"]
+
+        transmission_df["effective_bw_sum"] = transmission_df["nr_effective_bw_sum"] + transmission_df[
+            "lte_effective_bw_sum"]
+
+        transmission_df = transmission_df.filter(["goodput", "effective_bw_sum", "srtt"])
+        transmission_df = transmission_df.reset_index(drop=True)
+
+        if concat_frame is None:
+            concat_frame = transmission_df
+        else:
+            concat_frame = pd.concat([concat_frame, transmission_df])
+
+    concat_frame.to_csv("{}_concat_bw_gp.csv".format(args.save))
+
+

plot_gps_csv.py

@@ -8,6 +8,12 @@ import geopandas as gpd
 import contextily as cx
 import matplotlib.pyplot as plt
 
+import seaborn as sns
+
+sns.set()
+#sns.set(font_scale=1.5)
+
+
 if __name__ == "__main__":
     parser = ArgumentParser()
     parser.add_argument("-f", "--file", required=True, help="Messfahrt csv")
@@ -39,14 +45,20 @@ if __name__ == "__main__":
 
     df_wm = gdf.to_crs(epsg=3857)
     ax2 = df_wm.plot()
-    ax2 = df_wm.plot(args.column, cmap="hot", legend=args.no_legend, ax=ax2)
+    ax2 = df_wm.plot(column=args.column, cmap="hot", legend=args.no_legend, ax=ax2, legend_kwds={"label": args.label},)
     # ax2 = df_wm.plot.scatter(x="longitude", y="latitude", c="kmh", cmap="hot")
     # zoom 17 is pretty
     cx.add_basemap(ax2, source=cx.providers.OpenStreetMap.Mapnik, zoom=17)
 
     #    gdf.plot()
     ax2.set_axis_off()
-    ax2.set_title(args.label if args.label else args.column)
+    if not args.no_legend:
+        ax2.set_title(args.label if args.label else args.column)
+    else:
+        fig = ax2.figure
+        cb_ax = fig.axes[0]
+        cb_ax.set_label(args.label)
+        cb_ax.tick_params(labelsize=30)
 
     if args.show_providerinfo:
         #####################################
@@ -68,6 +80,6 @@ if __name__ == "__main__":
     # df.plot(x="longitude", y="latitude", kind="scatter", colormap="YlOrRd")
 
     if args.save:
-        plt.savefig("{}gps_plot.pdf".format(args.save))
+        plt.savefig("{}gps_plot.eps".format(args.save), bbox_inches="tight")
     else:
         plt.show()

plot_single_transmission_EM9190.py

@@ -9,8 +9,10 @@ import numpy as np
 import pandas as pd
 import matplotlib.pyplot as plt
 
-# Using seaborn's style
-# plt.style.use('seaborn')
+import seaborn as sns
+
+sns.set()
+#sns.set(font_scale=1.5)
 
 tex_fonts = {
     "pgf.texsystem": "lualatex",
@@ -25,6 +27,7 @@ tex_fonts = {
     "mathtext.fontset": "dejavusans",
 }
 
+
 # plt.rcParams.update(tex_fonts)
 
 
@@ -46,6 +49,7 @@ if __name__ == "__main__":
         "-p", "--pcap_csv_folder", required=True, help="PCAP csv folder."
     )
     parser.add_argument("--save", required=True, help="Location to save pdf file.")
+    parser.add_argument("--fancy", action="store_true", help="Create fancy plot.")
     parser.add_argument(
         "-i",
         "--interval",
@@ -88,7 +92,7 @@ if __name__ == "__main__":
         transmission_df = transmission_df.sort_index()
 
         # srtt to [s]
-        transmission_df["srtt"] = transmission_df["srtt"].apply(lambda x: x / 10**6)
+        transmission_df["srtt"] = transmission_df["srtt"].apply(lambda x: x / 10 ** 6)
 
         # key for columns and level for index
         transmission_df["goodput"] = (
@@ -97,14 +101,14 @@ if __name__ == "__main__":
             .transform("sum")
         )
         transmission_df["goodput"] = transmission_df["goodput"].apply(
-            lambda x: ((x * 8) / args.interval) / 10**6
+            lambda x: ((x * 8) / args.interval) / 10 ** 6
         )
 
         transmission_df["goodput_rolling"] = (
             transmission_df["payload_size"].rolling("{}s".format(args.interval)).sum()
         )
         transmission_df["goodput_rolling"] = transmission_df["goodput_rolling"].apply(
-            lambda x: ((x * 8) / args.interval) / 10**6
+            lambda x: ((x * 8) / args.interval) / 10 ** 6
         )
 
         # set meta values and remove all not needed columns
@@ -145,10 +149,8 @@ if __name__ == "__main__":
             0, np.NaN
         )
         transmission_df["RSRQ_(dB)"] = transmission_df["RSRQ_(dB)"].replace(0, np.NaN)
-        # stacked plot for bandwidth
-        # transmission_df["lte_bw_sum"] = transmission_df["bw_sum"] - transmission_df["NR5G_dl_bw"]
-        # transmission_df["nr_bw_sum"] = transmission_df["NR5G_dl_bw"]
 
+        # filter active state
         for i in range(1, 5):
             transmission_df["LTE_SCC{}_effective_bw".format(i)] = transmission_df[
                 "LTE_SCC{}_bw".format(i)
@@ -159,29 +161,37 @@ if __name__ == "__main__":
                 "LTE_SCC{}_effective_bw".format(i)
             ].where(mask, other=0)
 
-            # df = df.filter(["LTE_SCC1_state", "LTE_SCC1_bw", "LTE_SCC1_effective_bw"])
+        # filter if sc is usesd for uplink
+        for i in range(1, 5):
+            mask = transmission_df["LTE_SCC{}_UL_Configured".format(i)].isin([False])
+            transmission_df["LTE_SCC{}_effective_bw".format(i)] = transmission_df[
+                "LTE_SCC{}_effective_bw".format(i)
+            ].where(mask, other=0)
+
+        # sum all effective bandwidth for 5G and 4G
         transmission_df["SCC1_NR5G_effective_bw"] = transmission_df[
             "SCC1_NR5G_bw"
         ].fillna(0)
-        transmission_df["effective_bw_sum"] = (
-            transmission_df["SCC1_NR5G_effective_bw"]
-            + transmission_df["LTE_SCC1_effective_bw"]
-            + transmission_df["LTE_SCC2_effective_bw"]
-            + transmission_df["LTE_SCC3_effective_bw"]
-            + transmission_df["LTE_SCC4_effective_bw"]
-            + transmission_df["LTE_bw"]
-        )
 
-        transmission_df["lte_effective_bw_sum"] = transmission_df["effective_bw_sum"] - transmission_df["SCC1_NR5G_effective_bw"]
+        transmission_df["lte_effective_bw_sum"] = (
+                transmission_df["LTE_SCC1_effective_bw"].fillna(0)
+                + transmission_df["LTE_SCC2_effective_bw"].fillna(0)
+                + transmission_df["LTE_SCC3_effective_bw"].fillna(0)
+                + transmission_df["LTE_SCC4_effective_bw"].fillna(0)
+                + transmission_df["LTE_bw"].fillna(0))
         transmission_df["nr_effective_bw_sum"] = transmission_df["SCC1_NR5G_effective_bw"]
 
+        transmission_df["effective_bw_sum"] = transmission_df["nr_effective_bw_sum"] + transmission_df[
+            "lte_effective_bw_sum"]
+
         # transmission timeline
         scaley = 1.5
         scalex = 1.0
-        plt.title("{} with {}".format(transmission_direction, cc_algo))
         fig, ax = plt.subplots(2, 1, figsize=[6.4 * scaley, 4.8 * scalex])
         fig.subplots_adjust(right=0.75)
-        fig.suptitle("{} with {}".format(transmission_direction, cc_algo))
+        if not args.fancy:
+            plt.title("{} with {}".format(transmission_direction, cc_algo))
+            fig.suptitle("{} with {}".format(transmission_direction, cc_algo))
         ax0 = ax[0]
         ax1 = ax0.twinx()
         ax2 = ax0.twinx()
@@ -194,10 +204,11 @@ if __name__ == "__main__":
         # Plot vertical lines
         first = True
         lte_handovers = transmission_df["Cell_ID"].dropna().diff()
+        lte_hanover_plot = None
         for index, value in lte_handovers.items():
             if value > 0:
                 if first:
-                    ax00.axvline(
+                    lte_hanover_plot = ax00.axvline(
                         index, ymin=0, ymax=1, color="skyblue", label="4G Handover"
                     )
                     first = False
@@ -208,29 +219,31 @@ if __name__ == "__main__":
         nr_handovers = (
             transmission_df["NR5G_Cell_ID"].replace(0, np.NaN).dropna().diff()
         )
+
+        nr_hanover_plot = None
         for index, value in nr_handovers.items():
             if value > 0:
                 if first:
-                    ax00.axvline(
+                    nr_hanover_plot = ax00.axvline(
                         index, ymin=0, ymax=1, color="greenyellow", label="5G Handover"
                     )
                     first = False
                 else:
                     ax00.axvline(index, ymin=0, ymax=1, color="greenyellow")
 
-        ax0.plot(
+        snd_plot = ax0.plot(
             transmission_df["snd_cwnd"].dropna(),
             color="lime",
             linestyle="dashed",
             label="cwnd",
         )
-        ax1.plot(
+        srtt_plot = ax1.plot(
             transmission_df["srtt"].dropna(),
             color="red",
             linestyle="dashdot",
             label="sRTT",
         )
-        ax2.plot(
+        goodput_plot = ax2.plot(
             transmission_df["goodput_rolling"],
             color="blue",
             linestyle="solid",
@@ -238,20 +251,20 @@ if __name__ == "__main__":
         )
         # ax2.plot(transmission_df["goodput"], color="blue", linestyle="solid", label="goodput")
 
-        ax01.plot(
+        eff_bw_plot = ax01.plot(
             transmission_df["effective_bw_sum"].dropna(),
             color="peru",
             linestyle="solid",
             label="bandwidth",
         )
-        ax01.plot(
+        lte_eff_bw_plot = ax01.plot(
             transmission_df["lte_effective_bw_sum"].dropna(),
             color="lightsteelblue",
             linestyle="solid",
             label="4G bandwidth",
             alpha=0.5,
         )
-        ax01.plot(
+        nr_eff_bw_plot = ax01.plot(
             transmission_df["nr_effective_bw_sum"].dropna(),
             color="cornflowerblue",
             linestyle="solid",
@@ -265,13 +278,13 @@ if __name__ == "__main__":
         #               labels=["4G bandwidth", "5G bandwidth"]
         #               )
 
-        ax02.plot(
+        lte_rsrq_plot = ax02.plot(
             transmission_df["RSRQ_(dB)"].dropna(),
             color="purple",
             linestyle="dotted",
             label="LTE RSRQ",
         )
-        ax00.plot(
+        nr_rsrq_plot = ax00.plot(
             transmission_df["NR5G_RSRQ_(dB)"].dropna(),
             color="magenta",
             linestyle="dotted",
@@ -283,7 +296,7 @@ if __name__ == "__main__":
 
         ax0.set_ylim(0, 5000)
         ax1.set_ylim(0, 0.3)
-        ax2.set_ylim(0, 500)
+        ax2.set_ylim(0, 600)
         ax00.set_ylim(-25, 0)
         ax01.set_ylim(0, 250)
         # second dB axis
@@ -296,12 +309,31 @@ if __name__ == "__main__":
         ax00.set_ylabel("LTE/NR RSRQ [dB]")
         # ax02.set_ylabel("LTE RSRQ [dB]")
         ax1.set_ylabel("sRTT [s]")
-        ax0.set_ylabel("cwnd")
+        ax0.set_ylabel("cwnd [MSS]")
         ax01.set_ylabel("Bandwidth [MHz]")
 
-        fig.legend(loc="lower right")
+        if args.fancy:
+            legend_frame = False
+            ax0.set_xlim([0, transmission_df.index[-1]])
+            ax00.set_xlim([0, transmission_df.index[-1]])
+            # added these three lines
+            lns_ax0 = snd_plot + srtt_plot + goodput_plot
+            labs_ax0 = [l.get_label() for l in lns_ax0]
+            ax2.legend(lns_ax0, labs_ax0, ncols=9, fontsize=9, loc="upper right", frameon=legend_frame)
+            #ax0.set_zorder(100)
 
-        plt.savefig("{}{}_plot.pdf".format(args.save, csv.replace(".csv", "")))
+            lns_ax00 = eff_bw_plot + lte_eff_bw_plot + nr_eff_bw_plot + lte_rsrq_plot + nr_rsrq_plot
+            if lte_hanover_plot:
+                lns_ax00.append(lte_hanover_plot)
+            if nr_hanover_plot:
+                lns_ax00.append(nr_hanover_plot)
+            labs_ax00 = [l.get_label() for l in lns_ax00]
+            ax02.legend(lns_ax00, labs_ax00, ncols=3, fontsize=9, loc="upper center", frameon=legend_frame)
+            #ax00.set_zorder(100)
+            plt.savefig("{}{}_plot.eps".format(args.save, csv.replace(".csv", "")), bbox_inches="tight")
+        else:
+            fig.legend(loc="lower right")
+            plt.savefig("{}{}_plot.pdf".format(args.save, csv.replace(".csv", "")), bbox_inches="tight")
         # except Exception as e:
         #    print("Error processing file: {}".format(csv))
         #    print(str(e))

plot_single_transmission_paper.py

@@ -0,0 +1,269 @@
+#!/usr/bin/env python3
+import math
+import multiprocessing
+import os
+from argparse import ArgumentParser
+
+import matplotlib
+import numpy as np
+import pandas as pd
+import matplotlib.pyplot as plt
+
+import seaborn as sns
+
+sns.set()
+#sns.set(font_scale=1.5)
+
+tex_fonts = {
+    "pgf.texsystem": "lualatex",
+    # "legend.fontsize": "x-large",
+    # "figure.figsize": (15, 5),
+    "axes.labelsize": 15,  # "small",
+    # "axes.titlesize": "x-large",
+    "xtick.labelsize": 15,  # "small",
+    "ytick.labelsize": 15,  # "small",
+    "legend.fontsize": 15,
+    "axes.formatter.use_mathtext": True,
+    "mathtext.fontset": "dejavusans",
+}
+
+
+# plt.rcParams.update(tex_fonts)
+
+
+def convert_cellid(value):
+    if isinstance(value, str):
+        try:
+            r = int(value.split(" ")[-1].replace("(", "").replace(")", ""))
+            return r
+        except Exception as e:
+            return -1
+    else:
+        return int(-1)
+
+
+if __name__ == "__main__":
+    parser = ArgumentParser()
+    parser.add_argument("-s", "--serial_file", required=True, help="Serial csv file.")
+    parser.add_argument(
+        "-p", "--pcap_csv_folder", required=True, help="PCAP csv folder."
+    )
+    parser.add_argument("--save", required=True, help="Location to save pdf file.")
+    parser.add_argument("--fancy", action="store_true", help="Create fancy plot.")
+    parser.add_argument(
+        "-i",
+        "--interval",
+        default=10,
+        type=int,
+        help="Time interval for rolling window.",
+    )
+
+    args = parser.parse_args()
+
+    pcap_csv_list = list()
+    for filename in os.listdir(args.pcap_csv_folder):
+        if filename.endswith(".csv") and "tcp" in filename:
+            pcap_csv_list.append(filename)
+
+    counter = 1
+    if len(pcap_csv_list) == 0:
+        print("No CSV files found.")
+
+    pcap_csv_list.sort(key=lambda x: int(x.split("_")[-1].replace(".csv", "")))
+
+    for csv in pcap_csv_list:
+        print(
+            "\rProcessing {} out of {} CSVs.\t({}%)\t".format(
+                counter, len(pcap_csv_list), math.floor(counter / len(pcap_csv_list))
+            )
+        )
+
+        # try:
+        transmission_df = pd.read_csv(
+            "{}{}".format(args.pcap_csv_folder, csv),
+            dtype=dict(is_retranmission=bool, is_dup_ack=bool),
+        )
+
+        transmission_df = transmission_df.set_index("datetime")
+        transmission_df.index = pd.to_datetime(transmission_df.index)
+        transmission_df = transmission_df.sort_index()
+
+        # srtt to [s]
+        transmission_df["srtt"] = transmission_df["srtt"].apply(lambda x: x / 10 ** 6)
+
+        # key for columns and level for index
+        transmission_df["goodput"] = (
+            transmission_df["payload_size"]
+            .groupby(pd.Grouper(level="datetime", freq="{}s".format(args.interval)))
+            .transform("sum")
+        )
+        transmission_df["goodput"] = transmission_df["goodput"].apply(
+            lambda x: ((x * 8) / args.interval) / 10 ** 6
+        )
+
+        transmission_df["goodput_rolling"] = (
+            transmission_df["payload_size"].rolling("{}s".format(args.interval)).sum()
+        )
+        transmission_df["goodput_rolling"] = transmission_df["goodput_rolling"].apply(
+            lambda x: ((x * 8) / args.interval) / 10 ** 6
+        )
+
+        # set meta values and remove all not needed columns
+        cc_algo = transmission_df["congestion_control"].iloc[0]
+        cc_algo = cc_algo.upper()
+        transmission_direction = transmission_df["direction"].iloc[0]
+
+        # transmission_df = transmission_df.filter(["goodput", "datetime", "ack_rtt", "goodput_rolling", "snd_cwnd"])
+
+        # read serial csv
+        serial_df = pd.read_csv(
+            args.serial_file, converters={"Cell_ID": convert_cellid}
+        )
+        serial_df = serial_df.set_index("datetime")
+        serial_df.index = pd.to_datetime(serial_df.index)
+        serial_df.sort_index()
+
+        # Select DataFrame rows between two dates
+        mask = (serial_df.index >= transmission_df.index[0]) & (serial_df.index <= transmission_df.index[-1])
+        serial_df = serial_df.loc[mask]
+
+        serial_df["arrival_time"] = (serial_df["time"] - serial_df["time"].iloc[0]) * 60
+        serial_df.index = serial_df["arrival_time"]
+
+        transmission_df.index = transmission_df["arrival_time"]
+
+        # filter active state
+        for i in range(1, 5):
+            serial_df["LTE_SCC{}_effective_bw".format(i)] = serial_df[
+                "LTE_SCC{}_bw".format(i)
+            ]
+
+            mask = serial_df["LTE_SCC{}_state".format(i)].isin(["ACTIVE"])
+            serial_df["LTE_SCC{}_effective_bw".format(i)] = serial_df[
+                "LTE_SCC{}_effective_bw".format(i)
+            ].where(mask, other=0)
+
+        # filter if sc is usesd for uplink
+        for i in range(1, 5):
+            mask = serial_df["LTE_SCC{}_UL_Configured".format(i)].isin([False])
+            serial_df["LTE_SCC{}_effective_bw".format(i)] = serial_df[
+                "LTE_SCC{}_effective_bw".format(i)
+            ].where(mask, other=0)
+
+        # sum all effective bandwidth for 5G and 4G
+        serial_df["SCC1_NR5G_effective_bw"] = serial_df["SCC1_NR5G_bw"].fillna(0)
+        serial_df["effective_bw_sum"] = (
+                serial_df["SCC1_NR5G_effective_bw"]
+                + serial_df["LTE_SCC1_effective_bw"]
+                + serial_df["LTE_SCC2_effective_bw"]
+                + serial_df["LTE_SCC3_effective_bw"]
+                + serial_df["LTE_SCC4_effective_bw"]
+                + serial_df["LTE_bw"]
+        )
+        bw_cols = [
+            "SCC1_NR5G_effective_bw",
+            "LTE_bw",
+            "LTE_SCC1_effective_bw",
+            "LTE_SCC2_effective_bw",
+            "LTE_SCC3_effective_bw",
+            "LTE_SCC4_effective_bw",
+        ]
+
+
+        # transmission timeline
+        scaley = 1.5
+        scalex = 1.0
+        fig, ax = plt.subplots(2, 1, figsize=[6.4 * scaley, 4.8 * scalex])
+        fig.subplots_adjust(right=0.75)
+        if not args.fancy:
+            plt.title("{} with {}".format(transmission_direction, cc_algo))
+            fig.suptitle("{} with {}".format(transmission_direction, cc_algo))
+        ax0 = ax[0]
+        ax1 = ax0.twinx()
+        ax2 = ax0.twinx()
+        # ax2.spines.right.set_position(("axes", 1.22))
+
+        ax00 = ax[1]
+
+        snd_plot = ax0.plot(
+            transmission_df["snd_cwnd"].dropna(),
+            color="darkorange",
+            linestyle="dashed",
+            label="cwnd",
+        )
+        srtt_plot = ax1.plot(
+            transmission_df["srtt"].dropna(),
+            color="maroon",
+            linestyle="dotted",
+            label="sRTT",
+        )
+        goodput_plot = ax2.plot(
+            transmission_df["goodput_rolling"],
+            color="blue",
+            linestyle="solid",
+            label="goodput",
+        )
+
+        serial_df["time_rel"] = serial_df["time"] - serial_df["time"].iloc[0]
+        serial_df.index = serial_df["time_rel"]
+
+        ax_stacked = serial_df[bw_cols].plot.area(stacked=True, linewidth=0, ax=ax00)
+        ax00.set_ylabel("bandwidth [MHz]")
+        ax00.set_ylim(0, 200)
+        #ax.set_xlabel("time [minutes]")
+        #ax00.set_xlim([0, transmission_df.index[-1]])
+        ax00.xaxis.grid(True)
+
+
+
+        ax2.spines.right.set_position(("axes", 1.1))
+
+
+        ax0.set_ylim(0, 5000) #2500
+        ax1.set_ylim(0, 2) #0.3
+        ax2.set_ylim(0, 500)
+        #ax00.set_ylim(-25, 0)
+
+        ax00.set_xlabel("time [s]")
+
+        ax2.set_ylabel("goodput [mbps]")
+        #ax00.set_ylabel("LTE/NR RSRQ [dB]")
+        # ax02.set_ylabel("LTE RSRQ [dB]")
+        ax1.set_ylabel("sRTT [s]")
+        ax0.set_ylabel("cwnd [MSS]")
+
+        if args.fancy:
+            legend_frame = False
+            ax0.set_xlim([0, 60])
+            ax00.set_xlim([0, 60])
+            # added these three lines
+            lns_ax0 = snd_plot + srtt_plot + goodput_plot
+            labs_ax0 = [l.get_label() for l in lns_ax0]
+            ax2.legend(lns_ax0, labs_ax0, ncols=9, fontsize=9, loc="upper right", frameon=legend_frame)
+            #ax0.set_zorder(100)
+
+            #lns_ax00 = [ax_stacked]
+            #labs_ax00 = ["5G bandwidth", "4G bandwidth"]
+            #ax00.legend(lns_ax00, labs_ax00, ncols=3, fontsize=9, loc="upper center", frameon=legend_frame)
+
+            L = ax00.legend(ncols=3, fontsize=9, frameon=False)
+            L.get_texts()[0].set_text("5G main")
+            L.get_texts()[1].set_text("4G main")
+            L.get_texts()[2].set_text("4G SCC 1")
+            L.get_texts()[3].set_text("4G SCC 2")
+            L.get_texts()[4].set_text("4G SCC 3")
+            L.get_texts()[5].set_text("4G SCC 4")
+
+            #ax00.set_zorder(100)
+            plt.savefig("{}{}_plot.eps".format(args.save, csv.replace(".csv", "")), bbox_inches="tight")
+            #serial_df.to_csv("{}{}_plot.csv".format(args.save, csv.replace(".csv", "")))
+        else:
+            fig.legend(loc="lower right")
+            plt.savefig("{}{}_plot.pdf".format(args.save, csv.replace(".csv", "")), bbox_inches="tight")
+        # except Exception as e:
+        #    print("Error processing file: {}".format(csv))
+        #    print(str(e))
+        counter += 1
+
+        plt.close(fig)
+        plt.clf()

plot_stacked_bandwidth.py

@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+
+from argparse import ArgumentParser
+
+
+import pandas as pd
+import matplotlib.pyplot as plt
+
+
+plt_params = {
+    "pgf.texsystem": "lualatex",
+    #"legend.fontsize": "x-large",
+    #"figure.figsize": (15, 5),
+    "axes.labelsize": 15,  # "small",
+    "axes.titlesize": "x-large",
+    "xtick.labelsize": 15,  # "small",
+    "ytick.labelsize": 15,  # "small",
+    "legend.fontsize": 15,
+    "axes.formatter.use_mathtext": True,
+    "mathtext.fontset": "dejavusans",
+}
+
+#plt.rcParams.update(plt_params)
+
+import seaborn as sns
+sns.set()
+sns.set(font_scale=1.5)
+
+plt.rcParams["figure.figsize"] = (10, 3)
+
+if __name__ == "__main__":
+    parser = ArgumentParser()
+    parser.add_argument("-f", "--file", required=True, help="Serial CSV")
+    parser.add_argument("--save", default=None, help="Location to save pdf file.")
+
+    args = parser.parse_args()
+
+    df = pd.read_csv(args.file)
+    df["time_rel"] = df["time"] - df["time"].iloc[0]
+    df.index = df["time_rel"] / 60
+
+    # filter active state
+    for i in range(1, 5):
+        df["LTE_SCC{}_effective_bw".format(i)] = df["LTE_SCC{}_bw".format(i)]
+
+        mask = df["LTE_SCC{}_state".format(i)].isin(["ACTIVE"])
+        df["LTE_SCC{}_effective_bw".format(i)] = df[
+            "LTE_SCC{}_effective_bw".format(i)
+        ].where(mask, other=0)
+
+    # filter if sc is usesd for uplink
+    for i in range(1, 5):
+        mask = df["LTE_SCC{}_UL_Configured".format(i)].isin([False])
+        df["LTE_SCC{}_effective_bw".format(i)] = df[
+            "LTE_SCC{}_effective_bw".format(i)
+        ].where(mask, other=0)
+
+    # sum all effective bandwidth for 5G and 4G
+    df["SCC1_NR5G_effective_bw"] = df["SCC1_NR5G_bw"].fillna(0)
+    df["effective_bw_sum"] = (
+        df["SCC1_NR5G_effective_bw"]
+        + df["LTE_SCC1_effective_bw"]
+        + df["LTE_SCC2_effective_bw"]
+        + df["LTE_SCC3_effective_bw"]
+        + df["LTE_SCC4_effective_bw"]
+        + df["LTE_bw"]
+    )
+    bw_cols = [
+        "SCC1_NR5G_effective_bw",
+        "LTE_bw",
+        "LTE_SCC1_effective_bw",
+        "LTE_SCC2_effective_bw",
+        "LTE_SCC3_effective_bw",
+        "LTE_SCC4_effective_bw",
+    ]
+
+    ax = df[bw_cols].plot.area(stacked=True, linewidth=0)
+    ax.set_ylabel("bandwidth [MHz]")
+    ax.set_xlabel("time [minutes]")
+    ax.set_xlim([0, df.index[-1]])
+    ax.xaxis.grid(False)
+
+
+    L = plt.legend(ncols=2, fontsize=12, frameon=False)
+    L.get_texts()[0].set_text("5G main")
+    L.get_texts()[1].set_text("4G main")
+    L.get_texts()[2].set_text("4G SCC 1")
+    L.get_texts()[3].set_text("4G SCC 2")
+    L.get_texts()[4].set_text("4G SCC 3")
+    L.get_texts()[5].set_text("4G SCC 4")
+
+    if args.save:
+        plt.savefig("{}-used_bandwidth.eps".format(args.save), bbox_inches="tight")
+    else:
+        plt.show()