AMReX_EB2_IndexSpaceI.H

Go to the documentation of this file.

 
template <typename G>
IndexSpaceImp<G>::IndexSpaceImp (const G& gshop, const Geometry& geom,
                                 int required_coarsening_level,
                                 int max_coarsening_level,
                                 int ngrow, bool build_coarse_level_by_coarsening,
                                 bool extend_domain_face, int num_coarsen_opt)
    : m_gshop(gshop),
      m_build_coarse_level_by_coarsening(build_coarse_level_by_coarsening),
      m_extend_domain_face(extend_domain_face),
      m_num_coarsen_opt(num_coarsen_opt)
{
    // build finest level (i.e., level 0) first
    AMREX_ALWAYS_ASSERT(required_coarsening_level >= 0 && required_coarsening_level <= 30);
    max_coarsening_level = std::max(required_coarsening_level,max_coarsening_level);
    max_coarsening_level = std::min(30,max_coarsening_level);
 
    int ngrow_finest = std::max(ngrow,0);
    for (int i = 1; i <= required_coarsening_level; ++i) {
        ngrow_finest *= 2;
    }
 
    m_geom.push_back(geom);
    m_domain.push_back(geom.Domain());
    m_ngrow.push_back(ngrow_finest);
    m_gslevel.reserve(max_coarsening_level+1);
    m_gslevel.emplace_back(this, gshop, geom, EB2::max_grid_size, ngrow_finest, extend_domain_face,
                           num_coarsen_opt);
 
    for (int ilev = 1; ilev <= max_coarsening_level; ++ilev)
    {
        bool coarsenable = m_geom.back().Domain().coarsenable(2,2);
        if (!coarsenable) {
            if (ilev <= required_coarsening_level) {
                amrex::Abort("IndexSpaceImp: domain is not coarsenable at level "+std::to_string(ilev));
            } else {
                break;
            }
        }
 
        int ng = (ilev > required_coarsening_level) ? 0 : m_ngrow.back()/2;
 
        Box cdomain = amrex::coarsen(m_geom.back().Domain(),2);
        Geometry cgeom = amrex::coarsen(m_geom.back(),2);
        m_gslevel.emplace_back(this, ilev, EB2::max_grid_size, ng, cgeom, m_gslevel[ilev-1]);
        if (!m_gslevel.back().isOK()) {
            m_gslevel.pop_back();
            if (ilev <= required_coarsening_level) {
                if (build_coarse_level_by_coarsening) {
                    amrex::Abort("Failed to build required coarse EB level "+std::to_string(ilev));
                } else {
                    m_gslevel.emplace_back(this, gshop, cgeom, EB2::max_grid_size, ng, extend_domain_face,
                                           num_coarsen_opt-ilev);
                }
            } else {
                break;
            }
        }
        m_geom.push_back(cgeom);
        m_domain.push_back(cdomain);
        m_ngrow.push_back(ng);
    }
}
 
 
template <typename G>
IndexSpaceImp<G>::IndexSpaceImp (const G& gshop, const Vector<Geometry>& geom,
                                 int ngrow,
                                 bool extend_domain_face, int num_coarsen_opt)
    : m_gshop(gshop),
      m_build_coarse_level_by_coarsening(false),
      m_extend_domain_face(extend_domain_face),
      m_num_coarsen_opt(num_coarsen_opt)
{
 
    auto nlevels = int(geom.size());
    m_gslevel.reserve(nlevels);
 
    for (int ilev = 0; ilev < nlevels; ++ilev)
    {
        int ng = ngrow;
        Geometry cgeom = geom[ilev];
        Box cdomain = cgeom.Domain();
        m_gslevel.emplace_back(this, gshop, cgeom, EB2::max_grid_size, ng, extend_domain_face,
                                           num_coarsen_opt-ilev);
        m_geom.push_back(cgeom);
        m_domain.push_back(cdomain);
        m_ngrow.push_back(ng);
    }
}
 
template <typename G>
const Level&
IndexSpaceImp<G>::getLevel (const Geometry& geom) const
{
    auto it = std::ranges::find(m_domain, geom.Domain());
    AMREX_ALWAYS_ASSERT_WITH_MESSAGE(it != std::end(m_domain),
                                     "IndexSpaceImp::getLevel: Geometry not found");
    int i = std::distance(m_domain.begin(), it);
    return m_gslevel[i];
}
 
template <typename G>
const Geometry&
IndexSpaceImp<G>::getGeometry (const Box& dom) const
{
    auto it = std::ranges::find(m_domain, dom);
    AMREX_ALWAYS_ASSERT_WITH_MESSAGE(it != std::end(m_domain),
                                     "IndexSpaceImp::getLevel: domain not found");
    int i = std::distance(m_domain.begin(), it);
    return m_geom[i];
}
 
template <typename G>
void
IndexSpaceImp<G>::addFineLevels (int num_new_fine_levels)
{
    if (num_new_fine_levels <= 0) { return; }
    AMREX_ALWAYS_ASSERT_WITH_MESSAGE(num_new_fine_levels <= 30,
                                     "IndexSpaceImp::addFineLevels: num_new_fine_levels too large");
 
    if (m_num_coarsen_opt > 0) {
        m_num_coarsen_opt += num_new_fine_levels;
    }
 
    int rr_fine = static_cast<int>(1U << num_new_fine_levels);
    IndexSpaceImp<G> fine_isp(m_gshop, amrex::refine(m_geom[0], rr_fine),
                              num_new_fine_levels-1, num_new_fine_levels-1,
                              m_ngrow[0], m_build_coarse_level_by_coarsening,
                              m_extend_domain_face, m_num_coarsen_opt);
 
    fine_isp.m_gslevel.reserve(m_domain.size()+num_new_fine_levels);
    for (int i = 0; i < m_domain.size(); ++i) {
        fine_isp.m_gslevel.emplace_back(std::move(m_gslevel[i]));
    }
    std::swap(fine_isp.m_gslevel, m_gslevel);
 
    // The new levels were built by fine_isp, which is about to be destroyed.
    for (auto& lev : m_gslevel) {
        lev.m_parent = this;
    }
 
    m_geom.insert(m_geom.begin(), fine_isp.m_geom.begin(), fine_isp.m_geom.end());
    m_domain.insert(m_domain.begin(), fine_isp.m_domain.begin(), fine_isp.m_domain.end());
    m_ngrow.insert(m_ngrow.begin(), fine_isp.m_ngrow.begin(), fine_isp.m_ngrow.end());
}
 
template <typename G>
void
IndexSpaceImp<G>::addRegularCoarseLevels (int num_new_coarse_levels)
{
    if (num_new_coarse_levels <= 0) { return; }
    AMREX_ALWAYS_ASSERT_WITH_MESSAGE(num_new_coarse_levels <= 30,
                                     "IndexSpaceImp::addRegularCoarseLevels: num_new_coarse_levels too large");
 
    auto nlevs_old = int(m_gslevel.size());
    int nlevs_new = nlevs_old + num_new_coarse_levels;
 
    Vector<GShopLevel<G>> new_gslevel;
    new_gslevel.reserve(nlevs_new);
 
    Vector<Geometry> new_geom;
    new_geom.reserve(nlevs_new);
 
    Vector<Box> new_domain;
    new_domain.reserve(nlevs_new);
 
    Vector<int> new_ngrow;
    new_ngrow.reserve(nlevs_new);
 
    for (int ilev = 0; ilev < num_new_coarse_levels; ++ilev) {
        int rr = static_cast<int>(1U << (num_new_coarse_levels-ilev)); // 2^(num_new_coarse_levels-ilev)
        new_geom.push_back(amrex::coarsen(m_geom[0], rr));
        new_domain.push_back(new_geom.back().Domain());
        new_ngrow.push_back(m_ngrow[0]);
        new_gslevel.push_back(GShopLevel<G>::makeAllRegular(this, new_geom.back()));
    }
 
    for (int ilev = 0; ilev < nlevs_old; ++ilev) {
        new_gslevel.emplace_back(std::move(m_gslevel[ilev]));
        new_geom.push_back  (m_geom  [ilev]);
        new_domain.push_back(m_domain[ilev]);
        new_ngrow.push_back (m_ngrow [ilev]);
    }
 
    std::swap(new_gslevel, m_gslevel);
    std::swap(new_geom   , m_geom);
    std::swap(new_domain , m_domain);
    std::swap(new_ngrow  , m_ngrow);
 
    for (int ilev = num_new_coarse_levels-1; ilev >= 0; --ilev) {
        m_gslevel[ilev].buildCutCellMask(m_gslevel[ilev+1]);
    }
}
 
template <typename G>
void
IndexSpaceImp<G>::setShift (int direction, int ncells)
{
    auto nlevs = int(m_gslevel.size());
    for (int ilev = nlevs-1; ilev >= 0; --ilev) {
        m_gslevel[ilev].setShift(direction, ncells);
        ncells *= 2;
    }
}
 
template <typename G>
void
IndexSpaceImp<G>::buildAllFCData ()
{
    for (int face_dir = 0; face_dir < AMREX_SPACEDIM; ++face_dir) {
        for (int ilev = 0; ilev < int(m_gslevel.size()); ++ilev) {
            m_gslevel[ilev].buildFCData(m_gshop, face_dir, EB2::max_grid_size);
        }
    }
}