AMReX_EB2.H

Go to the documentation of this file.

#ifndef AMREX_EB2_H_
#define AMREX_EB2_H_
#include <AMReX_Config.H>
 
#include <AMReX_Geometry.H>
#include <AMReX_Vector.H>
#include <AMReX_EB2_GeometryShop.H>
#include <AMReX_EB2_Level.H>
 
#include <cmath>
#include <algorithm>
#include <memory>
#include <type_traits>
#include <string>
 
namespace amrex::EB2 {
 
extern AMREX_EXPORT int max_grid_size;
 
void useEB2 (bool);
 
void Initialize ();
void Finalize ();
 
class IndexSpace
{
public:
    virtual ~IndexSpace() = default;
 
    IndexSpace () noexcept = default;
    IndexSpace (IndexSpace const&) = delete;
    IndexSpace (IndexSpace &&) = delete;
    IndexSpace& operator= (IndexSpace const&) = delete;
    IndexSpace& operator= (IndexSpace &&) = delete;
 
    // This function will take the ownership of the IndexSpace
    // pointer, and put it on the top of the stack (i.e., back of the
    // vector).  If the pointer is already in the stack, it will be
    // moved to the top.
    static void push (IndexSpace* ispace);
 
    static void push (std::unique_ptr<IndexSpace> ispace);
 
    // This erases `ispace` from the stack.
    static void erase (IndexSpace* ispace);
 
    static void pop () noexcept { m_instance.pop_back(); }
    static void clear () noexcept { m_instance.clear(); }
    static IndexSpace& top () {
        AMREX_ALWAYS_ASSERT_WITH_MESSAGE(!m_instance.empty(),
                                         "Have you forgot to call EB2::build? It's required even if the geometry is all regular.");
        return *(m_instance.back());
    }
    static bool empty () noexcept { return m_instance.empty(); }
    static int size () noexcept { return static_cast<int>(m_instance.size()); }
 
    [[nodiscard]] virtual const Level& getLevel (const Geometry & geom) const = 0;
    [[nodiscard]] virtual const Geometry& getGeometry (const Box& domain) const = 0;
    [[nodiscard]] virtual const Box& coarsestDomain () const = 0;
    virtual void addFineLevels (int num_new_fine_levels) = 0;
    virtual void addRegularCoarseLevels (int num_new_coarse_levels) = 0;
 
    virtual void setShift (int, int) {
        amrex::Abort("IndexSpace::setShift: not supported");
    }
 
protected:
    static AMREX_EXPORT Vector<std::unique_ptr<IndexSpace> > m_instance;
};
 
const IndexSpace* TopIndexSpaceIfPresent () noexcept;
inline const IndexSpace* TopIndexSpace () noexcept { return TopIndexSpaceIfPresent(); }
 
template <typename G>
class IndexSpaceImp
    : public IndexSpace
{
public:
 
    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);
 
    IndexSpaceImp (const G& gshop, const Vector<Geometry>& geom,
                   int ngrow,
                   bool extend_domain_face, int num_coarsen_opt);
 
    IndexSpaceImp (IndexSpaceImp<G> const&) = delete;
    IndexSpaceImp (IndexSpaceImp<G> &&) = delete;
    void operator= (IndexSpaceImp<G> const&) = delete;
    void operator= (IndexSpaceImp<G> &&) = delete;
 
    ~IndexSpaceImp () override = default;
 
    [[nodiscard]] const Level& getLevel (const Geometry& geom) const final;
    [[nodiscard]] const Geometry& getGeometry (const Box& dom) const final;
    [[nodiscard]] const Box& coarsestDomain () const final {
        return m_geom.back().Domain();
    }
    void addFineLevels (int num_new_fine_levels) final;
    void addRegularCoarseLevels (int num_new_coarse_levels) final;
 
    void setShift (int direction, int ncells) override;
 
    using F = typename G::FunctionType;
 
private:
 
    G m_gshop;
    bool m_build_coarse_level_by_coarsening;
    bool m_extend_domain_face;
    int m_num_coarsen_opt;
 
    Vector<GShopLevel<G> > m_gslevel;
    Vector<Geometry> m_geom;
    Vector<Box> m_domain;
    Vector<int> m_ngrow;
};
 
#include <AMReX_EB2_IndexSpaceI.H>
 
bool ExtendDomainFace ();
int NumCoarsenOpt ();
 
template <typename G>
void
Build (const G& gshop, const Geometry& geom,
       int required_coarsening_level, int max_coarsening_level,
       int ngrow = 4, bool build_coarse_level_by_coarsening = true,
       bool extend_domain_face = ExtendDomainFace(),
       int num_coarsen_opt = NumCoarsenOpt())
{
    BL_PROFILE("EB2::Initialize()");
    IndexSpace::push(std::make_unique<IndexSpaceImp<G>>
                     (gshop, geom, required_coarsening_level, max_coarsening_level,
                      ngrow, build_coarse_level_by_coarsening,extend_domain_face,
                      num_coarsen_opt));
}
 
template <typename G>
void
Build (const G& gshop, Vector<Geometry> geom,
       int ngrow = 4,
       bool extend_domain_face = ExtendDomainFace(),
       int num_coarsen_opt = NumCoarsenOpt())
{
    BL_PROFILE("EB2::Initialize()");
    std::sort(geom.begin(), geom.end(), [] (Geometry const& a, Geometry const& b) { return a.Domain().numPts() > b.Domain().numPts(); });
    IndexSpace::push(std::make_unique<IndexSpaceImp<G>>
                     (gshop, geom, ngrow, extend_domain_face, num_coarsen_opt));
}
 
void Build (const Geometry& geom,
            int required_coarsening_level,
            int max_coarsening_level,
            int ngrow = 4,
            bool build_coarse_level_by_coarsening = true,
            bool extend_domain_face = ExtendDomainFace(),
            int num_coarsen_opt = NumCoarsenOpt(),
            bool support_mvmc = false);
 
void BuildMultiValuedMultiCut (const Geometry& geom,
                               int required_coarsening_level,
                               int max_coarsening_level,
                               int ngrow = 4,
                               bool build_coarse_level_by_coarsening = true,
                               bool extend_domain_face = ExtendDomainFace(),
                               int num_coarsen_opt = NumCoarsenOpt());
 
void BuildFromChkptFile (std::string const& fname,
                         const Geometry& geom,
                         int required_coarsening_level,
                         int max_coarsening_level,
                         int ngrow = 4,
                         bool build_coarse_level_by_coarsening = true,
                         bool extend_domain_face = ExtendDomainFace());
 
int maxCoarseningLevel (const Geometry& geom);
int maxCoarseningLevel (IndexSpace const* ebis, const Geometry& geom);
 
void addFineLevels (int num_new_fine_levels);
 
void addRegularCoarseLevels (int num_new_coarse_levels);
 
}
 
#endif