PixelAlignCondAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "PixelAlignCondAlg.h"
 
#include "PixelReadoutGeometry/PixelDetectorManager.h"
#include "InDetReadoutGeometry/SiDetectorElement.h"
 
#include <memory>
 
PixelAlignCondAlg::PixelAlignCondAlg(const std::string& name, ISvcLocator* pSvcLocator)
  : ::AthCondAlgorithm(name, pSvcLocator)
{
}
 
StatusCode PixelAlignCondAlg::initialize()
{
  ATH_MSG_DEBUG("initialize " << name());
 
  // Read Handles
  // Static
  ATH_CHECK(m_readKeyStatic.initialize(!m_useDynamicAlignFolders.value()));
  // Dynamic
  ATH_CHECK(m_readKeyDynamicL1.initialize(m_useDynamicAlignFolders.value()));
  ATH_CHECK(m_readKeyDynamicL2.initialize(m_useDynamicAlignFolders.value()));
  ATH_CHECK(m_readKeyDynamicL3.initialize(m_useDynamicAlignFolders.value()));
  ATH_CHECK(m_readKeyIBLDist.initialize(!m_readKeyIBLDist.empty()));
 
  // Write Handles
  ATH_CHECK(m_writeKey.initialize());
 
  ATH_CHECK(detStore()->retrieve(m_detManager, m_detManagerName));
 
  return StatusCode::SUCCESS;
}
 
StatusCode PixelAlignCondAlg::execute(const EventContext& ctx) const
{
  ATH_MSG_DEBUG("execute " << name());
 
  // ____________ Construct Write Cond Handle and check its validity ____________
  SG::WriteCondHandle<GeoAlignmentStore> writeHandle{m_writeKey, ctx};
 
  // Do we have a valid Write Cond Handle for current time?
  if (writeHandle.isValid()) {
    ATH_MSG_DEBUG("CondHandle " << writeHandle.fullKey() << " is already valid."
                  << ". In theory this should not be called, but may happen"
                  << " if multiple concurrent events are being processed out of order.");
    return StatusCode::SUCCESS;
  }
 
  // Get SiDetectorElements
  const InDetDD::SiDetectorElementCollection* oldColl{m_detManager->getDetectorElementCollection()};
  if (oldColl==nullptr) {
    ATH_MSG_FATAL("Null pointer is returned by getDetectorElementCollection()");
    return StatusCode::FAILURE;
  }
 
  // ____________ Construct new Write Cond Object ____________
  std::unique_ptr<GeoAlignmentStore> writeCdo{std::make_unique<GeoAlignmentStore>()};
 
  if (not m_useDynamicAlignFolders.value()) { // Static
    // ____________ Get Read Cond Object ____________
    SG::ReadCondHandle<AlignableTransformContainer> readHandleStatic{m_readKeyStatic, ctx};
    const AlignableTransformContainer* readCdoStatic{*readHandleStatic};
    if (readCdoStatic==nullptr) {
      ATH_MSG_FATAL("Null pointer to the read conditions object of " << m_readKeyStatic.key());
      return StatusCode::FAILURE;
    }
    writeHandle.addDependency(readHandleStatic);
 
    // ____________ Apply alignments to Pixel GeoModel ____________
    // Construct Container for read CDO.
    InDetDD::RawAlignmentObjects readCdoContainerStatic;
    readCdoContainerStatic.emplace(m_readKeyStatic.key(), readCdoStatic);
    ATH_CHECK(m_detManager->align(readCdoContainerStatic, writeCdo.get()));
 
    // IBL
    if (!m_readKeyIBLDist.empty()) {
      // This absolutely needs to go last, since it relies on deltas from other alignment
      // already be set in the alignment store!!!
      SG::ReadCondHandle<CondAttrListCollection> readHandleIBLDist{m_readKeyIBLDist, ctx};
      const CondAttrListCollection* readCdoIBLDist{*readHandleIBLDist};
      if (readCdoIBLDist==nullptr) {
        ATH_MSG_FATAL("Null pointer to the read conditions object of " << readHandleIBLDist.key());
        return StatusCode::FAILURE;
      }
      writeHandle.addDependency(readHandleIBLDist);
 
      InDetDD::RawAlignmentObjects readCdoContainerIBLDist;
      readCdoContainerIBLDist.emplace(m_readKeyIBLDist.key(), readCdoIBLDist);
      ATH_CHECK(m_detManager->align(readCdoContainerIBLDist, writeCdo.get()));
 
    }
  } else { // Dynamic
    // ____________ Get Read Cond Object ____________
    SG::ReadCondHandle<CondAttrListCollection> readHandleDynamicL1{m_readKeyDynamicL1, ctx};
    const CondAttrListCollection* readCdoDynamicL1{*readHandleDynamicL1};
    if (readCdoDynamicL1==nullptr) {
      ATH_MSG_FATAL("Null pointer to the read conditions object of " << m_readKeyDynamicL1.key());
      return StatusCode::FAILURE;
    }
    writeHandle.addDependency(readHandleDynamicL1);
 
    SG::ReadCondHandle<CondAttrListCollection> readHandleDynamicL2{m_readKeyDynamicL2, ctx};
    const CondAttrListCollection* readCdoDynamicL2{*readHandleDynamicL2};
    if (readCdoDynamicL2==nullptr) {
      ATH_MSG_FATAL("Null pointer to the read conditions object of " << readHandleDynamicL2.key());
      return StatusCode::FAILURE;
    }
    writeHandle.addDependency(readHandleDynamicL2);
 
    SG::ReadCondHandle<AlignableTransformContainer> readHandleDynamicL3{m_readKeyDynamicL3, ctx};
    const AlignableTransformContainer* readCdoDynamicL3{*readHandleDynamicL3};
    if (readCdoDynamicL3==nullptr) {
      ATH_MSG_FATAL("Null pointer to the read conditions object of " << readHandleDynamicL3.key());
      return StatusCode::FAILURE;
    }
    writeHandle.addDependency(readHandleDynamicL3);
 
    // ____________ Apply alignments to Pixel GeoModel ____________
    // Construct Container for read CDO-s.
    InDetDD::RawAlignmentObjects readCdoContainerDynamicL1;
    readCdoContainerDynamicL1.emplace(m_readKeyDynamicL1.key(), readCdoDynamicL1);
    ATH_CHECK(m_detManager->align(readCdoContainerDynamicL1, writeCdo.get()));
 
    InDetDD::RawAlignmentObjects readCdoContainerDynamicL2;
    readCdoContainerDynamicL2.emplace(m_readKeyDynamicL2.key(), readCdoDynamicL2);
    ATH_CHECK(m_detManager->align(readCdoContainerDynamicL2, writeCdo.get()));
 
    InDetDD::RawAlignmentObjects readCdoContainerDynamicL3;
    readCdoContainerDynamicL3.emplace(m_readKeyDynamicL3.key(), readCdoDynamicL3);
    ATH_CHECK(m_detManager->align(readCdoContainerDynamicL3, writeCdo.get()));
 
 
    // IBL
    if (!m_readKeyIBLDist.empty()) {
      // This absolutely needs to go last, since it relies on deltas from other alignment
      // already be set in the alignment store!!!
      SG::ReadCondHandle<CondAttrListCollection> readHandleIBLDist{m_readKeyIBLDist, ctx};
      const CondAttrListCollection* readCdoIBLDist{*readHandleIBLDist};
      if (readCdoIBLDist==nullptr) {
        ATH_MSG_FATAL("Null pointer to the read conditions object of " << readHandleIBLDist.key());
        return StatusCode::FAILURE;
      }
      writeHandle.addDependency(readHandleIBLDist);
      InDetDD::RawAlignmentObjects readCdoContainerIBLDist;
      readCdoContainerIBLDist.emplace(m_readKeyIBLDist.key(), readCdoIBLDist);
      ATH_CHECK(m_detManager->align(readCdoContainerIBLDist, writeCdo.get()));  
    }
    
  }
  
  // Set (default) absolute transforms in alignment store by calling them.
  for (const InDetDD::SiDetectorElement* oldEl: *oldColl) {
    oldEl->getMaterialGeom()->getAbsoluteTransform(writeCdo.get());
    oldEl->getMaterialGeom()->getDefAbsoluteTransform(writeCdo.get());
  }
  writeCdo->lockDelta();
  writeCdo->lockPosCache();
 
  if (writeHandle.record(std::move(writeCdo)).isFailure()) {
    ATH_MSG_FATAL("Could not record GeoAlignmentStore " << writeHandle.key()
                  << " with EventRange " << writeHandle.getRange()
                  << " into Conditions Store");
    return StatusCode::FAILURE;
  }
  ATH_MSG_INFO("recorded new CDO " << writeHandle.key() << " with range " << writeHandle.getRange() << " into Conditions Store");
 
  return StatusCode::SUCCESS;
}