From 4fb69efe4c87596b8f0d4a36620f6f63612c7aa1 Mon Sep 17 00:00:00 2001 From: David Grote Date: Mon, 24 Aug 2020 14:30:20 -0700 Subject: For PICMI, added more arguments for psatd solver (#1259) --- Python/pywarpx/picmi.py | 22 ++++++++++++++++------ 1 file changed, 16 insertions(+), 6 deletions(-) (limited to 'Python/pywarpx/picmi.py') diff --git a/Python/pywarpx/picmi.py b/Python/pywarpx/picmi.py index 0579b4337..940e1d017 100644 --- a/Python/pywarpx/picmi.py +++ b/Python/pywarpx/picmi.py @@ -458,9 +458,14 @@ class ElectromagneticSolver(picmistandard.PICMI_ElectromagneticSolver): self.pml_ncell = kw.pop('warpx_pml_ncell', None) if self.method == 'PSATD': - self.periodic_single_box_fft = kw.pop('warpx_periodic_single_box_fft', None) - self.fftw_plan_measure = kw.pop('warpx_fftw_plan_measure', None) - self.current_correction = kw.pop('warpx_current_correction', None) + self.psatd_periodic_single_box_fft = kw.pop('warpx_periodic_single_box_fft', None) + self.psatd_fftw_plan_measure = kw.pop('warpx_fftw_plan_measure', None) + self.psatd_current_correction = kw.pop('warpx_current_correction', None) + self.psatd_update_with_rho = kw.pop('warpx_psatd_update_with_rho', None) + self.psatd_do_time_averaging = kw.pop('warpx_psatd_do_time_averaging', None) + self.psatd_nx_guard = kw.pop('warpx_psatd_nx_guard', None) + self.psatd_ny_guard = kw.pop('warpx_psatd_ny_guard', None) + self.psatd_nz_guard = kw.pop('warpx_psatd_nz_guard', None) def initialize_inputs(self): @@ -471,9 +476,14 @@ class ElectromagneticSolver(picmistandard.PICMI_ElectromagneticSolver): pywarpx.warpx.do_nodal = self.l_nodal if self.method == 'PSATD': - pywarpx.psatd.periodic_single_box_fft = self.periodic_single_box_fft - pywarpx.psatd.fftw_plan_measure = self.fftw_plan_measure - pywarpx.psatd.current_correction = self.current_correction + pywarpx.psatd.periodic_single_box_fft = self.psatd_periodic_single_box_fft + pywarpx.psatd.fftw_plan_measure = self.psatd_fftw_plan_measure + pywarpx.psatd.current_correction = self.psatd_current_correction + pywarpx.psatd.update_with_rho = self.psatd_update_with_rho + pywarpx.psatd.do_time_averaging = self.psatd_do_time_averaging + pywarpx.psatd.nx_guard = self.psatd_nx_guard + pywarpx.psatd.ny_guard = self.psatd_ny_guard + pywarpx.psatd.nz_guard = self.psatd_nz_guard if self.stencil_order is not None: pywarpx.psatd.nox = self.stencil_order[0] -- cgit v1.2.3