EIC code displayed by LXR master/​acts/​Core/​src/​Seeding/​GbtsGeometry.cpp	Source navigation Diff markup Identifier search General search
	Version: master test Revert   Change

File indexing completed on 2025-12-13 09:21:25

 // This file is part of the ACTS project.
 //
 // Copyright (C) 2016 CERN for the benefit of the ACTS project
 //
 // This Source Code Form is subject to the terms of the Mozilla Public
 // License, v. 2.0. If a copy of the MPL was not distributed with this
 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
 #include "Acts/Seeding/GbtsGeometry.hpp"
 
 #include <cmath>
 #include <cstring>
 #include <iostream>
 
 namespace Acts::Experimental {
 
 GbtsLayer::GbtsLayer(const TrigInDetSiLayer& ls, float ew, int bin0)
     : m_layer(ls), m_etaBinWidth(ew) {
   if (m_layer.m_type == 0) {  // barrel
     m_r1 = m_layer.m_refCoord;
     m_r2 = m_layer.m_refCoord;
     m_z1 = m_layer.m_minBound;
     m_z2 = m_layer.m_maxBound;
   } else {  // endcap
     m_r1 = m_layer.m_minBound;
     m_r2 = m_layer.m_maxBound;
     m_z1 = m_layer.m_refCoord;
     m_z2 = m_layer.m_refCoord;
   }
 
   float t1 = m_z1 / m_r1;
   float eta1 = -std::log(std::sqrt(1 + t1 * t1) - t1);
 
   float t2 = m_z2 / m_r2;
   float eta2 = -std::log(std::sqrt(1 + t2 * t2) - t2);
 
   m_minEta = eta1;
   m_maxEta = eta2;
 
   if (m_maxEta < m_minEta) {
     m_minEta = eta2;
     m_maxEta = eta1;
   }
 
   m_maxEta += 1e-6;  // increasing them slightly to avoid range_check exceptions
   m_minEta -= 1e-6;
 
   float deltaEta = m_maxEta - m_minEta;
 
   int binCounter = bin0;
 
   if (deltaEta < m_etaBinWidth) {
     m_nBins = 1;
     m_bins.push_back(binCounter++);
     m_etaBin = deltaEta;
     if (m_layer.m_type == 0) {  // barrel
       m_minRadius.push_back(m_layer.m_refCoord - 2.0);
       m_maxRadius.push_back(m_layer.m_refCoord + 2.0);
       m_minBinCoord.push_back(m_layer.m_minBound);
       m_maxBinCoord.push_back(m_layer.m_maxBound);
     } else {  // endcap
       m_minRadius.push_back(m_layer.m_minBound - 2.0);
       m_maxRadius.push_back(m_layer.m_maxBound + 2.0);
       m_minBinCoord.push_back(m_layer.m_minBound);
       m_maxBinCoord.push_back(m_layer.m_maxBound);
     }
   } else {
     int nB = static_cast<int>(deltaEta / m_etaBinWidth);
     m_nBins = nB;
     if (deltaEta - m_etaBinWidth * nB > 0.5 * m_etaBinWidth) {
       m_nBins++;
     }
 
     m_etaBin = deltaEta / m_nBins;
 
     if (m_nBins == 1) {
       m_bins.push_back(binCounter++);
       if (m_layer.m_type == 0) {  // barrel
         m_minRadius.push_back(m_layer.m_refCoord - 2.0);
         m_maxRadius.push_back(m_layer.m_refCoord + 2.0);
         m_minBinCoord.push_back(m_layer.m_minBound);
         m_maxBinCoord.push_back(m_layer.m_maxBound);
       } else {  // endcap
         m_minRadius.push_back(m_layer.m_minBound - 2.0);
         m_maxRadius.push_back(m_layer.m_maxBound + 2.0);
         m_minBinCoord.push_back(m_layer.m_minBound);
         m_maxBinCoord.push_back(m_layer.m_maxBound);
       }
     } else {
       float eta = m_minEta + 0.5 * m_etaBin;
 
       for (int i = 1; i <= m_nBins; i++) {
         m_bins.push_back(binCounter++);
 
         float e1 = eta - 0.5 * m_etaBin;
         float e2 = eta + 0.5 * m_etaBin;
 
         if (m_layer.m_type == 0) {  // barrel
           m_minRadius.push_back(m_layer.m_refCoord - 2.0);
           m_maxRadius.push_back(m_layer.m_refCoord + 2.0);
           float z1 = m_layer.m_refCoord * std::sinh(e1);
           m_minBinCoord.push_back(z1);
           float z2 = m_layer.m_refCoord * std::sinh(e2);
           m_maxBinCoord.push_back(z2);
         } else {  // endcap
           float r = m_layer.m_refCoord / std::sinh(e1);
           m_minBinCoord.push_back(r);
           m_minRadius.push_back(r - 2.0);
           r = m_layer.m_refCoord / std::sinh(e2);
           m_maxBinCoord.push_back(r);
           m_maxRadius.push_back(r + 2.0);
         }
 
         eta += m_etaBin;
       }
     }
   }
 }
 
 bool GbtsLayer::verifyBin(const GbtsLayer* pL, int b1, int b2, float min_z0,
                           float max_z0) const {
   float z1min = m_minBinCoord.at(b1);
   float z1max = m_maxBinCoord.at(b1);
   float r1 = m_layer.m_refCoord;
 
   if (m_layer.m_type == 0 && pL->m_layer.m_type == 0) {  // barrel -> barrel
 
     const float tol = 5.0;
 
     float min_b2 = pL->m_minBinCoord.at(b2);
     float max_b2 = pL->m_maxBinCoord.at(b2);
 
     float r2 = pL->m_layer.m_refCoord;
 
     float A = r2 / (r2 - r1);
     float B = r1 / (r2 - r1);
 
     float z0_min = z1min * A - max_b2 * B;
     float z0_max = z1max * A - min_b2 * B;
 
     if (z0_max < min_z0 - tol || z0_min > max_z0 + tol) {
       return false;
     }
 
     return true;
   }
 
   if (m_layer.m_type == 0 && pL->m_layer.m_type != 0) {  // barrel -> endcap
 
     const float tol = 10.0;
 
     float z2 = pL->m_layer.m_refCoord;
     float r2max = pL->m_maxBinCoord.at(b2);
     float r2min = pL->m_minBinCoord.at(b2);
 
     if (r2max <= r1) {
       return false;
     }
 
     if (r2min <= r1) {
       r2min = r1 + 1e-3;
     }
 
     float z0_max = 0.0;
     float z0_min = 0.0;
 
     if (z2 > 0) {
       z0_max = (z1max * r2max - z2 * r1) / (r2max - r1);
       z0_min = (z1min * r2min - z2 * r1) / (r2min - r1);
     } else {
       z0_max = (z1max * r2min - z2 * r1) / (r2min - r1);
       z0_min = (z1min * r2max - z2 * r1) / (r2max - r1);
     }
 
     if (z0_max < min_z0 - tol || z0_min > max_z0 + tol) {
       return false;
     }
     return true;
   }
 
   return true;
 }
 
 int GbtsLayer::getEtaBin(float zh, float rh) const {
   if (m_bins.size() == 1) {
     return m_bins.at(0);
   }
 
   float t1 = zh / rh;
   float eta = -std::log(std::sqrt(1 + t1 * t1) - t1);
 
   int idx = static_cast<int>((eta - m_minEta) / m_etaBin);
 
   if (idx < 0) {
     idx = 0;
   }
   if (idx >= static_cast<int>(m_bins.size())) {
     idx = static_cast<int>(m_bins.size()) - 1;
   }
 
   return m_bins.at(idx);  // index in the global storage
 }
 
 float GbtsLayer::getMinBinRadius(int idx) const {
   if (idx >= static_cast<int>(m_minRadius.size())) {
     idx = idx - 1;
   }
   if (idx < 0) {
     idx = 0;
   }
 
   return m_minRadius.at(idx);
 }
 
 float GbtsLayer::getMaxBinRadius(int idx) const {
   if (idx >= static_cast<int>(m_maxRadius.size())) {
     idx = idx - 1;
   }
   if (idx < 0) {
     idx = 0;
   }
 
   return m_maxRadius.at(idx);
 }
 
 GbtsLayer::~GbtsLayer() {
   m_bins.clear();
 }
 
 GbtsGeometry::GbtsGeometry(const std::vector<TrigInDetSiLayer>& layers,
                            const std::unique_ptr<GbtsConnector>& conn) {
   const float min_z0 = -168.0;
   const float max_z0 = 168.0;
 
   m_etaBinWidth = conn->m_etaBin;
 
   for (const auto& layer : layers) {
     const GbtsLayer* pL = addNewLayer(layer, m_nEtaBins);
     m_nEtaBins += pL->num_bins();
   }
 
   // calculating bin tables in the connector...
   // calculate bin pairs for graph edge building
 
   int lastBin1 = -1;
 
   for (std::map<int, std::vector<GbtsConnection*> >::const_iterator it =
            conn->m_connMap.begin();
        it != conn->m_connMap.end(); ++it) {
     const std::vector<GbtsConnection*>& vConn = (*it).second;
 
     for (std::vector<GbtsConnection*>::const_iterator cIt = vConn.begin();
          cIt != vConn.end(); ++cIt) {
       unsigned int src = (*cIt)->m_src;  // n2 : the new connectors
       unsigned int dst = (*cIt)->m_dst;  // n1
 
       const GbtsLayer* pL1 = getGbtsLayerByKey(dst);
       const GbtsLayer* pL2 = getGbtsLayerByKey(src);
 
       if (pL1 == nullptr) {
         std::cout << " skipping invalid dst layer " << dst << std::endl;
         continue;
       }
       if (pL2 == nullptr) {
         std::cout << " skipping invalid src layer " << src << std::endl;
         continue;
       }
       int nSrcBins = pL2->m_bins.size();
       int nDstBins = pL1->m_bins.size();
 
       (*cIt)->m_binTable.resize(nSrcBins * nDstBins, 0);
 
       for (int b1 = 0; b1 < nDstBins; b1++) {    // loop over bins in Layer 1
         for (int b2 = 0; b2 < nSrcBins; b2++) {  // loop over bins in Layer 2
           if (!pL1->verifyBin(pL2, b1, b2, min_z0, max_z0)) {
             continue;
           }
           int address = b1 + b2 * nDstBins;
           (*cIt)->m_binTable.at(address) = 1;
 
           int bin1_idx = pL1->m_bins.at(b1);
           int bin2_idx = pL2->m_bins.at(b2);
 
           if (bin1_idx != lastBin1) {  // adding a new group
 
             std::vector<int> v2(1, bin2_idx);
             m_binGroups.push_back(std::make_pair(bin1_idx, v2));
             lastBin1 = bin1_idx;
 
           } else {  // extend the last group
             (*m_binGroups.rbegin()).second.push_back(bin2_idx);
           }
         }
       }
     }
   }
 }
 
 GbtsGeometry::~GbtsGeometry() {
   for (std::vector<GbtsLayer*>::iterator it = m_layArray.begin();
        it != m_layArray.end(); ++it) {
     delete (*it);
   }
   m_layMap.clear();
   m_layArray.clear();
 }
 
 const GbtsLayer* GbtsGeometry::getGbtsLayerByKey(unsigned int key) const {
   std::map<unsigned int, GbtsLayer*>::const_iterator it = m_layMap.find(key);
   if (it == m_layMap.end()) {
     return nullptr;
   }
   return (*it).second;
 }
 
 const GbtsLayer* GbtsGeometry::getGbtsLayerByIndex(int idx) const {
   return m_layArray.at(idx);
 }
 
 const GbtsLayer* GbtsGeometry::addNewLayer(const TrigInDetSiLayer& l,
                                            int bin0) {
   unsigned int layerKey = l.m_subdet;
 
   float ew = m_etaBinWidth;
 
   GbtsLayer* pHL = new GbtsLayer(l, ew, bin0);
 
   m_layMap.insert(std::pair<unsigned int, GbtsLayer*>(layerKey, pHL));
   m_layArray.push_back(pHL);
   return pHL;
 }
 
 }  // namespace Acts::Experimental

This page was automatically generated by the 2.3.7 LXR engine. The LXR team