From 55c1e774599a012cea9965c8833e019d6d8557fa Mon Sep 17 00:00:00 2001 From: Giovanni Malfattore Date: Thu, 17 Nov 2022 12:56:52 +0100 Subject: [PATCH] Add format choice --- PerformancePlots/plot_tof_mass.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/PerformancePlots/plot_tof_mass.py b/PerformancePlots/plot_tof_mass.py index d62bc67..0a43d0e 100755 --- a/PerformancePlots/plot_tof_mass.py +++ b/PerformancePlots/plot_tof_mass.py @@ -103,9 +103,9 @@ def draw_tofmass(h, return h, latex -def main(filename, tag, ymin, xrange): +def main(filename, tag, ymin, xrange, format): mass = get_for_tofmass(filename, tag=tag, subdir="EvTimeTOF") - d = draw_tofmass(mass, saveas="/tmp/TOFMass.pdf", + d = draw_tofmass(mass, saveas="/tmp/TOFMass."+format, rebin=6, ymin=ymin, xrange=xrange, @@ -131,6 +131,8 @@ def main(filename, tag, ymin, xrange): help="Start y range of the plot.") parser.add_argument("--xrange", "-x", type=float, nargs=2, default=[0, 2.5], help="Start x range of the plot.") + parser.add_argument("--format", "-f", type=str, default="pdf", + help="Output format for the plot.") args = parser.parse_args() - main(args.data_file, args.tag, ymin=args.ymin, xrange=args.xrange) \ No newline at end of file + main(args.data_file, args.tag, ymin=args.ymin, xrange=args.xrange, format=args.format) \ No newline at end of file