CopyEventStreamInfo.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/

 
#include "CopyEventStreamInfo.h"
 
#include "EventInfo/EventStreamInfo.h"
#include "StoreGate/StoreGateSvc.h"
#include <algorithm>
 
//___________________________________________________________________________
CopyEventStreamInfo::CopyEventStreamInfo(const std::string& type,
                                         const std::string& name,
                                         const IInterface* parent) :
  base_class(type, name, parent) {
}
//___________________________________________________________________________
StatusCode CopyEventStreamInfo::initialize() {
   ATH_MSG_DEBUG("Initializing " << name());
   // Locate the MetaDataSvc and InputMetaDataStore
   ATH_CHECK( m_metaDataSvc.retrieve() );
   ATH_CHECK( m_inputMetaDataStore.retrieve() );
 
   return StatusCode::SUCCESS;
}
 
 
StatusCode CopyEventStreamInfo::beginInputFile(const SG::SourceID&)
{
   std::vector<std::string> keys = m_keys;
   if (keys.empty()) {
      m_inputMetaDataStore->keys<EventStreamInfo>(keys);
   } else {
     // remove keys not in the InputMetaDataStore
     std::erase_if(keys, [this](const std::string& key) {
               return !m_inputMetaDataStore->contains<EventStreamInfo>(key);
             });
   }
 
   // If the input file doesn't have any event stream info metadata,
   // then finish right away:
   if (keys.empty()) return StatusCode::SUCCESS;
 
   for (const auto &key : keys) {
      const EventStreamInfo* evtStrInfo_in = nullptr;
      ATH_CHECK( m_inputMetaDataStore->retrieve(evtStrInfo_in, key) );
 
      EventStreamInfo* evtStrInfo_out = m_metaDataSvc->tryRetrieve<EventStreamInfo>(key);
      if( !evtStrInfo_out ) {
         auto esinfo_up = std::make_unique<EventStreamInfo>(*evtStrInfo_in);
         ATH_CHECK( m_metaDataSvc->record( std::move(esinfo_up), key ) );
      } else {
         evtStrInfo_out->addEvent(evtStrInfo_in->getNumberOfEvents());
         for (const auto& elem : evtStrInfo_in->getRunNumbers()) {
            evtStrInfo_out->insertRunNumber(elem);
         }
         for (const auto& elem : evtStrInfo_in->getLumiBlockNumbers()) {
            evtStrInfo_out->insertLumiBlockNumber(elem);
         }
         for (const auto& elem : evtStrInfo_in->getProcessingTags()) {
            evtStrInfo_out->insertProcessingTag(elem);
         }
         for (const auto& [classId, key] : evtStrInfo_in->getItemList()) {
            evtStrInfo_out->insertItemList(classId, key);
         }
         for (const auto& elem : evtStrInfo_in->getEventTypes()) {
            evtStrInfo_out->insertEventType(elem);
         }
      }
   }
   return StatusCode::SUCCESS;
}
StatusCode CopyEventStreamInfo::endInputFile(const SG::SourceID&)
{
   return StatusCode::SUCCESS;
}
StatusCode CopyEventStreamInfo::metaDataStop()
{
   return StatusCode::SUCCESS;
}