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
|
# 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
from . import Particles
from .Algo import algo
from .Amr import amr
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 .Interpolation import interpolation
from .Langmuirwave import langmuirwave
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):
argv = []
argv += warpx.attrlist()
argv += my_constants.attrlist()
argv += amr.attrlist()
argv += geometry.attrlist()
argv += boundary.attrlist()
argv += algo.attrlist()
argv += langmuirwave.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():
diagnostic.species = diagnostic._species_dict.keys()
argv += diagnostic.attrlist()
for species_diagnostic in diagnostic._species_dict.values():
argv += species_diagnostic.attrlist()
return argv
def init(self, mpi_comm=None):
argv = ['warpx'] + self.create_argv_list()
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()
for k, v in kw.items():
argv.append(f'{k} = {v}')
# 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')
|
