diff options
Diffstat (limited to 'Python/pywarpx/WarpX.py')
| -rw-r--r-- | Python/pywarpx/WarpX.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/Python/pywarpx/WarpX.py b/Python/pywarpx/WarpX.py index 8c4900fab..ece837816 100644 --- a/Python/pywarpx/WarpX.py +++ b/Python/pywarpx/WarpX.py @@ -5,7 +5,7 @@ # # License: BSD-3-Clause-LBNL -from . import Particles, _libwarpx +from . import Particles from .Algo import algo from .Amr import amr from .Boundary import boundary @@ -19,6 +19,7 @@ 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): @@ -75,19 +76,19 @@ class WarpX(Bucket): def init(self, mpi_comm=None): argv = ['warpx'] + self.create_argv_list() - _libwarpx.initialize(argv, mpi_comm=mpi_comm) + libwarpx.initialize(argv, mpi_comm=mpi_comm) def evolve(self, nsteps=-1): - _libwarpx.evolve(nsteps) + libwarpx.evolve(nsteps) def finalize(self, finalize_mpi=1): - _libwarpx.finalize(finalize_mpi) + libwarpx.finalize(finalize_mpi) def getProbLo(self, direction): - return _libwarpx.libwarpx.libwarpx_so.warpx_getProbLo(direction) + return libwarpx.libwarpx_so.warpx_getProbLo(direction) def getProbHi(self, direction): - return _libwarpx.libwarpx.libwarpx_so.warpx_getProbHi(direction) + return libwarpx.libwarpx_so.warpx_getProbHi(direction) def write_inputs(self, filename='inputs', **kw): argv = self.create_argv_list() |