SeedingToolBase Class Reference

#include <SeedingToolBase.h>

Inheritance diagram for SeedingToolBase:

Public Member Functions
	SeedingToolBase (const std::string &t, const std::string &n, const IInterface *p)
	SeedingToolBase (const std::string &t, const std::string &n, const IInterface *p)
ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.
virtual StatusCode	sysInitialize () override
	Perform system initialization for an algorithm.
virtual StatusCode	sysStart () override
	Handle START transition.
virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles.
Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T, V, H > &t)
void	updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream &	msg () const
bool	msgLvl (const MSG::Level lvl) const

Protected Types
typedef TrigFTF_GNN_Node	GNN_Node
typedef TrigFTF_GNN_DataStorage	GNN_DataStorage
typedef TrigFTF_GNN_Edge	GNN_Edge
typedef TrigFTF_GNNR3_Node	GNNR3_Node
typedef TrigFTF_GNNR3_DataStorage	GNNR3_DataStorage
typedef TrigFTF_GNNR3_Edge	GNNR3_Edge

Protected Member Functions
virtual StatusCode	initialize ()
virtual StatusCode	finalize ()
std::pair< int, int >	buildTheGraph (const IRoiDescriptor &, const std::unique_ptr< GNN_DataStorage > &, std::vector< GNN_Edge > &) const
int	runCCA (int, std::vector< GNN_Edge > &) const
void	extractSeedsFromTheGraph (int, int, int, std::vector< GNN_Edge > &, std::vector< std::pair< float, std::vector< unsigned int > > > &) const
bool	check_z0_bitmask (const unsigned short &, const float &, const float &, const float &) const
float	estimate_curvature (const std::array< const GNN_Node *, 3 > &) const
bool	validate_triplet (std::array< const GNN_Node *, 3 > &, const float min_pt, const float tau_ratio, const float tau_ratio_cut) const
virtual StatusCode	initialize ()
virtual StatusCode	finalize ()
std::pair< int, int >	buildTheGraph (const IRoiDescriptor &, const std::unique_ptr< GNNR3_DataStorage > &, std::vector< GNNR3_Edge > &) const
int	runCCA (int, std::vector< GNNR3_Edge > &) const
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void >	renounce (T &h)
void	extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
	Add StoreName to extra input/output deps as needed.

Protected Attributes
ToolHandle< ITrigL2LayerNumberTool >	m_layerNumberTool {this, "layerNumberTool", "TrigL2LayerNumberToolITk"}
const AtlasDetectorID *	m_atlasId = nullptr
const SCT_ID *	m_sctId = nullptr
const PixelID *	m_pixelId = nullptr
BooleanProperty	m_filter_phi {this, "DoPhiFiltering", true}
BooleanProperty	m_useBeamTilt {this, "UseBeamTilt", false}
BooleanProperty	m_LRTmode {this, "LRTMode",false}
BooleanProperty	m_useML {this, "useML", true}
UnsignedIntegerProperty	m_nMaxPhiSlice {this, "nMaxPhiSlice", 53}
BooleanProperty	m_doubletFilterRZ {this, "Doublet_FilterRZ", true}
BooleanProperty	m_useEtaBinning {this, "UseEtaBinning", true}
BooleanProperty	m_matchBeforeCreate {this, "MatchBeforeCreate", true}
FloatProperty	m_minPt {this, "pTmin", 1000.0}
FloatProperty	m_etaBinOverride {this, "etaBin", 0.0f, "specify non-zero to override eta bin width from connection file (default 0.2 in createLinkingScheme.py)"}
IntegerProperty	m_nMaxEdges {this, "MaxGraphEdges", 2000000}
IntegerProperty	m_nBufferEdges {this, "BufferEdges", 200000}
StringProperty	m_connectionFile {this, "ConnectionFileName", "binTables_ITK_RUN4.txt"}
StringProperty	m_lutFile {this, "MLpredictorLutFileName", "gbts_ml_pixel_barrel_loose.lut"}
BooleanProperty	m_useGPUseedExtraction {this, "UseGPUseedExtraction", true}
BooleanProperty	m_useOldTunings {this, "UseOldTunings", false}
BooleanProperty	m_validateTriplets {this, "ValidateTriplets", true}
BooleanProperty	m_useAdaptiveCuts {this, "UseAdaptiveCuts", true}
FloatProperty	m_tau_ratio_cut {this, "tau_ratio_cut", 0.007}
FloatProperty	m_tau_ratio_corr {this, "tau_ratio_correction", 0.006}
FloatProperty	m_d0_max {this, "d0_max", 3.0}
float	m_phiSliceWidth = 0.
std::unique_ptr< GNN_FasTrackConnector >	m_connector = nullptr
std::vector< TrigInDetSiLayer >	m_layerGeometry
std::unique_ptr< const TrigFTF_GNN_Geometry >	m_geo = nullptr
std::vector< std::array< float, 5 > >	m_mlLUT
std::unique_ptr< GNNR3_FasTrackConnector >	m_connector = nullptr
std::unique_ptr< const TrigFTF_GNNR3_Geometry >	m_geo = nullptr

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes
StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default)
StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default)
std::vector< SG::VarHandleKeyArray * >	m_vhka
bool	m_varHandleArraysDeclared

Detailed Description

Definition at line 29 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

Member Typedef Documentation

GNN_DataStorage

typedef TrigFTF_GNN_DataStorage SeedingToolBase::GNN_DataStorage

protected

Definition at line 36 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

GNN_Edge

typedef TrigFTF_GNN_Edge SeedingToolBase::GNN_Edge

protected

Definition at line 37 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

GNN_Node

typedef TrigFTF_GNN_Node SeedingToolBase::GNN_Node

protected

Definition at line 35 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

GNNR3_DataStorage

typedef TrigFTF_GNNR3_DataStorage SeedingToolBase::GNNR3_DataStorage

protected

Definition at line 33 of file TrigInDetR3PattRecoTools/src/SeedingToolBase.h.

GNNR3_Edge

typedef TrigFTF_GNNR3_Edge SeedingToolBase::GNNR3_Edge

protected

Definition at line 34 of file TrigInDetR3PattRecoTools/src/SeedingToolBase.h.

GNNR3_Node

typedef TrigFTF_GNNR3_Node SeedingToolBase::GNNR3_Node

protected

Definition at line 32 of file TrigInDetR3PattRecoTools/src/SeedingToolBase.h.

StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< AlgTool > >::StoreGateSvc_t

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

SeedingToolBase() [1/2]

SeedingToolBase::SeedingToolBase ( const std::string & t, const std::string & n, const IInterface * p )

inline

Definition at line 31 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

: AthAlgTool(t,n,p){}

SeedingToolBase() [2/2]

SeedingToolBase::SeedingToolBase ( const std::string & t, const std::string & n, const IInterface * p )

inline

Definition at line 28 of file TrigInDetR3PattRecoTools/src/SeedingToolBase.h.

: AthAlgTool(t,n,p){}

Member Function Documentation

buildTheGraph() [1/2]

std::pair< int, int > SeedingToolBase::buildTheGraph ( const IRoiDescriptor & roi, const std::unique_ptr< GNN_DataStorage > & storage, std::vector< GNN_Edge > & edgeStorage ) const

protected

Definition at line 98 of file TrigInDetPattRecoTools/src/SeedingToolBase.cxx.

                                                                                                                                                                                 {
 
  
  struct GBTS_SlidingWindow {
 
    GBTS_SlidingWindow() : m_first_it(0), m_deltaPhi(0.0), m_has_nodes(false), m_bin(nullptr) {};
    
    unsigned int m_first_it;// sliding window position
    float m_deltaPhi;       // window half-width;
    bool m_has_nodes;       // active or not
 
    const TrigFTF_GNN_EtaBin* m_bin;//associated eta bin
  };
  
  constexpr float M_2PI = 2.0*M_PI;
  
  const float cut_dphi_max      = m_LRTmode ? 0.07f : 0.012f;
  const float cut_dcurv_max     = m_LRTmode ? 0.015f : 0.001f;
  const float cut_tau_ratio_max = m_LRTmode ? 0.015f : static_cast<float>(m_tau_ratio_cut);
  const float min_z0            = m_LRTmode ? -600.0 : roi.zedMinus();
  const float max_z0            = m_LRTmode ? 600.0 : roi.zedPlus();
  const float min_deltaPhi      = m_LRTmode ? 0.01f : 0.001f;
  const float tau_ratio_precut  = 0.009f;
  
  const float maxOuterRadius    = m_LRTmode ? 1050.0 : 550.0;
 
  const float cut_zMinU = min_z0 + maxOuterRadius*roi.dzdrMinus();
  const float cut_zMaxU = max_z0 + maxOuterRadius*roi.dzdrPlus();
 
  constexpr float ptCoeff = 0.29997*1.9972/2.0;// ~0.3*B/2 - assuming nominal field of 2*T
 
  float tripletPtMin = 0.8f*m_minPt;//correction due to limited pT resolution
  const float pt_scale     = 900.0f/m_minPt;//to re-scale original tunings done for the 900 MeV pT cut
  
  float maxCurv = ptCoeff/tripletPtMin;
 
  float maxKappa_high_eta          = m_LRTmode ? 1.0f*maxCurv : std::sqrt(0.8f)*maxCurv;
  float maxKappa_low_eta           = m_LRTmode ? 1.0f*maxCurv : std::sqrt(0.6f)*maxCurv;
  
  if(!m_useOldTunings && !m_LRTmode) {//new settings for curvature cuts
    maxKappa_high_eta          = 4.75e-4f*pt_scale;
    maxKappa_low_eta           = 3.75e-4f*pt_scale;
  }
 
  const float dphi_coeff                 = m_LRTmode ? 1.0f*maxCurv : 0.68f*maxCurv;
  
  const float minDeltaRadius = 2.0;
    
  float deltaPhi0 = 0.5f*m_phiSliceWidth;//the default sliding window along phi
  
  unsigned int nConnections = 0;
  
  edgeStorage.reserve(m_nMaxEdges);
  
  int nEdges = 0;
 
  float z0_histo_coeff = 16/(max_z0 - min_z0 + 1e-6);//assuming 16-bit z0 bitmask
 
  for(const auto& bg : m_geo->bin_groups()) {//loop over bin groups
    
    TrigFTF_GNN_EtaBin& B1 = storage->getEtaBin(bg.first);
 
    if(B1.empty()) continue;
 
    float rb1 = B1.getMinBinRadius();
    
    const unsigned int lk1 = B1.m_layerKey;
 
    const bool isBarrel1 = (lk1 / 10000) == 8;
    
    //prepare a sliding window for each bin2 in the group 
 
    std::vector<GBTS_SlidingWindow> vSLW;
 
    vSLW.resize(bg.second.size());//initialization using default ctor
 
    int win_idx = 0;
    
    for(const auto& b2_idx : bg.second) { //loop over n2 eta-bins in L2 layers
 
      const TrigFTF_GNN_EtaBin& B2 = storage->getEtaBin(b2_idx);
 
      if(B2.empty()) {
        win_idx++;
        continue;
      }
      
      float rb2 = B2.getMaxBinRadius();
 
      float deltaPhi = deltaPhi0;//the default
      
      if(m_useEtaBinning) { //override the default window width
        float abs_dr = std::fabs(rb2-rb1);
        if (m_useOldTunings) {
          deltaPhi = min_deltaPhi + dphi_coeff*abs_dr;
        }
        else {
          if(abs_dr < 60.0) {
            deltaPhi = 0.002f + 4.33e-4f*pt_scale*abs_dr;
          } else {
            deltaPhi = 0.015f + 2.2e-4f*pt_scale*abs_dr;
          }
        }
      }
 
      vSLW[win_idx].m_bin = &B2;
      vSLW[win_idx].m_has_nodes = true;
      vSLW[win_idx].m_deltaPhi = deltaPhi;
      win_idx++;
    }
 
    for(unsigned int n1Idx = 0;n1Idx<B1.m_vn.size();n1Idx++) {//in GBTSv3 the outer loop goes over n1 nodes in the Layer 1 bin
 
      B1.m_vFirstEdge[n1Idx] = nEdges;//initialization using the top watermark of the edge storage
 
      unsigned short num_created_edges = 0;//the counter for the incoming graph edges created for n1
 
      bool is_connected = false;
 
      std::array<unsigned char, 16> z0_histo = {};
  
      const std::array<float, 5>& n1pars = B1.m_params[n1Idx];
 
      float phi1 = n1pars[2];
      float r1 = n1pars[3];
      float z1 = n1pars[4];
 
      for(unsigned int winIdx = 0; winIdx < vSLW.size(); winIdx++) {//the intermediate loop over sliding windows
 
        GBTS_SlidingWindow& slw = vSLW[winIdx];
 
        if (!slw.m_has_nodes) continue;
 
        const TrigFTF_GNN_EtaBin& B2 = *slw.m_bin;
 
    const unsigned int lk2 = B2.m_layerKey;
 
    const bool isBarrel2 = (lk2 / 10000) == 8;
 
        float deltaPhi = slw.m_deltaPhi;
      
        //sliding window phi1 +/- deltaPhi
      
        float minPhi = phi1 - deltaPhi;
        float maxPhi = phi1 + deltaPhi;
      
    for(unsigned int n2PhiIdx = slw.m_first_it; n2PhiIdx<B2.m_vPhiNodes.size();n2PhiIdx++) {//the inner loop over n2 nodes using sliding window
    
      float phi2 = B2.m_vPhiNodes[n2PhiIdx].first;
 
      if(phi2 < minPhi) {
            slw.m_first_it = n2PhiIdx; //update the window position
            continue;
          }
          if(phi2 > maxPhi) break; //break and go to the next window
      
      unsigned int n2Idx = B2.m_vPhiNodes[n2PhiIdx].second;
 
      unsigned short node_info = B2.m_vIsConnected[n2Idx];
 
      if ((lk1 == 80000) && (node_info == 0) ) continue;//skip isolated nodes as their incoming edges lead to nowhere
 
      unsigned int   n2_first_edge = B2.m_vFirstEdge[n2Idx];
          unsigned short n2_num_edges  = B2.m_vNumEdges[n2Idx];
      unsigned int   n2_last_edge  = n2_first_edge + n2_num_edges;
      
      const std::array<float, 5>& n2pars = B2.m_params[n2Idx];
    
      float r2 = n2pars[3];
      
      float dr = r2 - r1;
    
      if(dr < minDeltaRadius) {
        continue;
      }
    
      float z2 = n2pars[4];
 
      float dz = z2 - z1;
      float tau = dz/dr;
      float ftau = std::fabs(tau);
      if (ftau > 36.0) {
        continue;
      }
    
      if(ftau < n1pars[0]) continue;
      if(ftau > n1pars[1]) continue;
 
      if(ftau < n2pars[0]) continue;
      if(ftau > n2pars[1]) continue;
 
      float z0 = z1 - r1*tau;
 
      if (lk1 == 80000) {//check against non-empty z0 histogram
        
        if ( !check_z0_bitmask(node_info, z0, min_z0, z0_histo_coeff) ) {
          continue;
        }
      }
      
      if (m_doubletFilterRZ) {
        
        if(z0 < min_z0 || z0 > max_z0) continue;
      
        float zouter = z0 + maxOuterRadius*tau;
      
        if(zouter < cut_zMinU || zouter > cut_zMaxU) continue;                
      }
        
      float curv = (phi2-phi1)/dr;
      float abs_curv = std::abs(curv);
        
      if(ftau < 4.0) {//eta = 2.1
        if(abs_curv > maxKappa_low_eta) {
          continue;
        }
      }
      else {
        if(abs_curv > maxKappa_high_eta) {
          continue;
        }
      }
 
      float exp_eta = std::sqrt(1.f+tau*tau) - tau;
        
      if (m_matchBeforeCreate && (lk1 == 80000 || lk1 == 81000) ) {//match edge candidate against edges incoming to n2
 
        bool isGood = n2_num_edges <= 2;//we must have enough incoming edges to decide
        
        if(!isGood) {
 
          float uat_1 = 1.0f/exp_eta;
            
          for(unsigned int n2_in_idx = n2_first_edge; n2_in_idx < n2_last_edge; n2_in_idx++) {
            
        float tau2 = edgeStorage.at(n2_in_idx).m_p[0]; 
        float tau_ratio = tau2*uat_1 - 1.0f;
        
        if(std::fabs(tau_ratio) > tau_ratio_precut){//bad match
          continue;
        }
        isGood = true;//good match found
        break;
          }
        }
        
        if(!isGood) {//no match found, skip creating [n1 <- n2] edge
          continue;
        }
      }
      
      float dPhi2 = curv*r2;
      float dPhi1 = curv*r1;
    
      if(nEdges < m_nMaxEdges) {
      
        edgeStorage.emplace_back(B1.m_vn[n1Idx], B2.m_vn[n2Idx], exp_eta, curv, phi1 + dPhi1);
 
        num_created_edges++;
          
        int outEdgeIdx = nEdges;
      
        float uat_2  = 1.f/exp_eta;
        float Phi2  = phi2 + dPhi2;
        float curv2 = curv;
        
        for(unsigned int inEdgeIdx = n2_first_edge; inEdgeIdx < n2_last_edge; inEdgeIdx++) {//looking for neighbours of the new edge
          
          TrigFTF_GNN_Edge* pS = &(edgeStorage.at(inEdgeIdx));
          
          if(pS->m_nNei >= N_SEG_CONNS) continue;
 
          const unsigned int lk3 = m_geo->getTrigFTF_GNN_LayerKeyByIndex(pS->m_n2->m_layer);
 
          const bool isBarrel3 = (lk3 / 10000) == 8;
          
          float abs_tau_ratio = std::abs(pS->m_p[0]*uat_2 - 1.0f);
          float add_tau_ratio_corr = 0;
          
          if (m_useAdaptiveCuts) {
 
        if (isBarrel1 && isBarrel2 && isBarrel3) {
          bool no_gap = ((lk3-lk2) == 1000) && ((lk2-lk1) == 1000);
          if(!no_gap) {
            add_tau_ratio_corr = m_tau_ratio_corr;//assume more scattering due to the layer in between
          }
        }
        else {
          bool mixed_triplet = isBarrel1 && isBarrel2 && !isBarrel3;
          if (mixed_triplet) {
            add_tau_ratio_corr = m_tau_ratio_corr;
          }
        }
          }
          
          if(abs_tau_ratio > cut_tau_ratio_max + add_tau_ratio_corr){//bad match
        continue;
          }
          
          float dPhi =  Phi2 - pS->m_p[2];
          
          if(dPhi<-M_PI) dPhi += M_2PI;
          else if(dPhi>M_PI) dPhi -= M_2PI;
          
          if(std::abs(dPhi) > cut_dphi_max) {
        continue;
          }
            
          float dcurv = curv2 - pS->m_p[1];
            
          if(dcurv < -cut_dcurv_max || dcurv > cut_dcurv_max) {
        continue;
          }
 
          //final check: cuts on pT and d0
          
          if (isBarrel1 && isBarrel2 && isBarrel3) {//Pixel barrel
 
                std::array<const GNN_Node*, 3> sps = {B1.m_vn[n1Idx], B2.m_vn[n2Idx], pS->m_n2};
 
                if (!validate_triplet(sps, tripletPtMin, abs_tau_ratio, cut_tau_ratio_max) ) continue;
        
              }
            
          pS->m_vNei[pS->m_nNei++] = outEdgeIdx;
 
          is_connected = true;//there is at least one good match
 
          //edge confirmed - update z0 histogram
 
          int z0_bin_index = z0_histo_coeff*(z0 - min_z0);
 
          ++z0_histo[z0_bin_index];
          
          nConnections++;
        
        }
        nEdges++;       
      }
    } //loop over n2 (outer) nodes inside a sliding window on n2 bin
      } //loop over sliding windows associated with n2 bins
 
      //updating the n1 node attributes
      
      B1.m_vNumEdges[n1Idx] = num_created_edges;
 
      if (is_connected) {
 
        unsigned short z0_bitmask = 0x0;
 
        for(unsigned int bIdx = 0; bIdx < 16; bIdx++) {
 
      if (z0_histo[bIdx] == 0) continue;
 
      z0_bitmask |= (1 << bIdx);
        }
 
        B1.m_vIsConnected[n1Idx] = z0_bitmask;//non-zero mask indicates that there is at least one connected edge
      }
      
    } //loop over n1 (inner) nodes
  } //loop over bin groups: a single n1 bin and multiple n2 bins
 
  if(nEdges >= m_nMaxEdges) {
    ATH_MSG_WARNING("Maximum number of graph edges exceeded - possible efficiency loss "<< nEdges);
  }
 
  return std::make_pair(nEdges, nConnections);
}

buildTheGraph() [2/2]

std::pair< int, int > SeedingToolBase::buildTheGraph ( const IRoiDescriptor & roi, const std::unique_ptr< GNNR3_DataStorage > & storage, std::vector< GNNR3_Edge > & edgeStorage ) const

protected

Definition at line 71 of file TrigInDetR3PattRecoTools/src/SeedingToolBase.cxx.

                                                                                                                                                                                     {
 
  const float M_2PI = 2.0*M_PI;
  
  const float cut_dphi_max      = m_LRTmode ? 0.07 : 0.012;
  const float cut_dcurv_max     = m_LRTmode ? 0.015 : 0.001;
  const float cut_tau_ratio_max = m_LRTmode ? 0.015 : 0.007;
  const float min_z0            = m_LRTmode ? -600.0 : roi.zedMinus();
  const float max_z0            = m_LRTmode ? 600.0 : roi.zedPlus();
  const float min_deltaPhi      = m_LRTmode ? 0.01f : 0.001f;
  
  const float maxOuterRadius    = m_LRTmode ? 1050.0 : 550.0;
 
  const float cut_zMinU = min_z0 + maxOuterRadius*roi.dzdrMinus();
  const float cut_zMaxU = max_z0 + maxOuterRadius*roi.dzdrPlus();
 
  const float ptCoeff = 0.29997*1.9972/2.0;// ~0.3*B/2 - assuming nominal field of 2*T
 
  float tripletPtMin = 0.8*m_minPt;//correction due to limited pT resolution
  
  float maxCurv = ptCoeff/tripletPtMin;
 
  const float maxKappa_high_eta          = m_LRTmode ? 1.0*maxCurv : std::sqrt(0.8)*maxCurv;
  const float maxKappa_low_eta           = m_LRTmode ? 1.0*maxCurv : std::sqrt(0.6)*maxCurv;
  const float dphi_coeff                 = m_LRTmode ? 1.0*maxCurv : 0.68*maxCurv;
  
  const float minDeltaRadius = 2.0;
    
  float deltaPhi = 0.5f*m_phiSliceWidth;//the default sliding window along phi
 
  unsigned int nConnections = 0;
  
  edgeStorage.reserve(m_nMaxEdges);
  
  int nEdges = 0;
 
  for(const auto& bg : m_geo->bin_groups()) {//loop over bin groups
    
    TrigFTF_GNNR3_EtaBin& B1 = storage->getEtaBin(bg.first);
 
    if(B1.empty()) continue;
 
    float rb1 = B1.getMinBinRadius();
 
    for(const auto& b2_idx : bg.second) {
 
      const TrigFTF_GNNR3_EtaBin& B2 = storage->getEtaBin(b2_idx);
 
      if(B2.empty()) continue;
      
      float rb2 = B2.getMaxBinRadius();
    
      if(m_useEtaBinning) {
    deltaPhi = min_deltaPhi + dphi_coeff*std::fabs(rb2-rb1);    
      }
 
      unsigned int first_it = 0;
 
      for(unsigned int n1Idx = 0;n1Idx<B1.m_vn.size();n1Idx++) {//loop over nodes in Layer 1
 
    std::vector<unsigned int>& v1In = B1.m_in[n1Idx];   
 
    if(v1In.size() >= MAX_SEG_PER_NODE) continue;
      
    const std::array<float, 5>& n1pars = B1.m_params[n1Idx];
 
    float phi1 = n1pars[2];
    float r1 = n1pars[3];
    float z1 = n1pars[4];
      
    //sliding window phi1 +/- deltaPhi
      
    float minPhi = phi1 - deltaPhi;
    float maxPhi = phi1 + deltaPhi;
      
    for(unsigned int n2PhiIdx = first_it; n2PhiIdx<B2.m_vPhiNodes.size();n2PhiIdx++) {//sliding window over nodes in Layer 2
    
      float phi2 = B2.m_vPhiNodes[n2PhiIdx].first;
    
      if(phi2 < minPhi) {
        first_it = n2PhiIdx;
        continue;
      }
      if(phi2 > maxPhi) break;
    
      unsigned int n2Idx = B2.m_vPhiNodes[n2PhiIdx].second;
    
      const std::vector<unsigned int>& v2In = B2.m_in[n2Idx];
        
      if(v2In.size() >= MAX_SEG_PER_NODE) continue;
        
      const std::array<float, 5>& n2pars = B2.m_params[n2Idx];
    
      float r2 = n2pars[3];
      
      float dr = r2 - r1;
    
      if(dr < minDeltaRadius) {
        continue;
      }
    
      float z2 = n2pars[4];
 
      float dz = z2 - z1;
      float tau = dz/dr;
      float ftau = std::fabs(tau);
      if (ftau > 36.0) {
        continue;
      }
    
      if(ftau < n1pars[0]) continue;
      if(ftau > n1pars[1]) continue;
 
      if(ftau < n2pars[0]) continue;
      if(ftau > n2pars[1]) continue;
        
      if (m_doubletFilterRZ) {
          
        float z0 = z1 - r1*tau;
      
        if(z0 < min_z0 || z0 > max_z0) continue;
      
        float zouter = z0 + maxOuterRadius*tau;
      
        if(zouter < cut_zMinU || zouter > cut_zMaxU) continue;                
      }
        
      float curv = (phi2-phi1)/dr;
      float abs_curv = std::abs(curv);
        
      if(ftau < 4.0) {//eta = 2.1
        if(abs_curv > maxKappa_low_eta) {
          continue;
        }
      }
      else {
        if(abs_curv > maxKappa_high_eta) {
          continue;
        }
      }
 
      float exp_eta = std::sqrt(1+tau*tau)-tau;
      
      if (m_matchBeforeCreate) {//match edge candidate against edges incoming to n2
 
        bool isGood = v2In.size() <= 2;//we must have enough incoming edges to decide
 
        if(!isGood) {
 
          float uat_1 = 1.0f/exp_eta;
            
          for(const auto& n2_in_idx : v2In) {
            
        float tau2 = edgeStorage.at(n2_in_idx).m_p[0]; 
        float tau_ratio = tau2*uat_1 - 1.0f;
        
        if(std::fabs(tau_ratio) > cut_tau_ratio_max){//bad match
          continue;
        }
        isGood = true;//good match found
        break;
          }
        }
        
        if(!isGood) {//no match found, skip creating [n1 <- n2] edge
          continue;
        }
      }
      
      float dPhi2 = curv*r2;
      float dPhi1 = curv*r1;
    
      if(nEdges < m_nMaxEdges) {
      
        edgeStorage.emplace_back(B1.m_vn[n1Idx], B2.m_vn[n2Idx], exp_eta, curv, phi1 + dPhi1);
        
        if(v1In.size() < MAX_SEG_PER_NODE) v1In.push_back(nEdges);
          
        int outEdgeIdx = nEdges;
      
        float uat_2  = 1/exp_eta;
        float Phi2  = phi2 + dPhi2;
        float curv2 = curv;
        
        for(const auto& inEdgeIdx : v2In) {//looking for neighbours of the new edge
        
          TrigFTF_GNNR3_Edge* pS = &(edgeStorage.at(inEdgeIdx));
          
          if(pS->m_nNei >= N_SEG_CONNS) continue;
        
          float tau_ratio = pS->m_p[0]*uat_2 - 1.0f;
          
          if(std::abs(tau_ratio) > cut_tau_ratio_max){//bad match
        continue;
          }
          
          float dPhi =  Phi2 - pS->m_p[2];
          
          if(dPhi<-M_PI) dPhi += M_2PI;
          else if(dPhi>M_PI) dPhi -= M_2PI;
          
          if(dPhi < -cut_dphi_max || dPhi > cut_dphi_max) {
        continue;
          }
            
          float dcurv = curv2 - pS->m_p[1];
            
          if(dcurv < -cut_dcurv_max || dcurv > cut_dcurv_max) {
        continue;
          }
            
          pS->m_vNei[pS->m_nNei++] = outEdgeIdx;
        
          nConnections++;
        
        }
        nEdges++;       
      }
    } //loop over n2 (outer) nodes
      } //loop over n1 (inner) nodes
    } //loop over bins in Layer 2
  } //loop over bin groups
 
  return std::make_pair(nEdges, nConnections);
}

check_z0_bitmask()

bool SeedingToolBase::check_z0_bitmask ( const unsigned short & z0_bitmask, const float & z0, const float & min_z0, const float & z0_histo_coeff ) const

protected

Definition at line 788 of file TrigInDetPattRecoTools/src/SeedingToolBase.cxx.

                                                                                                                                                {
 
  if (z0_bitmask == 0) return true;
 
  float dz = z0 - min_z0; 
  int z0_bin_index = z0_histo_coeff*dz;
 
  if ((z0_bitmask >> z0_bin_index) & 1) return true;
 
  //check adjacent bins as well
            
  const float z0_resolution = 2.5;
  
  float dzm = dz - z0_resolution;
 
  int next_bin  = z0_histo_coeff*dzm;
 
  if (next_bin >= 0 && next_bin != z0_bin_index) {
      
    if ((z0_bitmask >> next_bin) & 1) return true;
 
  }               
 
  float dzp = dz + z0_resolution;
 
  next_bin  = z0_histo_coeff*dzp;
 
  if (next_bin < 16 && next_bin != z0_bin_index) {
            
    if ((z0_bitmask >> next_bin) & 1) return true;
      
  }
    
  return false;
}

declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty ( Gaudi::Property< T, V, H > & hndl, const SG::VarHandleKeyType &  )

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

  {
    return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                       hndl.value(), 
                                                       hndl.documentation());
 
  }

declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty ( Gaudi::Property< T, V, H > & t )

inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

                                                                     {
    typedef typename SG::HandleClassifier<T>::type htype;
    return AthCommonDataStore<PBASE>::declareGaudiProperty(t, htype());
  }

detStore()

const ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< AlgTool > >::detStore ( ) const

inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

{ return m_detStore; }

estimate_curvature()

float SeedingToolBase::estimate_curvature ( const std::array< const GNN_Node *, 3 > & sps ) const

protected

Definition at line 825 of file TrigInDetPattRecoTools/src/SeedingToolBase.cxx.

                                                                                       {
 
  //conformal mapping with the center at the last spacepoint
 
  float u[2], v[2];
 
  float x0 = sps[2]->x();
  float y0 = sps[2]->y();
 
  float r0 = sps[2]->r();
  
  float cosA = x0/r0;
  
  float sinA = y0/r0;
 
  
  for(unsigned int k=0;k<2;k++) {
 
    float dx = sps[k]->x() - x0;
 
    float dy = sps[k]->y() - y0;
 
    float r2_inv = 1.0/(dx*dx+dy*dy);
    
    float xn = dx*cosA + dy*sinA;
    
    float yn =-dx*sinA + dy*cosA;
 
    u[k] = xn*r2_inv;
    v[k] = yn*r2_inv;    
  }
 
  float du = u[0] - u[1];
 
  if(du==0.0) return 0.0;
  
  float A = (v[0] - v[1])/du;
 
  float B = v[1] - A*u[1];
 
  return 1000.0*B/std::sqrt(1 + A*A); //inverse meters
  
}

evtStore()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( )

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

{ return m_evtStore; }

extractSeedsFromTheGraph()

void SeedingToolBase::extractSeedsFromTheGraph ( int maxLevel, int nEdges, int nHits, std::vector< GNN_Edge > & edgeStorage, std::vector< std::pair< float, std::vector< unsigned int > > > & vOutputSeeds ) const

protected

Definition at line 536 of file TrigInDetPattRecoTools/src/SeedingToolBase.cxx.

                                                                                                                                                                                              {
 
  const float edge_mask_min_eta      = 1.5;
  const float hit_share_threshold    = 0.49;
  const float max_eta_for_seed_split = 0.6;
  const float max_inv_rad_diff       = 0.7e-2;//in inverse meters
 
  int minLevel = 3;//a triplet + 2 confirmation
 
  if(m_LRTmode) {
    minLevel = 2;//a triplet + 1 confirmation
  }
 
  if(maxLevel < minLevel) return;
  
  std::vector<GNN_Edge*> vSeeds;
 
  vSeeds.reserve(nEdges/2);
 
  for(int edgeIndex = 0; edgeIndex < nEdges; edgeIndex++) {
    
    GNN_Edge* pS = &(edgeStorage.at(edgeIndex));
    
    if(pS->m_level < minLevel) continue;
    
    vSeeds.push_back(pS);
  }
  
  if(vSeeds.empty()) return;
  
  std::sort(vSeeds.begin(), vSeeds.end(), GNN_Edge::CompareLevel());
    
  //backtracking
 
  std::vector<std::tuple<float, int, std::vector<const GNN_Node*>, int > > vSeedCandidates;
 
  vSeedCandidates.reserve(vSeeds.size());
 
  std::vector<std::pair<float, unsigned int> > vArgSort;
 
  vArgSort.reserve(vSeeds.size());
 
  unsigned int seed_counter = 0;
  
  auto tFilter = std::make_unique<TrigFTF_GNN_TrackingFilter>(m_layerGeometry, edgeStorage);
 
  for(auto pS : vSeeds) {
 
    if(pS->m_level == -1) continue;
 
    TrigFTF_GNN_EdgeState rs(false);
 
    tFilter->followTrack(pS, rs);
 
    if(!rs.m_initialized) {
      continue;
    }
 
    if(static_cast<int>(rs.m_vs.size()) < minLevel) continue;
 
    float seed_eta = std::abs(-std::log(pS->m_p[0]));
    
    std::vector<const GNN_Node*> vN;
 
    for(std::vector<GNN_Edge*>::reverse_iterator sIt=rs.m_vs.rbegin();sIt!=rs.m_vs.rend();++sIt) {
 
      if (seed_eta > edge_mask_min_eta) {
    (*sIt)->m_level = -1;//mark as collected
      }
      
      if(sIt == rs.m_vs.rbegin()) {
    vN.push_back((*sIt)->m_n1);
      }
 
      vN.push_back((*sIt)->m_n2);
        
    }
 
    if(vN.size()<3) continue;
 
    unsigned int orig_seed_size = vN.size();
 
    float orig_seed_quality = -rs.m_J/orig_seed_size;
    
    int seed_split_flag = (seed_eta < max_eta_for_seed_split) && (orig_seed_size > 3) && (orig_seed_size <= 5) ? 1 : 0;
 
    if (seed_split_flag) {//split the seed by dropping spacepoints
      
      std::array< std::array<const GNN_Node*, 3>, 3> triplets;//2 "drop-outs" and the original seed candidate
      
      std::array<float, 3> inv_rads;//triplet parameter estimate
 
      triplets[0] = {vN[0], vN[orig_seed_size/2], vN[orig_seed_size-1]};
 
      std::vector<const GNN_Node*> drop_out1 = {vN.begin()+1, vN.end()}; //all but the first one
      
      triplets[1] = {drop_out1[0], drop_out1[(orig_seed_size-1)/2], drop_out1[orig_seed_size-2]};
      
      std::vector<const GNN_Node*> drop_out2;
 
      drop_out2.reserve(orig_seed_size-1);
      
      for(unsigned int k = 0; k < orig_seed_size; k++) {
 
        if (k == orig_seed_size/2) continue;//drop the middle SP in the original seed
        
        drop_out2.emplace_back(vN[k]);
      }
 
      triplets[2] = {drop_out2[0], drop_out2[(orig_seed_size-1)/2], drop_out2[orig_seed_size-2]};
 
      for (unsigned int k = 0; k < inv_rads.size(); k++) {
 
        inv_rads[k] = estimate_curvature(triplets[k]);
    
      }
 
      float diffs[3] = {std::abs(inv_rads[1] - inv_rads[0]), std::abs(inv_rads[2] - inv_rads[0]), std::abs(inv_rads[2] - inv_rads[1])};
 
      bool confirmed = diffs[0] < max_inv_rad_diff && diffs[1] < max_inv_rad_diff && diffs[2] < max_inv_rad_diff;
 
      if (confirmed) {
        seed_split_flag = 0;//reset the flag
      }
      
    }
        
    vSeedCandidates.emplace_back(orig_seed_quality, 0, vN, seed_split_flag);
    
    vArgSort.emplace_back(orig_seed_quality, seed_counter);
 
    ++seed_counter;
    
  }
  
  //clone removal code goes below ...
 
  std::sort(vArgSort.begin(), vArgSort.end());
  
  std::vector<int> H2T(nHits + 1, 0);//hit to track associations
 
  int trackId = 0;
 
  
  for(const auto& ags : vArgSort) {
 
    const auto& seed = vSeedCandidates[ags.second];
    
    trackId++;
    
    for(const auto& h : std::get<2>(seed) ) {//loop over spacepoints indices
    
      unsigned int hit_id = h->sp_idx() + 1;
      
      int tid     = H2T[hit_id];
      
      if(tid == 0 || tid > trackId) {//unused hit or used by a lesser track
    
    H2T[hit_id] = trackId;//overwrite
    
      }
    }      
  }
 
  unsigned int trackIdx = 0;
 
  for(const auto& ags : vArgSort) {
 
    const auto& seed = std::get<2>(vSeedCandidates[ags.second]);
    
    int nTotal = seed.size();
    
    int nOther = 0;
    
    int trackId = trackIdx + 1;
 
    ++trackIdx;
 
    for(const auto& h : seed ) {
 
      unsigned int hit_id = h->sp_idx() + 1;
      
      int tid = H2T[hit_id];
 
      if(tid != trackId) {//taken by a better candidate
    nOther++;
      }
    }
 
    if (nOther > hit_share_threshold*nTotal) {
      std::get<1>(vSeedCandidates[ags.second]) = -1;//reject
    }
 
  }
 
  vOutputSeeds.reserve(vSeedCandidates.size());
  
  //drop the clones and split seeds if need be
 
  for(const auto& ags : vArgSort) {
 
    const auto& seed = vSeedCandidates[ags.second];
    
    if (std::get<1>(seed) != 0) continue;//identified as a clone of a better candidate
 
    const auto& vN = std::get<2>(seed);
 
    if (std::get<3>(seed) == 0) {
      
      //add seed to output
 
      std::vector<unsigned int> vSpIdx;
      
      vSpIdx.resize(vN.size());
    
      for(unsigned int k = 0; k < vSpIdx.size(); k++) {
    vSpIdx[k] = vN[k]->sp_idx();
      }
 
      vOutputSeeds.emplace_back(std::get<0>(seed), vSpIdx);
 
      continue;
 
    }
 
    //seed split into "drop-out" seeds 
 
    unsigned int seedSize = vN.size();
        
    std::array<std::size_t, 2> indices2drop = {0, seedSize / 2ul};//the first and the middle
 
    for(const auto& skipIdx : indices2drop) {
 
      std::vector<unsigned int> new_seed;
 
      new_seed.reserve(seedSize-1);
        
      for (unsigned int k = 0; k < seedSize; k++) {
          
    if (k ==  skipIdx) continue;
          
    new_seed.emplace_back(vN[k]->sp_idx());         
      }
 
      vOutputSeeds.emplace_back(std::get<0>(seed), new_seed);        
 
    }
    
  }
  
}

extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps )

protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

finalize() [1/2]

StatusCode SeedingToolBase::finalize ( )

protectedvirtual

Reimplemented in GbtsFtfActsSeedingTool, TrigInDetR3TrackSeedingTool, and TrigInDetTrackSeedingTool.

Definition at line 93 of file TrigInDetPattRecoTools/src/SeedingToolBase.cxx.

                                     {
  StatusCode sc = AthAlgTool::finalize(); 
  return sc;
}

finalize() [2/2]

virtual StatusCode SeedingToolBase::finalize ( )

protectedvirtual

Reimplemented in GbtsFtfActsSeedingTool, TrigInDetR3TrackSeedingTool, and TrigInDetTrackSeedingTool.

initialize() [1/2]

StatusCode SeedingToolBase::initialize ( )

protectedvirtual

Reimplemented in GbtsFtfActsSeedingTool, TrigInDetR3TrackSeedingTool, and TrigInDetTrackSeedingTool.

Definition at line 24 of file TrigInDetPattRecoTools/src/SeedingToolBase.cxx.

                                       {
  ATH_CHECK(AthAlgTool::initialize());
 
  ATH_CHECK(m_layerNumberTool.retrieve());
 
 
  ATH_CHECK(detStore()->retrieve(m_atlasId, "AtlasID"));
 
  ATH_CHECK(detStore()->retrieve(m_pixelId, "PixelID"));
 
  ATH_CHECK(detStore()->retrieve(m_sctId, "SCT_ID"));
 
  std::string conn_fileName = PathResolver::find_file(m_connectionFile, "DATAPATH");
  if (conn_fileName.empty()) {
    ATH_MSG_FATAL("Cannot find layer connections file " << conn_fileName);
    return StatusCode::FAILURE;
  }
  else {
 
    std::ifstream ifs(conn_fileName.c_str());
    
    m_connector = std::make_unique<GNN_FASTRACK_CONNECTOR>(ifs, m_LRTmode);
    if (m_etaBinOverride != 0.0f) {
      m_connector->m_etaBin = m_etaBinOverride;
    }
    
    ATH_MSG_INFO("Layer connections are initialized from file " << conn_fileName);
  }
 
  const std::vector<TrigInDetSiLayer>* pVL = m_layerNumberTool->layerGeometry();
  
  std::copy(pVL->begin(),pVL->end(), std::back_inserter(m_layerGeometry));
 
  m_geo = std::make_unique<TrigFTF_GNN_Geometry>(m_layerGeometry, m_connector);
 
 
  if (m_useML) {
    std::string lut_fileName = PathResolver::find_file(m_lutFile, "DATAPATH");
    if (lut_fileName.empty()) {
      ATH_MSG_FATAL("Cannot find ML predictor LUT file " << lut_fileName);
      return StatusCode::FAILURE;
    }
    else {
      m_mlLUT.reserve(100);
      std::ifstream ifs(lut_fileName.c_str());
      while (!ifs.eof()) {
    float cl_width, min1, max1, min2, max2;
    ifs >> cl_width >> min1 >> max1 >> min2 >> max2;
    if (ifs.eof()) break;
    std::array<float, 5> lut_line = {cl_width, min1, max1, min2, max2};
    m_mlLUT.emplace_back(lut_line);
      }
      ifs.close();
      ATH_MSG_INFO("ML predictor is initialized from file " << lut_fileName<<" LUT has "<<m_mlLUT.size()<<" entries");
    }
  }
  
  m_phiSliceWidth = 2*M_PI/m_nMaxPhiSlice;
  
  ATH_MSG_INFO("SeedingToolBase initialized ");
 
  ATH_MSG_DEBUG("Property useML "<< m_useML);
  ATH_MSG_DEBUG("Property DoPhiFiltering "<<m_filter_phi);
  ATH_MSG_DEBUG("Property pTmin "<<m_minPt);
  ATH_MSG_DEBUG("Property LRTmode "<<m_LRTmode);
 
  return StatusCode::SUCCESS;
}

initialize() [2/2]

virtual StatusCode SeedingToolBase::initialize ( )

protectedvirtual

Reimplemented in GbtsFtfActsSeedingTool, TrigInDetR3TrackSeedingTool, and TrigInDetTrackSeedingTool.

inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::inputHandles ( ) const

overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

msg()

MsgStream & AthCommonMsg< AlgTool >::msg ( ) const

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

msgLvl()

bool AthCommonMsg< AlgTool >::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                               {
    return this->msgLevel(lvl);
  }

outputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::outputHandles ( ) const

overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

renounce()

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > AthCommonDataStore< AthCommonMsg< AlgTool > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

  {
    h.renounce();
    PBASE::renounce (h);
  }

renounceArray()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::renounceArray ( SG::VarHandleKeyArray & handlesArray )

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                          {
    handlesArray.renounce();
  }

runCCA() [1/2]

int SeedingToolBase::runCCA ( int nEdges, std::vector< GNN_Edge > & edgeStorage ) const

protected

Definition at line 468 of file TrigInDetPattRecoTools/src/SeedingToolBase.cxx.

                                                                                      {
 
  constexpr int maxIter = 15;
 
  int maxLevel = 0;
 
  int iter = 0;
  
  std::vector<TrigFTF_GNN_Edge*> v_old;
  
  for(int edgeIndex=0;edgeIndex<nEdges;edgeIndex++) {
 
    TrigFTF_GNN_Edge* pS = &(edgeStorage[edgeIndex]);
    if(pS->m_nNei == 0) continue;
    
    v_old.push_back(pS);//TO-DO: increment level for segments as they already have at least one neighbour
  }
 
  std::vector<TrigFTF_GNN_Edge*> v_new;
  v_new.reserve(v_old.size());
 
  for(;iter<maxIter;iter++) {
 
    //generate proposals
    v_new.clear();
    
    for(auto pS : v_old) {
      
      int next_level = pS->m_level;
          
      for(int nIdx=0;nIdx<pS->m_nNei;nIdx++) {
    
        unsigned int nextEdgeIdx = pS->m_vNei[nIdx];
            
        TrigFTF_GNN_Edge* pN = &(edgeStorage[nextEdgeIdx]);
            
        if(pS->m_level == pN->m_level) {
          next_level = pS->m_level + 1;
          v_new.push_back(pS);
          break;
        }
      }
      
      pS->m_next = next_level;//proposal
    }
  
    //update
 
    int nChanges = 0;
      
    for(auto pS : v_new) {
      if(pS->m_next != pS->m_level) {
        nChanges++;
        pS->m_level = pS->m_next;
        if(maxLevel < pS->m_level) maxLevel = pS->m_level;
      }
    }
 
    if(nChanges == 0) break;
 
 
    v_old.swap(v_new);
    v_new.clear();
  }
 
  return maxLevel;  
}

runCCA() [2/2]

int SeedingToolBase::runCCA ( int nEdges, std::vector< GNNR3_Edge > & edgeStorage ) const

protected

Definition at line 297 of file TrigInDetR3PattRecoTools/src/SeedingToolBase.cxx.

                                                                                        {
 
  const int maxIter = 15;
 
  int maxLevel = 0;
 
  int iter = 0;
  
  std::vector<TrigFTF_GNNR3_Edge*> v_old;
  
  for(int edgeIndex=0;edgeIndex<nEdges;edgeIndex++) {
 
    TrigFTF_GNNR3_Edge* pS = &(edgeStorage[edgeIndex]);
    if(pS->m_nNei == 0) continue;
    
    v_old.push_back(pS);//TO-DO: increment level for segments as they already have at least one neighbour
  }
 
  for(;iter<maxIter;iter++) {
 
    //generate proposals
    std::vector<TrigFTF_GNNR3_Edge*> v_new;
    v_new.clear();
    v_new.reserve(v_old.size());
    
    for(auto pS : v_old) {
      
      int next_level = pS->m_level;
          
      for(int nIdx=0;nIdx<pS->m_nNei;nIdx++) {
    
        unsigned int nextEdgeIdx = pS->m_vNei[nIdx];
            
        TrigFTF_GNNR3_Edge* pN = &(edgeStorage[nextEdgeIdx]);
            
        if(pS->m_level == pN->m_level) {
          next_level = pS->m_level + 1;
          v_new.push_back(pS);
          break;
        }
      }
      
      pS->m_next = next_level;//proposal
    }
  
    //update
 
    int nChanges = 0;
      
    for(auto pS : v_new) {
      if(pS->m_next != pS->m_level) {
        nChanges++;
        pS->m_level = pS->m_next;
        if(maxLevel < pS->m_level) maxLevel = pS->m_level;
      }
    }
 
    if(nChanges == 0) break;
 
 
    v_old = std::move(v_new);
    v_new.clear();
  }
 
  return maxLevel;  
}

sysInitialize()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysInitialize ( )

overridevirtualinherited

Perform system initialization for an algorithm.

We override this to declare all the elements of handle key arrays at the end of initialization. See comments on updateVHKA.

Reimplemented in asg::AsgMetadataTool, AthCheckedComponent< AthAlgTool >, AthCheckedComponent<::AthAlgTool >, and DerivationFramework::CfAthAlgTool.

sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysStart ( )

overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

updateVHKA()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::updateVHKA ( Gaudi::Details::PropertyBase & )

inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

                                               {
    // debug() << "updateVHKA for property " << p.name() << " " << p.toString() 
    //         << "  size: " << m_vhka.size() << endmsg;
    for (auto &a : m_vhka) {
      std::vector<SG::VarHandleKey*> keys = a->keys();
      for (auto k : keys) {
        k->setOwner(this);
      }
    }
  }

validate_triplet()

bool SeedingToolBase::validate_triplet ( std::array< const GNN_Node *, 3 > & sps, const float min_pt, const float tau_ratio, const float tau_ratio_cut ) const

protected

Definition at line 869 of file TrigInDetPattRecoTools/src/SeedingToolBase.cxx.

                                                                                                                                                    {
  
  //conformal mapping with the center at the middle spacepoint
 
  float u[2], v[2];
 
  const float x0 = sps[1]->x();
  const float y0 = sps[1]->y();
 
  const float r0 = sps[1]->r();
  
  const float cosA = x0/r0;
  
  const float sinA = y0/r0;
  
  for(unsigned int k=0;k<2;k++) {
 
    int sp_idx = (k==1) ? 2 : k;
    
    const float dx = sps[sp_idx]->x() - x0;
 
    const float dy = sps[sp_idx]->y() - y0;
 
    const float r2_inv = 1.0/(dx*dx+dy*dy);
    
    const float xn = dx*cosA + dy*sinA;
    
    const float yn =-dx*sinA + dy*cosA;
 
    u[k] = xn*r2_inv;
    v[k] = yn*r2_inv;    
  }
 
  const float du = u[0] - u[1];
 
  if ( du == 0.0 ) return false;
  
  const float A = (v[0] - v[1])/du;
 
  const float B = v[1] - A*u[1];
 
  const float d0 = r0*(B*r0 - A);
 
  if (std::abs(d0) > m_d0_max) return false;
  
  if (B != 0.0) {//straight-line track is OK
  
    const float R = std::sqrt(1 + A*A)/B; //signed radius in mm
 
    const float pT = std::abs(0.3*R); //asssuming uniform 2T field
 
    if (pT < min_pT) return false;
 
    if (pT > 5*min_pT) {//relatively high-pT track
 
      if (tau_ratio > 0.9*tau_ratio_cut) return false;
 
    }
    
  }
 
  return true;
 
}

Member Data Documentation

m_atlasId

const AtlasDetectorID * SeedingToolBase::m_atlasId = nullptr

protected

Definition at line 56 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

m_connectionFile

StringProperty SeedingToolBase::m_connectionFile {this, "ConnectionFileName", "binTables_ITK_RUN4.txt"}

protected

Definition at line 73 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "ConnectionFileName", "binTables_ITK_RUN4.txt"};

m_connector [1/2]

std::unique_ptr<GNN_FasTrackConnector> SeedingToolBase::m_connector = nullptr

protected

Definition at line 88 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

m_connector [2/2]

std::unique_ptr<GNNR3_FasTrackConnector> SeedingToolBase::m_connector = nullptr

protected

Definition at line 67 of file TrigInDetR3PattRecoTools/src/SeedingToolBase.h.

m_d0_max

FloatProperty SeedingToolBase::m_d0_max {this, "d0_max", 3.0}

protected

Definition at line 84 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "d0_max", 3.0};

m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

m_doubletFilterRZ

BooleanProperty SeedingToolBase::m_doubletFilterRZ {this, "Doublet_FilterRZ", true}

protected

Definition at line 66 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "Doublet_FilterRZ", true};

m_etaBinOverride

FloatProperty SeedingToolBase::m_etaBinOverride {this, "etaBin", 0.0f, "specify non-zero to override eta bin width from connection file (default 0.2 in createLinkingScheme.py)"}

protected

Definition at line 70 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "etaBin", 0.0f, "specify non-zero to override eta bin width from connection file (default 0.2 in createLinkingScheme.py)"};

m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

m_filter_phi

BooleanProperty SeedingToolBase::m_filter_phi {this, "DoPhiFiltering", true}

protected

Definition at line 60 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "DoPhiFiltering", true};

m_geo [1/2]

std::unique_ptr<const TrigFTF_GNN_Geometry> SeedingToolBase::m_geo = nullptr

protected

Definition at line 90 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

m_geo [2/2]

std::unique_ptr<const TrigFTF_GNNR3_Geometry> SeedingToolBase::m_geo = nullptr

protected

Definition at line 69 of file TrigInDetR3PattRecoTools/src/SeedingToolBase.h.

m_layerGeometry

std::vector< TrigInDetSiLayer > SeedingToolBase::m_layerGeometry

protected

Definition at line 89 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

m_layerNumberTool

ToolHandle< ITrigL2LayerNumberTool > SeedingToolBase::m_layerNumberTool {this, "layerNumberTool", "TrigL2LayerNumberToolITk"}

protected

Definition at line 54 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "layerNumberTool", "TrigL2LayerNumberToolITk"};

m_LRTmode

BooleanProperty SeedingToolBase::m_LRTmode {this, "LRTMode",false}

protected

Definition at line 62 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "LRTMode",false};

m_lutFile

StringProperty SeedingToolBase::m_lutFile {this, "MLpredictorLutFileName", "gbts_ml_pixel_barrel_loose.lut"}

protected

Definition at line 74 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "MLpredictorLutFileName", "gbts_ml_pixel_barrel_loose.lut"};

m_matchBeforeCreate

BooleanProperty SeedingToolBase::m_matchBeforeCreate {this, "MatchBeforeCreate", true}

protected

Definition at line 68 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "MatchBeforeCreate", true};

m_minPt

FloatProperty SeedingToolBase::m_minPt {this, "pTmin", 1000.0}

protected

Definition at line 69 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "pTmin", 1000.0};

m_mlLUT

std::vector<std::array<float, 5> > SeedingToolBase::m_mlLUT

protected

Definition at line 91 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

m_nBufferEdges

IntegerProperty SeedingToolBase::m_nBufferEdges {this, "BufferEdges", 200000}

protected

Definition at line 72 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "BufferEdges", 200000};

m_nMaxEdges

IntegerProperty SeedingToolBase::m_nMaxEdges {this, "MaxGraphEdges", 2000000}

protected

Definition at line 71 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "MaxGraphEdges", 2000000};

m_nMaxPhiSlice

UnsignedIntegerProperty SeedingToolBase::m_nMaxPhiSlice {this, "nMaxPhiSlice", 53}

protected

Definition at line 65 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "nMaxPhiSlice", 53};

m_phiSliceWidth

float SeedingToolBase::m_phiSliceWidth = 0.

protected

Definition at line 86 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

m_pixelId

const PixelID * SeedingToolBase::m_pixelId = nullptr

protected

Definition at line 58 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

m_sctId

const SCT_ID * SeedingToolBase::m_sctId = nullptr

protected

Definition at line 57 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

m_tau_ratio_corr

FloatProperty SeedingToolBase::m_tau_ratio_corr {this, "tau_ratio_correction", 0.006}

protected

Definition at line 82 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "tau_ratio_correction", 0.006};

m_tau_ratio_cut

FloatProperty SeedingToolBase::m_tau_ratio_cut {this, "tau_ratio_cut", 0.007}

protected

Definition at line 81 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "tau_ratio_cut", 0.007};

m_useAdaptiveCuts

BooleanProperty SeedingToolBase::m_useAdaptiveCuts {this, "UseAdaptiveCuts", true}

protected

Definition at line 79 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "UseAdaptiveCuts", true};

m_useBeamTilt

BooleanProperty SeedingToolBase::m_useBeamTilt {this, "UseBeamTilt", false}

protected

Definition at line 61 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "UseBeamTilt", false};

m_useEtaBinning

BooleanProperty SeedingToolBase::m_useEtaBinning {this, "UseEtaBinning", true}

protected

Definition at line 67 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "UseEtaBinning", true};

m_useGPUseedExtraction

BooleanProperty SeedingToolBase::m_useGPUseedExtraction {this, "UseGPUseedExtraction", true}

protected

Definition at line 76 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "UseGPUseedExtraction", true};

m_useML

BooleanProperty SeedingToolBase::m_useML {this, "useML", true}

protected

Definition at line 63 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "useML", true};

m_useOldTunings

BooleanProperty SeedingToolBase::m_useOldTunings {this, "UseOldTunings", false}

protected

Definition at line 77 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "UseOldTunings", false};

m_validateTriplets

BooleanProperty SeedingToolBase::m_validateTriplets {this, "ValidateTriplets", true}

protected

Definition at line 78 of file TrigInDetPattRecoTools/src/SeedingToolBase.h.

{this, "ValidateTriplets", true};

m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< AlgTool > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< AlgTool > >::m_vhka

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

---

The documentation for this class was generated from the following files:

• TrigInDetPattRecoTools/src/SeedingToolBase.h
• TrigInDetR3PattRecoTools/src/SeedingToolBase.h
• TrigInDetPattRecoTools/src/SeedingToolBase.cxx
• TrigInDetR3PattRecoTools/src/SeedingToolBase.cxx