81 float shift_x = vertex.x() - beamSpotHandle->beamTilt(0)*vertex.z();
82 float shift_y = vertex.y() - beamSpotHandle->beamTilt(1)*vertex.z();
84 std::unique_ptr<GNN_DataStorage> storage = std::make_unique<GNN_DataStorage>(*
m_geo,
m_mlLUT);
92 std::map<short, std::vector<IdentifierHash> > detIdMap;
104 sctSpacePointsContainer = sctHandle.
ptr();
106 std::vector<IdentifierHash> listOfSctIds;
107 m_regsel_sct->lookup(ctx)->HashIDList( internalRoI, listOfSctIds );
111 for(
const auto& hashId : listOfSctIds) {
113 short layerIndex = h2l->at(
static_cast<int>(hashId));
115 auto it = detIdMap.find(layerIndex);
116 if(it != detIdMap.end()) (*it).second.push_back(hashId);
118 std::vector<IdentifierHash> v = {hashId};
119 detIdMap.insert(std::make_pair(layerIndex,v));
133 pixelSpacePointsContainer = pixHandle.
ptr();
135 std::vector<IdentifierHash> listOfPixIds;
137 m_regsel_pix->lookup(ctx)->HashIDList( internalRoI, listOfPixIds );
141 for(
const auto& hashId : listOfPixIds) {
143 short layerIndex = h2l->at(
static_cast<int>(hashId));
145 auto it = detIdMap.find(layerIndex);
146 if(it != detIdMap.end()) (*it).second.push_back(hashId);
148 std::vector<IdentifierHash> v = {hashId};
149 detIdMap.insert(std::make_pair(layerIndex,v));
156 std::unique_ptr<GNN_DataStorage> storage = std::make_unique<GNN_DataStorage>(*
m_geo,
m_mlLUT);
158 std::vector<const Trk::SpacePoint*> vSP;
162 std::vector<std::vector<GNN_Node> > trigSpStorage[2];
168 for(
const auto& lColl : detIdMap) {
170 short layerIndex = lColl.first;
172 int layerKey =
m_geo->getTrigFTF_GNN_LayerByIndex(layerIndex)->m_layer.m_subdet;
174 bool isPixel = layerKey > 20000;
176 auto pCont = isPixel ? pixelSpacePointsContainer : sctSpacePointsContainer;
178 int contIdx= isPixel ? 0 : 1;
182 for(
const auto& idx : lColl.second) {
184 std::vector<GNN_Node>& tmpColl = trigSpStorage[contIdx].at(
static_cast<int>(idx));
186 auto input_coll = pCont->indexFindPtr(idx);
188 if(input_coll ==
nullptr)
continue;
192 nNewNodes += (isPixel) ? storage->loadPixelGraphNodes(layerIndex, tmpColl,
m_useML) : storage->loadStripGraphNodes(layerIndex, tmpColl);
195 if(isPixel) nPixels += nNewNodes;
196 else nStrips += nNewNodes;
202 storage->sortByPhi();
203 storage->initializeNodes(
m_useML);
206 std::vector<GNN_Edge> edgeStorage;
208 std::pair<int, int> graphStats =
buildTheGraph(internalRoI, storage, edgeStorage);
210 ATH_MSG_DEBUG(
"Created graph with "<<graphStats.first<<
" edges and "<<graphStats.second<<
" edge links");
215 if (graphStats.second == 0)
return seedStats;
217 int maxLevel =
runCCA(graphStats.first, edgeStorage);
219 ATH_MSG_DEBUG(
"Reached Level "<<maxLevel<<
" after GNN iterations");
221 std::vector<std::pair<float, std::vector<unsigned int> > > vSeedCandidates;
225 if (vSeedCandidates.empty())
return seedStats;
227 for(
const auto& seed : vSeedCandidates) {
229 unsigned int lastIdx = output.size();
230 output.emplace_back(seed.first);
232 for(
const auto& sp_idx : seed.second) {
233 output[lastIdx].addSpacePoint(vSP[sp_idx]);
242 std::vector<const Trk::SpacePoint*> vSP;
243 std::vector<short> vL;
248 std::unique_ptr<TrigAccel::DATA_EXPORT_BUFFER> dataBuffer = std::unique_ptr<TrigAccel::DATA_EXPORT_BUFFER>(
new TrigAccel::DATA_EXPORT_BUFFER(5000));
251 const size_t bufferOffset = 256;
252 size_t totalSize = bufferOffset+dataTypeSize;
257 unsigned int spIdx = 0;
263 for(
const auto& lColl : detIdMap) {
265 short layerIndex = lColl.first;
271 bool isPixel = layerKey > 20000;
274 auto pCont = isPixel ? pixelSpacePointsContainer : sctSpacePointsContainer;
285 nEtaBins += pL->
m_bins.size();
287 for(
auto b : pL->
m_bins) {
288 if(b > MaxEtaBin) MaxEtaBin = b;
291 for(
const auto& idx : lColl.second) {
293 auto input_coll = pCont->indexFindPtr(idx);
295 if(input_coll ==
nullptr)
continue;
297 for(
const auto sp : *input_coll) {
305 if(cw > 0.2)
continue;
310 vSP.emplace_back(
sp);
311 vL.emplace_back(layerIndex);
313 const auto& p =
sp->globalPosition();
315 float params[4] = {(float)(p.x() - shift_x), float(p.y() - shift_y), (float)(p.z()), cw};
317 memcpy(&pJobData->
m_params[sp_offset], ¶ms[0],
sizeof(params));
328 if (isPixel) nPixels += spIdx - pJobData->
m_layerInfo[4*nLayers];
329 else nStrips += spIdx - pJobData->
m_layerInfo[4*nLayers];
334 if(spIdx == 0)
return seedStats;
340 pJobData->
m_nMaxEdges =
static_cast<unsigned int>(7*spIdx);
346 for(
const auto& bg :
m_geo->bin_groups()) {
348 int bin1_idx = bg.first;
350 for(
const auto& bin2_idx : bg.second) {
361 const float ptCoeff = 0.29997*1.9972/2.0;
363 float tripletPtMin = 0.8*
m_minPt;
365 float maxCurv = ptCoeff/tripletPtMin;
367 const float pt_scale = 900/
m_minPt;
369 const float min_deltaPhi_low_dr = 0.002*pt_scale;
370 const float dphi_coeff_low_dr = 4.33e-4*pt_scale;
371 const float min_deltaPhi = 0.015*pt_scale;
372 const float dphi_coeff = 2.2e-4*pt_scale;
374 const float cut_dphi_max =
m_LRTmode ? 0.07 : 0.012;
375 const float cut_dcurv_max =
m_LRTmode ? 0.015 : 0.001;
376 const float cut_tau_ratio_max =
m_LRTmode ? 0.015f : 0.01;
380 const float maxOuterRadius =
m_LRTmode ? 1050.0 : 550.0;
381 const float minDeltaRadius = 2.0;
383 const float cut_zMinU = min_z0 + maxOuterRadius*internalRoI.
dzdrMinus();
384 const float cut_zMaxU = max_z0 + maxOuterRadius*internalRoI.
dzdrPlus();
386 const float maxKappa =
m_LRTmode ? 1.0*maxCurv : 0.9*maxCurv;
416 ATH_MSG_DEBUG(
"Loaded "<<nPixels<<
" Pixel Spacepoints and "<<nStrips<<
" Strip SpacePoints");
418 std::shared_ptr<TrigAccel::OffloadBuffer> pBuff = std::make_shared<TrigAccel::OffloadBuffer>(dataBuffer.get());
427 bool workSuccess = pWork->run();
433 std::shared_ptr<TrigAccel::OffloadBuffer> pOutput = pWork->getOutput();
440 if(pSeeds->
m_nSeeds == 0)
return seedStats;
442 for(
unsigned int seed=0; seed<pSeeds->
m_nSeeds; seed++) {
444 unsigned int lastIdx = output.size();
447 output[lastIdx].addSpacePoint(vSP[pSeeds->
m_seedsArray[seed].m_nodes[
node]]);
453 unsigned int nEdges = pGraph->
m_nEdges;
458 std::vector<GNN_Node> nodes;
460 nodes.reserve(vSP.size());
462 for(
unsigned int idx = 0;idx < vSP.size(); idx++) {
464 nodes.emplace_back(vL[idx]);
466 const auto& pos = vSP[idx]->globalPosition();
467 float xs = pos.x() - shift_x;
468 float ys = pos.y() - shift_y;
474 nodes[idx].m_r = std::sqrt(xs*xs + ys*ys);
476 nodes[idx].m_idx = idx;
480 std::vector<GNN_Edge> edgeStorage;
482 std::pair<int, int> graphStats(0,0);
484 edgeStorage.resize(nEdges);
486 unsigned int edgeSize = nMaxNei + 1 + 2;
488 for(
unsigned int idx=0;idx<nEdges;idx++) {
489 unsigned int pos = idx*edgeSize;
497 edgeStorage[idx].m_n1 = &nodes[node1Idx];
498 edgeStorage[idx].m_n2 = &nodes[node2Idx];
499 edgeStorage[idx].m_level = 1;
500 edgeStorage[idx].m_nNei = nNei;
501 for(
int k=0;k<nNei;k++) {
506 graphStats.first = nEdges;
507 graphStats.second = pGraph->
m_nLinks;
511 int maxLevel =
runCCA(graphStats.first, edgeStorage);
513 ATH_MSG_DEBUG(
"Reached Level "<<maxLevel<<
" after GNN iterations");
515 if(maxLevel < minLevel)
return seedStats;
517 std::vector<GNN_Edge*> vSeeds;
519 vSeeds.reserve(graphStats.first/2);
521 for(
int edgeIndex=0;edgeIndex<graphStats.first;edgeIndex++) {
522 GNN_Edge* pS = &(edgeStorage.at(edgeIndex));
524 if(pS->
m_level < minLevel)
continue;
526 vSeeds.push_back(pS);
529 if(vSeeds.empty())
return seedStats;
537 output.reserve(vSeeds.size());
539 for(
auto pS : vSeeds) {
541 if(pS->m_level == -1)
continue;
547 if(!
rs.m_initialized) {
551 if(
static_cast<int>(
rs.m_vs.size()) < minLevel)
continue;
553 std::vector<const GNN_Node*> vN;
555 for(std::vector<GNN_Edge*>::reverse_iterator sIt=
rs.m_vs.rbegin();sIt!=
rs.m_vs.rend();++sIt) {
557 (*sIt)->m_level = -1;
559 if(sIt ==
rs.m_vs.rbegin()) {
560 vN.push_back((*sIt)->m_n1);
562 vN.push_back((*sIt)->m_n2);
565 if(vN.size()<3)
continue;
567 unsigned int lastIdx = output.size();
568 output.emplace_back(
rs.m_J);
570 for(
const auto& n : vN) {
571 output[lastIdx].addSpacePoint(vSP[n->m_idx]);