7 #ifndef ABLASTR_VECTOR_POISSON_SOLVER_H
8 #define ABLASTR_VECTOR_POISSON_SOLVER_H
21 #include <AMReX_Config.H>
81 typename T_BoundaryHandler,
82 typename T_PostACalculationFunctor = std::nullopt_t,
83 typename T_FArrayBoxFactory =
void
88 amrex::Real
const relative_tolerance,
89 amrex::Real absolute_tolerance,
95 T_BoundaryHandler
const boundary_handler,
96 bool const do_single_precision_comms =
false,
98 [[maybe_unused]] T_PostACalculationFunctor post_A_calculation = std::nullopt,
99 [[maybe_unused]] std::optional<amrex::Real const> current_time = std::nullopt,
103 using namespace amrex::literals;
105 if (!rel_ref_ratio.has_value()) {
107 "rel_ref_ratio must be set if mesh-refinement is used");
111 auto const finest_level =
static_cast<int>(curr.size()) - 1;
114 amrex::Real max_comp_J = 0.0;
115 for (
int lev=0; lev<=finest_level; lev++) {
116 for (
int adim=0; adim<3; adim++) {
118 max_comp_J =
amrex::max(max_comp_J, curr[lev][adim]->norm0());
123 const bool always_use_bnorm = (max_comp_J > 0);
124 if (!always_use_bnorm) {
125 if (absolute_tolerance == 0.0) { absolute_tolerance = amrex::Real(1e-6); }
127 "MagnetostaticSolver",
128 "Max norm of J is 0",
136 for (
int lev=0; lev<=finest_level; lev++) {
138 {geom[lev]}, {grids[lev]}, {dmap[lev]}, info
139 #if defined(AMREX_USE_EB)
140 , {eb_farray_box_factory.value()[lev]}
144 {geom[lev]}, {grids[lev]}, {dmap[lev]}, info
145 #if defined(AMREX_USE_EB)
146 , {eb_farray_box_factory.value()[lev]}
150 {geom[lev]}, {grids[lev]}, {dmap[lev]}, info
151 #if defined(AMREX_USE_EB)
152 , {eb_farray_box_factory.value()[lev]}
159 for (
int adim=0; adim<3; adim++) {
166 #if defined(AMREX_USE_EB)
168 linop[adim]->setEBDirichlet(0_rt);
172 linop[adim]->setRZ(
true);
175 linop[adim]->setDomainBC( boundary_handler.lobc[adim], boundary_handler.hibc[adim] );
177 mlmg[adim] = std::make_unique<amrex::MLMG>(*linop[adim]);
180 mlmg[adim]->setMaxIter(max_iters);
181 mlmg[adim]->setAlwaysUseBNorm(always_use_bnorm);
184 mlmg[adim]->solve( {A[lev][adim]}, {curr[lev][adim]},
185 relative_tolerance, absolute_tolerance );
189 *A[lev][adim], A[lev][adim]->
nGrowVect(),
190 do_single_precision_comms,
191 geom[lev].periodicity(),
false);
198 if (lev < finest_level) {
204 const int ncomp = linop[adim]->getNComp();
219 do_single_precision_comms,
225 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
234 amrex::Box const b = mfi.tilebox(A[lev + 1][adim]->ixType().toIntVect());
245 if (post_A_calculation.has_value()) {
246 post_A_calculation.value()(mlmg, lev);
#define AMREX_D_DECL(a, b, c)
#define ABLASTR_ALWAYS_ASSERT_WITH_MESSAGE(EX, MSG)
Definition: TextMsg.H:75
BoxArray & coarsen(int refinement_ratio)
Array4< typename FabArray< FArrayBox >::value_type const > array(const MFIter &mfi) const noexcept
AMREX_GPU_HOST_DEVICE static constexpr AMREX_FORCE_INLINE IntVect TheUnitVector() noexcept
void setSigma(Array< Real, AMREX_SPACEDIM > const &a_sigma) noexcept
AMREX_GPU_DEVICE AMREX_FORCE_INLINE void interp(int j, int k, int l, amrex::Array4< amrex::Real > const &fine, amrex::Array4< amrex::Real const > const &crse, const amrex::IntVect r_ratio, IntVect const &type) noexcept
Definition: Interpolate_K.H:9
static constexpr auto mu0
vacuum permeability: magnetic permeability of vacuum = 4.0e-7 * pi [H/m]
Definition: constant.H:48
Definition: IntegratedGreenFunctionSolver.cpp:33
void computeVectorPotential(amrex::Vector< amrex::Array< amrex::MultiFab *, 3 > > const &curr, amrex::Vector< amrex::Array< amrex::MultiFab *, 3 > > &A, amrex::Real const relative_tolerance, amrex::Real absolute_tolerance, int const max_iters, int const verbosity, amrex::Vector< amrex::Geometry > const &geom, amrex::Vector< amrex::DistributionMapping > const &dmap, amrex::Vector< amrex::BoxArray > const &grids, T_BoundaryHandler const boundary_handler, bool const do_single_precision_comms=false, std::optional< amrex::Vector< amrex::IntVect > > rel_ref_ratio=std::nullopt, [[maybe_unused]] T_PostACalculationFunctor post_A_calculation=std::nullopt, [[maybe_unused]] std::optional< amrex::Real const > current_time=std::nullopt, [[maybe_unused]] std::optional< amrex::Vector< T_FArrayBoxFactory const * > > eb_farray_box_factory=std::nullopt)
Definition: VectorPoissonSolver.H:86
void FillBoundary(amrex::MultiFab &mf, amrex::IntVect ng, bool do_single_precision_comms, const amrex::Periodicity &period, std::optional< bool > nodal_sync)
Definition: Communication.cpp:71
void ParallelCopy(amrex::MultiFab &dst, const amrex::MultiFab &src, int src_comp, int dst_comp, int num_comp, const amrex::IntVect &src_nghost, const amrex::IntVect &dst_nghost, bool do_single_precision_comms, const amrex::Periodicity &period, amrex::FabArrayBase::CpOp op)
Definition: Communication.cpp:28
void WMRecordWarning(const std::string &topic, const std::string &text, const WarnPriority &priority=WarnPriority::medium)
Helper function to abbreviate the call to WarnManager::GetInstance().RecordWarning (recording a warni...
Definition: WarnManager.cpp:318
void ReduceRealMax(Vector< std::reference_wrapper< Real > > const &)
DistributionMapping const & DistributionMap(FabArrayBase const &fa)
IntVect nGrowVect(FabArrayBase const &fa)
AMREX_GPU_HOST_DEVICE constexpr AMREX_FORCE_INLINE const T & max(const T &a, const T &b) noexcept
bool TilingIfNotGPU() noexcept
std::enable_if_t< std::is_integral_v< T > > ParallelFor(TypeList< CTOs... >, std::array< int, sizeof...(CTOs)> const &runtime_options, T N, F &&f)
value
Definition: updateAMReX.py:141
Definition: Interpolate.H:33