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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
#ifndef MS_STAT_READER_H
#define MS_STAT_READER_H
#include <complex>
#include <string>
#include <vector>
#include <utility>
#include "../structures/timefrequencydata.h"
#include "../structures/timefrequencymetadata.h"
#include "../imagesets/msoptions.h"
namespace aocommon {
class MultiBandData;
}
class AveragingMsReader {
public:
AveragingMsReader(const std::string& filename, const std::string& dataColumn);
size_t NSequences() const { return _sequenceStart.size() - 1; }
size_t NBands() const { return _nBands; }
typedef std::pair<TimeFrequencyData, TimeFrequencyMetaDataPtr> Result;
Result Read(BaselineIntegrationMode statType,
BaselineIntegrationDifferencing diffType, size_t sequenceIndex,
size_t bandIndex, bool includeAutos, bool includeFlags,
class ProgressListener& progress) {
switch (diffType) {
default:
case BaselineIntegrationDifferencing::NoDifference:
return readSamples(statType, sequenceIndex, bandIndex, includeAutos,
includeFlags, false, progress);
case BaselineIntegrationDifferencing::TimeDifference:
return readTimeDiff(statType, sequenceIndex, bandIndex, includeAutos,
includeFlags, progress);
case BaselineIntegrationDifferencing::FrequencyDifference:
return readSamples(statType, sequenceIndex, bandIndex, includeAutos,
includeFlags, true, progress);
}
}
std::string TelescopeName() const { return _telescopeName; }
void StoreFlags(const std::vector<Mask2DCPtr>& flags,
BaselineIntegrationDifferencing diffType,
size_t sequenceIndex, size_t bandIndex, bool includeAutos);
private:
Result readSamples(BaselineIntegrationMode statType, size_t sequenceIndex,
size_t bandIndex, bool includeAutos, bool includeFlags,
bool freqDiff, ProgressListener& progress);
Result readTimeDiff(BaselineIntegrationMode statType, size_t sequenceIndex,
size_t bandIndex, bool includeAutos, bool includeFlags,
ProgressListener& progress);
void fillBand(AveragingMsReader::Result& result, bool freqDiff,
const aocommon::MultiBandData& bands, size_t bandIndex);
struct Statistic {
double data1 = 0.0, data2 = 0.0, data3 = 0.0;
size_t count = 0;
void Add(BaselineIntegrationMode stat, std::complex<float> sample) {
++count;
switch (stat) {
case BaselineIntegrationMode::Count:
break;
case BaselineIntegrationMode::Average:
data1 += sample.real();
break;
case BaselineIntegrationMode::AverageAbs:
data1 += std::abs(sample);
break;
case BaselineIntegrationMode::Squared:
data1 += std::norm(sample);
break;
case BaselineIntegrationMode::Stddev:
data1 += sample.real();
data2 += sample.imag();
data3 += std::norm(sample);
break;
}
}
double Calculate(BaselineIntegrationMode stat) const {
switch (stat) {
case BaselineIntegrationMode::Count:
return count;
case BaselineIntegrationMode::Average:
case BaselineIntegrationMode::AverageAbs:
case BaselineIntegrationMode::Squared:
return data1 / count;
case BaselineIntegrationMode::Stddev: {
double normMeanSq =
std::norm(std::complex<double>(data1, data2) / double(count));
return std::sqrt(data3 / count - normMeanSq);
}
}
return 0.0;
}
};
Result makeResult(BaselineIntegrationMode statType,
const Statistic* statsData, size_t nPolarizations,
size_t nChannels, size_t nTimes);
void storeFlags(const std::vector<Mask2DCPtr>& flags, size_t sequenceIndex,
size_t bandIndex, bool includeAutos);
const std::string _filename;
const std::string _dataColumn;
size_t _nBands;
std::vector<size_t> _sequenceStart;
std::string _telescopeName;
};
#endif
|
