blob: 976d94c74da459aa06fa170e3eb7cd0a0b7e2703 (
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):
self.warpx = warpxC.WarpX.GetInstance()
self.warpx.InitData()
def evolve(self, nsteps=-1):
self.warpx.Evolve(nsteps)
def finalize(self):
warpxC.WarpX.ResetInstance()
def getProbLo(self, direction):
return self.warpx.Geom()[0].ProbLo(direction)
#return warpxC.warpx_getProbLo(direction)
def getProbHi(self, direction):
return self.warpx.Geom()[0].ProbHi(direction)
#return warpxC.warpx_getProbHi(direction)
warpx = WarpX('warpx')
|
