MakeEventStreamInfo.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MakeEventStreamInfo.h"
 
#include "GaudiKernel/IAlgorithm.h"
 
#include "PersistentDataModel/DataHeader.h"
#include "EventInfo/EventStreamInfo.h"
#include "xAODEventInfo/EventInfo.h"
#include "EventInfoUtils/EventInfoFromxAOD.h"
 
//___________________________________________________________________________
MakeEventStreamInfo::MakeEventStreamInfo(const std::string& type,
    const std::string& name,
    const IInterface* parent) : base_class(type, name, parent)
{
}
//___________________________________________________________________________
StatusCode MakeEventStreamInfo::initialize() {
   ATH_MSG_DEBUG("Initializing " << name());
   // Locate the MetaDataStore
   ATH_CHECK(m_metaDataSvc.retrieve());
   ATH_CHECK(m_eventStore.retrieve());
 
   // Autoconfigure data header key
   if (m_dataHeaderKey.empty()){
      if (const auto* parentAlg = dynamic_cast<const IAlgorithm*>(this->parent()); parentAlg) {
         m_dataHeaderKey.setValue(parentAlg->name());
      } else {
         ATH_MSG_ERROR("Unable to get parent Algorithm");
         return StatusCode::FAILURE;
      }
   }
 
   return StatusCode::SUCCESS;
}
//___________________________________________________________________________
StatusCode MakeEventStreamInfo::postInitialize() {
   // Remove EventStreamInfo with same key if it exists
   bool ignoreIfAbsent = true;
   if( !m_metaDataSvc->remove<EventStreamInfo>(m_key.value(), ignoreIfAbsent).isSuccess() ) {
      ATH_MSG_ERROR("Unable to remove EventStreamInfo with key " << m_key.value());
      return StatusCode::FAILURE;
   }
   return StatusCode::SUCCESS;
}
//___________________________________________________________________________
StatusCode MakeEventStreamInfo::preExecute() {
   return StatusCode::SUCCESS;
}
//___________________________________________________________________________
StatusCode MakeEventStreamInfo::preStream() {
   return StatusCode::SUCCESS;
}
//___________________________________________________________________________
StatusCode MakeEventStreamInfo::postExecute() {
   SG::ReadHandle<DataHeader> dataHeader(m_dataHeaderKey);
   if (!dataHeader.isValid()) {
      return StatusCode::SUCCESS;
   }
   // Retrieve the EventInfo object
   EventType evtype;
   unsigned long long runN = 0;
   unsigned lumiN = 0;
   SG::ReadHandle<xAOD::EventInfo> xEventInfo(m_eventInfoKey);
   if (xEventInfo.isValid()) {
      runN = xEventInfo->runNumber();
      lumiN = xEventInfo->lumiBlock();
      evtype = eventTypeFromxAOD(xEventInfo.get());
   } else {
      ATH_MSG_ERROR("Unable to retrieve xAOD::EventInfo object");
      return StatusCode::FAILURE;
   }
 
   EventStreamInfo* pEventStream = m_metaDataSvc->tryRetrieve<EventStreamInfo>(m_key.value());
   if( !pEventStream ) {
      auto esinfo_up = std::make_unique<EventStreamInfo>();
      pEventStream = esinfo_up.get();
      if( m_metaDataSvc->record(std::move(esinfo_up), m_key.value()).isFailure() ) {
         ATH_MSG_ERROR("Could not register EventStreamInfo object");
         return StatusCode::FAILURE;
      }
   }
   static std::once_flag resetNumberOfEventsFlag;
   std::call_once(resetNumberOfEventsFlag, [this, pEventStream]() -> void {
      ATH_MSG_DEBUG("Resetting the EventStreamInfo payload at the first event");
      pEventStream->reset();
   });
   pEventStream->addEvent();
   pEventStream->insertProcessingTag(dataHeader->getProcessTag());
   pEventStream->insertLumiBlockNumber( lumiN );
   pEventStream->insertRunNumber( runN );
   for (const DataHeaderElement& dhe : *dataHeader) {
      pEventStream->insertItemList(dhe.getPrimaryClassID(), dhe.getKey());
   }
   pEventStream->insertEventType( evtype );
 
   m_eventCounter++;
 
   return StatusCode::SUCCESS;
}
//___________________________________________________________________________
StatusCode MakeEventStreamInfo::preFinalize() {
  EventStreamInfo* pEventStream = m_metaDataSvc->tryRetrieve<EventStreamInfo>(m_key.value());
  if (!pEventStream) {
    auto esinfo_up = std::make_unique<EventStreamInfo>();
    pEventStream = esinfo_up.get();
    ATH_CHECK(m_metaDataSvc->record(std::move(esinfo_up), m_key.value()));
  }
  if (m_eventCounter > 0 &&
      m_eventCounter < pEventStream->getNumberOfEvents()) {
    // The number of events that have been processed by MakeEventStreamInfo
    // tool is less than the number of events in already existing
    // EventStreamInfo object. This can happen if the EventStreamInfo object was
    // updated by CopyEventStreamInfo, for example at beginning of reading the
    // 2nd (or next) file.
    ATH_MSG_DEBUG(
        "Event count mismatch in EventStreamInfo (likely due to "
        "multi-file processing). MakeEventStreamInfo processed: "
        << m_eventCounter << " events, existing EventStreamInfo object has "
        << pEventStream->getNumberOfEvents()
        << " events. Setting number of events to what "
           "MakeEventStreamInfo processed.");
    pEventStream->setNumberOfEvents(m_eventCounter);
  } else {
    // Insert processing tags when no events have been processed
    pEventStream->insertProcessingTag(m_dataHeaderKey.value());
  }
  return (StatusCode::SUCCESS);
}
//___________________________________________________________________________
StatusCode MakeEventStreamInfo::finalize() {
   ATH_MSG_DEBUG("in finalize()");
   return StatusCode::SUCCESS;
}