TrigInDetR3PattRecoTools/src/SeedingToolBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRIGINDETPATTRECOTOOLS_SEEDINGTOOLBASE_H
#define TRIGINDETPATTRECOTOOLS_SEEDINGTOOLBASE_H
 
#include "GaudiKernel/ToolHandle.h"
#include "TrigInDetToolInterfaces/ITrigInDetTrackSeedingTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "StoreGate/ReadHandleKey.h"
#include <string>
#include <vector>
 
#include "IRegionSelector/IRegSelTool.h"
#include "TrigInDetToolInterfaces/ITrigL2LayerNumberTool.h"
 
#include "GNNR3_FasTrackConnector.h"
#include "GNNR3_Geometry.h"
#include "GNNR3_DataStorage.h"
 
class AtlasDetectorID;
class SCT_ID;
class PixelID;
 
class SeedingToolBase: public AthAlgTool {
  public:
  SeedingToolBase(const std::string& t,const std::string& n,const IInterface* p): AthAlgTool(t,n,p){}
  
 protected:
 
  typedef TrigFTF_GNNR3_Node GNNR3_Node;
  typedef TrigFTF_GNNR3_DataStorage GNNR3_DataStorage;
  typedef TrigFTF_GNNR3_Edge GNNR3_Edge;
 
  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;
  
  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", false};
  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};
  StringProperty  m_connectionFile{this, "ConnectionFileName", "binTables_ITK_RUN4.txt"};
 
  BooleanProperty m_useGPUseedExtraction{this, "UseGPUseedExtraction", true};
 
  float m_phiSliceWidth = 0.;
 
  std::unique_ptr<GNNR3_FasTrackConnector> m_connector = nullptr;
  std::vector<TrigInDetSiLayer> m_layerGeometry;
  std::unique_ptr<const TrigFTF_GNNR3_Geometry> m_geo = nullptr;
};
 
#endif