1#ifndef AMREX_EB2_LEVEL_H_
2#define AMREX_EB2_LEVEL_H_
3#include <AMReX_Config.H>
16#include <AMReX_EB2_C.H>
27#include <unordered_map>
37template <
typename G>
class GShopLevel;
74 void write_to_chkpt_file (
const std::string& fname,
bool extend_domain_face,
int max_grid_size)
const;
79 void setShift (
int direction,
int ncells);
117 void setRegularLevel () {
130 int ngrow,
bool extend_domain_face,
int num_crse_opt);
135 int max_grid_size,
int ngrow,
bool extend_domain_face,
int num_crse_opt);
137#if (AMREX_SPACEDIM == 3)
138 template <
typename GS = G, std::enable_if_t<std::is_same_v<GS,STLtools>,
int> FOO = 0>
140 int max_grid_size,
int ngrow,
bool extend_domain_face,
int num_crse_opt);
155 void prepare_grids (G
const& gshop,
Geometry const& geom,
int max_grid_size,
int ngrow,
156 bool extend_domain_face,
int num_crse_opt);
167 int max_grid_size,
int ngrow,
bool extend_domain_face,
int num_crse_opt)
170 if (std::is_same<typename G::FunctionType, AllRegularIF>::value) {
176 define_fine(gshop, geom, max_grid_size, ngrow, extend_domain_face, num_crse_opt);
182 int max_grid_size,
int ngrow,
bool extend_domain_face,
int num_crse_opt)
185 amrex::Print() <<
"AMReX WARNING: extend_domain_face=false is not recommended!\n";
188 BL_PROFILE(
"EB2::GShopLevel()-prepare_grids");
191 m_ngrow =
IntVect{
static_cast<int>(std::ceil(ngrow/16.)) * 16};
194 Box domain_grown = domain;
195 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
199 m_ngrow[idim] = std::min(m_ngrow[idim], domain_grown.length(idim));
202 domain_grown.grow(m_ngrow);
203 m_bounding_box = (extend_domain_face) ? domain : domain_grown;
204 m_bounding_box.surroundingNodes();
207 BoxList covered_boxes;
212 num_crse_opt = std::max(0,std::min(8,num_crse_opt));
213 for (
int clev = num_crse_opt; clev >= 0; --clev) {
215 if (domain.coarsenable(crse_ratio)) {
219 if (cut_boxes.isEmpty()) {
220 covered_boxes.clear();
221 test_boxes = BoxList(crse_geom.Domain());
222 test_boxes.maxSize(max_grid_size);
224 test_boxes.swap(cut_boxes);
225 test_boxes.coarsen(crse_ratio);
226 test_boxes.maxSize(max_grid_size);
229 const Long nboxes = test_boxes.size();
230 const auto& boxes = test_boxes.data();
231 for (
Long i = iproc; i < nboxes; i += nprocs) {
232 const Box& vbx = boxes[i];
234 auto box_type = gshop.getBoxType(gbx&crse_bounding_box,crse_geom,
RunOn::Gpu);
235 if (box_type == gshop.allcovered) {
237 }
else if (box_type == gshop.mixedcells) {
249 auto grow_at_domain_boundary = [&] (BoxList& bl)
251 const IntVect& domlo = domain.smallEnd();
252 const IntVect& domhi = domain.bigEnd();
254 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
255 if (m_ngrow[idim] != 0) {
256 if (b.smallEnd(idim) == domlo[idim]) {
257 b.growLo(idim,m_ngrow[idim]);
259 if (b.bigEnd(idim) == domhi[idim]) {
260 b.growHi(idim,m_ngrow[idim]);
266 grow_at_domain_boundary(covered_boxes);
267 grow_at_domain_boundary(cut_boxes);
270 if ( cut_boxes.isEmpty() &&
271 !covered_boxes.isEmpty())
273 amrex::Abort(
"AMReX_EB2_Level.H: Domain is completely covered");
276 if (!covered_boxes.isEmpty()) {
277 if (num_crse_opt > 2) {
278 covered_boxes.maxSize(max_grid_size*4);
280 m_covered_grids = BoxArray(std::move(covered_boxes));
283 if (cut_boxes.isEmpty()) {
284 m_grids = BoxArray();
285 m_dmap = DistributionMapping();
289 m_grids = BoxArray(std::move(cut_boxes));
290 m_dmap = DistributionMapping(m_grids);
297 int max_grid_size,
int ngrow,
bool extend_domain_face,
int num_crse_opt)
301#ifdef AMREX_USE_FLOAT
302 Real small_volfrac = 1.e-5_rt;
304 Real small_volfrac = 1.e-14;
306 bool cover_multiple_cuts =
false;
311 pp.
queryAdd(
"cover_multiple_cuts", cover_multiple_cuts);
314 maxiter = std::min(100000, maxiter);
316 prepare_grids(gshop, geom, max_grid_size, ngrow, extend_domain_face, num_crse_opt);
318 if (m_allregular) {
return; }
320 m_mgf.define(m_grids, m_dmap);
323 mf_info.
SetTag(
"EB2::Level");
324 m_cellflag.define(m_grids, m_dmap, 1, ng, mf_info);
325 m_volfrac.define(m_grids, m_dmap, 1, ng, mf_info);
326 m_centroid.define(m_grids, m_dmap, AMREX_SPACEDIM, ng, mf_info);
327 m_bndryarea.define(m_grids, m_dmap, 1, ng, mf_info);
328 m_bndrycent.define(m_grids, m_dmap, AMREX_SPACEDIM, ng, mf_info);
329 m_bndrynorm.define(m_grids, m_dmap, AMREX_SPACEDIM, ng, mf_info);
330 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
332 m_dmap, 1, ng, mf_info);
334 m_dmap, AMREX_SPACEDIM-1, ng, mf_info);
335 IntVect edge_type{1}; edge_type[idim] = 0;
336 m_edgecent[idim].define(
amrex::convert(m_grids, edge_type), m_dmap, 1, ng, mf_info);
342 auto bounding_box = m_bounding_box;
343 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
344 if (!extend_domain_face || geom.
isPeriodic(idim)) {
356 for (; iter < maxiter; ++iter)
361#pragma omp parallel if (Gpu::notInLaunchRegion()) reduction(+:nsmallcells,nmulticuts)
364#if (AMREX_SPACEDIM == 3)
370 auto& gfab = m_mgf[mfi];
371 const Box& vbx = gfab.validbox();
373 auto&
levelset = gfab.getLevelSet();
375 gshop.fillFab(
levelset, geom, gshop_run_on, bounding_box);
402 auto& facetype = gfab.getFaceType();
410#if (AMREX_SPACEDIM == 3)
411 auto& edgetype = gfab.getEdgeType();
424 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
425 edgetype[idim].prefetchToHost();
426 m_edgecent[idim][mfi].prefetchToHost();
431 gshop.getIntercept({xip,yip,zip}, {xdg,ydg,zdg}, clst,
432 geom, gshop_run_on, bounding_box);
436 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
437 edgetype[idim].prefetchToDevice();
438 m_edgecent[idim][mfi].prefetchToDevice();
443 gshop.updateIntercept({xip,yip,zip}, {xdg,ydg,zdg}, clst, geom);
446 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
447 const Box& b = facetype[idim].box();
454 nmc = build_faces(vbx, cfg, ftx, fty, ftz, xdg, ydg, zdg, lst,
456 xm2, ym2, zm2, dx, problo, cover_multiple_cuts);
458 cellflagtmp.
resize(m_cellflag[mfi].box());
462 build_cells(vbx, cfg, ftx, fty, ftz,
apx,
apy,
apz,
463 fcx,
fcy,
fcz, xm2, ym2, zm2, dx, vfr, ctr,
464 bar, bct, bnm, cfgtmp, lst,
465 small_volfrac, geom, extend_domain_face, cover_multiple_cuts,
473#elif (AMREX_SPACEDIM == 2)
481 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
482 facetype[idim].prefetchToHost();
483 m_edgecent[idim][mfi].prefetchToHost();
489 gshop.getIntercept({xip,yip},
490 {facetype[1].const_array(), facetype[0].const_array()},
491 clst, geom, gshop_run_on, bounding_box);
495 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
496 facetype[idim].prefetchToDevice();
497 m_edgecent[idim][mfi].prefetchToDevice();
502 gshop.updateIntercept({xip,yip},
503 {facetype[1].const_array(), facetype[0].const_array()},
507 nmc = build_faces(vbx, cfg, ftx, fty, lst, xip, yip,
apx,
apy,
fcx,
fcy,
508 dx, problo, cover_multiple_cuts, nsm);
510 build_cells(vbx, cfg, ftx, fty,
apx,
apy, dx, vfr, ctr,
511 bar, bct, bnm, lst, small_volfrac, geom, extend_domain_face,
520 if (nsmallcells == 0 && nmulticuts == 0) {
523 auto ls = m_mgf.getLevelSet();
528 amrex::Print() <<
"AMReX EB: Iter. " << iter+1 <<
" fixed " << nsmallcells
529 <<
" small cells" <<
'\n';
532 amrex::Print() <<
"AMReX EB: Iter. " << iter+1 <<
" fixed " << nmulticuts
533 <<
" multicuts" <<
'\n';
542#pragma omp parallel if (Gpu::notInLaunchRegion())
546 auto& gfab = m_mgf[mfi];
547 auto const&
levelset = gfab.getLevelSet();
553#
if (AMREX_SPACEDIM == 3)
554 auto const& edgetype = gfab.getEdgeType();
559 intercept_to_edge_centroid(xip, yip, zip, xdg, ydg, zdg, clst, dx, problo);
561#elif (AMREX_SPACEDIM == 2)
562 auto& facetype = gfab.getFaceType();
566 intercept_to_edge_centroid(xip, yip, fty, ftx, clst, dx, problo);
570 m_levelset = m_mgf.getLevelSet();
575#if (AMREX_SPACEDIM == 3)
577template <
typename GS, std::enable_if_t<std::is_same_v<GS,STLtools>,
int> FOO>
580 int max_grid_size,
int ngrow,
bool extend_domain_face,
int num_crse_opt)
584 prepare_grids(gshop, geom, max_grid_size, ngrow, extend_domain_face, num_crse_opt);
586 if (m_allregular) {
return; }
591 gshop.fillSignedDistance(m_sdf, m_sdf.nGrowVect(), geom);
594 m_marching_cubes[mfi.index()] = std::make_unique<MC::MCFab>();
599#if defined(AMREX_USE_OMP) && !defined(AMREX_USE_GPU)
629 const int coarse_ratio = 2;
630 const int min_width = 8;
631 bool coarsenable = fine_grids.
coarsenable(coarse_ratio, min_width)
632 && (fine_covered_grids.
empty() || fine_covered_grids.
coarsenable(coarse_ratio));
#define BL_PROFILE(a)
Definition AMReX_BLProfiler.H:551
#define AMREX_ALWAYS_ASSERT_WITH_MESSAGE(EX, MSG)
Definition AMReX_BLassert.H:49
amrex::ParmParse pp
Input file parser instance for the given namespace.
Definition AMReX_HypreIJIface.cpp:15
#define AMREX_D_TERM(a, b, c)
Definition AMReX_SPACE.H:172
void resize(const Box &b, int N=1, Arena *ar=nullptr)
This function resizes a BaseFab so it covers the Box b with N components.
Definition AMReX_BaseFab.H:1628
Array4< T const > array() const noexcept
Definition AMReX_BaseFab.H:381
Elixir elixir() noexcept
Definition AMReX_BaseFab.H:1670
A collection of Boxes stored in an Array.
Definition AMReX_BoxArray.H:567
bool coarsenable(int refinement_ratio, int min_width=1) const
Coarsen each Box in the BoxArray to the specified ratio.
Definition AMReX_BoxArray.cpp:615
bool empty() const noexcept
Return whether the BoxArray is empty.
Definition AMReX_BoxArray.H:620
__host__ __device__ BoxND< new_dim > resize() const noexcept
Return a new BoxND of size new_dim by either shrinking or expanding this BoxND.
Definition AMReX_Box.H:893
GpuArray< Real, 3 > CellSizeArray() const noexcept
Definition AMReX_CoordSys.H:76
Calculates the distribution of FABs to MPI processes.
Definition AMReX_DistributionMapping.H:43
static constexpr int ng
Definition AMReX_EB2_MultiGFab.H:19
Definition AMReX_EB2_Level.H:127
GShopLevel(IndexSpace const *is, const Geometry &geom)
Definition AMReX_EB2_Level.H:161
void define_fine(G const &gshop, const Geometry &geom, int max_grid_size, int ngrow, bool extend_domain_face, int num_crse_opt)
Definition AMReX_EB2_Level.H:296
GShopLevel(IndexSpace const *is, int ilev, int max_grid_size, int ngrow, const Geometry &geom, GShopLevel< G > &fineLevel)
Definition AMReX_EB2_Level.H:614
void define_fine_mvmc(GS const &gshop, const Geometry &geom, int max_grid_size, int ngrow, bool extend_domain_face, int num_crse_opt)
Definition AMReX_EB2_Level.H:579
GShopLevel(IndexSpace const *is, G const &gshop, const Geometry &geom, int max_grid_size, int ngrow, bool extend_domain_face, int num_crse_opt)
Definition AMReX_EB2_Level.H:166
static GShopLevel< G > makeAllRegular(IndexSpace const *is, const Geometry &geom)
Definition AMReX_EB2_Level.H:144
Definition AMReX_EB2.H:28
Definition AMReX_EB2_Level.H:40
MultiFab m_bndryarea
Definition AMReX_EB2_Level.H:100
bool isOK() const noexcept
Definition AMReX_EB2_Level.H:44
void fillFaceCent(Array< MultiCutFab *, 3 > const &facecent, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:794
MultiFab m_volfrac
Definition AMReX_EB2_Level.H:98
MultiGFab m_mgf
Definition AMReX_EB2_Level.H:95
bool m_allregular
Definition AMReX_EB2_Level.H:107
IntVect m_shift
Definition AMReX_EB2_Level.H:110
iMultiFab m_cutcellmask
Definition AMReX_EB2_Level.H:106
Array< MultiFab, 3 > m_facecent
Definition AMReX_EB2_Level.H:104
Array< MultiFab, 3 > m_edgecent
Definition AMReX_EB2_Level.H:105
void prepareForCoarsening(const Level &rhs, int max_grid_size, IntVect const &ngrow)
Definition AMReX_EB2_Level.cpp:10
void fillCutCellMask(iMultiFab &cutcellmask, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:946
MultiFab m_centroid
Definition AMReX_EB2_Level.H:99
FabArray< EBCellFlagFab > m_cellflag
Definition AMReX_EB2_Level.H:97
const DistributionMapping & DistributionMap() const noexcept
Definition AMReX_EB2_Level.H:64
BoxArray m_covered_grids
Definition AMReX_EB2_Level.H:91
void fillLevelSet(MultiFab &levelset, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:907
Geometry m_geom
Definition AMReX_EB2_Level.H:88
Array< MultiFab, 3 > m_areafrac
Definition AMReX_EB2_Level.H:103
IntVect m_ngrow
Definition AMReX_EB2_Level.H:89
IndexSpace const * m_parent
Definition AMReX_EB2_Level.H:111
BoxArray m_grids
Definition AMReX_EB2_Level.H:90
const Geometry & Geom() const noexcept
Definition AMReX_EB2_Level.H:69
std::map< int, std::unique_ptr< MC::MCFab > > m_marching_cubes
Definition AMReX_EB2_Level.H:94
bool hasEBInfo() const noexcept
Definition AMReX_EB2_Level.H:76
void write_to_chkpt_file(const std::string &fname, bool extend_domain_face, int max_grid_size) const
Definition AMReX_EB2_Level.cpp:956
void fillEBCellFlag(FabArray< EBCellFlagFab > &cellflag, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:429
void fillEdgeCent(Array< MultiCutFab *, 3 > const &edgecent, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:835
int coarsenFromFine(Level &fineLevel, bool fill_boundary)
Definition AMReX_EB2_Level.cpp:84
MultiFab m_bndrynorm
Definition AMReX_EB2_Level.H:102
bool m_has_eb_info
Definition AMReX_EB2_Level.H:109
void fillAreaFrac(Array< MultiCutFab *, 3 > const &areafrac, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:640
MultiFab m_bndrycent
Definition AMReX_EB2_Level.H:101
MultiFab m_levelset
Definition AMReX_EB2_Level.H:96
void setShift(int direction, int ncells)
Definition AMReX_EB2_Level.cpp:1047
bool isAllRegular() const noexcept
Definition AMReX_EB2_Level.H:43
void fillBndryCent(MultiCutFab &bndrycent, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:592
void fillBndryNorm(MultiCutFab &bndrynorm, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:616
MultiFab m_sdf
Definition AMReX_EB2_Level.H:93
IntVect const & nGrowVect() const noexcept
Definition AMReX_EB2_Level.H:72
void fillVolFrac(MultiFab &vfrac, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:483
const BoxArray & boxArray() const noexcept
Definition AMReX_EB2_Level.H:63
Level(IndexSpace const *is, const Geometry &geom)
Definition AMReX_EB2_Level.H:66
friend class GShopLevel
Definition AMReX_EB2_Level.H:114
bool m_ok
Definition AMReX_EB2_Level.H:108
void fillCentroid(MultiCutFab ¢roid, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:544
DistributionMapping m_dmap
Definition AMReX_EB2_Level.H:92
void fillBndryArea(MultiCutFab &bndryarea, const Geometry &geom) const
Definition AMReX_EB2_Level.cpp:568
void buildCellFlag()
Definition AMReX_EB2_Level.cpp:403
void buildCutCellMask(Level const &fine_level)
Definition AMReX_EB2_Level.cpp:966
IndexSpace const * getEBIndexSpace() const noexcept
Definition AMReX_EB2_Level.H:70
Definition AMReX_EB2_MultiGFab.H:65
Definition AMReX_EBCellFlag.H:287
An Array of FortranArrayBox(FAB)-like Objects.
Definition AMReX_FabArray.H:347
Rectangular problem domain geometry.
Definition AMReX_Geometry.H:74
const Box & Domain() const noexcept
Returns our rectangular domain.
Definition AMReX_Geometry.H:211
Periodicity periodicity() const noexcept
Definition AMReX_Geometry.H:356
GpuArray< Real, 3 > ProbLoArray() const noexcept
Definition AMReX_Geometry.H:187
bool isPeriodic(int dir) const noexcept
Is the domain periodic in the specified direction?
Definition AMReX_Geometry.H:332
Definition AMReX_GpuElixir.H:13
__host__ static __device__ constexpr IntVectND< dim > TheZeroVector() noexcept
This static member function returns a reference to a constant IntVectND object, all of whose dim argu...
Definition AMReX_IntVect.H:679
__host__ static __device__ constexpr IntVectND< dim > TheDimensionVector(int d) noexcept
This static member function returns a reference to a constant IntVectND object, all of whose dim argu...
Definition AMReX_IntVect.H:698
Iterator for looping ever tiles and boxes of amrex::FabArray based containers.
Definition AMReX_MFIter.H:85
bool isValid() const noexcept
Is the iterator valid i.e. is it associated with a FAB?
Definition AMReX_MFIter.H:169
Definition AMReX_MultiCutFab.H:81
A collection (stored as an array) of FArrayBox objects.
Definition AMReX_MultiFab.H:40
Parse Parameters From Command Line and Input Files.
Definition AMReX_ParmParse.H:346
int queryAdd(std::string_view name, T &ref)
If name is found, the value in the ParmParse database will be stored in the ref argument....
Definition AMReX_ParmParse.H:1040
This class provides the user with a few print options.
Definition AMReX_Print.H:35
A Collection of IArrayBoxes.
Definition AMReX_iMultiFab.H:34
amrex_real Real
Floating Point Type for Fields.
Definition AMReX_REAL.H:79
amrex_long Long
Definition AMReX_INT.H:30
__host__ __device__ BoxND< dim > coarsen(const BoxND< dim > &b, int ref_ratio) noexcept
Coarsen BoxND by given (positive) coarsening ratio.
Definition AMReX_Box.H:1409
__host__ __device__ BoxND< dim > grow(const BoxND< dim > &b, int i) noexcept
Grow BoxND in all directions by given amount.
Definition AMReX_Box.H:1280
__host__ __device__ BoxND< dim > refine(const BoxND< dim > &b, int ref_ratio) noexcept
Definition AMReX_Box.H:1459
std::array< T, N > Array
Definition AMReX_Array.H:26
int MyProc() noexcept
Definition AMReX_ParallelDescriptor.H:128
int NProcs() noexcept
Definition AMReX_ParallelDescriptor.H:255
Definition AMReX_FabArrayBase.H:33
void streamSynchronize() noexcept
Definition AMReX_GpuDevice.H:263
bool inLaunchRegion() noexcept
Definition AMReX_GpuControl.H:92
void write_stl(std::string const &filename, std::map< int, std::unique_ptr< MCFab > > const &mc_fabs)
Definition AMReX_MarchingCubes.cpp:924
void marching_cubes(Geometry const &geom, FArrayBox &sdf_fab, MCFab &mc_fab)
Definition AMReX_MarchingCubes.cpp:693
void Initialize()
Definition AMReX_MarchingCubes.cpp:651
MPI_Comm CommunicatorSub() noexcept
sub-communicator for current frame
Definition AMReX_ParallelContext.H:70
Definition AMReX_Amr.cpp:49
__host__ __device__ void ignore_unused(const Ts &...)
This shuts up the compiler about unused variables.
Definition AMReX.H:138
__host__ __device__ BoxND< dim > convert(const BoxND< dim > &b, const IntVectND< dim > &typ) noexcept
Return a BoxND with different type.
Definition AMReX_Box.H:1558
__host__ __device__ BoxND< dim > surroundingNodes(const BoxND< dim > &b, int dir) noexcept
Return a BoxND with NODE based coordinates in direction dir that encloses BoxND b....
Definition AMReX_Box.H:1522
RunOn
Definition AMReX_GpuControl.H:69
BoxND< 3 > Box
Box is an alias for amrex::BoxND instantiated with AMREX_SPACEDIM.
Definition AMReX_BaseFwd.H:30
IntVectND< 3 > IntVect
IntVect is an alias for amrex::IntVectND instantiated with AMREX_SPACEDIM.
Definition AMReX_BaseFwd.H:33
__host__ __device__ constexpr IntVectND< dim > scale(const IntVectND< dim > &p, int s) noexcept
Returns a IntVectND obtained by multiplying each of the components of this IntVectND by s.
Definition AMReX_IntVect.H:1013
int Verbose() noexcept
Definition AMReX.cpp:169
void Abort(const std::string &msg)
Print out message to cerr and exit via abort().
Definition AMReX.cpp:230
void AllGatherBoxes(Vector< Box > &bxs, int n_extra_reserve)
Definition AMReX_Box.cpp:124
A multidimensional array accessor.
Definition AMReX_Array4.H:224
FabArray memory allocation information.
Definition AMReX_FabArray.H:66
MFInfo & SetTag() noexcept
Definition AMReX_FabArray.H:79
Definition AMReX_MFIter.H:20
MFItInfo & SetDynamic(bool f) noexcept
Definition AMReX_MFIter.H:40