From ae89e95c1636f28e2d88a93620c20f4c4acfbfeb Mon Sep 17 00:00:00 2001 From: David Grote Date: Tue, 1 Dec 2020 09:02:39 -0800 Subject: Update and fix to picmi.py (#1539) --- Python/pywarpx/picmi.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'Python/pywarpx/picmi.py') diff --git a/Python/pywarpx/picmi.py b/Python/pywarpx/picmi.py index fd4a608eb..784d9af01 100644 --- a/Python/pywarpx/picmi.py +++ b/Python/pywarpx/picmi.py @@ -70,6 +70,8 @@ class Species(picmistandard.PICMI_Species): if self.mass is None: self.mass = element.mass*periodictable.constants.atomic_mass_constant + self.boost_adjust_transverse_positions = kw.pop('warpx_boost_adjust_transverse_positions', None) + def initialize_inputs(self, layout, initialize_self_fields = False, injection_plane_position = None, @@ -85,7 +87,8 @@ class Species(picmistandard.PICMI_Species): mass = self.mass, charge = self.charge, injection_style = 'python', - initialize_self_fields = int(initialize_self_fields)) + initialize_self_fields = int(initialize_self_fields), + boost_adjust_transverse_positions = self.boost_adjust_transverse_positions) pywarpx.Particles.particles_list.append(self.species) if self.initial_distribution is not None: @@ -314,7 +317,7 @@ class BinomialSmoother(picmistandard.PICMI_BinomialSmoother): def initialize_inputs(self, solver): pywarpx.warpx.use_filter = 1 - pywarpx.warpx.use_filter_compensation = np.all(self.compensation) + pywarpx.warpx.use_filter_compensation = bool(np.all(self.compensation)) if self.n_pass is None: # If not specified, do at least one pass in each direction. self.n_pass = 1 -- cgit v1.2.3