Explorar el Código

Adds export to pickle format.

master
langspielplatte hace 3 años
padre
commit
4d09e0d2f1
Se han modificado 2 ficheros con 7 adiciones y 4 borrados
  1. +3
    -2
      plot_single_transmission_timeline.py
  2. +4
    -2
      plot_transmission_timeline.py

+ 3
- 2
plot_single_transmission_timeline.py Ver fichero

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 Ver fichero

#!/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()

Cargando…
Cancelar
Guardar