AthenaPoolAddressProviderSvc.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#include "AthenaPoolAddressProviderSvc.h"
#include "SGTools/CurrentEventStore.h"
#include "registerKeys.h"
 
#include "PersistentDataModel/AthenaAttributeList.h"
#include "PersistentDataModel/DataHeader.h"
#include "PersistentDataModel/TokenAddress.h"
 
// Framework
#include "GaudiKernel/IClassIDSvc.h"
#include "GaudiKernel/GenericAddress.h"
#include "GaudiKernel/StatusCode.h"
 
#include "StoreGate/StoreGateSvc.h"
#include "StoreGate/ReadHandle.h"
#include "StoreGate/WriteHandle.h"
#include "StorageSvc/DbType.h"
 
#include <vector>
 
//________________________________________________________________________________
AthenaPoolAddressProviderSvc::AthenaPoolAddressProviderSvc(const std::string& name, ISvcLocator* pSvcLocator) :
    base_class(name, pSvcLocator),
    m_metaDataStore("StoreGateSvc/MetaDataStore", name),
    m_clidSvc("ClassIDSvc", name),
    m_guid() {
}
//________________________________________________________________________________
AthenaPoolAddressProviderSvc::~AthenaPoolAddressProviderSvc() {
}
//________________________________________________________________________________
StoreGateSvc* AthenaPoolAddressProviderSvc::eventStore() const {
   return StoreGateSvc::currentStoreGate();
}
//________________________________________________________________________________
StatusCode AthenaPoolAddressProviderSvc::initialize() {
 
   ATH_CHECK( m_metaDataStore.retrieve() );
   ATH_CHECK( m_clidSvc.retrieve() );
 
   return StatusCode::SUCCESS;
}
//________________________________________________________________________________
StatusCode AthenaPoolAddressProviderSvc::preLoadAddresses(StoreID::type storeID,
        IAddressProvider::tadList& tads) {
   if (storeID != StoreID::DETECTOR_STORE) {
      return StatusCode::SUCCESS;
   }
   ServiceHandle<StoreGateSvc> detectorStoreSvc("DetectorStore", name());
   ATH_CHECK( detectorStoreSvc.retrieve() );
 
   SG::ReadHandle<DataHeader> dataHeader("CondProxyProvider", detectorStoreSvc->name());
   if (!dataHeader.isValid()) {
      ATH_MSG_DEBUG("Cannot retrieve DataHeader from DetectorStore.");
      return StatusCode::SUCCESS;
   }
   ATH_MSG_DEBUG("The current File contains: " << dataHeader->size() << " objects");
   for (const auto& element : *dataHeader) {
      SG::TransientAddress* tadd = element.getAddress(pool::POOL_StorageType.type());
      if (tadd->clID() == ClassID_traits<DataHeader>::ID()) {
         delete tadd; tadd = nullptr;
      } else {
         ATH_MSG_DEBUG("preLoadAddresses: DataObject address, clid = " << tadd->clID() << ", name = " << tadd->name());
         tads.push_back(tadd);
      }
      EventSelectorAthenaPoolUtil::registerKeys(element, &*detectorStoreSvc);
   }
 
   return StatusCode::SUCCESS;
}
//________________________________________________________________________________
StatusCode AthenaPoolAddressProviderSvc::loadAddresses(StoreID::type storeID,
        IAddressProvider::tadList& tads) {
   if (storeID != StoreID::EVENT_STORE && storeID != StoreID::PILEUP_STORE) {
      return StatusCode::SUCCESS;
   }
 
   Guid thisFile = Guid::null();
   const DataHeader* dataHeader = nullptr;
   if (dataHeader == nullptr) { // New file (or reading DataHeader)
      SG::ReadHandle<DataHeader> eventDataHeader(m_dataHeaderKey.value(), eventStore()->name());
      if (!eventDataHeader.isValid()) {
         ATH_MSG_ERROR("Cannot retrieve DataHeader from StoreGate: " << m_dataHeaderKey);
         return StatusCode::FAILURE;
      }
      dataHeader = eventDataHeader.cptr();
   }
   // second data header
   if (m_attrListKey.value() != "") {
      const AthenaAttributeList* attrList = nullptr;
      std::string tokenStr;
      if (eventStore()->retrieve(attrList, m_attrListKey.value()).isSuccess()) {
         try {
            tokenStr = (*attrList)["eventRef_secondary"].data<std::string>();
            ATH_MSG_DEBUG("found AthenaAttribute, name = eventRef_secondary = " << tokenStr);
         } catch (std::exception &e) {
            ATH_MSG_ERROR(e.what());
            return StatusCode::FAILURE;
         }
      } else {
         ATH_MSG_ERROR("Cannot find AthenaAttribute, key = " << m_attrListKey.value());
         return StatusCode::FAILURE;
      }
      IOpaqueAddress* iop = new GenericAddress(pool::POOL_StorageType.type(), ClassID_traits<DataHeader>::ID(), tokenStr, "SecondaryEventSelector");
      if (!eventStore()->recordAddress(iop).isSuccess()) {
         ATH_MSG_ERROR("Cannot record address to StoreGate with token string: " << tokenStr);
         return StatusCode::FAILURE;
      }
      ATH_MSG_DEBUG("Created dataHeader SecondaryEventSelector");
   }
   bool doRegister = dataHeader->dhFormToken().empty() ||
     m_dhFormKeys.emplace (dataHeader->dhFormToken()).second;
   ATH_MSG_DEBUG("The current Event contains: " << dataHeader->size() << " objects");
   for (const auto& element : *dataHeader) {
      if (doRegister) {
         EventSelectorAthenaPoolUtil::registerKeys(element, eventStore());
      }
      SG::TransientAddress* tadd = element.getAddress(pool::POOL_StorageType.type());
      if (tadd->clID() == ClassID_traits<DataHeader>::ID()) { // self reference
         delete tadd; tadd = nullptr;
      } else {
         ATH_MSG_DEBUG("loadAddresses: DataObject address, clid = " << tadd->clID() << ", name = " << tadd->name());
         tadd->setProvider(this, storeID);
         tads.push_back(tadd);
      }
   }
   m_guid = thisFile;
   return StatusCode::SUCCESS;
}
//________________________________________________________________________________
StatusCode AthenaPoolAddressProviderSvc::updateAddress(StoreID::type /*storeID*/,
                                                       SG::TransientAddress* /*tad*/,
                                                       const EventContext& /*ctx*/) {
   return StatusCode::FAILURE;
}