InDetToXAODClusterConversion.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#include "src/InDetToXAODClusterConversion.h"
#include "InDetMeasurementUtilities/ClusterConversionUtilities.h"
 
#include "Identifier/Identifier.h"
#include "InDetIdentifier/PixelID.h"
#include "InDetIdentifier/SCT_ID.h"
#include "HGTD_Identifier/HGTD_ID.h"
#include "InDetPrepRawData/PixelClusterCollection.h"
#include "InDetPrepRawData/SCT_ClusterCollection.h"
 
#include "StoreGate/WriteHandle.h"
#include "StoreGate/ReadHandle.h"
 
#include "xAODInDetMeasurement/PixelClusterAuxContainer.h"
#include "xAODInDetMeasurement/StripClusterAuxContainer.h"
 
#include "InDetReadoutGeometry/SiDetectorElement.h"
#include "SCT_ReadoutGeometry/StripStereoAnnulusDesign.h"
 
#include "AthContainers/DataVector.h"
#include "AthAllocators/DataPool.h"
#include <iterator>
 
 
 
using namespace InDet;
 
// Constructor with parameters:
InDetToXAODClusterConversion::InDetToXAODClusterConversion(const std::string &name, ISvcLocator *pSvcLocator) :
AthReentrantAlgorithm(name, pSvcLocator)
{}
 
 
//-----------------------------------------------------------------------------
// Initialize method:
StatusCode InDetToXAODClusterConversion::initialize() {
  ATH_MSG_INFO("Initializing " << name() << " ...");
 
  ATH_CHECK(detStore()->retrieve(m_pixelID,"PixelID"));
  ATH_CHECK(detStore()->retrieve(m_stripID,"SCT_ID"));
  if (m_processHgtd)
  {ATH_CHECK(detStore()->retrieve(m_hgtdID,"HGTD_ID"));}
  
  ATH_CHECK( m_pixelDetEleCollKey.initialize(m_processPixel) );
  ATH_CHECK( m_stripDetEleCollKey.initialize(m_processStrip) );
  ATH_CHECK( m_HGTDDetEleCollKey.initialize(m_processHgtd) );
 
  ATH_CHECK( m_inputPixelClusterContainerKey.initialize(m_processPixel) );
  ATH_CHECK( m_inputStripClusterContainerKey.initialize(m_processStrip) );
  ATH_CHECK( m_outputPixelClusterContainerKey.initialize(m_processPixel) );
  ATH_CHECK( m_outputStripClusterContainerKey.initialize(m_processStrip) );
 
  ATH_CHECK( m_inputHgtdClusterContainerKey.initialize(m_processHgtd) );
  ATH_CHECK( m_outputHgtdClusterContainerKey.initialize(m_processHgtd) );
 
  ATH_MSG_DEBUG( "Initialize done !" );
  return StatusCode::SUCCESS;
}
 
//----------------------------------------------------------------------------
//
StatusCode InDetToXAODClusterConversion::execute(const EventContext& ctx) const {
  ATH_MSG_DEBUG("Executing " << name() << " ...");
 
  if (m_processPixel.value()) {
    ATH_MSG_DEBUG("Converting Pixel Clusters: InDet -> xAOD");
    ATH_CHECK( convertPixelClusters(ctx) );
  }
 
  if (m_processStrip.value()) {
    ATH_MSG_DEBUG("Converting Strip Clusters: InDet -> xAOD");
    ATH_CHECK( convertStripClusters(ctx) );
  }
 
  if (m_processHgtd.value()) {
    ATH_MSG_DEBUG("Converting HGTD Clusters: InDet -> xAOD");
    ATH_CHECK( convertHgtdClusters(ctx) );
  }
 
  return StatusCode::SUCCESS;
}
 
StatusCode InDetToXAODClusterConversion::convertPixelClusters(const EventContext& ctx) const {
  SG::ReadCondHandle<InDetDD::SiDetectorElementCollection> pixelDetEleHandle(m_pixelDetEleCollKey, ctx);
  const InDetDD::SiDetectorElementCollection* pixElements(*pixelDetEleHandle);
  if (not pixelDetEleHandle.isValid() or pixElements==nullptr) {
    ATH_MSG_FATAL(m_pixelDetEleCollKey.fullKey() << " is not available.");
    return StatusCode::FAILURE;
  }
 
  SG::WriteHandle<xAOD::PixelClusterContainer> outputPixelClusterContainer(m_outputPixelClusterContainerKey, ctx);
  ATH_CHECK( outputPixelClusterContainer.record (std::make_unique<xAOD::PixelClusterContainer>(SG::VIEW_ELEMENTS, SG::ALWAYS_TRACK_INDICES),
                         std::make_unique<xAOD::PixelClusterAuxContainer>()) );
  ATH_MSG_DEBUG( "Recorded xAOD::PixelClusterContainer with key: " << m_outputPixelClusterContainerKey.key()  );
 
  SG::ReadHandle<InDet::PixelClusterContainer> inputPixelClusterContainer(m_inputPixelClusterContainerKey, ctx);
 
  static const SG::AuxElement::Accessor< ElementLink< InDet::PixelClusterCollection > > pixelLinkAcc("pixelClusterLink");
 
  size_t nclusters = 0;
  for (const auto *const clusterCollection : *inputPixelClusterContainer) {
    if (!clusterCollection) continue;
    nclusters += clusterCollection->size();
  }
  DataPool<xAOD::PixelCluster> pool (nclusters);
  outputPixelClusterContainer->push_new (nclusters, [&pool](){return pool.nextElementPtr();});
  size_t icluster = 0;
 
  for (const auto *const clusterCollection : *inputPixelClusterContainer) {
    if (!clusterCollection) continue;
    for(const auto *const theCluster : *clusterCollection)  {
      Identifier clusterId = theCluster->identify();
 
      const InDetDD::SiDetectorElement *element=pixElements->getDetectorElement(m_pixelID->wafer_hash(m_pixelID->wafer_id(clusterId)));
      if ( element==nullptr ) {
    ATH_MSG_FATAL( "Invalid pixel detector element for cluster identifier " << clusterId );
    return StatusCode::FAILURE;
      }
 
      xAOD::PixelCluster * pixelCl = outputPixelClusterContainer->at(icluster++);
      ATH_CHECK( TrackingUtilities::convertInDetToXaodCluster(*theCluster, *element, *pixelCl) );
 
      // Create auxiliary branches accessors
      ElementLink<InDet::PixelClusterCollection> pixelLink(theCluster, *clusterCollection);
      pixelLinkAcc( *pixelCl ) = pixelLink;
    }
  }
  if (icluster != outputPixelClusterContainer->size()) {
    ATH_MSG_ERROR("xAOD::PixelClusterContainer miscount; " << icluster << " versus " << outputPixelClusterContainer->size());
    return StatusCode::FAILURE;
  }
 
  ATH_MSG_DEBUG("xAOD::PixelClusterContainer with size: " << outputPixelClusterContainer->size());
  return StatusCode::SUCCESS;
}
 
StatusCode InDetToXAODClusterConversion::convertStripClusters(const EventContext& ctx) const {
  SG::ReadCondHandle<InDetDD::SiDetectorElementCollection> stripDetEleHandle(m_stripDetEleCollKey, ctx);
  const InDetDD::SiDetectorElementCollection* stripElements(*stripDetEleHandle);
  if (not stripDetEleHandle.isValid() or stripElements==nullptr) {
    ATH_MSG_FATAL(m_stripDetEleCollKey.fullKey() << " is not available.");
    return StatusCode::FAILURE;
  }
 
  SG::WriteHandle<xAOD::StripClusterContainer> outputStripClusterContainer(m_outputStripClusterContainerKey, ctx);
  ATH_CHECK( outputStripClusterContainer.record (std::make_unique<xAOD::StripClusterContainer>(SG::VIEW_ELEMENTS, SG::ALWAYS_TRACK_INDICES),
                         std::make_unique<xAOD::StripClusterAuxContainer>()) );
  ATH_MSG_DEBUG( "Recorded xAOD::StripClusterContainer with key: " << m_outputStripClusterContainerKey.key()  );
 
  SG::ReadHandle<InDet::SCT_ClusterContainer> inputStripClusterContainer(m_inputStripClusterContainerKey, ctx);
 
  size_t nclusters = 0;
  for (const auto *const clusterCollection : *inputStripClusterContainer) {
    if (!clusterCollection) continue;
    nclusters += clusterCollection->size();
  }
  DataPool<xAOD::StripCluster> pool (nclusters);
  outputStripClusterContainer->push_new (nclusters, [&pool](){return pool.nextElementPtr();});
  size_t icluster = 0;
 
  static const SG::AuxElement::Accessor< ElementLink< InDet::SCT_ClusterCollection > > stripLinkAcc("sctClusterLink");
  for (const auto *const clusterCollection : *inputStripClusterContainer) {
    if (!clusterCollection) continue;
    for(const auto *const theCluster : *clusterCollection)  {
      Identifier clusterId = theCluster->identify();
 
      const InDetDD::SiDetectorElement *element=stripElements->getDetectorElement(m_stripID->wafer_hash(m_stripID->wafer_id(clusterId)));
      if ( element==nullptr ) {
    ATH_MSG_FATAL( "Invalid strip detector element for cluster with identifier " << clusterId );
    return StatusCode::FAILURE;
      }
 
 
      xAOD::StripCluster * stripCl = outputStripClusterContainer->at(icluster++);
      ATH_CHECK( TrackingUtilities::convertInDetToXaodCluster(*theCluster, *element, *stripCl, m_isITk) );
      
      // Create auxiliary branches accessors
      ElementLink<InDet::SCT_ClusterCollection> stripLink(theCluster, *clusterCollection);
      stripLinkAcc( *stripCl ) = stripLink;
    }
  }
  if (icluster != outputStripClusterContainer->size()) {
    ATH_MSG_ERROR("xAOD::StripClusterContainer miscount; " << icluster << " versus " << outputStripClusterContainer->size());
    return StatusCode::FAILURE;
  }
 
  ATH_MSG_DEBUG("xAOD::StripClusterContainer with size: " << outputStripClusterContainer->size());
 
  return StatusCode::SUCCESS;
}
 
StatusCode InDetToXAODClusterConversion::convertHgtdClusters(const EventContext& ctx) const {
    SG::ReadCondHandle<InDetDD::HGTD_DetectorElementCollection> hgtdDetEleHandle(m_HGTDDetEleCollKey, ctx);
  const InDetDD::HGTD_DetectorElementCollection* hgtdElements(*hgtdDetEleHandle);
  if (not hgtdDetEleHandle.isValid() or hgtdElements==nullptr) {
    ATH_MSG_FATAL(m_HGTDDetEleCollKey.fullKey() << " is not available.");
    return StatusCode::FAILURE;
  }
 
  SG::WriteHandle<xAOD::HGTDClusterContainer> outputHgtdClusterContainer(m_outputHgtdClusterContainerKey, ctx);
  ATH_CHECK( outputHgtdClusterContainer.record (std::make_unique<xAOD::HGTDClusterContainer>(),
                        std::make_unique<xAOD::HGTDClusterAuxContainer>()) );
  ATH_MSG_DEBUG( "Recorded xAOD::HGTDClusterContainer with key: " << m_outputHgtdClusterContainerKey.key() );
 
  SG::ReadHandle<::HGTD_ClusterContainer> inputHgtdClusterContainer(m_inputHgtdClusterContainerKey, ctx);
  ATH_CHECK(inputHgtdClusterContainer.isValid());
 
  static const SG::AuxElement::Accessor< ElementLink< ::HGTD_ClusterCollection > > hgtdLinkAcc("hgtdClusterLink");
  for (const auto *const clusterCollection : *inputHgtdClusterContainer) {
    if (!clusterCollection) continue;
    for(const auto *const theCluster : *clusterCollection)  {
      Identifier clusterId = theCluster->identify();
 
      const auto *element = InDetDD::HGTDDetEl::getDetectorElement(m_hgtdID->wafer_hash(m_hgtdID->wafer_id(clusterId)),*hgtdElements);
      if ( element==nullptr ) {
        ATH_MSG_FATAL( "Invalid strip detector element for cluster with identifier " << clusterId );
        return StatusCode::FAILURE;
      }
 
      xAOD::HGTDCluster * hgtdCl = new xAOD::HGTDCluster();
      outputHgtdClusterContainer->push_back(hgtdCl);
      ATH_CHECK( TrackingUtilities::convertInDetToXaodCluster(*theCluster, *element, *hgtdCl) );
 
      // Create auxiliary branches accessors
      ElementLink<::HGTD_ClusterCollection> hgtdLink(theCluster, *clusterCollection);
      hgtdLinkAcc( *hgtdCl ) = hgtdLink;
    }
  }
 
  ATH_MSG_DEBUG("xAOD::HGTDClusterContainer with size: " << outputHgtdClusterContainer->size());
  return StatusCode::SUCCESS;
}