SiDetElementsRoadCondAlg_xk.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "SiDetElementsRoadCondAlg_xk.h"
 
#include "SiDetElementsRoadUtils_xk.h"
 
#include "SiDetElementsRoadTool_xk/SiDetElementsComparison.h"
#include "SiDetElementsRoadTool_xk/SiDetElementsLayer_xk.h"
 
#include <memory>
#include <utility>

// Constructor
 
InDet::SiDetElementsRoadCondAlg_xk::SiDetElementsRoadCondAlg_xk(const std::string& name, ISvcLocator* pSvcLocator)
  : ::AthCondAlgorithm(name, pSvcLocator)
{
}

// Initialisation
 
StatusCode InDet::SiDetElementsRoadCondAlg_xk::initialize()
{
  if ((not m_usePIX) and (not m_useSCT)) {
    ATH_MSG_FATAL("Please don't call this tool if usePixel and useSCT are false");
    return StatusCode::FAILURE;
  }
 
  ATH_CHECK(m_pixelDetEleCollKey.initialize(m_usePIX));
  ATH_CHECK(m_SCTDetEleCollKey.initialize(m_useSCT));
 
  ATH_CHECK(m_writeKey.initialize());
 
  return StatusCode::SUCCESS;
}

// Finalize
 
StatusCode InDet::SiDetElementsRoadCondAlg_xk::finalize()
{
  return StatusCode::SUCCESS;
}

// Map of detector elements production
// Taken from InDet::SiDetElementsRoadMaker_xk::mapDetectorElementsProduction()
 
StatusCode InDet::SiDetElementsRoadCondAlg_xk::execute(const EventContext& ctx) const
{
  const double pi2=2.*M_PI;
  const double pi=M_PI;
 
  SG::WriteCondHandle<SiDetElementsLayerVectors_xk> writeHandle{m_writeKey, ctx};
  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;
  }
 
  std::unique_ptr<SiDetElementsLayerVectors_xk> writeCdo{std::make_unique<SiDetElementsLayerVectors_xk>(3)};
 
  std::vector<const InDetDD::SiDetectorElement*> pW[3];
 
  if (m_usePIX) {
    // Loop over each wafer of pixels
    SG::ReadCondHandle<InDetDD::SiDetectorElementCollection> pixelDetEleHandle(m_pixelDetEleCollKey, ctx);
    const InDetDD::SiDetectorElementCollection* pixelDetEle{*pixelDetEleHandle};
    if (not pixelDetEleHandle.isValid() or pixelDetEle==nullptr) {
      ATH_MSG_FATAL(m_pixelDetEleCollKey.fullKey() << " is not available.");
      return StatusCode::FAILURE;
    }
    for (const InDetDD::SiDetectorElement* s: *pixelDetEle) {
      if      (s->isBarrel()       ) pW[1].push_back(s); // Barrel
      else if (s->center().z() > 0.) pW[2].push_back(s); // Right endcap
      else                           pW[0].push_back(s); // Left  endcap
    }
 
    writeHandle.addDependency(pixelDetEleHandle);
  }
 
  if (m_useSCT) {
    // Loop over each wafer of sct
    SG::ReadCondHandle<InDetDD::SiDetectorElementCollection> sctDetEleHandle(m_SCTDetEleCollKey, ctx);
    const InDetDD::SiDetectorElementCollection* sctDetEle{*sctDetEleHandle};
    if (not sctDetEleHandle.isValid() or sctDetEle==nullptr) {
      ATH_MSG_FATAL(m_SCTDetEleCollKey.fullKey() << " is not available.");
      return StatusCode::FAILURE;
    }
    for (const InDetDD::SiDetectorElement* s: *sctDetEle) {
      if      (s->isBarrel()       ) pW[1].push_back(s); // Barrel
      else if (s->center().z() > 0.) pW[2].push_back(s); // Right endcap
      else                           pW[0].push_back(s); // Left  endcap
    }
 
    writeHandle.addDependency(sctDetEleHandle);
  }
 
  int nel = pW[0].size()+pW[1].size()+pW[2].size();
  if (nel==0) {
    ATH_MSG_WARNING("The number of elements is zero.");
    return StatusCode::SUCCESS;
  }
 
  std::sort(pW[1].begin(), pW[1].end(), InDet::compDetElements_RAZ());
  std::sort(pW[0].begin(), pW[0].end(), InDet::compDetElements_ZRA());
  std::sort(pW[2].begin(), pW[2].end(), InDet::compDetElements_ZRA());
 
  for(int N=0; N<3; ++N) {
 
    double P[40];
    int im    = static_cast<int>(pW[N].size())-1;
    int If    = 0      ;
    double z0 = 0.     ;
    double r0 = 0.     ;
    
    for (int i = 0; i<= im; ++i) {
     
      InDet::SiDetElementsRoadUtils_xk::detElementInformation(*(pW[N][i]), P);
 
      double r    = P[0];
      double z    = P[1];
      bool newl = false;
      if (N==1) {
        if (fabs(r-r0) > 10.) {
          newl=true;
          r0=r;
        }
      } else {
        if (fabs(z-z0) > 10.) {
          newl=true; 
          r0=r; 
          z0=z;
        }
      }
 
      if (newl || i==im) {
 
        int Il = i-1;
        if (i==im) ++Il;
 
        if (If<=Il) {
 
          double rmin = 100000., rmax =-100000.;
          double zmin = 100000., zmax =-100000.;
          double dfm  = 0.;
 
          std::vector<const InDetDD::SiDetectorElement*> pE;
          for (int j=If; j<=Il; ++j) pE.push_back(pW[N][j]);
          std::sort(pE.begin(), pE.end(), InDet::compDetElements_A());
 
          InDet::SiDetElementsLayer_xk layer;
 
          for (auto & j : pE) {
 
            if (j) {
 
              InDet::SiDetElementsRoadUtils_xk::detElementInformation(*j,P);
 
              if ( P[ 9] < rmin ) rmin = P[ 9];
              if ( P[10] > rmax ) rmax = P[10];
              if ( P[11] < zmin ) zmin = P[11];
              if ( P[12] > zmax ) zmax = P[12];
 
              double df1 = fabs(P[13]-P[2]);
              if (df1>pi) df1 = fabs(df1-pi2);
              double df2 = fabs(P[14]-P[2]);
              if (df2>pi) df2 = fabs(df2-pi2);
              if (df1>dfm) dfm = df1;
              if (df2>dfm) dfm = df2;
 
              InDet::SiDetElementLink_xk link(j, P, m_ITkGeometry);
              layer.add(link);
            }
          }
          layer.sortDetectorElements();
          double r  =(rmax+rmin)*.5;
          double dr =(rmax-rmin)*.5;
          double z  =(zmax+zmin)*.5;
          double dz =(zmax-zmin)*.5;
          layer.set(r, dr, z, dz, dfm);
          (writeCdo->at(N)).push_back(layer);
        }
        If = i;
      }
    }
  }
 
  if (writeHandle.record(std::move(writeCdo)).isFailure()) {
    ATH_MSG_FATAL("Could not record " << writeHandle.key()
                  << " with EventRange " << writeHandle.getRange()
                  << " into Conditions Store");
    return StatusCode::FAILURE;
  }
  ATH_MSG_DEBUG("recorded new CDO " << writeHandle.key() << " with range " << writeHandle.getRange() << " into Conditions Store");
 
  return StatusCode::SUCCESS;
}