blob: 2b1fbc7038be46b294ad6bcbf3672cfee48e4214 (
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#!/usr/bin/env python3
# Copyright 2019 Luca Fedeli
#
# This file is part of WarpX.
#
# License: BSD-3-Clause-LBNL
# -*- coding: utf-8 -*-
import sys
import analysis_core as ac
import openpmd_api as io
#sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
#import checksumAPI
# This script is a frontend for the analysis routines
# in analysis_core.py (please refer to this file for
# a full description). It reads output files in openPMD
# format and extracts the data needed for
# the analysis routines.
def main():
print("Opening openPMD output")
prefix = sys.argv[1]
series = io.Series(prefix+"/openpmd_%T.h5", io.Access.read_only)
data_set_end = series.iterations[2]
# get simulation time
sim_time = data_set_end.time
# no particles can be created on the first timestep so we have 2 timesteps in the test case,
# with only the second one resulting in particle creation
dt = sim_time/2.
# get particle data
particle_data = {}
names, types = ac.get_all_species_names_and_types()
for spec_name_type in zip(names, types):
spec_name = spec_name_type[0]
is_photon = spec_name_type[1]
data = {}
px = data_set_end.particles[spec_name]["momentum"]["x"][:]
py = data_set_end.particles[spec_name]["momentum"]["y"][:]
pz = data_set_end.particles[spec_name]["momentum"]["z"][:]
w = data_set_end.particles[spec_name]["weighting"][io.Mesh_Record_Component.SCALAR][:]
if is_photon :
opt = data_set_end.particles[spec_name]["opticalDepthBW"][io.Mesh_Record_Component.SCALAR][:]
else:
opt = data_set_end.particles[spec_name]["opticalDepthQSR"][io.Mesh_Record_Component.SCALAR][:]
series.flush()
data["px"] = px
data["py"] = py
data["pz"] = pz
data["w"] = w
data["opt"] = opt
particle_data[spec_name] = data
ac.check(dt, particle_data)
#test_name = os.path.split(os.getcwd())[1]
#checksumAPI.evaluate_checksum(test_name, filename_end)
if __name__ == "__main__":
main()
|
