|
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
#ifndef DataFormats_PortableTestObjects_interface_TestSoA_h
#define DataFormats_PortableTestObjects_interface_TestSoA_h
#include <array>
#include <Eigen/Core>
#include <Eigen/Dense>
#include "DataFormats/Common/interface/StdArray.h"
#include "DataFormats/SoATemplate/interface/SoACommon.h"
#include "DataFormats/SoATemplate/interface/SoALayout.h"
#include "DataFormats/SoATemplate/interface/SoAView.h"
namespace portabletest {
// the type aliases are needed because commas confuse macros
using Array = edm::StdArray<short, 4>;
//using Array = std::array<short, 4>;
using Matrix = Eigen::Matrix<double, 3, 6>;
// SoA layout with x, y, z, id fields
GENERATE_SOA_LAYOUT(TestSoALayout,
// columns: one value per element
SOA_COLUMN(double, x),
SOA_COLUMN(double, y),
SOA_COLUMN(double, z),
SOA_COLUMN(int32_t, id),
// scalars: one value for the whole structure
SOA_SCALAR(double, r),
// column of arrays: one fixed-size array per element
SOA_COLUMN(Array, flags),
// Eigen columns: each matrix element is stored in a separate column
SOA_EIGEN_COLUMN(Matrix, m))
using TestSoA = TestSoALayout<>;
GENERATE_SOA_LAYOUT(TestSoALayout2,
// columns: one value per element
SOA_COLUMN(double, x2),
SOA_COLUMN(double, y2),
SOA_COLUMN(double, z2),
SOA_COLUMN(int32_t, id2),
// scalars: one value for the whole structure
SOA_SCALAR(double, r2),
// Eigen columns
// the typedef is needed because commas confuse macros
SOA_EIGEN_COLUMN(Matrix, m2))
using TestSoA2 = TestSoALayout2<>;
GENERATE_SOA_LAYOUT(TestSoALayout3,
// columns: one value per element
SOA_COLUMN(double, x3),
SOA_COLUMN(double, y3),
SOA_COLUMN(double, z3),
SOA_COLUMN(int32_t, id3),
// scalars: one value for the whole structure
SOA_SCALAR(double, r3),
// Eigen columns
// the typedef is needed because commas confuse macros
SOA_EIGEN_COLUMN(Matrix, m3))
using TestSoA3 = TestSoALayout3<>;
} // namespace portabletest
#endif // DataFormats_PortableTestObjects_interface_TestSoA_h
|