diff options
Diffstat (limited to 'Examples/Modules')
| -rwxr-xr-x | Examples/Modules/RigidInjection/analysis_rigid_injection_BoostedFrame.py | 16 |
1 files changed, 2 insertions, 14 deletions
diff --git a/Examples/Modules/RigidInjection/analysis_rigid_injection_BoostedFrame.py b/Examples/Modules/RigidInjection/analysis_rigid_injection_BoostedFrame.py index 497a30097..3dcf40ac8 100755 --- a/Examples/Modules/RigidInjection/analysis_rigid_injection_BoostedFrame.py +++ b/Examples/Modules/RigidInjection/analysis_rigid_injection_BoostedFrame.py @@ -18,24 +18,12 @@ import scipy.constants as scc import read_raw_data yt.funcs.mylog.setLevel(0) -# filename = sys.argv[1] - -def get_particle_field(snapshot, species, field): - fn = snapshot + '/' + species - files = glob.glob(os.path.join(fn, field + '_*')) - files.sort() - all_data = np.array([]) - for f in files: - data = np.fromfile(f) - all_data = np.concatenate((all_data, data)) - return all_data - # Read data from back-transformed diagnostics snapshot = './lab_frame_data/snapshot00001' header = './lab_frame_data/Header' allrd, info = read_raw_data.read_lab_snapshot(snapshot, header) -z = np.mean( get_particle_field(snapshot, 'beam', 'z') ) -w = np.std ( get_particle_field(snapshot, 'beam', 'x') ) +z = np.mean( read_raw_data.get_particle_field(snapshot, 'beam', 'z') ) +w = np.std ( read_raw_data.get_particle_field(snapshot, 'beam', 'x') ) # initial parameters z0 = 20.e-6 |