Просмотр исходного кода

Adds export to pickle format.

master
langspielplatte 3 лет назад
Родитель
Сommit
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()

Загрузка…
Отмена
Сохранить