diff options
Diffstat (limited to 'Python/pywarpx/timestepper.py')
| -rw-r--r-- | Python/pywarpx/timestepper.py | 38 |
1 files changed, 19 insertions, 19 deletions
diff --git a/Python/pywarpx/timestepper.py b/Python/pywarpx/timestepper.py index 180bd845e..f75bc8c2b 100644 --- a/Python/pywarpx/timestepper.py +++ b/Python/pywarpx/timestepper.py @@ -12,7 +12,7 @@ # 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 ._libwarpx import libwarpx +from . import libwarpx from . import callbacks @@ -26,8 +26,8 @@ class TimeStepper(object): callbacks._beforestep() - self.cur_time = libwarpx.warpx_gett_new(0) - self.istep = libwarpx.warpx_getistep(0) + 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)) @@ -35,48 +35,48 @@ class TimeStepper(object): #if (ParallelDescriptor::NProcs() > 1) # if (okToRegrid(step)) RegridBaseLevel(); - dt = libwarpx.warpx_getdt(0) + 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.warpx_FillBoundaryE() - libwarpx.warpx_EvolveB(0.5*dt,1) # We now B^{n} + libwarpx.libwarpx_so.warpx_FillBoundaryE() + libwarpx.libwarpx_so.warpx_EvolveB(0.5*dt,1) # We now B^{n} - libwarpx.warpx_FillBoundaryB() - libwarpx.warpx_UpdateAuxilaryData() + 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.warpx_PushParticlesandDepose(self.cur_time) + libwarpx.libwarpx_so.warpx_PushParticlesandDepose(self.cur_time) callbacks._afterdeposition() - libwarpx.mypc_Redistribute() # Redistribute particles + libwarpx.libwarpx_so.mypc_Redistribute() # Redistribute particles - libwarpx.warpx_FillBoundaryE() - libwarpx.warpx_EvolveB(0.5*dt,2) # We now B^{n+1/2} + libwarpx.libwarpx_so.warpx_FillBoundaryE() + libwarpx.libwarpx_so.warpx_EvolveB(0.5*dt,2) # We now B^{n+1/2} - libwarpx.warpx_SyncCurrent() + libwarpx.libwarpx_so.warpx_SyncCurrent() - libwarpx.warpx_FillBoundaryB() + libwarpx.libwarpx_so.warpx_FillBoundaryB() callbacks._beforeEsolve() - libwarpx.warpx_EvolveE(dt,0) # We now have E^{n+1} + libwarpx.libwarpx_so.warpx_EvolveE(dt,0) # We now have E^{n+1} callbacks._afterEsolve() self.istep += 1 self.cur_time += dt - libwarpx.warpx_MoveWindow(self.istep,True); + 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.warpx_finestLevel()+1): - libwarpx.warpx_sett_new(i, self.cur_time) - libwarpx.warpx_setistep(i, self.istep) + 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() |