blob: 2ae52c0eb1aabb60f4e33eb7cb0f2e7086adc9ec (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
# Copyright 2018-2019 Andrew Myers, Maxence Thevenet
#
# This file is part of WarpX.
#
# License: BSD-3-Clause-LBNL
import numpy as np
from glob import glob
import os
it = 1
fn = "./lab_frame_data/" + 'snapshot' + str(it).zfill(5) + "/particle1/"
print(fn)
def get_particle_field(field):
files = glob(os.path.join(fn, field + '_*'))
all_data = np.array([])
files.sort()
for f in files:
data = np.fromfile(f)
all_data = np.concatenate((all_data, data))
return all_data
x = get_particle_field('x')
z = get_particle_field('z')
ux = get_particle_field('ux')
uz = get_particle_field('uz')
|
