Loading [MathJax]/extensions/tex2jax.js
ATLAS Offline Software
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
FPGATrackSimWindowExtensionTool.h
Go to the documentation of this file.
1 // Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
2 
3 #ifndef FPGATrackSimWINDOWEXTENSION_H
4 #define FPGATrackSimWINDOWEXTENSION_H
5 
13 #include "GaudiKernel/ServiceHandle.h"
15 
16 // add public header directory to algorithms for this?
18 
30 
31 #include <vector>
32 
33 class FPGATrackSimWindowExtensionTool : public extends <AthAlgTool, IFPGATrackSimTrackExtensionTool>
34 {
35  public:
36 
37  FPGATrackSimWindowExtensionTool(const std::string&, const std::string&, const IInterface*);
38 
39  virtual StatusCode initialize() override;
40 
41  virtual StatusCode extendTracks(const std::vector<std::shared_ptr<const FPGATrackSimHit>> & hits,
42  const std::vector<std::shared_ptr<const FPGATrackSimTrack>> & tracks,
43  std::vector<std::shared_ptr<const FPGATrackSimRoad>> & roads) override;
44 
45  // We don't have a "union" tool that sits in front of the extension tool, so this is needed here.
46  virtual StatusCode setupSlices(FPGATrackSimLogicalEventInputHeader *slicedHitHeader) override {
47  m_slicedHitHeader = slicedHitHeader;
48  return StatusCode::SUCCESS;
49  };
50 
51 
52  private:
53 
54  ServiceHandle<IFPGATrackSimBankSvc> m_FPGATrackSimBankSvc {this, "FPGATrackSimBankSvc", "FPGATrackSimBankSvc"};
55  ServiceHandle<IFPGATrackSimMappingSvc> m_FPGATrackSimMapping {this, "FPGATrackSimMappingSvc", "FPGATrackSimMappingSvc"};
56 
57  // We'll definitely need properties, but I don't know which ones.
58  Gaudi::Property<int> m_threshold { this, "threshold", 11, "Minimum number of hits to fire a road"};
59  Gaudi::Property<std::vector<float>> m_windows {this, "phiWindow", {}, "Default window settings for phi, must be size nlayers."};
60  Gaudi::Property<std::vector<float>> m_zwindows {this, "zWindow", {}, "Default window settings for z, must be size nlayers."};
61  Gaudi::Property<bool> m_fieldCorrection {this, "fieldCorrection", true, "Use magnetic field correction for Hough transform"};
62  Gaudi::Property<bool> m_idealGeoRoads {this, "IdealGeoRoads", true, "Do sector assignment of second stage roads"};
63 
64  // Options only needed for sector assignment.
65  // The eta pattern option here should probably be dropped, because we're not using it
66  // and supporting it requires having two sets of eta patterns (one for the first stage, one for the second)
67  // and then running the eta pattern filter a second time.
68  Gaudi::Property <bool> m_doRegionalMapping { this, "RegionalMapping", false, "Use the sub-region maps to define the sector"};
69  Gaudi::Property <bool> m_doEtaPatternConsts { this, "doEtaPatternConsts", false, "Whether to use the eta pattern tool for constant generation"};
70  Gaudi::Property <bool> m_useSpacePoints { this, "useSpacePoints", false, "Whether we are using spacepoints."};
71 
72  std::vector<FPGATrackSimRoad> m_roads;
73  //This is a map(dict python equivalent) of slice IDs that have a map of layer IDs in it. That map has a vector of hits associated with it
74  std::map<unsigned, std::map<unsigned, std::vector<std::shared_ptr<const FPGATrackSimHit>>>> m_phits_atLayer;
75  unsigned m_nLayers_1stStage = 0;
76  unsigned m_nLayers_2ndStage = 0;
77  unsigned m_maxMiss = 0;
78 
79  // Internal storage for the sliced hits (implemented as a LogicalEventInputHeader,
80  // so we can easily copy to the output ROOT file).
82 
83 };
84 
85 #endif
FPGATrackSimLogicalEventInputHeader
Definition: FPGATrackSimLogicalEventInputHeader.h:21
FPGATrackSimWindowExtensionTool::setupSlices
virtual StatusCode setupSlices(FPGATrackSimLogicalEventInputHeader *slicedHitHeader) override
Definition: FPGATrackSimWindowExtensionTool.h:46
TRTCalib_Extractor.hits
hits
Definition: TRTCalib_Extractor.py:35
FPGATrackSimWindowExtensionTool::m_fieldCorrection
Gaudi::Property< bool > m_fieldCorrection
Definition: FPGATrackSimWindowExtensionTool.h:61
FPGATrackSimWindowExtensionTool::m_useSpacePoints
Gaudi::Property< bool > m_useSpacePoints
Definition: FPGATrackSimWindowExtensionTool.h:70
FPGATrackSimPlaneMap.h
Maps physical layers to logical layers.
FPGATrackSimNNTrackTool.h
Utilize NN score to build track candidates.
FPGATrackSimTowerInputHeader.h
IFPGATrackSimMappingSvc.h
IFPGATrackSimTrackExtensionTool.h
Interface declaration for track extension tools (tracks + hits -> roads)
FPGATrackSimWindowExtensionTool::m_slicedHitHeader
FPGATrackSimLogicalEventInputHeader * m_slicedHitHeader
Definition: FPGATrackSimWindowExtensionTool.h:81
FPGATrackSimWindowExtensionTool::m_FPGATrackSimBankSvc
ServiceHandle< IFPGATrackSimBankSvc > m_FPGATrackSimBankSvc
Definition: FPGATrackSimWindowExtensionTool.h:54
FPGATrackSimRegionMap.h
Maps ITK module indices to FPGATrackSim regions.
FPGATrackSimWindowExtensionTool::extendTracks
virtual StatusCode extendTracks(const std::vector< std::shared_ptr< const FPGATrackSimHit >> &hits, const std::vector< std::shared_ptr< const FPGATrackSimTrack >> &tracks, std::vector< std::shared_ptr< const FPGATrackSimRoad >> &roads) override
Definition: FPGATrackSimWindowExtensionTool.cxx:54
FPGATrackSimWindowExtensionTool::m_doRegionalMapping
Gaudi::Property< bool > m_doRegionalMapping
Definition: FPGATrackSimWindowExtensionTool.h:68
FPGATrackSimWindowExtensionTool
Definition: FPGATrackSimWindowExtensionTool.h:34
FPGATrackSimWindowExtensionTool::initialize
virtual StatusCode initialize() override
Definition: FPGATrackSimWindowExtensionTool.cxx:30
FPGATrackSimWindowExtensionTool::m_roads
std::vector< FPGATrackSimRoad > m_roads
Definition: FPGATrackSimWindowExtensionTool.h:72
FPGATrackSimWindowExtensionTool::m_maxMiss
unsigned m_maxMiss
Definition: FPGATrackSimWindowExtensionTool.h:77
EL::StatusCode
::StatusCode StatusCode
StatusCode definition for legacy code.
Definition: PhysicsAnalysis/D3PDTools/EventLoop/EventLoop/StatusCode.h:22
AthAlgTool.h
FPGATrackSimWindowExtensionTool::m_windows
Gaudi::Property< std::vector< float > > m_windows
Definition: FPGATrackSimWindowExtensionTool.h:59
FPGATrackSimWindowExtensionTool::FPGATrackSimWindowExtensionTool
FPGATrackSimWindowExtensionTool(const std::string &, const std::string &, const IInterface *)
Definition: FPGATrackSimWindowExtensionTool.cxx:24
FPGATrackSimWindowExtensionTool::m_FPGATrackSimMapping
ServiceHandle< IFPGATrackSimMappingSvc > m_FPGATrackSimMapping
Definition: FPGATrackSimWindowExtensionTool.h:55
FPGATrackSimFunctions.h
FPGATrackSimHit.h
: FPGATrackSim-specific class to represent an hit in the detector.
FPGATrackSimWindowExtensionTool::m_nLayers_2ndStage
unsigned m_nLayers_2ndStage
Definition: FPGATrackSimWindowExtensionTool.h:76
FPGATrackSimWindowExtensionTool::m_nLayers_1stStage
unsigned m_nLayers_1stStage
Definition: FPGATrackSimWindowExtensionTool.h:75
FPGATrackSimWindowExtensionTool::m_phits_atLayer
std::map< unsigned, std::map< unsigned, std::vector< std::shared_ptr< const FPGATrackSimHit > > > > m_phits_atLayer
Definition: FPGATrackSimWindowExtensionTool.h:74
FPGATrackSimWindowExtensionTool::m_threshold
Gaudi::Property< int > m_threshold
Definition: FPGATrackSimWindowExtensionTool.h:58
FPGATrackSimWindowExtensionTool::m_idealGeoRoads
Gaudi::Property< bool > m_idealGeoRoads
Definition: FPGATrackSimWindowExtensionTool.h:62
FPGATrackSimWindowExtensionTool::m_zwindows
Gaudi::Property< std::vector< float > > m_zwindows
Definition: FPGATrackSimWindowExtensionTool.h:60
IFPGATrackSimBankSvc.h
FPGATrackSimWindowExtensionTool::m_doEtaPatternConsts
Gaudi::Property< bool > m_doEtaPatternConsts
Definition: FPGATrackSimWindowExtensionTool.h:69
FPGATrackSimRoad.h
Defines a class for roads.
FPGATrackSimTypes.h
FPGATrackSimTrack.h
ServiceHandle< IFPGATrackSimBankSvc >