Allow to change output file name

Add parameter -o --output to specify output file name (default still works)
ondrejtichacek · Jul 22, 2017 · 7b7d10e · 7b7d10e
1 parent 03fb4a8
commit 7b7d10e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examplescripts/pycorn-bin.py b/examplescripts/pycorn-bin.py
@@ -297,7 +297,10 @@ def plotterX(inp,fname):
  host.yaxis.set_minor_locator(AutoMinorLocator())
  if not args.no_title:
  plt.title(fname, loc='left', size=9)
- plot_file = fname[:-4] + "_" + inp.run_name + "_plot." + args.format
+ if not args.output:
+ plot_file = fname[:-4] + "_" + inp.run_name + "_plot." + args.format
+ else:
+ plot_file = args.output
  plt.savefig(plot_file, bbox_inches='tight', dpi=args.dpi)
  print("Plot saved to: " + plot_file)
  plt.clf()