|
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#ifndef DataFormats_VertexSoA_interface_ZVertexSoA_h
#define DataFormats_VertexSoA_interface_ZVertexSoA_h
#include <alpaka/alpaka.hpp>
#include <Eigen/Core>
#include "DataFormats/SoATemplate/interface/SoALayout.h"
namespace reco {
GENERATE_SOA_LAYOUT(ZVertexLayout,
SOA_COLUMN(float, zv), // output z-posistion of found vertices
SOA_COLUMN(float, wv), // output weight (1/error^2) on the above
SOA_COLUMN(float, chi2), // vertices chi2
SOA_COLUMN(float, ptv2), // vertices pt^2
SOA_COLUMN(uint16_t, sortInd), // sorted index (by pt2) ascending
SOA_SCALAR(uint32_t, nvFinal)) // the number of vertices
GENERATE_SOA_LAYOUT(ZVertexTracksLayout,
SOA_COLUMN(int16_t, idv), // vertex index for each associated (original) track
// (-1 == not associate)
SOA_COLUMN(int32_t, ndof)) // vertices number of dof
// FIXME: reused as workspace for the number of nearest neighbours
// Common types for both Host and Device code
using ZVertexSoA = ZVertexLayout<>;
using ZVertexSoAView = ZVertexSoA::View;
using ZVertexSoAConstView = ZVertexSoA::ConstView;
// Common types for both Host and Device code
using ZVertexTracksSoA = ZVertexTracksLayout<>;
using ZVertexTracksSoAView = ZVertexTracksSoA::View;
using ZVertexTracksSoAConstView = ZVertexTracksSoA::ConstView;
ALPAKA_FN_HOST_ACC ALPAKA_FN_INLINE void init(ZVertexSoAView &vertices) { vertices.nvFinal() = 0; }
} // namespace reco
#endif // DataFormats_VertexSoA_interface_ZVertexSoA_h
|