diff options
Diffstat (limited to 'Source/WarpX.cpp')
| -rw-r--r-- | Source/WarpX.cpp | 47 |
1 files changed, 42 insertions, 5 deletions
diff --git a/Source/WarpX.cpp b/Source/WarpX.cpp index d4f9b5de1..a31c0c1e3 100644 --- a/Source/WarpX.cpp +++ b/Source/WarpX.cpp @@ -171,6 +171,9 @@ Real WarpX::self_fields_required_precision = 1.e-11_rt; int WarpX::self_fields_max_iters = 200; int WarpX::do_subcycling = 0; +int WarpX::do_multi_J = 0; +int WarpX::do_multi_J_n_depositions; +int WarpX::J_linear_in_time = 0; bool WarpX::safe_guard_cells = 0; IntVect WarpX::filter_npass_each_dir(1); @@ -447,6 +450,11 @@ WarpX::ReadParameters () pp_warpx.query("verbose", verbose); pp_warpx.query("regrid_int", regrid_int); pp_warpx.query("do_subcycling", do_subcycling); + pp_warpx.query("do_multi_J", do_multi_J); + if (do_multi_J) + { + pp_warpx.get("do_multi_J_n_depositions", do_multi_J_n_depositions); + } pp_warpx.query("use_hybrid_QED", use_hybrid_QED); pp_warpx.query("safe_guard_cells", safe_guard_cells); std::vector<std::string> override_sync_intervals_string_vec = {"1"}; @@ -616,12 +624,18 @@ WarpX::ReadParameters () pp_warpx.query("do_pml_j_damping", do_pml_j_damping); pp_warpx.query("do_pml_in_domain", do_pml_in_domain); + if (do_multi_J && do_pml) + { + amrex::Abort("Multi-J algorithm not implemented with PMLs"); + } + // div(E) cleaning not implemented for PSATD solver if (maxwell_solver_id == MaxwellSolverAlgo::PSATD) { - AMREX_ALWAYS_ASSERT_WITH_MESSAGE( - do_dive_cleaning == 0, - "warpx.do_dive_cleaning = 1 not implemented for PSATD solver"); + if (do_multi_J == 0 && do_dive_cleaning == 1) + { + amrex::Abort("warpx.do_dive_cleaning = 1 not implemented for PSATD solver"); + } } // Default values of WarpX::do_pml_dive_cleaning and WarpX::do_pml_divb_cleaning: @@ -958,6 +972,7 @@ WarpX::ReadParameters () pp_psatd.query("current_correction", current_correction); pp_psatd.query("v_comoving", m_v_comoving); pp_psatd.query("do_time_averaging", fft_do_time_averaging); + pp_psatd.query("J_linear_in_time", J_linear_in_time); if (!fft_periodic_single_box && current_correction) amrex::Abort( @@ -1021,6 +1036,24 @@ WarpX::ReadParameters () "psatd.update_with_rho must be equal to 1 for comoving PSATD"); } + if (do_multi_J) + { + if (m_v_galilean[0] != 0. || m_v_galilean[1] != 0. || m_v_galilean[2] != 0.) + { + amrex::Abort("Multi-J algorithm not implemented with Galilean PSATD"); + } + } + + if (J_linear_in_time) + { + AMREX_ALWAYS_ASSERT_WITH_MESSAGE(update_with_rho, + "psatd.update_with_rho must be set to 1 when psatd.J_linear_in_time = 1"); + AMREX_ALWAYS_ASSERT_WITH_MESSAGE(do_dive_cleaning, + "warpx.do_dive_cleaning must be set to 1 when psatd.J_linear_in_time = 1"); + AMREX_ALWAYS_ASSERT_WITH_MESSAGE(do_divb_cleaning, + "warpx.do_divb_cleaning must be set to 1 when psatd.J_linear_in_time = 1"); + } + constexpr int zdir = AMREX_SPACEDIM - 1; if (WarpX::field_boundary_lo[zdir] == FieldBoundaryType::Damped || WarpX::field_boundary_hi[zdir] == FieldBoundaryType::Damped ) { @@ -1776,6 +1809,9 @@ void WarpX::AllocLevelSpectralSolver (amrex::Vector<std::unique_ptr<SpectralSolv RealVect dx_vect(dx[0], dx[2]); #endif + amrex::Real solver_dt = dt[lev]; + if (WarpX::do_multi_J) solver_dt /= static_cast<amrex::Real>(WarpX::do_multi_J_n_depositions); + auto pss = std::make_unique<SpectralSolver>(lev, realspace_ba, dm, @@ -1786,11 +1822,12 @@ void WarpX::AllocLevelSpectralSolver (amrex::Vector<std::unique_ptr<SpectralSolv m_v_galilean, m_v_comoving, dx_vect, - dt[lev], + solver_dt, pml_flag, fft_periodic_single_box, update_with_rho, - fft_do_time_averaging); + fft_do_time_averaging, + J_linear_in_time); spectral_solver[lev] = std::move(pss); } # endif |