blob: e8950419f25ae5ca8348b69030d5d4ab504a323d (
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
|
#!/usr/bin/env python3
# Copyright 2019 Luca Fedeli
#
# This file is part of WarpX.
#
# License: BSD-3-Clause-LBNL
# -*- coding: utf-8 -*-
import os
import sys
import yt
sys.path.insert(1, '../../../../warpx/Regression/Checksum/')
import analysis_core as ac
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 yt
# format and extracts the data needed for
# the analysis routines.
yt
def main():
print("Opening yt output")
filename_end = sys.argv[1]
data_set_end = yt.load(filename_end)
# get simulation time
sim_time = data_set_end.current_time.to_value()
# 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
all_data_end = data_set_end.all_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 = {}
data["px"] = all_data_end[spec_name,"particle_momentum_x"].v
data["py"] = all_data_end[spec_name,"particle_momentum_y"].v
data["pz"] = all_data_end[spec_name,"particle_momentum_z"].v
data["w"] = all_data_end[spec_name,"particle_weighting"].v
if is_photon :
data["opt"] = all_data_end[spec_name, "particle_opticalDepthBW"].v
else:
data["opt"] = all_data_end[spec_name, "particle_opticalDepthQSR"].v
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()
|
