diff options
Diffstat (limited to 'Python/pywarpx/picmi.py')
-rw-r--r-- | Python/pywarpx/picmi.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/Python/pywarpx/picmi.py b/Python/pywarpx/picmi.py index 69fa25ac0..f197afbc2 100644 --- a/Python/pywarpx/picmi.py +++ b/Python/pywarpx/picmi.py @@ -859,6 +859,27 @@ class Mirror(picmistandard.PICMI_Mirror): pywarpx.warpx.mirror_z_npoints.append(self.number_of_cells) +class CoulombCollisions(picmistandard.base._ClassWithInit): + """Custom class to handle setup of binary Coulmb collisions in WarpX. If + collision initialization is added to picmistandard this can be changed to + inherit that functionality.""" + + def __init__(self, name, species, CoulombLog=None, ndt=None, **kw): + self.name = name + self.species = species + self.CoulombLog = CoulombLog + self.ndt = ndt + + self.handle_init(kw) + + def initialize_inputs(self): + collision = pywarpx.Collisions.newcollision(self.name) + collision.type = 'pairwisecoulomb' + collision.species = [species.name for species in self.species] + collision.CoulombLog = self.CoulombLog + collision.ndt = self.ndt + + class MCCCollisions(picmistandard.base._ClassWithInit): """Custom class to handle setup of MCC collisions in WarpX. If collision initialization is added to picmistandard this can be changed to inherit |