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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#include <boost/test/unit_test.hpp>
#include "../../math/imageoperations.h"
namespace wsclean {
using aocommon::HMC4x4;
using aocommon::Image;
using aocommon::MC2x2;
BOOST_AUTO_TEST_SUITE(image_operations)
void CheckConstantImage(const Image& image, float expected,
const std::string& description) {
for (float value : image) {
if (std::fabs(value - expected) > 1e-6) {
std::ostringstream str;
str << description << ", expecting " << expected << ", got " << value;
BOOST_CHECK_EQUAL(str.str(), "");
break;
}
BOOST_CHECK_LT(std::fabs(value - expected), 1e-6);
}
}
BOOST_AUTO_TEST_CASE(correct_images_for_mueller_matrix) {
Image i(3, 2, 77.0);
Image q(3, 2, 0.0);
Image u(3, 2, 0.0);
Image v(3, 2, 0.0);
std::array<aocommon::Image*, 4> images{&i, &q, &u, &v};
math::CorrectImagesForMuellerMatrix(HMC4x4::Unit(), images);
CheckConstantImage(i, 77.0, "Stokes I");
CheckConstantImage(q, 0.0, "Stokes Q");
CheckConstantImage(u, 0.0, "Stokes U");
CheckConstantImage(v, 0.0, "Stokes V");
const HMC4x4 i_to_q =
HMC4x4::KroneckerProduct(MC2x2{2, 0, 0, 0}, MC2x2{1, 0, 0, 0});
math::CorrectImagesForMuellerMatrix(i_to_q, images);
CheckConstantImage(i, 77.0, "Stokes I");
CheckConstantImage(q, 77.0, "Stokes Q");
CheckConstantImage(u, 0.0, "Stokes U");
CheckConstantImage(v, 0.0, "Stokes V");
i = 10.0;
q = 0.0;
u = 0.0;
v = 0.0;
constexpr std::complex j(0.0, 1.0);
const HMC4x4 i_to_v{
0.0, -j, j, 0.0, // XX=0
j, 0.0, 0.0, 0.0, // XY=iXX
-j, 0.0, 0.0, 0.0, // YX=-iXX
0.0, 0.0, 0.0, 0.0 // YY=0
};
math::CorrectImagesForMuellerMatrix(i_to_v, images);
CheckConstantImage(i, 0.0, "Stokes I");
CheckConstantImage(q, 0.0, "Stokes Q");
CheckConstantImage(u, 0.0, "Stokes U");
// V = -i(XY - YX)/2 = -i((10i) - (-10i)) = (10 + 10) / 2 = 10
CheckConstantImage(v, 10.0, "Stokes V");
}
BOOST_AUTO_TEST_CASE(correct_dual_images_for_mueller_matrix) {
Image xx(3, 2, 37.0);
Image yy(3, 2, 42.0);
std::array<aocommon::Image*, 2> images{&xx, &yy};
math::CorrectDualImagesForMuellerMatrix(HMC4x4::Unit(), images);
CheckConstantImage(xx, 37.0, "XX");
CheckConstantImage(yy, 42.0, "YY");
const HMC4x4 zero_yy =
HMC4x4::KroneckerProduct(MC2x2{2, 0, 0, 0}, MC2x2{1, 0, 0, 0});
math::CorrectDualImagesForMuellerMatrix(zero_yy, images);
CheckConstantImage(xx, 74.0, "XX");
CheckConstantImage(yy, 0.0, "YY");
xx = 10.0;
yy = 3.0;
const HMC4x4 swap{
0.0, 0.0, 0.0, {2.0, 4.0}, // XX=(2+4i) YY
0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, {2.0, -4.0}, 0.0, 0.0, 0.0 // YY=(2-4i) XX
};
math::CorrectDualImagesForMuellerMatrix(swap, images);
CheckConstantImage(xx, 6.0, "XX");
CheckConstantImage(yy, 20.0, "YY");
}
BOOST_AUTO_TEST_SUITE_END()
} // namespace wsclean
|
