AMReX.H

Go to the documentation of this file.

 
#ifndef BL_AMREX_H
#define BL_AMREX_H
#include <AMReX_Config.H>
 
#include <AMReX_GpuQualifiers.H>
#include <AMReX_GpuPrint.H>
#include <AMReX_GpuAssert.H>
#include <AMReX_ccse-mpi.H>
#include <AMReX_Exception.H>
#include <AMReX_Extension.H>
 
#include <cstdint>
#include <cstring>
#include <functional>
#include <iostream>
#include <memory>
#include <string>
#include <type_traits>
#include <vector>
 
//
// Initialize, Finalize, Error Reporting, and Version String Functions
 
/*
 This class consists of initialize, finalize, error-reporting, and version
 functions that are used throughout AMReX.  Note that all the output functions
 add a terminating exclamation mark, so there is no need to add any punctuation
 to sentences that get output explicitly.
*/
 
namespace amrex
{
    class AMReX;
 
    using PTR_TO_VOID_FUNC = void (*)();
    using ErrorHandler = void (*)(const char*);
 
    namespace system
    {
#ifndef AMREX_DEBUG
        constexpr bool NDebug = true;
        constexpr bool  Debug = false;
#else
        constexpr bool NDebug = false;
        constexpr bool  Debug = true;
#endif
 
        extern AMREX_EXPORT std::string exename;
 
        extern AMREX_EXPORT int verbose;
 
        extern AMREX_EXPORT bool signal_handling;
        extern AMREX_EXPORT bool handle_sigsegv;
        extern AMREX_EXPORT bool handle_sigterm;
        extern AMREX_EXPORT bool handle_sigint;
        extern AMREX_EXPORT bool handle_sigabrt;
        extern AMREX_EXPORT bool handle_sigfpe;
 
        extern AMREX_EXPORT bool call_addr2line;
        extern AMREX_EXPORT bool throw_exception;
 
        extern AMREX_EXPORT bool regtest_reduction;
 
        extern AMREX_EXPORT std::ostream* osout;
        extern AMREX_EXPORT std::ostream* oserr;
 
        extern AMREX_EXPORT ErrorHandler error_handler;
        extern AMREX_EXPORT bool abort_on_unused_inputs;
 
        extern AMREX_EXPORT bool init_snan;
    }
 
    [[nodiscard]] std::string Version ();
 
    // The returned AMReX* is non-owning! To delete it, call Finalize(AMReX*).
    AMReX* Initialize (MPI_Comm mpi_comm,
                       std::ostream& a_osout = std::cout,
                       std::ostream& a_oserr = std::cerr,
                       ErrorHandler a_errhandler = nullptr,
                       int a_device_id = -1);
 
    // The returned AMReX* is non-owning! To delete it, call Finalize(AMReX*).
    AMReX* Initialize (int& argc, char**& argv,
                       const std::function<void()>& func_parm_parse,
                       std::ostream& a_osout = std::cout,
                       std::ostream& a_oserr = std::cerr,
                       ErrorHandler a_errhandler = nullptr,
                       int a_device_id = -1);
 
    // The returned AMReX* is non-owning! To delete it, call Finalize(AMReX*).
    AMReX* Initialize (int& argc, char**& argv, bool build_parm_parse=true,
                       MPI_Comm mpi_comm = MPI_COMM_WORLD,
                       const std::function<void()>& func_parm_parse = {},
                       std::ostream& a_osout = std::cout,
                       std::ostream& a_oserr = std::cerr,
                       ErrorHandler a_errhandler = nullptr,
                       int a_device_id = -1);
 
    // \brief Minimal version of initialization.
    //
    // This version is intended for users who only need AMReX for some
    // specific functionalities such as FFT. It's the user's responsibility
    // to initialize MPI. For multiple-GPU systems, it's the user's
    // responsibility to properly set the GPU devices to be used. We will not
    // try to pre-allocate memory arenas. We will not install a signal
    // handler. Functionalities like random number generator and async I/O
    // will be work. However, functionalities like FFT and linear solvers do
    // work.
    void Init_minimal (MPI_Comm mpi_comm = MPI_COMM_WORLD);
 
    [[nodiscard]] bool Initialized ();
 
    void Finalize (AMReX* pamrex);
    void Finalize (); // Finalize the current top
    // For initialization with Init_minimal, Finalize_minimal should be used
    // for finalization.
    void Finalize_minimal();
 
    void ExecOnFinalize (std::function<void()>);
    void ExecOnInitialize (std::function<void()>);
 
    template <class... Ts>
    AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
    void ignore_unused (const Ts&...) {}
 
    void Error (const std::string& msg);
 
    void Error_host (const char* type, const char* msg);
 
    AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
    void Error (const char* msg = nullptr) {
#if defined(NDEBUG)
        AMREX_IF_ON_DEVICE((amrex::ignore_unused(msg);))
#else
        AMREX_IF_ON_DEVICE((
                if (msg) { AMREX_DEVICE_PRINTF("Error %s\n", msg); }
                AMREX_DEVICE_ASSERT(0);
        ))
#endif
        AMREX_IF_ON_HOST((Error_host("Error", msg);))
    }
 
    void Warning (const std::string& msg);
 
    void Warning_host (const char * msg);
 
    AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
    void Warning (const char * msg) {
#if defined(NDEBUG)
        AMREX_IF_ON_DEVICE((amrex::ignore_unused(msg);))
#else
        AMREX_IF_ON_DEVICE((if (msg) { AMREX_DEVICE_PRINTF("Warning %s\n", msg); }))
#endif
        AMREX_IF_ON_HOST((Warning_host(msg);))
    }
 
    void Abort (const std::string& msg);
 
    AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
    void Abort (const char * msg = nullptr) {
#if defined(NDEBUG)
        AMREX_IF_ON_DEVICE((amrex::ignore_unused(msg);))
#else
        AMREX_IF_ON_DEVICE((
                if (msg) { AMREX_DEVICE_PRINTF("Abort %s\n", msg); }
                AMREX_DEVICE_ASSERT(0);
        ))
#endif
        AMREX_IF_ON_HOST((Error_host("Abort", msg);))
    }
 
    void Assert_host (const char* EX, const char* file, int line, const char* msg,
                      std::size_t msg_size=0);
 
    AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
    void Assert (const char* EX, const char* file, int line) {
#if defined(NDEBUG) && !defined(AMREX_USE_ASSERTION)
        AMREX_IF_ON_DEVICE((amrex::ignore_unused(EX,file,line);))
#else
        AMREX_IF_ON_DEVICE((
            AMREX_DEVICE_PRINTF("Assertion `%s' failed, file \"%s\", line %d",
                                    EX, file, line);
            AMREX_DEVICE_ASSERT(0);
        ))
#endif
        AMREX_IF_ON_HOST((Assert_host(EX,file,line,nullptr,0U);))
    }
 
    AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
    void Assert (const char* EX, const char* file, int line, const char* msg) {
#if defined(NDEBUG) && !defined(AMREX_USE_ASSERTION)
        AMREX_IF_ON_DEVICE((amrex::ignore_unused(EX,file,line,msg);))
#else
        AMREX_IF_ON_DEVICE((
            AMREX_DEVICE_PRINTF("Assertion `%s' failed, file \"%s\", line %d, Msg: %s",
                                    EX, file, line, msg);
            AMREX_DEVICE_ASSERT(0);
        ))
#endif
        AMREX_IF_ON_HOST((Assert_host(EX,file,line,msg,std::strlen(msg));))
    }
 
    AMREX_FORCE_INLINE
    void Assert (const char* EX, const char* file, int line, const std::string& msg) {
        Assert_host(EX,file,line,msg.c_str(),msg.size());
    }
 
    void write_to_stderr_without_buffering (const char* str);
 
    void SetErrorHandler (ErrorHandler f);
 
    std::ostream& OutStream ();
    std::ostream& ErrorStream ();
 
    [[nodiscard]] int Verbose () noexcept;
    void SetVerbose (int v) noexcept;
 
    [[nodiscard]] bool InitSNaN () noexcept;
    void SetInitSNaN (bool v) noexcept;
 
    // ! Get the entire command line including the executable
    [[nodiscard]] std::string get_command ();
 
    // ! Get number of command line arguments after the executable
    [[nodiscard]] int command_argument_count ();
 
    [[nodiscard]] std::string get_command_argument (int number);
 
#ifndef _MSC_VER
    inline void GccPlacater ()
    {
        std::allocator<bool> a_b;
        std::allocator<char> a_c;
        std::allocator<int> a_i;
        std::allocator<long> a_l;
        std::allocator<long long> a_ll;
        std::allocator<unsigned char> a_uc;
        std::allocator<unsigned int> a_ui;
        std::allocator<unsigned long> a_ul;
        std::allocator<unsigned long long> a_ull;
        std::allocator<float> a_f;
        std::allocator<double> a_d;
        std::allocator<std::string> a_s;
 
        amrex::ignore_unused(a_b);
        amrex::ignore_unused(a_c);
        amrex::ignore_unused(a_i);
        amrex::ignore_unused(a_l);
        amrex::ignore_unused(a_ll);
        amrex::ignore_unused(a_uc);
        amrex::ignore_unused(a_ui);
        amrex::ignore_unused(a_ul);
        amrex::ignore_unused(a_ull);
        amrex::ignore_unused(a_ll);
        amrex::ignore_unused(a_f);
        amrex::ignore_unused(a_d);
        amrex::ignore_unused(a_s);
    }
#endif
 
    class Geometry;
 
    class AMReX
    {
    public:
        AMReX ();
        ~AMReX ();
        AMReX (AMReX const&) = delete;
        AMReX (AMReX &&) = delete;
        AMReX& operator= (AMReX const&) = delete;
        AMReX& operator= (AMReX &&) = delete;
 
        static bool empty () noexcept { return m_instance.empty(); }
 
        static int size () noexcept { return static_cast<int>(m_instance.size()); }
 
        static AMReX* top () noexcept { return m_instance.back().get(); }
 
        // Thisfunction will take the ownership of the AMReX 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 (AMReX* pamrex);
 
        // This erases `pamrex` from the stack.
        static void erase (AMReX* pamrex);
 
        [[nodiscard]] Geometry* getDefaultGeometry () noexcept { return m_geom; }
 
    private:
 
        static AMREX_EXPORT std::vector<std::unique_ptr<AMReX> > m_instance;
 
        Geometry* m_geom = nullptr;
    };
 
    enum struct FPExcept : std::uint8_t {
        none     = 0B0000,
        invalid  = 0B0001,
        zero     = 0B0010,
        overflow = 0B0100,
        all      = 0B0111
    };
 
    [[nodiscard]] inline bool any (FPExcept a) { return a != FPExcept::none; }
 
    [[nodiscard]] inline FPExcept operator| (FPExcept a, FPExcept b)
    {
        using T = std::underlying_type_t<FPExcept>;
        return static_cast<FPExcept>(static_cast<T>(a) | static_cast<T>(b));
    }
 
    [[nodiscard]] inline FPExcept operator& (FPExcept a, FPExcept b)
    {
        using T = std::underlying_type_t<FPExcept>;
        return static_cast<FPExcept>(static_cast<T>(a) & static_cast<T>(b));
    }
 
    [[nodiscard]] FPExcept getFPExcept ();
 
    FPExcept setFPExcept (FPExcept excepts);
 
    [[nodiscard]] FPExcept disableFPExcept (FPExcept excepts);
 
    [[nodiscard]] FPExcept enableFPExcept (FPExcept excepts);
}
 
#endif /*BL_AMREX_H*/