diff options
Diffstat (limited to '')
| -rw-r--r-- | Examples/Modules/embedded_boundary_cube/inputs_3d | 3 | ||||
| -rw-r--r-- | Source/EmbeddedBoundary/WarpXInitEB.cpp | 55 | ||||
| -rw-r--r-- | Source/WarpX.H | 3 |
3 files changed, 56 insertions, 5 deletions
diff --git a/Examples/Modules/embedded_boundary_cube/inputs_3d b/Examples/Modules/embedded_boundary_cube/inputs_3d index 481707540..e1bf83c95 100644 --- a/Examples/Modules/embedded_boundary_cube/inputs_3d +++ b/Examples/Modules/embedded_boundary_cube/inputs_3d @@ -15,6 +15,9 @@ eb2.geom_type = box eb2.box_lo = -0.5 -0.5 -0.5 eb2.box_hi = 0.5 0.5 0.5 eb2.box_has_fluid_inside = true +# Alternatively one could use parser to build EB +# Note that for amrex EB implicit function, >0 is covered, =0 is boundary and <0 is regular. +# warpx.eb_implicit_function = "max(max(max(x-0.5,-0.5-x), max(y-0.5,-0.5-y)), max(z-0.5,-0.5-z))" warpx.B_ext_grid_init_style = parse_B_ext_grid_function my_constants.m = 0 diff --git a/Source/EmbeddedBoundary/WarpXInitEB.cpp b/Source/EmbeddedBoundary/WarpXInitEB.cpp index 8214eec9b..b0de40136 100644 --- a/Source/EmbeddedBoundary/WarpXInitEB.cpp +++ b/Source/EmbeddedBoundary/WarpXInitEB.cpp @@ -6,6 +6,40 @@ # include <AMReX_ParmParse.H> #endif +#ifdef AMREX_USE_EB +namespace { + class ParserIF + : public amrex::GPUable + { + public: + ParserIF (const ParserWrapper<3>& a_parser) + : m_parser(a_parser.getParser()) + {} + + ParserIF (const ParserIF& rhs) noexcept = default; + ParserIF (ParserIF&& rhs) noexcept = default; + ParserIF& operator= (const ParserIF& rhs) = delete; + ParserIF& operator= (ParserIF&& rhs) = delete; + + AMREX_GPU_HOST_DEVICE inline + amrex::Real operator() (AMREX_D_DECL(amrex::Real x, amrex::Real y, + amrex::Real z)) const noexcept { +#if (AMREX_SPACEDIM == 2) + return m_parser(x,amrex::Real(0.0),y); +#else + return m_parser(x,y,z); +#endif + } + + inline amrex::Real operator() (const amrex::RealArray& p) const noexcept { + return this->operator()(AMREX_D_DECL(p[0],p[1],p[2])); + } + + private: + HostDeviceParser<3> m_parser; + }; +} +#endif void WarpX::InitEB () @@ -13,12 +47,23 @@ WarpX::InitEB () #ifdef AMREX_USE_EB BL_PROFILE("InitEB"); - amrex::ParmParse pp_eb2("eb2"); - if (!pp_eb2.contains("geom_type")) { - std::string geom_type = "all_regular"; - pp_eb2.add("geom_type", geom_type); // use all_regular by default + amrex::ParmParse pp_warpx("warpx"); + std::string impf; + pp_warpx.query("eb_implicit_function", impf); + if (! impf.empty()) { + WarpXParser wp = makeParser(impf, {"x", "y", "z"}); + m_eb_if_parser = std::make_unique<ParserWrapper<3> >(wp); + ParserIF pif(*m_eb_if_parser); + auto gshop = amrex::EB2::makeShop(pif); + amrex::EB2::Build(gshop, Geom(maxLevel()), maxLevel(), maxLevel()); + } else { + amrex::ParmParse pp_eb2("eb2"); + if (!pp_eb2.contains("geom_type")) { + std::string geom_type = "all_regular"; + pp_eb2.add("geom_type", geom_type); // use all_regular by default + } + amrex::EB2::Build(Geom(maxLevel()), maxLevel(), maxLevel()); } - amrex::EB2::Build(Geom(maxLevel()), maxLevel(), maxLevel()); #endif } diff --git a/Source/WarpX.H b/Source/WarpX.H index 82a3e1483..9141df0e9 100644 --- a/Source/WarpX.H +++ b/Source/WarpX.H @@ -1099,6 +1099,9 @@ private: amrex::EBFArrayBoxFactory const& fieldEBFactory (int lev) const noexcept { return static_cast<amrex::EBFArrayBoxFactory const&>(*m_field_factory[lev]); } + + // EB implicit funciton + std::unique_ptr<ParserWrapper<3> > m_eb_if_parser; #endif void InitEB (); |