Python/pywarpx/WarpX.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

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()

warpx = WarpX('warpx')