add beam_plot() method that accepts command line arguments

fnalacceleratormodeling · Jun 6, 2024 · 3ecf9f5 · 3ecf9f5
1 parent 7a8c98d
commit 3ecf9f5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/analysis_tools/diag_plot.py b/src/analysis_tools/diag_plot.py
@@ -189,6 +189,12 @@ def do_plots(options, plotparams):
  if options.show:
  pyplot.show()
 
+def diag_plot(command):
+ plotparams = generate_plotparams()
+ plotargs = command.split()
+ options = handle_args(plotargs, plotparams)
+ do_plots(options, plotparams)
+
 if __name__ == '__main__':
  plotparams = generate_plotparams()
  options = handle_args(sys.argv[1:], plotparams)