Loading [MathJax]/extensions/tex2jax.js
 |
ATLAS Offline Software
|
#include <Gbts2ActsSeedingTool.h>
|
| Gbts2ActsSeedingTool (const std::string &, const std::string &, const IInterface *) |
|
virtual | ~Gbts2ActsSeedingTool () |
|
virtual StatusCode | initialize () override |
|
virtual StatusCode | finalize () override |
|
virtual StatusCode | createSeeds (const EventContext &ctx, const Acts::SpacePointContainer< ActsTrk::SpacePointCollector, Acts::detail::RefHolder > &spContainer, const Acts::Vector3 &beamSpotPos, const Acts::Vector3 &bField, ActsTrk::SeedContainer &seedContainer) const override |
|
ServiceHandle< StoreGateSvc > & | evtStore () |
| The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc . More...
|
|
const ServiceHandle< StoreGateSvc > & | evtStore () const |
| The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc . More...
|
|
const ServiceHandle< StoreGateSvc > & | detStore () const |
| The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc . More...
|
|
virtual StatusCode | sysInitialize () override |
| Perform system initialization for an algorithm. More...
|
|
virtual StatusCode | sysStart () override |
| Handle START transition. More...
|
|
virtual std::vector< Gaudi::DataHandle * > | inputHandles () const override |
| Return this algorithm's input handles. More...
|
|
virtual std::vector< Gaudi::DataHandle * > | outputHandles () const override |
| Return this algorithm's output handles. More...
|
|
Gaudi::Details::PropertyBase & | declareProperty (Gaudi::Property< T > &t) |
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, SG::VarHandleKey &hndl, const std::string &doc, const SG::VarHandleKeyType &) |
| Declare a new Gaudi property. More...
|
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, SG::VarHandleBase &hndl, const std::string &doc, const SG::VarHandleType &) |
| Declare a new Gaudi property. More...
|
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, SG::VarHandleKeyArray &hndArr, const std::string &doc, const SG::VarHandleKeyArrayType &) |
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, T &property, const std::string &doc, const SG::NotHandleType &) |
| Declare a new Gaudi property. More...
|
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, T &property, const std::string &doc="none") |
| Declare a new Gaudi property. More...
|
|
void | updateVHKA (Gaudi::Details::PropertyBase &) |
|
MsgStream & | msg () const |
|
MsgStream & | msg (const MSG::Level lvl) const |
|
bool | msgLvl (const MSG::Level lvl) const |
|
| DeclareInterfaceID (ISeedingTool, 1, 0) |
|
virtual StatusCode | createSeeds (const EventContext &ctx, const Acts::SpacePointContainer< ActsTrk::SpacePointCollector, Acts::detail::RefHolder > &spContainer, const Acts::Vector3 &beamSpotPos, const Acts::Vector3 &bField, ActsTrk::SeedContainer &seedContainer) const =0 |
|
|
SG::ReadCondHandleKey< InDet::BeamSpotData > | m_beamSpotKey { this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot" } |
|
const std::vector< short > * | m_sct_h2l {nullptr} |
|
const std::vector< short > * | m_pix_h2l {nullptr} |
|
std::vector< bool > | m_are_pixels |
|
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} |
|
FloatProperty | m_minPt {this, "pTmin", 1000.0} |
|
IntegerProperty | m_nMaxEdges {this, "MaxGraphEdges", 2000000} |
|
StringProperty | m_connectionFile {this, "ConnectionFileName", "binTables_ITK_RUN4.txt"} |
|
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 |
|
Definition at line 14 of file Gbts2ActsSeedingTool.h.
◆ GNN_DataStorage
◆ GNN_Edge
◆ GNN_Node
◆ StoreGateSvc_t
◆ Gbts2ActsSeedingTool()
Gbts2ActsSeedingTool::Gbts2ActsSeedingTool |
( |
const std::string & |
t, |
|
|
const std::string & |
n, |
|
|
const IInterface * |
p |
|
) |
| |
◆ ~Gbts2ActsSeedingTool()
virtual Gbts2ActsSeedingTool::~Gbts2ActsSeedingTool |
( |
| ) |
|
|
inlinevirtual |
◆ buildTheGraph()
Definition at line 68 of file SeedingToolBase.cxx.
72 const float cut_dphi_max =
m_LRTmode ? 0.07 : 0.012;
73 const float cut_dcurv_max =
m_LRTmode ? 0.015 : 0.001;
74 const float cut_tau_ratio_max =
m_LRTmode ? 0.015 : 0.007;
77 const float min_deltaPhi =
m_LRTmode ? 0.01f : 0.001f;
79 const float maxOuterRadius =
m_LRTmode ? 1050.0 : 550.0;
81 const float cut_zMinU = min_z0 + maxOuterRadius*roi.
dzdrMinus();
82 const float cut_zMaxU = max_z0 + maxOuterRadius*roi.
dzdrPlus();
84 const float ptCoeff = 0.29997*1.9972/2.0;
86 float tripletPtMin = 0.8*
m_minPt;
88 float maxCurv = ptCoeff/tripletPtMin;
90 const float maxKappa_high_eta =
m_LRTmode ? 1.0*maxCurv : std::sqrt(0.8)*maxCurv;
91 const float maxKappa_low_eta =
m_LRTmode ? 1.0*maxCurv : std::sqrt(0.6)*maxCurv;
92 const float dphi_coeff =
m_LRTmode ? 1.0*maxCurv : 0.68*maxCurv;
94 const float minDeltaRadius = 2.0;
98 unsigned int nConnections = 0;
104 for(
const auto&
bg :
m_geo->bin_groups()) {
108 if(B1.
empty())
continue;
112 for(
const auto& b2_idx :
bg.second) {
116 if(B2.
empty())
continue;
121 deltaPhi = min_deltaPhi + dphi_coeff*std::fabs(rb2-rb1);
124 unsigned int first_it = 0;
126 for(
unsigned int n1Idx = 0;n1Idx<B1.
m_vn.size();n1Idx++) {
128 std::vector<unsigned int>& v1In = B1.
m_in[n1Idx];
132 const std::array<float, 5>& n1pars = B1.
m_params[n1Idx];
134 float phi1 = n1pars[2];
135 float r1 = n1pars[3];
136 float z1 = n1pars[4];
143 for(
unsigned int n2PhiIdx = first_it; n2PhiIdx<B2.
m_vPhiNodes.size();n2PhiIdx++) {
151 if(phi2 > maxPhi)
break;
153 unsigned int n2Idx = B2.
m_vPhiNodes[n2PhiIdx].second;
155 const std::vector<unsigned int>& v2In = B2.
m_in[n2Idx];
159 const std::array<float, 5>& n2pars = B2.
m_params[n2Idx];
161 float r2 = n2pars[3];
165 if(
dr < minDeltaRadius) {
169 float z2 = n2pars[4];
173 float ftau = std::fabs(tau);
178 if(ftau < n1pars[0])
continue;
179 if(ftau > n1pars[1])
continue;
181 if(ftau < n2pars[0])
continue;
182 if(ftau > n2pars[1])
continue;
186 float z0 = z1 -
r1*tau;
188 if(z0 < min_z0 || z0 > max_z0)
continue;
190 float zouter =
z0 + maxOuterRadius*tau;
192 if(zouter < cut_zMinU || zouter > cut_zMaxU)
continue;
195 float curv = (phi2-phi1)/
dr;
196 float abs_curv = std::abs(curv);
199 if(abs_curv > maxKappa_low_eta) {
204 if(abs_curv > maxKappa_high_eta) {
211 float exp_eta = std::sqrt(1+tau*tau)-tau;
213 bool isGood = v2In.size() <= 2;
217 float uat_1 = 1.0f/exp_eta;
219 for(
const auto& n2_in_idx : v2In) {
221 float tau2 = edgeStorage.at(n2_in_idx).m_p[0];
222 float tau_ratio = tau2*uat_1 - 1.0f;
224 if(std::fabs(tau_ratio) > cut_tau_ratio_max){
236 float dPhi2 = curv*
r2;
237 float dPhi1 = curv*
r1;
241 edgeStorage.emplace_back(B1.
m_vn[n1Idx], B2.
m_vn[n2Idx], exp_eta, curv, phi1 + dPhi1);
245 int outEdgeIdx = nEdges;
247 float uat_2 = 1/exp_eta;
248 float Phi2 = phi2 + dPhi2;
251 for(
const auto& inEdgeIdx : v2In) {
257 float tau_ratio = pS->
m_p[0]*uat_2 - 1.0f;
259 if(std::abs(tau_ratio) > cut_tau_ratio_max){
263 float dPhi = Phi2 - pS->
m_p[2];
268 if(dPhi < -cut_dphi_max || dPhi > cut_dphi_max) {
272 float dcurv = curv2 - pS->
m_p[1];
274 if(dcurv < -cut_dcurv_max || dcurv > cut_dcurv_max) {
290 return std::make_pair(nEdges, nConnections);
◆ createSeeds() [1/2]
◆ createSeeds() [2/2]
Definition at line 35 of file Gbts2ActsSeedingTool.cxx.
37 std::unique_ptr<GNN_DataStorage> storage = std::make_unique<GNN_DataStorage>(*
m_geo);
43 float shift_x =
vertex.x() - beamSpotHandle->beamTilt(0)*
vertex.z();
44 float shift_y =
vertex.y() - beamSpotHandle->beamTilt(1)*
vertex.z();
46 std::vector<std::vector<GNN_Node> > node_storage;
49 for(
auto&
v : node_storage)
v.reserve(100000);
51 unsigned int nPixelLoaded = 0;
52 unsigned int nStripLoaded = 0;
54 for(
size_t idx=0;
idx<spContainer.size();
idx++){
55 const auto & sp = spContainer.at(
idx);
56 const auto & extSP = sp.externalSpacePoint();
57 const std::vector<xAOD::DetectorIDHashType>& elementlist = extSP.elementIdList() ;
59 bool isPixel(elementlist.size() == 1);
67 const auto&
pos = extSP.globalPosition();
84 for(
size_t l = 0;
l < node_storage.size();
l++) {
86 const std::vector<GNN_Node>& nodes = node_storage[
l];
88 if(nodes.size() == 0)
continue;
96 ATH_MSG_DEBUG(
"Loaded "<<nPixelLoaded<<
" pixel spacepoints and "<<nStripLoaded<<
" strip spacepoints");
104 std::vector<GNN_Edge> edgeStorage;
108 std::pair<int, int> graphStats =
buildTheGraph(fakeRoi, storage, edgeStorage);
110 ATH_MSG_DEBUG(
"Created graph with "<<graphStats.first<<
" edges and "<<graphStats.second<<
" edge links");
112 int maxLevel =
runCCA(graphStats.first, edgeStorage);
114 ATH_MSG_DEBUG(
"Reached Level "<<maxLevel<<
" after GNN iterations");
122 if(maxLevel < minLevel)
return StatusCode::SUCCESS;
124 std::vector<GNN_Edge*> vSeeds;
126 vSeeds.reserve(graphStats.first/2);
128 for(
int edgeIndex = 0; edgeIndex < graphStats.first; edgeIndex++) {
130 GNN_Edge* pS = &(edgeStorage.at(edgeIndex));
132 if(pS->m_level < minLevel)
continue;
134 vSeeds.push_back(pS);
137 if(vSeeds.empty())
return StatusCode::SUCCESS;
139 std::sort(vSeeds.begin(), vSeeds.end(), GNN_Edge::CompareLevel());
145 for(
auto pS : vSeeds) {
146 if(pS->m_level == -1)
continue;
150 tFilter.followTrack(pS, rs);
152 if(!rs.m_initialized) {
156 if(
static_cast<int>(rs.m_vs.size()) < minLevel)
continue;
158 std::vector<const GNN_Node*> vN;
160 for(std::vector<GNN_Edge*>::reverse_iterator sIt=rs.m_vs.rbegin();sIt!=rs.m_vs.rend();++sIt) {
161 (*sIt)->m_level = -1;
163 if(sIt == rs.m_vs.rbegin()) {
164 vN.push_back((*sIt)->m_n1);
167 vN.push_back((*sIt)->m_n2);
171 if(vN.size()<3)
continue;
173 const auto & sp1 = spContainer.at(vN[0]->sp_idx()).externalSpacePoint();
174 const auto & sp2 = spContainer.at(vN[1]->sp_idx()).externalSpacePoint();
175 const auto & sp3 = spContainer.at(vN[2]->sp_idx()).externalSpacePoint();
179 std::unique_ptr<ActsTrk::Seed> to_add = std::make_unique<ActsTrk::Seed>(sp1, sp2, sp3);
181 seedContainer.
push_back(std::move(to_add));
186 return StatusCode::SUCCESS;
◆ declareGaudiProperty() [1/4]
specialization for handling Gaudi::Property<SG::VarHandleKeyArray>
Definition at line 170 of file AthCommonDataStore.h.
175 hndl.documentation());
◆ declareGaudiProperty() [2/4]
specialization for handling Gaudi::Property<SG::VarHandleKey>
Definition at line 156 of file AthCommonDataStore.h.
161 hndl.documentation());
◆ declareGaudiProperty() [3/4]
specialization for handling Gaudi::Property<SG::VarHandleBase>
Definition at line 184 of file AthCommonDataStore.h.
189 hndl.documentation());
◆ declareGaudiProperty() [4/4]
◆ DeclareInterfaceID()
ActsTrk::ISeedingTool::DeclareInterfaceID |
( |
ISeedingTool |
, |
|
|
1 |
, |
|
|
0 |
|
|
) |
| |
|
inherited |
◆ declareProperty() [1/6]
Declare a new Gaudi property.
- Parameters
-
name | Name of the property. |
hndl | Object holding the property value. |
doc | Documentation string for the property. |
This is the version for types that derive from SG::VarHandleBase
. The property value object is put on the input and output lists as appropriate; then we forward to the base class.
Definition at line 245 of file AthCommonDataStore.h.
250 this->declare(hndl.
vhKey());
251 hndl.
vhKey().setOwner(
this);
253 return PBASE::declareProperty(
name,hndl,
doc);
◆ declareProperty() [2/6]
Declare a new Gaudi property.
- Parameters
-
name | Name of the property. |
hndl | Object holding the property value. |
doc | Documentation string for the property. |
This is the version for types that derive from SG::VarHandleKey
. The property value object is put on the input and output lists as appropriate; then we forward to the base class.
Definition at line 221 of file AthCommonDataStore.h.
229 return PBASE::declareProperty(
name,hndl,
doc);
◆ declareProperty() [3/6]
◆ declareProperty() [4/6]
Declare a new Gaudi property.
- Parameters
-
name | Name of the property. |
property | Object holding the property value. |
doc | Documentation string for the property. |
This is the generic version, for types that do not derive from SG::VarHandleKey
. It just forwards to the base class version of declareProperty
.
Definition at line 333 of file AthCommonDataStore.h.
338 return PBASE::declareProperty(
name, property,
doc);
◆ declareProperty() [5/6]
Declare a new Gaudi property.
- Parameters
-
name | Name of the property. |
property | Object holding the property value. |
doc | Documentation string for the property. |
This dispatches to either the generic declareProperty
or the one for VarHandle/Key/KeyArray.
Definition at line 352 of file AthCommonDataStore.h.
◆ declareProperty() [6/6]
◆ detStore()
◆ evtStore() [1/2]
◆ evtStore() [2/2]
◆ extraDeps_update_handler()
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()
StatusCode Gbts2ActsSeedingTool::finalize |
( |
| ) |
|
|
overridevirtual |
◆ initialize()
StatusCode Gbts2ActsSeedingTool::initialize |
( |
| ) |
|
|
overridevirtual |
◆ inputHandles()
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() [1/2]
◆ msg() [2/2]
◆ msgLvl()
◆ outputHandles()
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()
◆ renounceArray()
◆ runCCA()
int SeedingToolBase::runCCA |
( |
int |
nEdges, |
|
|
std::vector< GNN_Edge > & |
edgeStorage |
|
) |
| const |
|
protectedinherited |
Definition at line 293 of file SeedingToolBase.cxx.
295 const int maxIter = 15;
301 std::vector<TrigFTF_GNN_Edge*> v_old;
303 for(
int edgeIndex=0;edgeIndex<nEdges;edgeIndex++) {
306 if(pS->
m_nNei == 0)
continue;
311 for(;iter<maxIter;iter++) {
314 std::vector<TrigFTF_GNN_Edge*> v_new;
316 v_new.reserve(v_old.size());
318 for(
auto pS : v_old) {
322 for(
int nIdx=0;nIdx<pS->
m_nNei;nIdx++) {
324 unsigned int nextEdgeIdx = pS->
m_vNei[nIdx];
342 for(
auto pS : v_new) {
346 if(maxLevel < pS->m_level) maxLevel = pS->
m_level;
350 if(nChanges == 0)
break;
353 v_old = std::move(v_new);
◆ sysInitialize()
◆ sysStart()
Handle START transition.
We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.
◆ updateVHKA()
◆ m_are_pixels
std::vector<bool> Gbts2ActsSeedingTool::m_are_pixels |
|
protected |
◆ m_atlasId
◆ m_beamSpotKey
◆ m_connectionFile
StringProperty SeedingToolBase::m_connectionFile {this, "ConnectionFileName", "binTables_ITK_RUN4.txt"} |
|
protectedinherited |
◆ m_connector
◆ m_detStore
◆ m_doubletFilterRZ
BooleanProperty SeedingToolBase::m_doubletFilterRZ {this, "Doublet_FilterRZ", true} |
|
protectedinherited |
◆ m_evtStore
◆ m_filter_phi
BooleanProperty SeedingToolBase::m_filter_phi {this, "DoPhiFiltering", true} |
|
protectedinherited |
◆ m_geo
◆ m_layerGeometry
◆ m_layerNumberTool
◆ m_LRTmode
BooleanProperty SeedingToolBase::m_LRTmode {this, "LRTMode",false} |
|
protectedinherited |
◆ m_minPt
FloatProperty SeedingToolBase::m_minPt {this, "pTmin", 1000.0} |
|
protectedinherited |
◆ m_nMaxEdges
IntegerProperty SeedingToolBase::m_nMaxEdges {this, "MaxGraphEdges", 2000000} |
|
protectedinherited |
◆ m_nMaxPhiSlice
UnsignedIntegerProperty SeedingToolBase::m_nMaxPhiSlice {this, "nMaxPhiSlice", 53} |
|
protectedinherited |
◆ m_phiSliceWidth
float SeedingToolBase::m_phiSliceWidth = 0. |
|
protectedinherited |
◆ m_pix_h2l
const std::vector<short>* Gbts2ActsSeedingTool::m_pix_h2l {nullptr} |
|
protected |
◆ m_pixelId
◆ m_sct_h2l
const std::vector<short>* Gbts2ActsSeedingTool::m_sct_h2l {nullptr} |
|
protected |
◆ m_sctId
◆ m_useBeamTilt
BooleanProperty SeedingToolBase::m_useBeamTilt {this, "UseBeamTilt", false} |
|
protectedinherited |
◆ m_useEtaBinning
BooleanProperty SeedingToolBase::m_useEtaBinning {this, "UseEtaBinning", true} |
|
protectedinherited |
◆ m_useML
BooleanProperty SeedingToolBase::m_useML {this, "useML", true} |
|
protectedinherited |
◆ m_varHandleArraysDeclared
◆ m_vhka
The documentation for this class was generated from the following files:
std::vector< std::vector< unsigned int > > m_in
float widthInEta() const
Returns the width of the cluster in phi (x) and eta (y) directions, respectively.
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T > &t)
setSAddress setEtaMS setDirPhiMS setDirZMS setBarrelRadius setEndcapAlpha setEndcapRadius setInterceptInner setEtaMap setEtaBin setIsTgcFailure setDeltaPt deltaPhi
unsigned int m_vNei[N_SEG_CONNS]
StoreGateSvc_t m_evtStore
Pointer to StoreGate (event store by default)
std::vector< SG::VarHandleKeyArray * > m_vhka
nope - should be used for standalone also, perhaps need to protect the class def bits #ifndef XAOD_AN...
virtual double dzdrMinus() const =0
return the gradients
virtual double dzdrPlus() const =0
virtual void setOwner(IDataHandleHolder *o)=0
void generatePhiIndexing(float)
std::vector< std::pair< float, unsigned int > > m_vPhiNodes
bool dPhi(const xAOD::TauJet &tau, const xAOD::TauTrack &track, double &out)
StoreGateSvc_t m_detStore
Pointer to StoreGate (detector store by default)
TrigFTF_GNN_EtaBin & getEtaBin(int idx)
int loadPixelGraphNodes(short, const std::vector< TrigFTF_GNN_Node > &, bool)
float getMaxBinRadius() const
virtual void renounce()=0
std::conditional< std::is_base_of< SG::VarHandleKeyArray, T >::value, VarHandleKeyArrayType, type2 >::type type
void initializeNodes(bool)
value_type push_back(value_type pElem)
Add an element to the end of the collection.
StatusCode initialize(bool used=true)
Eigen::Matrix< double, 3, 1 > Vector3D
virtual double zedPlus() const =0
the zed and eta values at the most forward and most rear ends of the RoI
void sort(typename std::reverse_iterator< DataModel_detail::iterator< DVL > > beg, typename std::reverse_iterator< DataModel_detail::iterator< DVL > > end, const Compare &comp)
Specialization of sort for DataVector/List.
SG::VarHandleKey & vhKey()
Return a non-const reference to the HandleKey.
int loadStripGraphNodes(short, const std::vector< TrigFTF_GNN_Node > &)
virtual double zedMinus() const =0
Athena::TPCnvVers::Current TrigRoiDescriptor
std::vector< const TrigFTF_GNN_Node * > m_vn
constexpr int pow(int base, int exp) noexcept
size_type size() const noexcept
Returns the number of elements in the collection.
Gaudi::Details::PropertyBase & declareGaudiProperty(Gaudi::Property< T > &hndl, const SG::VarHandleKeyType &)
specialization for handling Gaudi::Property<SG::VarHandleKey>
float getMinBinRadius() const
std::vector< std::array< float, 5 > > m_params