diff --git a/plot_single_transmission_EM9190.py b/plot_single_transmission_EM9190.py index 0205b27..0490d72 100755 --- a/plot_single_transmission_EM9190.py +++ b/plot_single_transmission_EM9190.py @@ -46,6 +46,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", @@ -182,10 +183,11 @@ if __name__ == "__main__": # 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() @@ -303,9 +305,12 @@ if __name__ == "__main__": ax0.set_ylabel("cwnd") ax01.set_ylabel("Bandwidth [MHz]") - fig.legend(loc="lower right") - - plt.savefig("{}{}_plot.pdf".format(args.save, csv.replace(".csv", "")), bbox_inches="tight") + 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") + 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))