RootAuxDynStore.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "AthContainers/exceptions.h"
#include "AthContainersInterfaces/IAuxTypeVector.h"
 
#include "RootAuxDynStore.h"
#include "RootAuxDynIO/RootAuxDynIO.h"
 
 
RootAuxDynStore::RootAuxDynStore(RootAuxDynIO::IRootAuxDynReader& reader,
                                 long long entry, bool standalone, std::recursive_mutex* iomtx)
  : SG::AuxStoreInternal( standalone ),
    m_entry(entry),
    m_iomutex(iomtx)
{
   for( auto id : reader.auxIDs() ) {
      addAuxID(id);
   }
   lock();
}
 
      
const void* RootAuxDynStore::getData(SG::auxid_t auxid) const
{
  const SG::IAuxTypeVector* v = getVector( auxid );
  if( v ) {
     return v->toPtr();
  }
  return nullptr;
}
 
 
const SG::IAuxTypeVector* RootAuxDynStore::getVector(SG::auxid_t auxid) const
{
  guard_t guard (m_mutex);
  // lock
  const SG::IAuxTypeVector* ret = SG::AuxStoreInternal::getVector (auxid);
  if (!ret) {
    auto this_nc ATLAS_THREAD_SAFE = const_cast<RootAuxDynStore*>(this); // locked above
    if( this_nc->readData(auxid) ) {
       ret = SG::AuxStoreInternal::getVector (auxid);
    }
  }
  return ret;
}
 
 
void* RootAuxDynStore::getData(SG::auxid_t /*auxid*/, size_t /*size*/, size_t /*capacity*/)
{
  // MN: how do we add new attributes to this store?  A:for now we don't
  throw("Non-const access to RootAuxDynStore is not supported");
}
 
 
const void* RootAuxDynStore::getIOData(SG::auxid_t auxid) const
{
  guard_t guard (m_mutex);
  const void* ret = SG::AuxStoreInternal::getIODataInternal (auxid, true);
  if (!ret) {
    auto this_nc ATLAS_THREAD_SAFE = const_cast<RootAuxDynStore*>(this); // locked above
    if( this_nc->readData(auxid) ) {
       ret = SG::AuxStoreInternal::getIOData (auxid);
    }
  }
  return ret;
}
 
void*
RootAuxDynStore::getDecoration (SG::auxid_t auxid, size_t size, size_t capacity)
{
  guard_t guard (m_mutex);
  if (SG::AuxStoreInternal::getIODataInternal (auxid, true) == 0 &&
      SG::AuxStoreInternal::getAuxIDs().count (auxid) > 0)
  {
    throw SG::ExcStoreLocked (auxid);
  }
  return SG::AuxStoreInternal::getDecoration (auxid, size, capacity);
}
 
 
bool RootAuxDynStore::isDecoration (SG::auxid_t auxid) const
{
  guard_t guard( m_mutex );
  if (SG::AuxStoreInternal::getIODataInternal (auxid, true) == 0 &&
      SG::AuxStoreInternal::getAuxIDs().count (auxid) > 0)
  {
    return false;
  }
  return SG::AuxStoreInternal::isDecoration (auxid);
}
 
size_t RootAuxDynStore::size() const
{
  const std::size_t s = SG::AuxStoreInternal::size();
  if( s != 0 ) {
    return s;
  }
 
  for( SG::auxid_t id : getAuxIDs() ) {
    if( getData( id ) != nullptr ) {
      return SG::AuxStoreInternal::size();
    }
  }
 
  return 0;
}