synchronized_memory_resource.hpp

/* VecMem project, part of the ACTS project (R&D line)
 *
 * (c) 2023-2025 CERN for the benefit of the ACTS project
 *
 * Mozilla Public License Version 2.0
 */
#pragma once
 
// Local include(s).
#include "vecmem/memory/memory_resource.hpp"
#include "vecmem/vecmem_core_export.hpp"
 
// System include(s).
#include <functional>
#include <mutex>
 
namespace vecmem {
 
class synchronized_memory_resource final : public memory_resource {
 
public:
    VECMEM_CORE_EXPORT explicit synchronized_memory_resource(
        memory_resource& upstream);
    VECMEM_CORE_EXPORT
    synchronized_memory_resource(const synchronized_memory_resource& parent);
 
    VECMEM_CORE_EXPORT
    ~synchronized_memory_resource() override;
 
    VECMEM_CORE_EXPORT
    synchronized_memory_resource& operator=(
        const synchronized_memory_resource& rhs);
 
private:
 
    VECMEM_CORE_EXPORT
    void* do_allocate(std::size_t, std::size_t) override;
    VECMEM_CORE_EXPORT
    void do_deallocate(void* p, std::size_t, std::size_t) override;
    VECMEM_CORE_EXPORT
    bool do_is_equal(const memory_resource& other) const noexcept override;
 
 
    std::reference_wrapper<memory_resource> m_upstream;
    std::mutex m_mutex;
 
};  // class synchronized_memory_resource
 
}  // namespace vecmem