ATLAS Offline Software
Loading...
Searching...
No Matches
TrigInDetTrackSeedingTool.cxx
Go to the documentation of this file.
1/*
2 Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
3*/
4
7
13
15
17
19
20#include "GNN_TrackingFilter.h"
21
22//for GPU offloading
23
26
28 const std::string& n,
29 const IInterface* p ) : SeedingToolBase(t,n,p)
30{
31
32}
33
35
37
38 ATH_CHECK(m_regsel_pix.retrieve());
39 ATH_CHECK(m_regsel_sct.retrieve());
40
41 ATH_CHECK(m_beamSpotKey.initialize());
42
43 if(m_useGPU) {//for GPU offloading
44 ATH_CHECK(m_accelSvc.retrieve());
45 ATH_CHECK(m_accelSvc->isReady());
46 }
47
49 ATH_MSG_FATAL("Both usePixelSpacePoints and useSctSpacePoints set to False. At least one needs to be True");
50 return StatusCode::FAILURE;
51 }
52
53 if (!m_useSctSpacePoints) ATH_MSG_INFO("Only using Pixel spacepoints => Pixel seeds only");
54
55 if (!m_usePixelSpacePoints) ATH_MSG_INFO("Only using SCT spacepoints => Strip seeds only");
56
57 if (m_usePixelSpacePoints && m_useSctSpacePoints) ATH_MSG_INFO("Using SCT and Pixel spacepoints");
58
60
62
63 ATH_MSG_INFO("TrigInDetTrackSeedingTool initialized with " << m_useGPU);
64
65 return StatusCode::SUCCESS;
66}
67
71
72
73TrigInDetTrackSeedingResult TrigInDetTrackSeedingTool::findSeeds(const IRoiDescriptor& internalRoI, std::vector<TrigInDetTracklet>& output, const EventContext& ctx) const {
74
76
77 output.clear();
78
80 const Amg::Vector3D &vertex = beamSpotHandle->beamPos();
81 float shift_x = vertex.x() - beamSpotHandle->beamTilt(0)*vertex.z();
82 float shift_y = vertex.y() - beamSpotHandle->beamTilt(1)*vertex.z();
83
84
85 int nPixels = 0;
86 int nStrips = 0;
87
88 const SpacePointContainer* sctSpacePointsContainer = nullptr;
89 const SpacePointContainer* pixelSpacePointsContainer = nullptr;
90
91 std::map<short, std::vector<IdentifierHash> > detIdMap;//relates global detector layer ID to constituent detector elements
92
94
96
97 if(!sctHandle.isValid()) {
98 ATH_MSG_WARNING("Invalid Strip Spacepoints handle: "<<sctHandle);
99 return seedStats;
100
101 }
102
103 sctSpacePointsContainer = sctHandle.ptr();
104
105 std::vector<IdentifierHash> listOfSctIds;
106 m_regsel_sct->lookup(ctx)->HashIDList( internalRoI, listOfSctIds );
107
108 const std::vector<short>* h2l = m_layerNumberTool->sctLayers();
109
110 for(const auto& hashId : listOfSctIds) {
111
112 short layerIndex = h2l->at(static_cast<int>(hashId));
113
114 auto it = detIdMap.find(layerIndex);
115 if(it != detIdMap.end()) (*it).second.push_back(hashId);
116 else {
117 std::vector<IdentifierHash> v = {hashId};
118 detIdMap.insert(std::make_pair(layerIndex,v));
119 }
120 }
121 }
122
124
126
127 if(!pixHandle.isValid()) {
128 ATH_MSG_WARNING("Invalid Pixel Spacepoints handle: "<<pixHandle);
129 return seedStats;
130 }
131
132 pixelSpacePointsContainer = pixHandle.ptr();
133
134 std::vector<IdentifierHash> listOfPixIds;
135
136 m_regsel_pix->lookup(ctx)->HashIDList( internalRoI, listOfPixIds );
137
138 const std::vector<short>* h2l = m_layerNumberTool->pixelLayers();
139
140 for(const auto& hashId : listOfPixIds) {
141
142 short layerIndex = h2l->at(static_cast<int>(hashId));
143
144 auto it = detIdMap.find(layerIndex);
145 if(it != detIdMap.end()) (*it).second.push_back(hashId);
146 else {
147 std::vector<IdentifierHash> v = {hashId};
148 detIdMap.insert(std::make_pair(layerIndex,v));
149 }
150 }
151 }
152
153 if(!m_useGPU) {
154
155 std::unique_ptr<GNN_DataStorage> storage = std::make_unique<GNN_DataStorage>(*m_geo, m_mlLUT);
156
157 std::vector<const Trk::SpacePoint*> vSP;
158
159 vSP.reserve(m_nMaxEdges);
160
161 std::vector<std::vector<GNN_Node> > trigSpStorage[2];
162
163 trigSpStorage[1].resize(m_layerNumberTool->sctLayers()->size());
164
165 trigSpStorage[0].resize(m_layerNumberTool->pixelLayers()->size());
166
167 for(const auto& lColl : detIdMap) {
168
169 short layerIndex = lColl.first;
170
171 int layerKey = m_geo->getTrigFTF_GNN_LayerByIndex(layerIndex)->m_layer.m_subdet;
172
173 bool isPixel = layerKey > 20000;
174
175 auto pCont = isPixel ? pixelSpacePointsContainer : sctSpacePointsContainer;
176
177 int contIdx= isPixel ? 0 : 1;
178
179 int nNewNodes = 0;
180
181 for(const auto& idx : lColl.second) {
182
183 std::vector<GNN_Node>& tmpColl = trigSpStorage[contIdx].at(static_cast<int>(idx));
184
185 auto input_coll = pCont->indexFindPtr(idx);
186
187 if(input_coll == nullptr) continue;
188
189 createGraphNodes(input_coll, tmpColl, vSP, layerIndex, shift_x, shift_y);//TO-DO: if(m_useBeamTilt) SP full transform functor
190
191 nNewNodes += (isPixel) ? storage->loadPixelGraphNodes(layerIndex, tmpColl, m_useML) : storage->loadStripGraphNodes(layerIndex, tmpColl);
192 }
193
194 if(isPixel) nPixels += nNewNodes;
195 else nStrips += nNewNodes;
196 }
197
198 seedStats.m_nPixelSPs = nPixels;
199 seedStats.m_nStripSPs = nStrips;
200
201 storage->sortByPhi();
202 storage->initializeNodes(m_useML);
203 storage->generatePhiIndexing(1.5*m_phiSliceWidth);
204
205 std::vector<GNN_Edge> edgeStorage;
206
207 std::pair<int, int> graphStats = buildTheGraph(internalRoI, storage, edgeStorage);
208
209 ATH_MSG_DEBUG("Created graph with "<<graphStats.first<<" edges and "<<graphStats.second<< " edge links");
210
211 seedStats.m_nGraphEdges = graphStats.first;
212 seedStats.m_nEdgeLinks = graphStats.second;
213
214 if (graphStats.second == 0) return seedStats;
215
216 int maxLevel = runCCA(graphStats.first, edgeStorage);
217
218 ATH_MSG_DEBUG("Reached Level "<<maxLevel<<" after GNN iterations");
219
220 std::vector<std::pair<float, std::vector<unsigned int> > > vSeedCandidates;
221
222 extractSeedsFromTheGraph(maxLevel, graphStats.first, vSP.size(), edgeStorage, vSeedCandidates);
223
224 if (vSeedCandidates.empty()) return seedStats;
225
226 for(const auto& seed : vSeedCandidates) {
227
228 unsigned int lastIdx = output.size();
229 output.emplace_back(seed.first);
230
231 for(const auto& sp_idx : seed.second) {
232 output[lastIdx].addSpacePoint(vSP[sp_idx]);
233 }
234 }
235
236 ATH_MSG_DEBUG("Found "<<output.size()<<" tracklets");
237 }
238 else {//GPU-accelerated graph building
239 //1. data export
240
241 std::vector<const Trk::SpacePoint*> vSP;
242 std::vector<short> vL;
243
246
247 std::unique_ptr<TrigAccel::DATA_EXPORT_BUFFER> dataBuffer = std::unique_ptr<TrigAccel::DATA_EXPORT_BUFFER>(new TrigAccel::DATA_EXPORT_BUFFER(5000));
248
249 size_t dataTypeSize = sizeof(TrigAccel::ITk::GRAPH_MAKING_INPUT_DATA);
250 const size_t bufferOffset = 256;
251 size_t totalSize = bufferOffset+dataTypeSize;//make room for the header
252 if(!dataBuffer->fit(totalSize)) dataBuffer->reallocate(totalSize);
253
254 TrigAccel::ITk::GRAPH_MAKING_INPUT_DATA* pJobData = reinterpret_cast<TrigAccel::ITk::GRAPH_MAKING_INPUT_DATA*>(dataBuffer->m_buffer + bufferOffset);
255
256 unsigned int spIdx = 0;
257 int sp_offset = 0;
258 int nLayers = 0;
259 int MaxEtaBin = 0;
260 int nEtaBins = 0;
261
262 for(const auto& lColl : detIdMap) {
263
264 short layerIndex = lColl.first;
265
266 const TrigFTF_GNN_Layer* pL = m_geo->getTrigFTF_GNN_LayerByIndex(layerIndex);
267
268 int layerKey = pL->m_layer.m_subdet;
269
270 bool isPixel = layerKey > 20000;
271 bool isBarrel = (pL->m_layer.m_type == 0);
272
273 auto pCont = isPixel ? pixelSpacePointsContainer : sctSpacePointsContainer;
274
275 pJobData->m_layerIdx[nLayers] = layerIndex;
276
277 pJobData->m_layerInfo[4*nLayers ] = spIdx;//layerInfo.x
278 pJobData->m_layerInfo[4*nLayers + 2] = pL->num_bins();//layerInfo.z
279 pJobData->m_layerInfo[4*nLayers + 3] = pL->m_bins[0];//layerInfo.w
280
281 pJobData->m_layerGeo[2*nLayers ] = pL->m_minEta;//layerGeo.x
282 pJobData->m_layerGeo[2*nLayers + 1] = pL->m_etaBin;//layerGeo.y
283
284 nEtaBins += pL->m_bins.size();
285
286 for(auto b : pL->m_bins) {
287 if(b > MaxEtaBin) MaxEtaBin = b;
288 }
289
290 for(const auto& idx : lColl.second) {
291
292 auto input_coll = pCont->indexFindPtr(idx);
293
294 if(input_coll == nullptr) continue;
295
296 for(const auto sp : *input_coll) {
297
298 float cw = -1.0;
299
300 const InDet::PixelCluster* pCL = dynamic_cast<const InDet::PixelCluster*>(sp->clusterList().first);
301 if(pCL != nullptr) {
302 cw = pCL->width().widthPhiRZ().y();
303 if(!isBarrel && m_useML) {
304 if(cw > 0.2) continue;
305 cw = -1.0;//set it to -1 so that it can be skipped later in the ML code on GPU
306 }
307 }
308
309 vSP.emplace_back(sp);
310 vL.emplace_back(layerIndex);
311
312 const auto& p = sp->globalPosition();
313
314 float params[4] = {(float)(p.x() - shift_x), float(p.y() - shift_y), (float)(p.z()), cw};
315
316 memcpy(&pJobData->m_params[sp_offset], &params[0], sizeof(params));
317
318 sp_offset += 4;
319 spIdx++;
321 }
323 }
324
325 pJobData->m_layerInfo[4*nLayers + 1] = spIdx;//layerInfo.y
326
327 if (isPixel) nPixels += spIdx - pJobData->m_layerInfo[4*nLayers];
328 else nStrips += spIdx - pJobData->m_layerInfo[4*nLayers];
329
330 nLayers++;
331 }
332
333 if(spIdx == 0) return seedStats;
334
335 pJobData->m_nSpacepoints = spIdx;
336 pJobData->m_nLayers = nLayers;
337 pJobData->m_nEtaBins = nEtaBins;
338 pJobData->m_maxEtaBin = MaxEtaBin;
339 pJobData->m_nMaxEdges = static_cast<unsigned int>(7*spIdx);
340
341 //load bin pairs
342
343 int pairIdx = 0;
344
345 for(const auto& bg : m_geo->bin_groups()) {//loop over bin groups
346
347 int bin1_idx = bg.first;
348
349 for(const auto& bin2_idx : bg.second) {
350 pJobData->m_bin_pairs[2*pairIdx ] = bin1_idx;
351 pJobData->m_bin_pairs[2*pairIdx+1] = bin2_idx;
352 pairIdx++;
353 }
354 }
355
356 pJobData->m_nBinPairs = pairIdx;
357
358 //add algorithm parameters
359
360 const float ptCoeff = 0.29997*1.9972/2.0;// ~0.3*B/2 - assuming nominal field of 2*T
361
362 float tripletPtMin = 0.8*m_minPt;//correction due to limited pT resolution
363
364 float maxCurv = ptCoeff/tripletPtMin;
365
366 const float pt_scale = 900/m_minPt;
367
368 const float min_deltaPhi_low_dr = 0.002*pt_scale;
369 const float dphi_coeff_low_dr = 4.33e-4*pt_scale;
370 const float min_deltaPhi = 0.015*pt_scale;
371 const float dphi_coeff = 2.2e-4*pt_scale;
372
373 const float cut_dphi_max = m_LRTmode ? 0.07 : 0.012;
374 const float cut_dcurv_max = m_LRTmode ? 0.015 : 0.001;
375 const float cut_tau_ratio_max = m_LRTmode ? 0.015f : 0.01;
376 const float min_z0 = m_LRTmode ? -600.0 : internalRoI.zedMinus();
377 const float max_z0 = m_LRTmode ? 600.0 : internalRoI.zedPlus();
378
379 const float maxOuterRadius = m_LRTmode ? 1050.0 : 550.0;
380 const float minDeltaRadius = 2.0;
381
382 const float cut_zMinU = min_z0 + maxOuterRadius*internalRoI.dzdrMinus();
383 const float cut_zMaxU = max_z0 + maxOuterRadius*internalRoI.dzdrPlus();
384
385 const float maxKappa = m_LRTmode ? 1.0*maxCurv : 0.9*maxCurv;
386
387 pJobData->m_algo_params[0] = min_deltaPhi;
388 pJobData->m_algo_params[1] = dphi_coeff;
389 pJobData->m_algo_params[2] = min_deltaPhi_low_dr;
390 pJobData->m_algo_params[3] = dphi_coeff_low_dr;
391 pJobData->m_algo_params[4] = minDeltaRadius;
392 pJobData->m_algo_params[5] = min_z0;
393 pJobData->m_algo_params[6] = max_z0;
394 pJobData->m_algo_params[7] = maxOuterRadius;
395 pJobData->m_algo_params[8] = cut_zMinU;
396 pJobData->m_algo_params[9] = cut_zMaxU;
397 pJobData->m_algo_params[10] = maxKappa;
398 pJobData->m_algo_params[11] = cut_dphi_max;
399 pJobData->m_algo_params[12] = cut_dcurv_max;
400 pJobData->m_algo_params[13] = cut_tau_ratio_max;
401
402 int minLevel = 3;//a triplet + 1 confirmation
403
404 if(m_LRTmode) {
405 minLevel = 2;//a triplet
406 }
407
408 pJobData->m_minLevel = minLevel;
409
411
412 seedStats.m_nPixelSPs = nPixels;
413 seedStats.m_nStripSPs = nStrips;
414
415 ATH_MSG_DEBUG("Loaded "<<nPixels<< " Pixel Spacepoints and "<<nStrips<< " Strip SpacePoints");
416
417 std::shared_ptr<TrigAccel::OffloadBuffer> pBuff = std::make_shared<TrigAccel::OffloadBuffer>(dataBuffer.get());
418
419 std::unique_ptr<TrigAccel::Work> pWork = std::unique_ptr<TrigAccel::Work>(m_accelSvc->createWork(TrigAccel::InDetJobControlCode::RUN_GBTS, std::move(pBuff)));
420
421 if(!pWork) {
422 ATH_MSG_WARNING("Failed to create a work item for task "<<TrigAccel::InDetJobControlCode::RUN_GBTS);
423 return seedStats;
424 }
425
426 bool workSuccess = pWork->run();
427 if(!workSuccess) {
428 ATH_MSG_WARNING("Work item failed to complete");
429 return seedStats;
430 }
431
432 std::shared_ptr<TrigAccel::OffloadBuffer> pOutput = pWork->getOutput();
433
434 TrigAccel::ITk::GRAPH_AND_SEEDS_OUTPUT* pGraphAndSeeds = reinterpret_cast<TrigAccel::ITk::GRAPH_AND_SEEDS_OUTPUT*>(pOutput->m_rawBuffer);
435 TrigAccel::ITk::COMPRESSED_GRAPH* pGraph = &pGraphAndSeeds->m_CompressedGraph;
436 TrigAccel::ITk::OUTPUT_SEEDS* pSeeds = &pGraphAndSeeds->m_OutputSeeds;
437
439 if(pSeeds->m_nSeeds == 0) return seedStats;
440 //converting tracklet into GBTS-CPU format
441 for(unsigned int seed=0; seed<pSeeds->m_nSeeds; seed++) {
442
443 unsigned int lastIdx = output.size();
444 output.emplace_back(pSeeds->m_seedsArray[seed].m_Q);
445 for(int node=0; node<pSeeds->m_seedsArray[seed].m_size; node++) {
446 output[lastIdx].addSpacePoint(vSP[pSeeds->m_seedsArray[seed].m_nodes[node]]);
447 }
448 }
449 }
450 else {
451
452 unsigned int nEdges = pGraph->m_nEdges;
453 unsigned int nMaxNei = pGraph->m_nMaxNeighbours;
454
455 //populating the edgeStorage
456
457 std::vector<GNN_Node> nodes;
458 nodes.reserve(vSP.size());
459
460 for(unsigned int idx = 0;idx < vSP.size(); idx++) {
461 nodes.emplace_back(static_cast<unsigned short>(vL[idx]));
462 auto &node = nodes.back();
463
464 const auto& pos = vSP[idx]->globalPosition();
465 float xs = pos.x() - shift_x;
466 float ys = pos.y() - shift_y;
467
468 node.m_x = xs;
469 node.m_y = ys;
470 node.m_z = pos.z();
471 node.m_r = std::sqrt(xs*xs + ys*ys);
472 node.m_idx = idx;
473
474 }
475
476 std::vector<GNN_Edge> edgeStorage;
477
478 std::pair<int, int> graphStats(0,0);
479
480 edgeStorage.resize(nEdges);
481
482 unsigned int edgeSize = nMaxNei + 1 + 2;//neigbours, num_neighbours, 2 nodes
483
484 for(unsigned int idx=0;idx<nEdges;idx++) {
485 unsigned int pos = idx*edgeSize;
486
487 int node1Idx = pGraph->m_graphArray[pos + TrigAccel::ITk::node1];
488 int node2Idx = pGraph->m_graphArray[pos + TrigAccel::ITk::node2];
489 int nNei = pGraph->m_graphArray[pos + TrigAccel::ITk::nNei];
490
491 if(nNei > N_SEG_CONNS) nNei = N_SEG_CONNS;
492 //coverity[WRAPPER_ESCAPE]
493 edgeStorage[idx].m_n1 = &nodes[node1Idx];
494 //coverity[WRAPPER_ESCAPE]
495 edgeStorage[idx].m_n2 = &nodes[node2Idx];
496 edgeStorage[idx].m_level = 1;
497 edgeStorage[idx].m_nNei = nNei;
498 for(int k=0;k<nNei;k++) {
499 edgeStorage[idx].m_vNei[k] = pGraph->m_graphArray[pos + TrigAccel::ITk::nei_idx_start + k];
500 }
501 }
502
503 graphStats.first = nEdges;
504 graphStats.second = pGraph->m_nLinks;
505
506 //run the rest of the GBTS workflow
507
508 int maxLevel = runCCA(graphStats.first, edgeStorage);
509
510 ATH_MSG_DEBUG("Reached Level "<<maxLevel<<" after GNN iterations");
511
512 if(maxLevel < minLevel) return seedStats;
513
514 std::vector<GNN_Edge*> vSeeds;
515
516 vSeeds.reserve(graphStats.first/2);
517
518 for(int edgeIndex=0;edgeIndex<graphStats.first;edgeIndex++) {
519 GNN_Edge* pS = &(edgeStorage.at(edgeIndex));
520
521 if(pS->m_level < minLevel) continue;
522
523 vSeeds.push_back(pS);
524 }
525
526 if(vSeeds.empty()) return seedStats;
527
528 std::sort(vSeeds.begin(), vSeeds.end(), GNN_Edge::CompareLevel());
529
530 //backtracking
531 //Local variable tFilter uses 320048 bytes of stack space;
532 //Total stack use for this function is 331508 bytes
533 //coverity[STACK_USE]
534 TrigFTF_GNN_TrackingFilter tFilter(m_layerGeometry, edgeStorage);
535
536 output.reserve(vSeeds.size());
537
538 for(auto pS : vSeeds) {
539
540 if(pS->m_level == -1) continue;
541
543
544 tFilter.followTrack(pS, rs);
545
546 if(!rs.m_initialized) {
547 continue;
548 }
549
550 if(static_cast<int>(rs.m_vs.size()) < minLevel) continue;
551
552 std::vector<const GNN_Node*> vN;
553
554 for(std::vector<GNN_Edge*>::reverse_iterator sIt=rs.m_vs.rbegin();sIt!=rs.m_vs.rend();++sIt) {
555
556 (*sIt)->m_level = -1;//mark as collected
557
558 if(sIt == rs.m_vs.rbegin()) {
559 vN.push_back((*sIt)->m_n1);
560 }
561 vN.push_back((*sIt)->m_n2);
562 }
563
564 if(vN.size()<3) continue;
565
566 unsigned int lastIdx = output.size();
567 output.emplace_back(rs.m_J);
568
569 for(const auto& n : vN) {
570 output[lastIdx].addSpacePoint(vSP[n->m_idx]);
571 }
572 }
573 }
574 ATH_MSG_DEBUG("Found "<<output.size()<<" tracklets");
575 }
576 return seedStats;
577}
578
579void TrigInDetTrackSeedingTool::createGraphNodes(const SpacePointCollection* spColl, std::vector<GNN_Node>& tmpColl, std::vector<const Trk::SpacePoint*>& vSP, unsigned short layer, float shift_x, float shift_y) const {
580
581 tmpColl.resize(spColl->size(), GNN_Node(layer));//all nodes belong to the same layer
582
583 int idx = 0;
584 int init_size = vSP.size();
585 for(const auto sp : *spColl) {
586 const auto& pos = sp->globalPosition();
587 vSP.emplace_back(sp);
588 float xs = pos.x() - shift_x;
589 float ys = pos.y() - shift_y;
590 float zs = pos.z();
591 tmpColl[idx].m_x = xs;
592 tmpColl[idx].m_y = ys;
593 tmpColl[idx].m_z = zs;
594 tmpColl[idx].m_r = std::sqrt(xs*xs + ys*ys);
595 tmpColl[idx].m_phi = std::atan2(ys,xs);
596 tmpColl[idx].m_idx = init_size + idx;
597
598 const InDet::PixelCluster* pCL = dynamic_cast<const InDet::PixelCluster*>(sp->clusterList().first);
599 if(pCL != nullptr){
600 tmpColl[idx].m_pcw = pCL->width().z();
601 tmpColl[idx].m_locPosY = pCL->localPosition().y();
602 }
603
604 idx++;
605 }
606}
#define ATH_CHECK
Evaluate an expression and check for errors.
#define ATH_MSG_DEBUG(x,...)
#define ATH_MSG_WARNING(x,...)
#define ATH_MSG_INFO(x,...)
#define ATH_MSG_FATAL(x,...)
This class provides an interface to generate or decode an identifier for the upper levels of the dete...
#define N_SEG_CONNS
static Double_t sp
static Double_t rs
static unsigned int totalSize(const MultiDimArray< T, N > &ht)
This is an Identifier helper class for the Pixel subdetector.
This is an Identifier helper class for the SCT subdetector.
size_type size() const noexcept
Returns the number of elements in the collection.
Describes the API of the Region of Ineterest geometry.
virtual double zedPlus() const =0
the zed and eta values at the most forward and most rear ends of the RoI
virtual double dzdrMinus() const =0
return the gradients
virtual double zedMinus() const =0
virtual double dzdrPlus() const =0
const InDet::SiWidth & width() const
return width class reference
double z() const
Definition SiWidth.h:131
const Amg::Vector2D & widthPhiRZ() const
Definition SiWidth.h:121
const_pointer_type ptr()
Dereference the pointer.
virtual bool isValid() override final
Can the handle be successfully dereferenced?
int runCCA(int, std::vector< GNN_Edge > &) const
ToolHandle< ITrigL2LayerNumberTool > m_layerNumberTool
std::vector< std::array< float, 5 > > m_mlLUT
std::vector< TrigInDetSiLayer > m_layerGeometry
void extractSeedsFromTheGraph(int, int, int, std::vector< GNN_Edge > &, std::vector< std::pair< float, std::vector< unsigned int > > > &) const
std::pair< int, int > buildTheGraph(const IRoiDescriptor &, const std::unique_ptr< GNN_DataStorage > &, std::vector< GNN_Edge > &) const
SeedingToolBase(const std::string &t, const std::string &n, const IInterface *p)
std::unique_ptr< const TrigFTF_GNN_Geometry > m_geo
const TrigInDetSiLayer & m_layer
std::vector< int > m_bins
int num_bins() const
void followTrack(TrigFTF_GNN_Edge *, TrigFTF_GNN_EdgeState &)
TrigInDetTrackSeedingTool(const std::string &, const std::string &, const IInterface *)
void createGraphNodes(const SpacePointCollection *, std::vector< GNN_Node > &, std::vector< const Trk::SpacePoint * > &, unsigned short, float, float) const
SG::ReadHandleKey< SpacePointContainer > m_pixelSpacePointsContainerKey
SG::ReadHandleKey< SpacePointContainer > m_sctSpacePointsContainerKey
virtual StatusCode finalize() override
ToolHandle< IRegSelTool > m_regsel_sct
ToolHandle< IRegSelTool > m_regsel_pix
region selector tools
ServiceHandle< ITrigInDetAccelerationSvc > m_accelSvc
virtual TrigInDetTrackSeedingResult findSeeds(const IRoiDescriptor &, std::vector< TrigInDetTracklet > &, const EventContext &) const override final
virtual StatusCode initialize() override
SG::ReadCondHandleKey< InDet::BeamSpotData > m_beamSpotKey
const Amg::Vector2D & localPosition() const
return the local position reference
Definition node.h:24
Eigen::Matrix< double, 3, 1 > Vector3D
struct TrigAccel::ITk::GraphMakingInputData GRAPH_MAKING_INPUT_DATA
static constexpr unsigned int GBTS_MAX_NUMBER_SPACEPOINTS
static constexpr unsigned char nNei
static constexpr unsigned char node1
static constexpr unsigned char node2
struct TrigAccel::ITk::OutputSeeds OUTPUT_SEEDS
struct TrigAccel::ITk::GraphAndSeedsOutput GRAPH_AND_SEEDS_OUTPUT
static constexpr unsigned char nei_idx_start
struct TrigAccel::ITk::CompressedGraph COMPRESSED_GRAPH
struct TrigAccel::DataExportBuffer DATA_EXPORT_BUFFER
void sort(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end)
Specialization of sort for DataVector/List.
std::unique_ptr< int[]> m_graphArray
int m_bin_pairs[2 *GBTS_MAX_ETA_BIN_PAIR]
int m_layerInfo[4 *GBTS_MAX_SILICON_LAYERS]
float m_layerGeo[2 *GBTS_MAX_SILICON_LAYERS]
int m_layerIdx[GBTS_MAX_SILICON_LAYERS]
float m_params[4 *GBTS_MAX_NUMBER_SPACEPOINTS]
std::unique_ptr< Tracklet[]> m_seedsArray