|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
#ifndef CUDADataFormats_Vertex_ZVertexSoA_h
#define CUDADataFormats_Vertex_ZVertexSoA_h
#include <cstdint>
#include "HeterogeneousCore/CUDAUtilities/interface/cudaCompat.h"
// SOA for vertices
// These vertices are clusterized and fitted only along the beam line (z)
// to obtain their global coordinate the beam spot position shall be added (eventually correcting for the beam angle as well)
struct ZVertexSoA {
static constexpr uint32_t MAXTRACKS = 128 * 1024;
static constexpr uint32_t MAXVTX = 1024;
int16_t idv[MAXTRACKS]; // vertex index for each associated (original) track (-1 == not associate)
float zv[MAXVTX]; // output z-posistion of found vertices
float wv[MAXVTX]; // output weight (1/error^2) on the above
float chi2[MAXVTX]; // vertices chi2
float ptv2[MAXVTX]; // vertices pt^2
int32_t ndof[MAXTRACKS]; // vertices number of dof (reused as workspace for the number of nearest neighbours FIXME)
uint16_t sortInd[MAXVTX]; // sorted index (by pt2) ascending
uint32_t nvFinal; // the number of vertices
__host__ __device__ void init() { nvFinal = 0; }
};
#endif // CUDADataFormats_Vertex_ZVertexSoA_h
|