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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
# Copyright 2016-2022 Andrew Myers, David Grote, Maxence Thevenet
# Remi Lehe, Lorenzo Giacomel
#
# This file is part of WarpX.
#
# License: BSD-3-Clause-LBNL
import re
import sys
from . import Particles
from .Algo import algo
from .Amr import amr
from .Amrex import amrex
from .Boundary import boundary
from .Bucket import Bucket
from .Collisions import collisions, collisions_list
from .Constants import my_constants
from .Diagnostics import diagnostics, reduced_diagnostics
from .EB2 import eb2
from .Geometry import geometry
from .HybridPICModel import hybridpicmodel
from .Interpolation import interpolation
from .Lasers import lasers, lasers_list
from .PSATD import psatd
from .Particles import particles, particles_list
from ._libwarpx import libwarpx
class WarpX(Bucket):
"""
A Python wrapper for the WarpX C++ class
"""
def create_argv_list(self, **kw):
argv = []
for k, v in kw.items():
if v is not None:
argv.append(f'{k} = {v}')
argv += warpx.attrlist()
argv += my_constants.attrlist()
argv += amr.attrlist()
argv += amrex.attrlist()
argv += geometry.attrlist()
argv += hybridpicmodel.attrlist()
argv += boundary.attrlist()
argv += algo.attrlist()
argv += interpolation.attrlist()
argv += psatd.attrlist()
argv += eb2.attrlist()
# --- Search through species_names and add any predefined particle objects in the list.
particles_list_names = [p.instancename for p in particles_list]
for pstring in particles.species_names:
if pstring in particles_list_names:
# --- The species is already included in particles_list
continue
elif hasattr(Particles, pstring):
# --- Add the predefined species to particles_list
particles_list.append(getattr(Particles, pstring))
particles_list_names.append(pstring)
else:
raise Exception('Species %s listed in species_names not defined'%pstring)
argv += particles.attrlist()
for particle in particles_list:
argv += particle.attrlist()
argv += collisions.attrlist()
for collision in collisions_list:
argv += collision.attrlist()
argv += lasers.attrlist()
for laser in lasers_list:
argv += laser.attrlist()
diagnostics.diags_names = diagnostics._diagnostics_dict.keys()
argv += diagnostics.attrlist()
for diagnostic in diagnostics._diagnostics_dict.values():
diagnostic.species = diagnostic._species_dict.keys()
argv += diagnostic.attrlist()
for species_diagnostic in diagnostic._species_dict.values():
argv += species_diagnostic.attrlist()
reduced_diagnostics.reduced_diags_names = reduced_diagnostics._diagnostics_dict.keys()
argv += reduced_diagnostics.attrlist()
for diagnostic in reduced_diagnostics._diagnostics_dict.values():
argv += diagnostic.attrlist()
return argv
def init(self, mpi_comm=None, **kw):
# note: argv[0] needs to be an absolute path so it works with AMReX backtraces
# https://github.com/AMReX-Codes/amrex/issues/3435
argv = [sys.executable] + self.create_argv_list(**kw)
libwarpx.initialize(argv, mpi_comm=mpi_comm)
def evolve(self, nsteps=-1):
libwarpx.evolve(nsteps)
def finalize(self, finalize_mpi=1):
libwarpx.finalize(finalize_mpi)
def getProbLo(self, direction):
return libwarpx.libwarpx_so.warpx_getProbLo(direction)
def getProbHi(self, direction):
return libwarpx.libwarpx_so.warpx_getProbHi(direction)
def write_inputs(self, filename='inputs', **kw):
argv = self.create_argv_list(**kw)
# Sort the argv list to make it more human readable
argv.sort()
with open(filename, 'w') as ff:
prefix_old = ''
for arg in argv:
# This prints the name of the input group (prefix) as a header
# before each group to make the input file more human readable
prefix_new = re.split(' |\.', arg)[0]
if prefix_new != prefix_old:
if prefix_old != '':
ff.write('\n')
ff.write(f'# {prefix_new}\n')
prefix_old = prefix_new
ff.write(f'{arg}\n')
warpx = WarpX('warpx')
|
