dynamicdata.H

Go to the documentation of this file.

/* Copyright 2022-2026 The Regents of the University of California, through Lawrence
 *           Berkeley National Laboratory (subject to receipt of any required
 *           approvals from the U.S. Dept. of Energy). All rights reserved.
 *
 * This file is part of ImpactX.
 *
 * Authors: Axel Huebl
 * License: BSD-3-Clause-LBNL
 */
#ifndef IMPACTX_ELEMENTS_MIXIN_DYNAMICDATA_H
#define IMPACTX_ELEMENTS_MIXIN_DYNAMICDATA_H
 
#include "impactx_export.H"
 
#include <AMReX_Gpu.H>
 
#include <map>
#include <memory>
#include <stdexcept>
#include <string>
#include <type_traits>
#include <vector>
 
 
namespace impactx::elements::mixin
{

    template<typename T>
    struct GPUDataRegistry
    {
        using DataType = T;  
 
        IMPACTX_EXPORT static int& next_id () {
            static int id = 0;
            return id;
        }
 
        IMPACTX_EXPORT static std::map<int, std::shared_ptr<T>>& registry () {
            static std::map<int, std::shared_ptr<T>> reg;
            return reg;
        }

        static int allocate_id () { return next_id()++; }

        static std::shared_ptr<T> const &
        get (int id)
        {
            auto& reg = registry();
            auto it = reg.find(id);
            if (it == reg.end() || !it->second)
                throw std::runtime_error(
                    "GPUDataRegistry::get failed for id=" + std::to_string(id));
            return it->second;
        }

        static void store (int id, std::shared_ptr<T> data)
        {
            registry()[id] = std::move(data);
        }

        template<typename... Args>
        static T& emplace (int id, Args&&... args)
        {
            auto ptr = std::shared_ptr<T>(new T{std::forward<Args>(args)...});  // NOLINT(modernize-make-shared)
            registry()[id] = ptr;
            return *ptr;
        }

        static void clear ()
        {
            registry().clear();
            next_id() = 0;
        }
    };
 
} // namespace impactx::elements::mixin
 

#define IMPACTX_GPUDATA_EXTERN(ElementType) \
    extern template struct impactx::elements::mixin::GPUDataRegistry< \
        ElementType::DynamicData::DataType>;

#define IMPACTX_GPUDATA_INSTANTIATE(ElementType) \
    template struct impactx::elements::mixin::GPUDataRegistry< \
        ElementType::DynamicData::DataType>;
 
#endif // IMPACTX_ELEMENTS_MIXIN_DYNAMICDATA_H