diff options
Diffstat (limited to 'Tools/plot_parallel.py')
| -rw-r--r-- | Tools/plot_parallel.py | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/Tools/plot_parallel.py b/Tools/plot_parallel.py index f2f2c7ecf..56b243aca 100644 --- a/Tools/plot_parallel.py +++ b/Tools/plot_parallel.py @@ -20,7 +20,7 @@ between MPI ranks. This script also proposes an option to plot one quantity over all timesteps. The data of all plotfiles are gathered to rank 0, and the quantity evolution is plotted and saved to file. For the illustration, this quantity is the max -of Ey. Use " --plot_Ey_max_evolution " to activate this option. +of Ey. Use " --plot_max_evolution Ey" to activate this option. To get help, run > python plot_parallel --help @@ -47,10 +47,11 @@ parser.add_argument('--parallel', dest='parallel', action='store_true', default= help='whether or not to do the analysis in parallel (e.g., 1 plotfile per MPI rank)') parser.add_argument('--species', dest='pslist', nargs='+', type=str, default=None, help='Species to be plotted, e.g., " --species beam plasma_e ". By default, all species in the simulation are shown') -parser.add_argument('--plot_Ey_max_evolution', dest='plot_Ey_max_evolution', action='store_true', default=False, - help='Whether to plot evolution of max(Ey) to illustrate how to plot one quantity across all plotfiles.') +parser.add_argument('--plot_max_evolution', type=str, default=None, + help='Quantity to plot the max of across all data files') args = parser.parse_args() +plot_max_evolution = args.plot_max_evolution # Sanity check if int(sys.version[0]) != 3: print('WARNING: Parallel analysis was only tested with Python3') @@ -162,15 +163,14 @@ def plot_field_max(zwin_arr, maxF_arr): plt.figure() plt.plot(zwin_arr, maxF_arr) plt.xlabel('z (m)') - plt.ylabel('Field (S.I.)') + plt.ylabel('%s (S.I.)'%plot_max_evolution) plt.title('Field max evolution') - plt.savefig('max_field_evolution.pdf', bbox_inches='tight') + plt.savefig('max_%s_evolution.pdf'%plot_max_evolution, bbox_inches='tight') ### Analysis ### # Get list of plotfiles plotlib = args.plotlib -plot_Ey_max_evolution = args.plot_Ey_max_evolution file_list = glob.glob(args.path + '/plt?????') file_list.sort() nfiles = len(file_list) @@ -194,7 +194,7 @@ if rank == 0: print('Number of plotfiles: %s'%nfiles) print('list of species: ', pslist) -if plot_Ey_max_evolution: +if plot_max_evolution is not None: # Fill with a value less than any possible value zwin = np.full(nfiles, np.finfo(float).min) maxF = np.full(nfiles, np.finfo(float).min) @@ -207,10 +207,10 @@ for count, filename in enumerate(file_list): continue plot_snapshot( filename ) - if plot_Ey_max_evolution: - zwin[count], maxF[count] = get_field_max( filename, 'Ey' ) + if plot_max_evolution is not None: + zwin[count], maxF[count] = get_field_max( filename, plot_max_evolution ) -if plot_Ey_max_evolution: +if plot_max_evolution is not None: if size > 1: global_zwin = np.empty_like(zwin) global_maxF = np.empty_like(maxF) |