|
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
|
#include "DataFormats/Math/interface/ProjectMatrix.h"
#include <functional>
int main() {
typedef double T;
typedef ROOT::Math::SMatrix<T, 5, 5, ROOT::Math::MatRepSym<T, 5> > SMat55;
typedef ROOT::Math::SMatrix<T, 2, 2, ROOT::Math::MatRepSym<T, 2> > SMatDD;
typedef ROOT::Math::SMatrix<T, 5, 5> SMatNN;
typedef ROOT::Math::SMatrix<T, 5, 2> SMatND;
typedef ROOT::Math::SMatrix<T, 2, 5> SMatDN;
double v[3] = {1., -0.5, 2.};
SMatDD S(v, 3);
std::cout << S << std::endl;
std::cout << std::endl;
{
SMatDN H;
H(0, 3) = 1;
H(1, 4) = 1;
SMatND K = ROOT::Math::Transpose(H) * S;
SMatNN V = K * H;
std::cout << H << std::endl;
std::cout << K << std::endl;
std::cout << V << std::endl;
std::cout << std::endl;
}
{
ProjectMatrix<double, 5, 2> H;
H.index[0] = 3;
H.index[1] = 4;
SMatND K = H.project(S);
SMatNN V = H.project(K);
std::cout << H.matrix() << std::endl;
std::cout << K << std::endl;
std::cout << V << std::endl;
std::cout << std::endl;
}
{
SMatDN HH;
HH(0, 3) = 1;
HH(1, 4) = 1;
ProjectMatrix<double, 5, 2> H;
H.fromH(HH);
SMatND K = H.project(S);
SMatNN V = H.project(K);
std::cout << H.matrix() << std::endl;
std::cout << K << std::endl;
std::cout << V << std::endl;
std::cout << std::endl;
}
return 0;
}
|