MemoryMonitorSvc.h

Go to the documentation of this file.

//
// Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHDEVICECOMPS_MEMORYMONITORSVC_H
#define ATHDEVICECOMPS_MEMORYMONITORSVC_H
 
// Framework include(s).
#include "AthenaBaseComps/AthService.h"
 
// AthDevice include(s).
#include "AthDeviceInterfaces/IMemoryMonitorSvc.h"
 
// System include(s).
#include <map>
#include <mutex>
 
namespace AthDevice {

class MemoryMonitorSvc : public extends<AthService, IMemoryMonitorSvc> {
 
 public:
  // Inherit the base class's constructor(s).
  using extends::extends;


  virtual StatusCode finalize() override;




  virtual void memoryAllocated(const std::string& component,
                               const std::string& device,
                               std::size_t bytes) override;

  virtual void memoryDeallocated(const std::string& component,
                                 const std::string& device,
                                 std::size_t bytes) override;

  virtual void memoryCopied(const std::string& component,
                            vecmem::copy::type::copy_type type,
                            std::size_t bytes) override;

 
 private:

  struct MemoryAllocKey {
    std::string component;
    std::string device;

    bool operator<(const MemoryAllocKey& other) const {
      if (device != other.device) {
        return device < other.device;
      } else {
        return component < other.component;
      }
    }
  };

  std::map<MemoryAllocKey, std::size_t> m_allocations;
  std::mutex m_allocationsMutex;
  std::map<MemoryAllocKey, std::size_t> m_deallocations;
  std::mutex m_deallocationsMutex;

  struct MemoryCopyKey {
    std::string component;
    vecmem::copy::type::copy_type type;

    bool operator<(const MemoryCopyKey& other) const {
      if (component != other.component) {
        return component < other.component;
      } else {
        return type < other.type;
      }
    }
  };

  std::map<MemoryCopyKey, std::size_t> m_copies;
  std::mutex m_copiesMutex;

 
};  // class MemoryMonitorSvc
 
}  // namespace AthDevice
 
#endif  // ATHDEVICECOMPS_MEMORYMONITORSVC_H