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

Adds fancy plot settings for eps export.

Browse Source
This commit is contained in:
Lukas Prause
2023-07-10 14:37:29 +02:00
parent 4c33e4872e
commit 14eca54f98
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
plot_single_transmission_EM9190.py
View File

@@ -46,6 +46,7 @@ if __name__ == "__main__":
"-p", "--pcap_csv_folder", required=True, help="PCAP csv folder." "-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("--save", required=True, help="Location to save pdf file.")
parser.add_argument("--fancy", action="store_true", help="Create fancy plot.")
parser.add_argument( parser.add_argument(
"-i", "-i",
"--interval", "--interval",
@@ -182,9 +183,10 @@ if __name__ == "__main__":
# transmission timeline # transmission timeline
scaley = 1.5 scaley = 1.5
scalex = 1.0 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, ax = plt.subplots(2, 1, figsize=[6.4 * scaley, 4.8 * scalex])
fig.subplots_adjust(right=0.75) 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)) fig.suptitle("{} with {}".format(transmission_direction, cc_algo))
ax0 = ax[0] ax0 = ax[0]
ax1 = ax0.twinx() ax1 = ax0.twinx()
@@ -303,8 +305,11 @@ if __name__ == "__main__":
ax0.set_ylabel("cwnd") ax0.set_ylabel("cwnd")
ax01.set_ylabel("Bandwidth [MHz]") ax01.set_ylabel("Bandwidth [MHz]")
if args.fancy:
fig.legend(ncols=4, fontsize=12, loc="lower right")
plt.savefig("{}{}_plot.eps".format(args.save, csv.replace(".csv", "")), bbox_inches="tight")
else:
fig.legend(loc="lower right") fig.legend(loc="lower right")
plt.savefig("{}{}_plot.pdf".format(args.save, csv.replace(".csv", "")), bbox_inches="tight") plt.savefig("{}{}_plot.pdf".format(args.save, csv.replace(".csv", "")), bbox_inches="tight")
# except Exception as e: # except Exception as e:
# print("Error processing file: {}".format(csv)) # print("Error processing file: {}".format(csv))