ATLAS Offline Software
Loading...
Searching...
No Matches
Public Member Functions | Private Member Functions | Private Attributes | List of all members
TrigITkAccelerationTool Class Reference

#include <TrigITkAccelerationTool.h>

Inheritance diagram for TrigITkAccelerationTool:
Collaboration diagram for TrigITkAccelerationTool:

Public Member Functions

 TrigITkAccelerationTool (const std::string &, const std::string &, const IInterface *)
virtual StatusCode initialize () override
virtual size_t exportSeedMakingJob (const TrigCombinatorialSettings &, const IRoiDescriptor *, const std::vector< TrigSiSpacePointBase > &, TrigAccel::DATA_EXPORT_BUFFER &) const override
virtual int extractTripletsFromOutput (std::shared_ptr< TrigAccel::OffloadBuffer >, const std::vector< TrigSiSpacePointBase > &, std::vector< TrigInDetTriplet > &) const override

Private Member Functions

float getCovR (bool isBarrel) const
float getCovZ (bool isBarrel) const

Private Attributes

ServiceHandle< ITrigInDetAccelerationSvc > m_accelSvc

Detailed Description

Definition at line 16 of file TrigITkAccelerationTool.h.

Constructor & Destructor Documentation

◆ TrigITkAccelerationTool()

TrigITkAccelerationTool::TrigITkAccelerationTool ( const std::string & t,
const std::string & n,
const IInterface * p )

Definition at line 11 of file TrigITkAccelerationTool.cxx.

13 : base_class(t,n,p),
14 m_accelSvc("TrigInDetAccelerationSvc",this->name()) {
15
16 declareInterface< ITrigInDetAccelerationTool >( this );
17}
TrigITkAccelerationTool::m_accelSvc
ServiceHandle< ITrigInDetAccelerationSvc > m_accelSvc
Definition TrigITkAccelerationTool.h:31

Member Function Documentation

◆ exportSeedMakingJob()

size_t TrigITkAccelerationTool::exportSeedMakingJob ( const TrigCombinatorialSettings & tcs,
const IRoiDescriptor * roi,
const std::vector< TrigSiSpacePointBase > & vsp,
TrigAccel::DATA_EXPORT_BUFFER & output ) const
overridevirtual

Definition at line 25 of file TrigITkAccelerationTool.cxx.

25 {
26
27 typedef struct SpIndexPair {
28 public:
29
30 struct compareZ {
31 public:
32 bool operator()(const std::pair<int, const TrigSiSpacePointBase*>& p1, const std::pair<int, const TrigSiSpacePointBase*>& p2) {
33 return p1.second->z() < p2.second->z();
34 }
35 };
36
37 struct compareR {
38 public:
39 bool operator()(const std::pair<int, const TrigSiSpacePointBase*>& p1, const std::pair<int, const TrigSiSpacePointBase*>& p2) {
40 return p1.second->r() < p2.second->r();
41 }
42 };
43
44 } SP_INDEX_PAIR;
45
46
47 //0. get InDet geometry information
48
49 const std::vector<short>& pixelLayers = m_accelSvc->getLayerInformation(1);
50 const std::vector<short>& layerTypes = m_accelSvc->getLayerInformation(0);
51
52
53 //1. check buffer size
54
55 size_t dataTypeSize = sizeof(TrigAccel::ITk::SEED_MAKING_JOB);
56 const size_t bufferOffset = 256;
57 size_t totalSize = bufferOffset+dataTypeSize;//make room for the header
58 if(!output.fit(totalSize)) output.reallocate(totalSize);
59
60 TrigAccel::ITk::SEED_MAKING_JOB* pJ = reinterpret_cast<TrigAccel::ITk::SEED_MAKING_JOB*>(output.m_buffer+bufferOffset);
61
62 // cppcheck-suppress memsetClassFloat; deliberate
63 memset(pJ,0,dataTypeSize);
64
65 TrigAccel::ITk::SEED_FINDER_SETTINGS& sfs = pJ->m_settings;
66
67 sfs.m_maxBarrelPix = tcs.m_maxBarrelPix;
68 sfs.m_minEndcapPix = tcs.m_minEndcapPix;
69 sfs.m_maxEndcapPix = tcs.m_maxEndcapPix;
70 sfs.m_maxSiliconLayer = tcs.m_maxSiliconLayer;
71 sfs.m_maxEta = roi->etaPlus();
72 sfs.m_minDoubletLength = tcs.m_seedRadBinWidth;
73 sfs.m_maxDoubletLength = tcs.m_doublet_dR_Max;
74
75 sfs.m_magFieldZ = tcs.m_magFieldZ;
76
77 sfs.m_tripletD0Max = tcs.m_tripletD0Max;
78 sfs.m_tripletD0_PPS_Max = tcs.m_tripletD0_PPS_Max;
79 sfs.m_tripletPtMin = tcs.m_tripletPtMin;
80 sfs.m_tripletDoPSS = tcs.m_tripletDoPSS ? 1 : 0;
81 sfs.m_tripletDoPPS = tcs.m_tripletDoPPS ? 1 : 0;
82 sfs.m_doubletFilterRZ = tcs.m_doubletFilterRZ ? 1 : 0;
83 sfs.m_nMaxPhiSlice = tcs.m_nMaxPhiSlice;
84 sfs.m_maxTripletBufferLength = tcs.m_maxTripletBufferLength;
85 sfs.m_isFullScan = 1;
86
87 // roi->isFullscan() check cannot be currently used for standalone ITk ROI configuration
88 bool isRoiFullscan = ((roi->etaPlus() >= 4) && (roi->etaMinus() <= -4));
89 if(!(isRoiFullscan || roi->composite() )){
90 //roi suitable for gpu
91 //composite rois are not supported at this point
92 sfs.m_isFullScan = 0;
93
94 sfs.m_phiMinus = roi->phiMinus();
95 sfs.m_phiPlus = roi->phiPlus();
96 }
97
98 sfs.m_zedMinus = roi->zedMinus();
99 sfs.m_zedPlus = roi->zedPlus();
100
101 TrigAccel::ITk::SPACEPOINT_STORAGE& sps = pJ->m_data;
102
103 unsigned int nSP = vsp.size();
104 if(nSP >= TrigAccel::ITk::MAX_NUMBER_SPACEPOINTS) {
105 nSP = TrigAccel::ITk::MAX_NUMBER_SPACEPOINTS-1;
106 ATH_MSG_WARNING("MAX_NUMBER_SPACEPOINTS exceeded, exported data truncated ...");
107 }
108
109 //2. arrange spacepoints into phi/Layer bins
110
111 double phiSliceWidth = 2*M_PI/tcs.m_nMaxPhiSlice;
112 int nLayers = static_cast<int>(layerTypes.size());
113 int nSlices = static_cast<int>(tcs.m_nMaxPhiSlice);
114
115 std::vector<std::vector<std::pair<int, const TrigSiSpacePointBase*> > > phiLArray;
116 phiLArray.resize(nLayers*nSlices);
117
118 for(unsigned int i=0;i<nSP;i++) {
119 const TrigSiSpacePointBase& sp = vsp[i];
120 const std::pair<IdentifierHash, IdentifierHash>& els = sp.offlineSpacePoint()->elementIdList();
121
122 IdentifierHash hashId = els.first;
123 short layerId = pixelLayers[hashId];
124
125 int phiIdx = (sp.phi()+M_PI)/phiSliceWidth;
126 if (phiIdx >= tcs.m_nMaxPhiSlice) {
127 phiIdx %= tcs.m_nMaxPhiSlice;
128 }
129 else if (phiIdx < 0) {
130 phiIdx += tcs.m_nMaxPhiSlice;
131 phiIdx %= tcs.m_nMaxPhiSlice;
132 }
133
134 std::vector<std::pair<int, const TrigSiSpacePointBase*> >& v = phiLArray[layerId + phiIdx*nLayers];
135 v.push_back(std::pair<int, const TrigSiSpacePointBase*>(i,&sp));//storing the original index of spacepoint
136 }
137
138 //sorting spacepoints in accordance with non-ref coordinate
139 int layerIdx=0;
140 for(std::vector<short>::const_iterator it = layerTypes.begin();it!=layerTypes.end();++it, layerIdx++) {
141 short barrel_ec = (*it);//0-barrel, !=0 - endcap
142 for(int slice = 0;slice<nSlices;slice++) {
143 std::vector<std::pair<int, const TrigSiSpacePointBase*> >& v = phiLArray[layerIdx + slice*nLayers];
144 if(barrel_ec==0) std::sort(v.begin(), v.end(), SP_INDEX_PAIR::compareZ());
145 else std::sort(v.begin(), v.end(), SP_INDEX_PAIR::compareR());
146 }
147 }
148
149 sps.m_nSpacepoints = nSP;
150 sps.m_nPhiSlices = nSlices;
151 sps.m_nLayers = nLayers;
152 sps.m_nMiddleLayers = m_accelSvc->getMiddleLayersSize();
153
154 int spIdx=0;
155 for(int slice = 0;slice<nSlices;slice++) {
156 for(int layer = 0;layer<nLayers;layer++) {
157 int layerStart = spIdx;
158 bool isBarrel = (layerTypes[layer] == 0); // barrel = 0, ec != 0
159 std::vector<std::pair<int, const TrigSiSpacePointBase*> >& v = phiLArray[layer + slice*nLayers];
160 for(std::vector<std::pair<int, const TrigSiSpacePointBase*> >::iterator it = v.begin();it!=v.end();++it) {
161 const TrigSiSpacePointBase* sp = (*it).second;
162 const Trk::SpacePoint* osp = sp->offlineSpacePoint();
163 const InDet::PixelCluster* pCL = dynamic_cast<const InDet::PixelCluster*>(osp->clusterList().first);
164 float clusterWidth = pCL->width().widthPhiRZ().y();
165 if (!isBarrel && clusterWidth > 0.2) continue;
166
167 sps.m_index[spIdx] = (*it).first;
168 sps.m_type[spIdx] = layerTypes[layer]; // store layer number
169 sps.m_x[spIdx] = sp->x();
170 sps.m_y[spIdx] = sp->y();
171 sps.m_z[spIdx] = sp->z();
172 sps.m_r[spIdx] = sp->r();
173 sps.m_phi[spIdx] = sp->phi();
174 sps.m_covR[spIdx] = getCovR(isBarrel);
175 sps.m_covZ[spIdx] = getCovZ(isBarrel);
176 sps.m_clusterWidth[spIdx] = clusterWidth;
177 spIdx++;
178 }
179 int layerEnd = spIdx;
180 sps.m_phiSlices[slice].m_layerBegin[layer] = layerStart;
181 sps.m_phiSlices[slice].m_layerEnd[layer] = layerEnd;
182 }
183 }
184
185 return totalSize;
186
187}
M_PI
#define M_PI
Definition ActiveFraction.h:11
ATH_MSG_WARNING
#define ATH_MSG_WARNING(x)
Definition AthMsgStreamMacros.h:32
sp
static Double_t sp
Definition LArPhysWaveHECTool.cxx:37
totalSize
static unsigned int totalSize(const MultiDimArray< T, N > &ht)
Definition MultiDimArray.h:310
IRoiDescriptor::phiPlus
virtual double phiPlus() const =0
extreme phi values
IRoiDescriptor::zedPlus
virtual double zedPlus() const =0
the zed and eta values at the most forward and most rear ends of the RoI
IRoiDescriptor::phiMinus
virtual double phiMinus() const =0
IRoiDescriptor::zedMinus
virtual double zedMinus() const =0
IRoiDescriptor::etaMinus
virtual double etaMinus() const =0
IRoiDescriptor::etaPlus
virtual double etaPlus() const =0
IRoiDescriptor::composite
virtual bool composite() const =0
Super RoI access methods.
InDet::SiCluster::width
const InDet::SiWidth & width() const
return width class reference
InDet::SiWidth::widthPhiRZ
const Amg::Vector2D & widthPhiRZ() const
Definition SiWidth.h:121
TrigITkAccelerationTool::getCovR
float getCovR(bool isBarrel) const
Definition TrigITkAccelerationTool.cxx:211
TrigITkAccelerationTool::getCovZ
float getCovZ(bool isBarrel) const
Definition TrigITkAccelerationTool.cxx:216
Trk::SpacePoint::clusterList
const std::pair< const PrepRawData *, const PrepRawData * > & clusterList() const
return the pair of cluster pointers by reference
Definition Tracking/TrkEvent/TrkSpacePoint/TrkSpacePoint/SpacePoint.h:127
TRTCalib_cfilter.p2
p2
Definition TRTCalib_cfilter.py:131
TRTCalib_cfilter.p1
p1
Definition TRTCalib_cfilter.py:130
TRT::Hit::layer
@ layer
Definition HitInfo.h:79
TrigAccel::ITk::SEED_MAKING_JOB
struct TrigAccel::ITk::SeedMakingJob SEED_MAKING_JOB
TrigAccel::ITk::SEED_FINDER_SETTINGS
struct TrigAccel::ITk::SeedFinderSettings SEED_FINDER_SETTINGS
TrigAccel::ITk::SPACEPOINT_STORAGE
struct TrigAccel::ITk::SpacePointStorage SPACEPOINT_STORAGE
TrigAccel::ITk::MAX_NUMBER_SPACEPOINTS
static constexpr unsigned int MAX_NUMBER_SPACEPOINTS
Definition TrigITkAccelEDM.h:20
merge.output
output
Definition merge.py:16
std::sort
void sort(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end)
Specialization of sort for DataVector/List.
Definition DVL_algorithms.h:554
TrigAccel::ITk::SeedMakingJob::m_settings
SEED_FINDER_SETTINGS m_settings
Definition TrigITkAccelEDM.h:91
TrigAccel::ITk::SpacePointLayerRange::m_layerBegin
int m_layerBegin[MAX_SILICON_LAYERS]
Definition TrigITkAccelEDM.h:49
TrigAccel::ITk::SpacePointStorage::m_type
int m_type[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:60
TrigAccel::ITk::SpacePointStorage::m_clusterWidth
float m_clusterWidth[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:68
TrigAccel::ITk::SpacePointStorage::m_z
float m_z[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:63
TrigAccel::ITk::SpacePointStorage::m_phiSlices
SPACEPOINT_LAYER_RANGE m_phiSlices[MAX_PHI_SLICES]
Definition TrigITkAccelEDM.h:69
TrigAccel::ITk::SpacePointStorage::m_index
int m_index[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:59
TrigAccel::ITk::SpacePointStorage::m_phi
float m_phi[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:65
TrigAccel::ITk::SpacePointStorage::m_x
float m_x[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:61
TrigAccel::ITk::SpacePointStorage::m_covZ
float m_covZ[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:67
TrigAccel::ITk::SpacePointStorage::m_covR
float m_covR[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:66
TrigAccel::ITk::SpacePointStorage::m_y
float m_y[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:62
TrigAccel::ITk::SpacePointStorage::m_r
float m_r[MAX_NUMBER_SPACEPOINTS]
Definition TrigITkAccelEDM.h:64

◆ extractTripletsFromOutput()

int TrigITkAccelerationTool::extractTripletsFromOutput ( std::shared_ptr< TrigAccel::OffloadBuffer > gpu_buffer,
const std::vector< TrigSiSpacePointBase > & vsp,
std::vector< TrigInDetTriplet > & output ) const
overridevirtual

Definition at line 189 of file TrigITkAccelerationTool.cxx.

189 {
190 TrigAccel::ITk::OUTPUT_SEED_STORAGE* pOutput = reinterpret_cast<TrigAccel::ITk::OUTPUT_SEED_STORAGE *>(gpu_buffer->m_rawBuffer);
191
192 int nTriplets = pOutput->m_nSeeds;
193
194 //copy seeds into the output buffer
195
196 output.clear();
197
198 for(int k=0;k<nTriplets;k++) {
199 // Check if a valid triplet was returned
200 if (pOutput->m_innerIndex[k] == pOutput->m_outerIndex[k]) continue;
201
202 const TrigSiSpacePointBase& SPi = vsp[pOutput->m_innerIndex[k]];
203 const TrigSiSpacePointBase& SPm = vsp[pOutput->m_middleIndex[k]];
204 const TrigSiSpacePointBase& SPo = vsp[pOutput->m_outerIndex[k]];
205 output.emplace_back(SPi, SPm, SPo, pOutput->m_Q[k]);
206 }
207
208 return nTriplets;
209}
TrigAccel::OffloadBuffer::m_rawBuffer
unsigned char * m_rawBuffer
Definition OffloadBuffer.h:39
TrigAccel::ITk::OUTPUT_SEED_STORAGE
struct TrigAccel::ITk::OutputSeedStorage OUTPUT_SEED_STORAGE
TrigAccel::ITk::OutputSeedStorage::m_middleIndex
int m_middleIndex[MAX_NUMBER_OUTPUT_SEEDS]
Definition TrigITkAccelEDM.h:102
TrigAccel::ITk::OutputSeedStorage::m_innerIndex
int m_innerIndex[MAX_NUMBER_OUTPUT_SEEDS]
Definition TrigITkAccelEDM.h:101
TrigAccel::ITk::OutputSeedStorage::m_outerIndex
int m_outerIndex[MAX_NUMBER_OUTPUT_SEEDS]
Definition TrigITkAccelEDM.h:103
TrigAccel::ITk::OutputSeedStorage::m_Q
float m_Q[MAX_NUMBER_OUTPUT_SEEDS]
Definition TrigITkAccelEDM.h:104

◆ getCovR()

float TrigITkAccelerationTool::getCovR ( bool isBarrel) const
private

Definition at line 211 of file TrigITkAccelerationTool.cxx.

211 {
212 if (isBarrel) return 0.01*0.01;
213 else return 0.02*0.02;
214}

◆ getCovZ()

float TrigITkAccelerationTool::getCovZ ( bool isBarrel) const
private

Definition at line 216 of file TrigITkAccelerationTool.cxx.

216 {
217 if (isBarrel) return 0.02*0.02;
218 else return 0.01*0.01;
219}

◆ initialize()

StatusCode TrigITkAccelerationTool::initialize ( )
overridevirtual

Definition at line 19 of file TrigITkAccelerationTool.cxx.

19 {
20
21 ATH_CHECK(m_accelSvc.retrieve());
22 return StatusCode::SUCCESS;
23}
ATH_CHECK
#define ATH_CHECK
Evaluate an expression and check for errors.
Definition AthCheckMacros.h:40

Member Data Documentation

◆ m_accelSvc

ServiceHandle<ITrigInDetAccelerationSvc> TrigITkAccelerationTool::m_accelSvc
private

Definition at line 31 of file TrigITkAccelerationTool.h.

The documentation for this class was generated from the following files:
Generated on for ATLAS Offline Software by doxygen 1.14.0