aboutsummaryrefslogtreecommitdiff
path: root/Python/pywarpx
diff options
context:
space:
mode:
Diffstat (limited to 'Python/pywarpx')
-rw-r--r--Python/pywarpx/BoxLib.py8
-rw-r--r--Python/pywarpx/Particles.py3
-rw-r--r--Python/pywarpx/WarpX.py16
-rw-r--r--Python/pywarpx/__init__.py1
4 files changed, 20 insertions, 8 deletions
diff --git a/Python/pywarpx/BoxLib.py b/Python/pywarpx/BoxLib.py
index 1a078d3c6..a15119250 100644
--- a/Python/pywarpx/BoxLib.py
+++ b/Python/pywarpx/BoxLib.py
@@ -6,8 +6,9 @@ from .Geometry import geometry
from .Algo import algo
from .Langmuirwave import langmuirwave
from .Interpolation import interpolation
+from .Particles import particles
-from . import _warpxC
+from . import warpxC
class BoxLib(object):
@@ -19,8 +20,9 @@ class BoxLib(object):
argv += algo.attrlist()
argv += langmuirwave.attrlist()
argv += interpolation.attrlist()
+ argv += particles.attrlist()
- _warpxC.boxlib_init(argv)
+ warpxC.boxlib_init(argv)
def finalize(self, finalize_mpi=1):
- _warpxC.boxlib_finalize(finalize_mpi)
+ warpxC.boxlib_finalize(finalize_mpi)
diff --git a/Python/pywarpx/Particles.py b/Python/pywarpx/Particles.py
new file mode 100644
index 000000000..a11da823d
--- /dev/null
+++ b/Python/pywarpx/Particles.py
@@ -0,0 +1,3 @@
+from .Bucket import Bucket
+
+particles = Bucket('particles')
diff --git a/Python/pywarpx/WarpX.py b/Python/pywarpx/WarpX.py
index a1c85e618..8b9ecefe5 100644
--- a/Python/pywarpx/WarpX.py
+++ b/Python/pywarpx/WarpX.py
@@ -1,18 +1,24 @@
from .Bucket import Bucket
-from . import _warpxC
+from . import warpxC
class WarpX(Bucket):
def init(self):
- _warpxC.warpx_init()
+ warpxC.warpx_init()
def evolve(self, nsteps=None):
if nsteps is None:
- _warpxC.warpx_evolve()
+ warpxC.warpx_evolve()
else:
- _warpxC.warpx_evolve(nsteps)
+ warpxC.warpx_evolve(nsteps)
def finalize(self):
- _warpxC.warpx_finalize()
+ warpxC.warpx_finalize()
+
+ def getProbLo(self, direction):
+ return warpxC.warpx_getProbLo(direction)
+
+ def getProbHi(self, direction):
+ return warpxC.warpx_getProbHi(direction)
warpx = WarpX('warpx')
diff --git a/Python/pywarpx/__init__.py b/Python/pywarpx/__init__.py
index 03d53d66e..98c26af36 100644
--- a/Python/pywarpx/__init__.py
+++ b/Python/pywarpx/__init__.py
@@ -5,5 +5,6 @@ from .Geometry import geometry
from .Algo import algo
from .Langmuirwave import langmuirwave
from .Interpolation import interpolation
+from .Particles import particles
from .BoxLib import BoxLib