ArenaBase.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#include "AthAllocators/ArenaBase.h"
#include "AthAllocators/ArenaAllocatorBase.h"
#include "AthAllocators/ArenaAllocatorRegistry.h"
#include <ostream>
 
 
namespace SG {
 
 
ArenaBase::ArenaBase (const std::string& name /*= ""*/)
  : m_name (name)
{
}
 
 
ArenaBase::~ArenaBase()
{
}
 
 
void ArenaBase::reset()
{
  lock_t l (m_mutex);
  for (AllocEntry& alloc : m_allocs) {
    if (alloc.m_alloc) {
      lock_t alloc_lock (*alloc.m_mutex);
      alloc.m_alloc->reset();
    }
  }
}
 
 
void ArenaBase::erase()
{
  lock_t l (m_mutex);
  for (AllocEntry& alloc : m_allocs) {
    if (alloc.m_alloc) {
      lock_t alloc_lock (*alloc.m_mutex);
      alloc.m_alloc->erase();
    }
  }
}
 
 
void ArenaBase::report (std::ostream& os) const
{
  lock_t l (m_mutex);
  bool first = true;
  for (const AllocEntry& alloc : m_allocs) {
    if (alloc.m_alloc) {
      if (first) {
        ArenaAllocatorBase::Stats::header (os);
        os << std::endl;
        first = false;
      }
      lock_t alloc_lock (*alloc.m_mutex);
      alloc.m_alloc->report (os);
    }
  }
}
 
 
ArenaAllocatorBase::Stats ArenaBase::stats () const
{
  ArenaAllocatorBase::Stats stats;
  lock_t l (m_mutex);
  for (const AllocEntry& alloc : m_allocs) {
    if (alloc.m_alloc) {
      lock_t alloc_lock (*alloc.m_mutex);
      stats += alloc.m_alloc->stats();
    }
  }
  return stats;
}
 
 
const std::string& ArenaBase::name() const
{
  return m_name;
}
 
 
ArenaAllocatorBase* ArenaBase::makeAllocator (size_t i)
{
  // We have to create a new Allocator.
  // Make sure there's room in the vector.
  if (m_allocs.size() <= i) {
    m_allocs.resize (i+1);
  }
 
  // Create the Allocator, using the Registry.
  std::unique_ptr<ArenaAllocatorBase> alloc =
    ArenaAllocatorRegistry::instance()->create (i);
 
  // Install it in the vector.
  m_allocs[i].m_alloc = std::move (alloc);
  m_allocs[i].m_mutex = std::make_unique<std::mutex>();
 
  return m_allocs[i].m_alloc.get();
}
 
 
} // namespace SG