diff options
Diffstat (limited to 'nemesis/nemesis.py')
| -rwxr-xr-x | nemesis/nemesis.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/nemesis/nemesis.py b/nemesis/nemesis.py index b7a9877..2e3837a 100755 --- a/nemesis/nemesis.py +++ b/nemesis/nemesis.py @@ -52,6 +52,9 @@ class Experiment: class Nemesis(object): + # the name of the target program + prog = None + # the (ideal) interval between samples signal_interval = 0.0 # the timestamp which the last sample was taken @@ -79,10 +82,11 @@ class Nemesis(object): prev_coro = defaultdict(lambda: None) @staticmethod - def __init__(e_duration, filename, signal_interval=0.01): + def __init__(e_duration, filename, prog, signal_interval=0.01): Nemesis.signal_interval = signal_interval Nemesis.e_duration = e_duration Nemesis.filename = filename + Nemesis.prog = prog @staticmethod def start(): @@ -111,7 +115,7 @@ class Nemesis(object): @staticmethod def stop(): signal.setitimer(signal.ITIMER_REAL, 0) - plot_results(Nemesis.results, Nemesis.filename) + plot_results(Nemesis.results, Nemesis.filename, Nemesis.prog) print(f"Wrote {Nemesis.filename}") @staticmethod @@ -338,6 +342,7 @@ if __name__ == "__main__": code = compile(fp.read(), args.prog[0], "exec") Nemesis(args.experiment_duration, args.filename, + args.prog[0], args.interval).start() exec(code, the_globals) Nemesis.stop() |