MonitoredMemoryResource.h

Go to the documentation of this file.

//
// Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHDEVICECOMPS_MONITOREDMEMORYRESOURCE_H
#define ATHDEVICECOMPS_MONITOREDMEMORYRESOURCE_H
 
// AthDevice include(s).
#include "AthDeviceInterfaces/IMemoryMonitorSvc.h"
 
// System include(s).
#include <functional>
#include <memory_resource>
 
namespace AthDevice {

class MonitoredMemoryResource : public std::pmr::memory_resource {
 
 public:
  MonitoredMemoryResource(std::pmr::memory_resource& upstream,
                          IMemoryMonitorSvc& svc, const std::string& component,
                          const std::string& device);
  virtual ~MonitoredMemoryResource();
 
 private:

  virtual void* do_allocate(std::size_t bytes, std::size_t alignment) override;

  virtual void do_deallocate(void* p, std::size_t bytes,
                             std::size_t alignment) override;

  virtual bool do_is_equal(
      const std::pmr::memory_resource& other) const noexcept override;



  std::reference_wrapper<std::pmr::memory_resource> m_upstream;
  std::reference_wrapper<IMemoryMonitorSvc> m_svc;
  std::string m_component;
  std::string m_device;

 
};  // class MonitoredMemoryResource
 
}  // namespace AthDevice
 
#endif  // ATHDEVICECOMPS_MONITOREDMEMORYRESOURCE_H