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
|
# Copyright 2017-2021 Andrew Myers, David Grote, Weiqun Zhang
#
#
# This file is part of WarpX.
#
# License: BSD-3-Clause-LBNL
# This is intended to be a Python level example of how one might write a customized
# time stepping loop. This would replace the functionality of the WarpX::Evolve routine.
# Wrappers are available for the major pieces of a time step and they would be called
# here in the appropriate order.
# Note that this is intended to be an example only and may not be functional. The
# onestep routine as written here is out of date and is not consistent with WarpX::Evolve.
from . import callbacks, libwarpx
class TimeStepper(object):
def step(self, nsteps=1):
for i in range(nsteps):
self.onestep()
def onestep(self):
callbacks._beforestep()
self.cur_time = libwarpx.libwarpx_so.warpx_gett_new(0)
self.istep = libwarpx.libwarpx_so.warpx_getistep(0)
#if mpi.rank == 0:
print("\nSTEP %d starts ..."%(self.istep + 1))
#if (ParallelDescriptor::NProcs() > 1)
# if (okToRegrid(step)) RegridBaseLevel();
dt = libwarpx.libwarpx_so.warpx_getdt(0)
# --- At the beginning, we have B^{n-1/2} and E^{n}.
# --- Particles have p^{n-1/2} and x^{n}.
libwarpx.libwarpx_so.warpx_FillBoundaryE()
libwarpx.libwarpx_so.warpx_EvolveB(0.5*dt,1) # We now B^{n}
libwarpx.libwarpx_so.warpx_FillBoundaryB()
libwarpx.libwarpx_so.warpx_UpdateAuxilaryData()
# --- Evolve particles to p^{n+1/2} and x^{n+1}
# --- Depose current, j^{n+1/2}
callbacks._particleinjection()
callbacks._particlescraper()
callbacks._beforedeposition()
libwarpx.libwarpx_so.warpx_PushParticlesandDepose(self.cur_time)
callbacks._afterdeposition()
libwarpx.libwarpx_so.mypc_Redistribute() # Redistribute particles
libwarpx.libwarpx_so.warpx_FillBoundaryE()
libwarpx.libwarpx_so.warpx_EvolveB(0.5*dt,2) # We now B^{n+1/2}
libwarpx.libwarpx_so.warpx_SyncCurrent()
libwarpx.libwarpx_so.warpx_FillBoundaryB()
callbacks._beforeEsolve()
libwarpx.libwarpx_so.warpx_EvolveE(dt,0) # We now have E^{n+1}
callbacks._afterEsolve()
self.istep += 1
self.cur_time += dt
libwarpx.libwarpx_so.warpx_MoveWindow(self.istep,True);
#if mpi.rank == 0:
print("STEP %d ends. TIME = %e DT = %e"%(self.istep, self.cur_time, dt))
# --- Sync up time
for i in range(libwarpx.libwarpx_so.warpx_finestLevel()+1):
libwarpx.libwarpx_so.warpx_sett_new(i, self.cur_time)
libwarpx.libwarpx_so.warpx_setistep(i, self.istep)
callbacks._afterstep()
|
