diff options
Diffstat (limited to 'Python/pywarpx/timestepper.py')
-rw-r--r-- | Python/pywarpx/timestepper.py | 81 |
1 files changed, 0 insertions, 81 deletions
diff --git a/Python/pywarpx/timestepper.py b/Python/pywarpx/timestepper.py deleted file mode 100644 index de341097a..000000000 --- a/Python/pywarpx/timestepper.py +++ /dev/null @@ -1,81 +0,0 @@ -# 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() |