diff options
Diffstat (limited to 'Python/pywarpx/picmi.py')
| -rw-r--r-- | Python/pywarpx/picmi.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/Python/pywarpx/picmi.py b/Python/pywarpx/picmi.py index d1f5136bc..c7124780a 100644 --- a/Python/pywarpx/picmi.py +++ b/Python/pywarpx/picmi.py @@ -277,6 +277,10 @@ class MultiSpecies(picmistandard.PICMI_MultiSpecies): class GaussianBunchDistribution(picmistandard.PICMI_GaussianBunchDistribution): + def init(self, kw): + self.do_symmetrize = kw.pop('warpx_do_symmetrize', None) + self.symmetrization_order = kw.pop('warpx_symmetrization_order', None) + def initialize_inputs(self, species_number, layout, species, density_scale): species.injection_style = "gaussian_beam" species.x_m = self.centroid_position[0] @@ -329,6 +333,9 @@ class GaussianBunchDistribution(picmistandard.PICMI_GaussianBunchDistribution): species.uy = self.centroid_velocity[1]/constants.c species.uz = self.centroid_velocity[2]/constants.c + species.do_symmetrize = self.do_symmetrize + species.symmetrization_order = self.symmetrization_order + class DensityDistributionBase(object): """This is a base class for several predefined density distributions. It |