ByteStreamAuxContainer_v1.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
// System include(s):
#include <iostream>
#include <atomic>
 
// EDM include(s):
#include "xAODCore/tools/AuxPersVector.h"
#include "AthContainers/AuxTypeRegistry.h"
#include "AthContainers/exceptions.h"
 
// Local include(s):
#include "xAODTrigger/versions/ByteStreamAuxContainer_v1.h"
 
namespace xAOD {
 
   ByteStreamAuxContainer_v1::ByteStreamAuxContainer_v1()
      : SG::IAuxStore(),
        m_int(), m_float(), m_vecInt(), m_vecFloat(),
        m_staticVecs(), m_dynamicVecs(),
        m_locked( false ),
        m_name( "UNKNOWN" ) {
 
   }
 
   ByteStreamAuxContainer_v1::
   ByteStreamAuxContainer_v1( const ByteStreamAuxContainer_v1& parent )
      : SG::IAuxStore( parent ),
        m_int( parent.m_int ), m_float( parent.m_float ),
        m_vecInt( parent.m_vecInt ), m_vecFloat( parent.m_vecFloat ),
        // The auxiliary IDs are copied:
        m_auxids( parent.m_auxids ),
        // But the internal pointers are not:
        m_staticVecs(),
        m_dynamicVecs(), m_locked( false ),
        m_name( parent.m_name ) {
 
   }
 
   ByteStreamAuxContainer_v1::~ByteStreamAuxContainer_v1() {
 
      // Clean up the helper objects:
      std::vector< SG::IAuxTypeVector* >::iterator itr = m_dynamicVecs.begin();
      std::vector< SG::IAuxTypeVector* >::iterator end = m_dynamicVecs.end();
      for( ; itr != end; ++itr ) {
         if( *itr ) delete *itr;
      }
      itr = m_staticVecs.begin();
      end = m_staticVecs.end();
      for( ; itr != end; ++itr ) {
         if( *itr ) delete *itr;
      }
   }
 
   ByteStreamAuxContainer_v1&
   ByteStreamAuxContainer_v1::
   operator=( const ByteStreamAuxContainer_v1& rhs ) {
 
      if (this != &rhs) {
        // Copy the persistent variables:
        m_int      = rhs.m_int;
        m_float    = rhs.m_float;
        m_vecInt   = rhs.m_vecInt;
        m_vecFloat = rhs.m_vecFloat;
 
        // Also copy the list of auxiliary IDs handled:
        m_auxids        = rhs.m_auxids;
 
        // The static variables should be left alone. Those should still
        // point to the correct places in memory. But the dynamic variables
        // need to be cleared out. Those will be re-created when/if needed.
        for( auto* ptr : m_dynamicVecs ) {
           delete ptr;
        }
        m_dynamicVecs.clear();
 
        // Copy the container's name:
        m_name = rhs.m_name;
      }
 
      return *this;
   }
 
   const void* ByteStreamAuxContainer_v1::getData( auxid_t auxid ) const {
 
      const SG::IAuxTypeVector* v = getVector( auxid );
      if( v ) {
         return v->toPtr();
      }
      return nullptr;
   }
 
   const SG::IAuxTypeVector* ByteStreamAuxContainer_v1::getVector( auxid_t auxid ) const {
 
      guard_t guard (m_mutex);
 
      // If it's a static auxiliary property:
      if( ( auxid < m_staticVecs.size() ) && ( m_staticVecs[ auxid ] ) ) {
         // We're done already:
         return m_staticVecs[ auxid ];
      }
      // If it's a dynamic one:
      else if( ( auxid < m_dynamicVecs.size() ) &&
               ( m_dynamicVecs[ auxid ] ) ) {
         // We're done already:
         return m_dynamicVecs[ auxid ];
      }
 
      // Try to retrieve the dynamic variable:
      // quiet on because this method may be called from isAccessible
      // and that shouldn't generate any output.
      return getVector1 (auxid, 0, 0, true, false);
   }
 
   const ByteStreamAuxContainer_v1::auxid_set_t&
   ByteStreamAuxContainer_v1::getAuxIDs() const {
 
      // Return the full list of IDs:
      return getWritableAuxIDs();
   }
 
   const ByteStreamAuxContainer_v1::auxid_set_t&
   ByteStreamAuxContainer_v1::getDecorIDs() const {
 
      // Return the full list of IDs:
      return m_decorations;
   }

   void* ByteStreamAuxContainer_v1::getDecoration (auxid_t auxid,
                                                   size_t size,
                                                   size_t capacity)
   {
      guard_t guard (m_mutex);
 
      // Does this variable already exist?
      void* ret = nullptr;
      if( ( auxid < m_staticVecs.size() ) && ( m_staticVecs[ auxid ] ) ) {
        ret = m_staticVecs[ auxid ]->toPtr();
      }
      // If it's a dynamic one:
      else if( ( auxid < m_dynamicVecs.size() ) &&
               ( m_dynamicVecs[ auxid ] ) ) {
        ret = m_dynamicVecs[ auxid ]->toPtr();
      }
 
      if (!ret) {
        SG::IAuxTypeVector* v = getVector1 (auxid, 0, 0, true, true);
        if( v ) {
           ret = v->toPtr();
        }
      }
 
      if (ret) {
        // Raise exception if locked and not a decoration.
        if (m_locked) {
          if ( ! m_decorations.test (auxid) ) {
            throw SG::ExcStoreLocked (auxid);
          }
        }
        return ret;
      }
 
      // Make a new variable.
      // If locked, mark as a decoration.
      if (m_locked) {
        m_decorations.insert (auxid);
        std::atomic_thread_fence (std::memory_order_seq_cst);
      }
      SG::IAuxTypeVector* v = getVector1 (auxid, size, capacity, false, true);
      if( v ) {
         ret = v->toPtr();
      }
 
      return ret;
   }
 

   bool ByteStreamAuxContainer_v1::isDecoration (auxid_t auxid) const
   {
      guard_t guard (m_mutex);
      return m_locked && m_decorations.test (auxid);
   }
 
 
   void ByteStreamAuxContainer_v1::lock()
   {
       guard_t guard (m_mutex);
       m_locked = true;
   }
 
   bool ByteStreamAuxContainer_v1::clearDecorations()
   {
     guard_t guard (m_mutex);
 
     const SG::AuxTypeRegistry& r = SG::AuxTypeRegistry::instance();
 
     bool anycleared = false;
     for (auxid_t auxid : m_decorations) {
       if (m_dynamicVecs[auxid]) {
         delete m_dynamicVecs[auxid];
         m_dynamicVecs[auxid] = nullptr;
         m_auxids.erase( auxid );
         anycleared = true;
 
         const std::string name = r.getName( auxid );
         const std::type_info* ti = r.getType( auxid );
         if (ti == &typeid(int))
           m_int.erase (name);
         else if (ti == &typeid(float))
           m_float.erase (name);
         else if (ti == &typeid(std::vector<int>))
           m_vecInt.erase (name);
         else if (ti == &typeid(std::vector<float>))
           m_vecFloat.erase (name);
       }
     }
     m_decorations.clear();
 
     return anycleared;
   }
 

   void ByteStreamAuxContainer_v1::lockDecoration (SG::auxid_t auxid)
   { 
     guard_t guard (m_mutex);
     m_decorations.reset (auxid);
   }
 
 
   size_t ByteStreamAuxContainer_v1::size_noLock() const
   {
     for (SG::auxid_t i : m_auxids) {
       if (i < m_staticVecs.size() && m_staticVecs[i]) {
         size_t sz = m_staticVecs[i]->size();
         if (sz > 0)
           return sz;
       }
       if (i < m_dynamicVecs.size() && m_dynamicVecs[i]) {
         size_t sz = m_dynamicVecs[i]->size();
         if (sz > 0)
           return sz;
       }
     }
 
     return 0;
   }
 
   size_t ByteStreamAuxContainer_v1::size() const
   {
     guard_t guard (m_mutex);
     return size_noLock();
   }
 
 
   void* ByteStreamAuxContainer_v1::getData( auxid_t auxid, size_t size,
                                             size_t capacity ) {
 
      guard_t guard (m_mutex);
 
      // Check if we have such a static variable:
      if( ( auxid < m_staticVecs.size() ) && ( m_staticVecs[ auxid ] ) ) {
         // Set it to the right size:
         m_staticVecs[ auxid ]->reserve( capacity );
         m_staticVecs[ auxid ]->resize( size );
         // We're done already:
         return m_staticVecs[ auxid ]->toPtr();
      }
      // Check if we already know about this dynamic variable:
      else if( ( auxid < m_dynamicVecs.size() ) &&
               ( m_dynamicVecs[ auxid ] ) ) {
         // Set it to the right size:
         m_dynamicVecs[ auxid ]->reserve( capacity );
         m_dynamicVecs[ auxid ]->resize( size );
         // We're done already:
         return m_dynamicVecs[ auxid ]->toPtr();
      }
 
      SG::IAuxTypeVector* v = getVector1 (auxid, size, capacity, false, false);
      if( v ) {
         return v->toPtr();
      }
      return nullptr;
   }
 
   const ByteStreamAuxContainer_v1::auxid_set_t&
   ByteStreamAuxContainer_v1::getWritableAuxIDs() const {
 
      return m_auxids;
   }
 
   bool ByteStreamAuxContainer_v1::resize( size_t size ) {
 
      guard_t guard (m_mutex);
 
      if (m_locked)
        throw SG::ExcStoreLocked ("resize");
 
      bool nomoves = true;
      for (SG::IAuxTypeVector* v : m_dynamicVecs) {
         if(v) {
           if (!v->resize( size ))
             nomoves = false;
         }
      }
 
      for (SG::IAuxTypeVector* v : m_staticVecs) {
         if(v) {
           if (!v->resize( size ))
             nomoves = false;
         }
      }
 
      return nomoves;
   }
 
   void ByteStreamAuxContainer_v1::reserve( size_t size ) {
 
      guard_t guard (m_mutex);
 
      if (m_locked)
        throw SG::ExcStoreLocked ("reserve");
 
      std::vector< SG::IAuxTypeVector* >::iterator itr = m_dynamicVecs.begin();
      std::vector< SG::IAuxTypeVector* >::iterator end = m_dynamicVecs.end();
      for( ; itr != end; ++itr ) {
         if( *itr ) ( *itr )->reserve( size );
      }
      itr = m_staticVecs.begin();
      end = m_staticVecs.end();
      for( ; itr != end; ++itr ) {
         if( *itr ) ( *itr )->reserve( size );
      }
 
      return;
   }
 
   void ByteStreamAuxContainer_v1::shift( size_t pos, ptrdiff_t offs ) {
 
      guard_t guard (m_mutex);
 
      if (m_locked)
        throw SG::ExcStoreLocked ("shift");
 
      std::vector< SG::IAuxTypeVector* >::iterator itr = m_dynamicVecs.begin();
      std::vector< SG::IAuxTypeVector* >::iterator end = m_dynamicVecs.end();
      for( ; itr != end; ++itr ) {
         if( *itr ) ( *itr )->shift( pos, offs );
      }
      itr = m_staticVecs.begin();
      end = m_staticVecs.end();
      for( ; itr != end; ++itr ) {
         if( *itr ) ( *itr )->shift( pos, offs );
      }
 
      return;
   }
 
   bool ByteStreamAuxContainer_v1::insertMove( size_t pos,
                                               IAuxStore& other,
                                               const SG::auxid_set_t& ignore_in) {
 
      guard_t guard( m_mutex );
 
      // This operation is not allowed on a locked container:
      if( m_locked ) {
         throw SG::ExcStoreLocked( "insertMove" );
      }
 
      bool nomove = true;
      size_t other_size = other.size();
 
      SG::auxid_set_t ignore = ignore_in;
 
      // Do the operation on the static variables:
      for (SG::auxid_t id : m_auxids) {
        SG::IAuxTypeVector* v_dst = nullptr;
        if (id < m_dynamicVecs.size())
          v_dst = m_dynamicVecs[id];
        if (!v_dst && id < m_staticVecs.size())
          v_dst = m_staticVecs[id];
        if (v_dst) {
          ignore.insert (id);
          if (other.getData (id)) {
            void* src_ptr = other.getData (id, other_size, other_size);
            if (src_ptr) {
              if (!v_dst->insertMove (pos, src_ptr, 0, other_size,
                                      other))
                nomove = false;
            }
          }
          else {
            const void* orig = v_dst->toPtr();
            v_dst->shift (pos, other_size);
            if (orig != v_dst->toPtr())
              nomove = false;
          }
        }
      }
 
      // Add any new variables not present in the original container.
      for (SG::auxid_t id : other.getAuxIDs()) {
        if (!m_auxids.test(id) &&
            !ignore.test(id))
        {
          if (other.getData (id)) {
            void* src_ptr = other.getData (id, other_size, other_size);
            if (src_ptr) {
              size_t sz = size_noLock();
              getVector1 (id, sz, sz, true, false);
              m_dynamicVecs[id]->resize (sz - other_size);
              (void)m_dynamicVecs[id]->insertMove (pos, src_ptr, 0, other_size,
                                                   other);
              nomove = false;
            }
          }
        }
      }
 
      return nomove;
   }
 
 
   void ByteStreamAuxContainer_v1::reset() {
 
      guard_t guard (m_mutex);
 
      for (SG::IAuxTypeVector* p : m_dynamicVecs)
         delete p;
      m_dynamicVecs.clear();
 
      SG::AuxTypeRegistry& r = SG::AuxTypeRegistry::instance();
#define ADD_IDS(VAR, TYP) \
      do { typedef std::map< std::string, std::vector< TYP > > CONT; \
          for (CONT::value_type& p : VAR)                                 \
             m_auxids.insert (r.getAuxID< TYP > (p.first, "", SG::AuxVarFlags::SkipNameCheck)); } while(0)
      ADD_IDS(m_int, int);
      ADD_IDS(m_float, float);
      ADD_IDS(m_vecInt, std::vector<int>);
      ADD_IDS(m_vecFloat, std::vector<float>);
#undef ADD_IDS
 
      return;
   }
 
   const char* ByteStreamAuxContainer_v1::name() const {
 
      return m_name.c_str();
   }
 
   void ByteStreamAuxContainer_v1::setName( const char* name ) {
 
      m_name = name;
      return;
   }

   template< typename T >
   SG::IAuxTypeVector*
   ByteStreamAuxContainer_v1::getVector1( auxid_t auxid,
                                          std::map< std::string,
                                                    std::vector< T > >& pers,
                                          size_t size, size_t capacity,
                                          bool quiet,
                                          bool forDecor) const {
 
      // Private method --- should hold lock before calling this.
 
      // Look up the name of the variable:
      const SG::AuxTypeRegistry& r = SG::AuxTypeRegistry::instance();
      const std::string name = r.getName( auxid );
 
      // Try to find the variable:
      typename std::map< std::string, std::vector< T > >::iterator itr =
         pers.find( name );
      if( itr == pers.end() ) {
         // Variable isn't there.  Fail if capacity==0.
         if (capacity == 0) {
           if (!quiet) {
             std::cerr << "ERROR xAOD::ByteStreamAuxContainer_v1::getData (const) "
                       << "Variable with unknown name (" << name << ") requested"
                       << std::endl;
           }
           return nullptr;
         }
 
         if (m_locked && !forDecor)
           throw SG::ExcStoreLocked ("getData");
 
         if (r.isLinked(auxid)) {
             std::cerr << "ERROR xAOD::ByteStreamAuxContainer_v1 doesn't implement linked variables"
                       << std::endl;
           return nullptr;
         }
 
         // Create the variable.
         itr = pers.insert (std::make_pair (name, std::vector<T>())).first;
      }
 
      // Get hold of the variable.
      std::vector< T >& var = itr->second;
 
      // Make sure that the internal vector is big enough:
      if( m_dynamicVecs.size() <= auxid ) {
         m_dynamicVecs.resize( auxid + 1 );
      }
 
      // Just an internal check...
      if( m_dynamicVecs[ auxid ] ) {
         std::cerr << "ERROR xAOD::ByteStreamAuxContainer_v1::getData "
                   << "Internal inconsistency detected!" << std::endl;
         return m_dynamicVecs[ auxid ];
      }
 
      // Register the variable:
      m_dynamicVecs[ auxid ] = new AuxPersVector< T >( auxid, var, false, nullptr );
 
      if (capacity > 0) {
        // Set it to the right size:
        m_dynamicVecs[ auxid ]->reserve( capacity );
        m_dynamicVecs[ auxid ]->resize( size );
      }
 
      // Remember that we are now handling this variable:
      m_auxids.insert( auxid );
 
      // Return the pointer to the array:
      return m_dynamicVecs[ auxid ];
   }
 
   SG::IAuxTypeVector*
   ByteStreamAuxContainer_v1::getVector1( auxid_t auxid,
                                          size_t size, size_t capacity,
                                          bool quiet,
                                          bool forDecor) const {
 
      // Private method --- should hold lock before calling this.
 
      if( *( SG::AuxTypeRegistry::instance().getType( auxid ) ) ==
          typeid( int ) ) {
 
         return getVector1( auxid, m_int, size, capacity, quiet, forDecor );
      }
      else if( *( SG::AuxTypeRegistry::instance().getType( auxid ) ) ==
               typeid( float ) ) {
 
         return getVector1( auxid, m_float, size, capacity, quiet, forDecor );
      }
      else if( *( SG::AuxTypeRegistry::instance().getType( auxid ) ) ==
               typeid( std::vector< int > ) ) {
 
         return getVector1( auxid, m_vecInt, size, capacity, quiet, forDecor );
      }
      else if( *( SG::AuxTypeRegistry::instance().getType( auxid ) ) ==
               typeid( std::vector< float > ) ) {
 
         return getVector1( auxid, m_vecFloat, size, capacity, quiet, forDecor );
      }
 
      // The object can't handle this variable type...
      std::cerr << "ERROR xAOD::ByteStreamAuxContainer_v1::getData "
                << "Unknown variable type ("
                << SG::AuxTypeRegistry::instance().getType( auxid )->name()
                << ") requested" << std::endl;
 
      return nullptr;
   }
 
} // namespace xAOD