ArenaHeader.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#include "AthAllocators/ArenaHeader.h"
#include "AthAllocators/ArenaAllocatorRegistry.h"
#include "AthAllocators/ArenaAllocatorBase.h"
#include "AthAllocators/ArenaBase.h"
#include "CxxUtils/checker_macros.h"
#include <algorithm>
#include <ostream>
#include <sstream>
#include <print>
#include <cassert>
 
 
namespace SG {
 
 
void null_arena_deleter (ArenaBase*) {}
 

ArenaHeader::ArenaHeader()
    // m_arena doesn't own the objects to which it points.
    // Need to pass in a dummy deleter to prevent them from being deleted.
  : m_arena (null_arena_deleter)
{
}
 

ArenaHeader::~ArenaHeader()
{
  // We don't own this.
  m_arena.release();
}
 

ArenaBase* ArenaHeader::setArena (ArenaBase* arena)
{
  ArenaBase* ret = m_arena.get();
  m_arena.release();
  m_arena.reset (arena);
  return ret;
}
 

void ArenaHeader::addArena (ArenaBase* a)
{
  std::lock_guard<std::mutex> lock (m_mutex);
  m_arenas.push_back (a);
}
 

void ArenaHeader::setArenaForSlot (int slot, ArenaBase* a)
{
  if (slot < 0) return;
  std::lock_guard<std::mutex> lock (m_mutex);
  if (slot >= static_cast<int> (m_slots.size())) {
    m_slots.resize (slot+1);
  }
  m_slots[slot] = a;
}
 

void ArenaHeader::delArena (ArenaBase* a)
{
  std::lock_guard<std::mutex> lock (m_mutex);
  std::vector<ArenaBase*>::iterator it =
    std::find (m_arenas.begin(), m_arenas.end(), a);
  assert (it != m_arenas.end());
  m_arenas.erase (it);
}
 

void ArenaHeader::report (std::ostream& os) const
{
  std::lock_guard<std::mutex> lock (m_mutex);
  // All Allocators in the group.
  for (ArenaBase* arena : m_arenas) {
    std::println (os, "=== {} ===", arena->name());
    arena->report (os);
  }
 
  // The default Arena.
  std::println (os, "=== default ===");
  m_defaultArena.report (os);
}
 

std::string ArenaHeader::reportStr() const
{
  std::ostringstream s;
  report (s);
  return s.str();
}
 

void ArenaHeader::reset()
{
  if (m_arena.get()) {
    m_arena->reset();
  }
}
 

ArenaHeader* ArenaHeader::defaultHeader()
{
  static ArenaHeader head ATLAS_THREAD_SAFE;
  return &head;
}
 
 
} // namespace SG