blob: 8b9ecefe5dccf7ed8fd7dcc56a8956217635bedc (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
from .Bucket import Bucket
from . import warpxC
class WarpX(Bucket):
def init(self):
warpxC.warpx_init()
def evolve(self, nsteps=None):
if nsteps is None:
warpxC.warpx_evolve()
else:
warpxC.warpx_evolve(nsteps)
def finalize(self):
warpxC.warpx_finalize()
def getProbLo(self, direction):
return warpxC.warpx_getProbLo(direction)
def getProbHi(self, direction):
return warpxC.warpx_getProbHi(direction)
warpx = WarpX('warpx')
|
