浏览代码

Adds export to pickle format.

master
langspielplatte 2 年前
父节点
当前提交
4d09e0d2f1
共有 2 个文件被更改,包括 7 次插入4 次删除
  1. +3
    -2
      plot_single_transmission_timeline.py
  2. +4
    -2
      plot_transmission_timeline.py

+ 3
- 2
plot_single_transmission_timeline.py 查看文件

import math import math
import multiprocessing import multiprocessing
import os import os
import pickle
from argparse import ArgumentParser from argparse import ArgumentParser


import matplotlib import matplotlib


if args.save: if args.save:
plt.savefig("{}{}_plot.pdf".format(args.save, csv.replace(".csv", ""))) plt.savefig("{}{}_plot.pdf".format(args.save, csv.replace(".csv", "")))
else:
plt.show()
if args.export:
pickle.dump(fig, open("{}{}_plot.pkl".format(args.save, csv.replace(".csv", "")), "wb"))
counter += 1 counter += 1


plt.clf() plt.clf()

+ 4
- 2
plot_transmission_timeline.py 查看文件

#!/usr/bin/env python3 #!/usr/bin/env python3
import multiprocessing import multiprocessing
import os import os
import pickle
from argparse import ArgumentParser from argparse import ArgumentParser
from math import ceil from math import ceil
from time import sleep from time import sleep
parser.add_argument("-s", "--serial_file", required=True, help="Serial csv file.") 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("-p", "--pcap_csv_folder", required=True, help="PCAP csv folder.")
parser.add_argument("--save", default=None, help="Location to save pdf file.") parser.add_argument("--save", default=None, help="Location to save pdf file.")
parser.add_argument("--export", default=None, help="Export figure as an pickle file.")
parser.add_argument( parser.add_argument(
"-c", "-c",
"--cores", "--cores",


if args.save: if args.save:
plt.savefig("{}timeline_plot.pdf".format(args.save)) plt.savefig("{}timeline_plot.pdf".format(args.save))
else:
plt.show()
if args.export:
pickle.dump(fig, open("{}timeline_plot.pkl".format(args.export), "wb"))


#goodput cdf #goodput cdf
plt.clf() plt.clf()

正在加载...
取消
保存