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
|
#include "baselinepagecontroller.h"
#include "../../quality/statisticscollection.h"
#include "../../quality/statisticsderivator.h"
using aocommon::Polarization;
std::pair<TimeFrequencyData, TimeFrequencyMetaDataCPtr>
BaselinePageController::constructImage(
QualityTablesFormatter::StatisticKind kind) {
if (HasStatistics()) {
const unsigned polarizationCount = _statCollection->PolarizationCount();
std::vector<std::pair<unsigned, unsigned>> baselines =
_statCollection->BaselineStatistics().BaselineList();
const StatisticsDerivator derivator(*_statCollection);
const unsigned antennaCount =
_statCollection->BaselineStatistics().AntennaCount();
std::vector<Image2DPtr> realImages(polarizationCount);
std::vector<Image2DPtr> imagImages(polarizationCount);
std::vector<Mask2DPtr> mask(polarizationCount);
for (unsigned p = 0; p < polarizationCount; ++p) {
realImages[p] = Image2D::CreateUnsetImagePtr(antennaCount, antennaCount);
realImages[p]->SetAll(std::numeric_limits<num_t>::quiet_NaN());
imagImages[p] = Image2D::CreateUnsetImagePtr(antennaCount, antennaCount);
imagImages[p]->SetAll(std::numeric_limits<num_t>::quiet_NaN());
mask[p] = Mask2D::CreateSetMaskPtr<true>(antennaCount, antennaCount);
}
for (std::vector<std::pair<unsigned, unsigned>>::const_iterator i =
baselines.begin();
i != baselines.end(); ++i) {
const unsigned antenna1 = i->first, antenna2 = i->second;
for (unsigned p = 0; p < polarizationCount; ++p) {
const std::complex<long double> val =
derivator.GetComplexBaselineStatistic(kind, antenna1, antenna2, p);
realImages[p]->SetValue(antenna1, antenna2, val.real());
imagImages[p]->SetValue(antenna1, antenna2, val.imag());
mask[p]->SetValue(antenna1, antenna2, false);
}
}
TimeFrequencyData data;
if (polarizationCount == 1) {
data = TimeFrequencyData(Polarization::StokesI, realImages[0],
imagImages[0]);
data.SetGlobalMask(mask[0]);
} else if (polarizationCount == 2) {
data = TimeFrequencyData(Polarization::XX, realImages[0], imagImages[0],
Polarization::YY, realImages[1], imagImages[1]);
data.SetIndividualPolarizationMasks(mask[0], mask[1]);
} else if (polarizationCount == 4) {
data = TimeFrequencyData::FromLinear(
realImages[0], imagImages[0], realImages[1], imagImages[1],
realImages[2], imagImages[2], realImages[3], imagImages[3]);
data.SetIndividualPolarizationMasks(mask[0], mask[1], mask[2], mask[3]);
} else {
std::stringstream s;
s << "Set has not 1, 2 or 4 polarizations (?!?) : "
"StatisticsCollection.PolarizationCount() == "
<< polarizationCount;
throw std::runtime_error(s.str());
}
return std::pair<TimeFrequencyData, TimeFrequencyMetaDataCPtr>(
data, TimeFrequencyMetaDataCPtr());
} else {
return std::pair<TimeFrequencyData, TimeFrequencyMetaDataCPtr>(
TimeFrequencyData(), TimeFrequencyMetaDataCPtr());
}
}
std::string BaselinePageController::AntennaName(size_t index) const {
std::string name;
if (_antennas == nullptr) {
std::stringstream s;
s << index;
name = s.str();
} else {
name = (*_antennas)[index].name;
}
return name;
}
|
