DiscOverlapDescriptor.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// DiscOverlapDescriptor.cxx, (c) ATLAS Detector software
 
// Amg
#include "GeoPrimitives/GeoPrimitives.h"
// iFatras
#include "InDetTrackingGeometry/DiscOverlapDescriptor.h"
#include "InDetReadoutGeometry/SiDetectorElement.h"
// Trk
#include "TrkSurfaces/Surface.h"
 
#include "StoreGate/StoreGateSvc.h"
#include "InDetIdentifier/SCT_ID.h"
#include "InDetIdentifier/PixelID.h"
#include "Identifier/Identifier.h"
#include <utility>
#include <exception>
 
namespace {
 
//Helpers to do "pseudo retrievals"
//if this was an alg/tool these would have been done
//once in init.
 
const StoreGateSvc*
getDetStore()
{
  SmartIF<StoreGateSvc> detStore{Gaudi::svcLocator()->service("DetectorStore")};
  if (!detStore.isValid()) {
    throw std::runtime_error("DiscOverlapDescriptor can not locate DetectorStore");
  }
 
  return detStore.get();
}
 
const PixelID*
getPixelID(StoreGateSvc const * const  detStore)
{
  const PixelID* pixIdHelper = nullptr;
  if (detStore->retrieve(pixIdHelper, "PixelID").isFailure()) {
    throw std::runtime_error("DiscOverlapDescriptor can not locate PixelID");
  }
  return pixIdHelper;
}
 
const SCT_ID*
getSCT_ID(StoreGateSvc const * const detStore)
{
  const SCT_ID* sctIdHelper = nullptr;
  if (detStore->retrieve(sctIdHelper, "SCT_ID").isFailure()) {
    throw std::runtime_error("DiscOverlapDescriptor can not locate SCT_ID");
  }
  return sctIdHelper;
}
 
} // end anonymous namespace
 
InDet::DiscOverlapDescriptor::DiscOverlapDescriptor(const Trk::BinnedArray<Trk::Surface>* bin_array,
                                                    const std::vector<Trk::BinUtility>& singleBinUtils,
                                                    bool isPixel):
  m_bin_array(bin_array),
  m_singleBinUtils(singleBinUtils),
  m_pixelCase(isPixel)
{}
 
bool InDet::DiscOverlapDescriptor::reachableSurfaces(std::vector<Trk::SurfaceIntersection>& surfaces,
                                                     const Trk::Surface& tsf,
                                                     const Amg::Vector3D& pos,
                                                     const Amg::Vector3D&) const
{ //These are done once no matter how many
  //instances of DiscOverlapDescriptor we have
  static const StoreGateSvc* const detStore = getDetStore();
  static const PixelID* const pixIdHelper = getPixelID(detStore);
  static const SCT_ID* const sctIdHelper = getSCT_ID(detStore);
 
  // get the according detector element
  const InDetDD::SiDetectorElement* pElement  = dynamic_cast<const InDetDD::SiDetectorElement*>(tsf.associatedDetectorElement());
 
  // first add the target surface
  surfaces.emplace_back(Trk::Intersection(pos, 0., true),&tsf);
  int etaModule =
    m_pixelCase
      ? pixIdHelper->eta_module(tsf.associatedDetectorElementIdentifier())
      : sctIdHelper->eta_module(tsf.associatedDetectorElementIdentifier());
 
  // return empty cell vector
  if (pElement) {
    size_t newCapacity = surfaces.size() + 19;
    if (pElement->otherSide()) newCapacity += 19;
    surfaces.reserve(newCapacity);
 
    addOtherSideDO(pElement,surfaces);
 
    addNextInPhiDO(pElement,surfaces);
    addPrevInPhiDO(pElement,surfaces);
 
    addNextInPhiDO(pElement->nextInPhi(),surfaces);
    addPrevInPhiDO(pElement->prevInPhi(),surfaces);
 
    // First use the bin utiliy to find the surfaces on the same disc but in different rings
    // This is done as follows:
    // 1 -> Find at which bin corresponds the surface, i.e. at which ring on the disc it corresponds
    // 2 -> If it is not 0 (i.e. the surface doesn't belong to the innermost ring),
    // look for the surfaces in the ring with smaller eta value (it returns 2 or 3 surfaces)
    // 3 -> If it is smaller than the number of rings (i.e. the surface doen't belong to the outermost ring),
    // look for the surfaces in the ring with bigger eta value (it returns 2 or 3 surfaces)//
 
    if (m_bin_array && !m_singleBinUtils.empty()){
 
      const Trk::Surface* samePhi_PrevEta     = nullptr;
      const Trk::Surface* samePhi_NextEta     = nullptr;
 
      const Trk::Surface* previousPhi_PrevEta = nullptr;
      const Trk::Surface* nextPhi_PrevEta     = nullptr;
 
      const Trk::Surface* previousPhi_NextEta = nullptr;
      const Trk::Surface* nextPhi_NextEta     = nullptr;
 
      std::span<Trk::Surface const * const> surf = m_bin_array->arrayObjects();
      size_t offset = 0;
      for (unsigned int bin = 0; bin < m_singleBinUtils.size(); bin++) {
        int etamod =
          m_pixelCase
            ? pixIdHelper->eta_module(
                (*(surf[offset])).associatedDetectorElementIdentifier())
            : sctIdHelper->eta_module(
                (*(surf[offset])).associatedDetectorElementIdentifier());
 
        if (etamod == etaModule || etamod < (etaModule - 1) ||
            etamod > (etaModule + 1)) {
          offset += (std::as_const(m_singleBinUtils).at(bin)).bins();
          continue;
        }
 
        double PrevDeltaPhi = 9999.;
        double NextDeltaPhi = -9999.;
        for (unsigned int ss = offset; ss < (offset+(std::as_const(m_singleBinUtils).at(bin)).bins()); ss++ ) {
          if (etamod == (etaModule-1) ) {
            if( tsf.center().phi() == (*(surf[ss])).center().phi() )
              samePhi_PrevEta = surf[ss];
            double DeltaPhi = tsf.center().phi() - (*(surf[ss])).center().phi();
            if( DeltaPhi < PrevDeltaPhi && DeltaPhi > 0) {
              previousPhi_PrevEta = surf[ss];
              PrevDeltaPhi = DeltaPhi;
            }
            if( DeltaPhi > NextDeltaPhi && DeltaPhi < 0) {
              nextPhi_PrevEta = surf[ss];
              NextDeltaPhi = DeltaPhi;
            }
          } else if (etamod == (etaModule+1) ) {
            if( tsf.center().phi() == (*(surf[ss])).center().phi() )
              samePhi_NextEta = surf[ss];
            double DeltaPhi = tsf.center().phi() - (*(surf[ss])).center().phi();
            if( DeltaPhi < PrevDeltaPhi && DeltaPhi > 0) {
              previousPhi_NextEta = surf[ss];
              PrevDeltaPhi = DeltaPhi;
            }
 
            if( DeltaPhi > NextDeltaPhi && DeltaPhi < 0) {
              nextPhi_NextEta = surf[ss];
              NextDeltaPhi = DeltaPhi;
            }
          }
        }
        offset += (std::as_const(m_singleBinUtils).at(bin)).bins();
      }
 
      if (samePhi_PrevEta) {
        const InDetDD::SiDetectorElement* PhiEta_Element = dynamic_cast<const InDetDD::SiDetectorElement*>(samePhi_PrevEta->associatedDetectorElement());
        if (PhiEta_Element) {
          addSurfaceDO(PhiEta_Element,surfaces);
          addOtherSideDO(PhiEta_Element, surfaces);
        }
      }
 
      if (previousPhi_PrevEta) {
        const InDetDD::SiDetectorElement* PhiEta_Element = dynamic_cast<const InDetDD::SiDetectorElement*>(previousPhi_PrevEta->associatedDetectorElement());
        if (PhiEta_Element) {
          addSurfaceDO(PhiEta_Element,surfaces);
          addOtherSideDO(PhiEta_Element, surfaces);
 
          addPrevInPhiDO(PhiEta_Element,surfaces);
          addPrevInPhiDO(PhiEta_Element->prevInPhi(),surfaces);
        }
      }
 
      if (nextPhi_PrevEta) {
        const InDetDD::SiDetectorElement* PhiEta_Element = dynamic_cast<const InDetDD::SiDetectorElement*>(nextPhi_PrevEta->associatedDetectorElement());
        if (PhiEta_Element) {
          addSurfaceDO(PhiEta_Element,surfaces);
          addOtherSideDO(PhiEta_Element, surfaces);
 
          addNextInPhiDO(PhiEta_Element,surfaces);
          addNextInPhiDO(PhiEta_Element->nextInPhi(),surfaces);
        }
      }
 
      if (samePhi_NextEta) {
        const InDetDD::SiDetectorElement* PhiEta_Element = dynamic_cast<const InDetDD::SiDetectorElement*>(samePhi_NextEta->associatedDetectorElement());
        if (PhiEta_Element) {
          addSurfaceDO(PhiEta_Element,surfaces);
          addOtherSideDO(PhiEta_Element, surfaces);
        }
      }
 
      if (previousPhi_NextEta) {
        const InDetDD::SiDetectorElement* PhiEta_Element = dynamic_cast<const InDetDD::SiDetectorElement*>(previousPhi_NextEta->associatedDetectorElement());
        if (PhiEta_Element) {
          addSurfaceDO(PhiEta_Element,surfaces);
          addOtherSideDO(PhiEta_Element, surfaces);
 
          addPrevInPhiDO(PhiEta_Element,surfaces);
          addPrevInPhiDO(PhiEta_Element->prevInPhi(),surfaces);
        }
      }
 
      if (nextPhi_NextEta) {
        const InDetDD::SiDetectorElement* PhiEta_Element = dynamic_cast<const InDetDD::SiDetectorElement*>(nextPhi_NextEta->associatedDetectorElement());
        if (PhiEta_Element) {
          addSurfaceDO(PhiEta_Element,surfaces);
          addOtherSideDO(PhiEta_Element, surfaces);
 
          addNextInPhiDO(PhiEta_Element,surfaces);
          addNextInPhiDO(PhiEta_Element->nextInPhi(),surfaces);
        }
      }
    }
  }
 
  return false;
 
}
 
bool InDet::DiscOverlapDescriptor::dumpSurfaces(std::vector<Trk::SurfaceIntersection>& surfaces) const {
  std::cout << "Dumping Surfaces for " << (m_pixelCase ? "Pixel " : "SCT ")
            << "with size = " << surfaces.size() << std::endl;
  static const StoreGateSvc* const detStore = getDetStore();
  static const PixelID* const pixIdHelper = getPixelID(detStore);
  static const SCT_ID* const sctIdHelper = getSCT_ID(detStore);
 
  for (auto & surface : surfaces) {
    Identifier hitId = (surface.object)->associatedDetectorElementIdentifier();
    if (m_pixelCase)
      std::cout <<  "barrel_ec " << pixIdHelper->barrel_ec(hitId)
                << ", layer_disk " << pixIdHelper->layer_disk(hitId)
                << ", phi_module " << pixIdHelper->phi_module(hitId)
                << ", eta_module " << pixIdHelper->eta_module(hitId) << std::endl;
    else
      std::cout <<  "barrel_ec " << sctIdHelper->barrel_ec(hitId)
                << ", layer_disk " << sctIdHelper->layer_disk(hitId)
                << ", phi_module " << sctIdHelper->phi_module(hitId)
                << ", eta_module " << sctIdHelper->eta_module(hitId)
                << ", side " << sctIdHelper->side(hitId) << std::endl;
  }
  return true;
}