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
|
#ifndef MS_SELECTION_H
#define MS_SELECTION_H
#include <casacore/tables/Tables/ArrayColumn.h>
#include <casacore/tables/Tables/ScalarColumn.h>
#include <casacore/ms/MeasurementSets/MeasurementSet.h>
#include <limits>
#include <map>
#include <vector>
#include "../util/progress/progresslistener.h"
class MSSelection {
public:
MSSelection(casacore::MeasurementSet& ms,
const std::vector<std::map<double, size_t>>& observationTimes,
ProgressListener& progress)
: _observationTimes(observationTimes), _ms(ms), _progress(progress) {}
template <typename Function>
void Process(Function function) {
casacore::ScalarColumn<double> timeColumn(_ms, "TIME");
casacore::ScalarColumn<int> fieldIdColumn(_ms, "FIELD_ID");
double prevTime = -1.0;
size_t prevFieldId = size_t(-1), sequenceId = size_t(-1),
timeIndexInSequence = size_t(-1);
for (size_t rowIndex = 0; rowIndex != _ms.nrow(); ++rowIndex) {
_progress.OnProgress(rowIndex, _ms.nrow());
double time = timeColumn(rowIndex);
bool newTime = time != prevTime;
size_t fieldId = fieldIdColumn(rowIndex);
if (fieldId != prevFieldId) {
prevFieldId = fieldId;
sequenceId++;
newTime = true;
}
if (newTime) {
const std::map<double, size_t>& observationTimes =
_observationTimes[sequenceId];
prevTime = time;
auto elem = observationTimes.find(time);
if (elem == observationTimes.end())
timeIndexInSequence = std::numeric_limits<size_t>::max();
else
timeIndexInSequence = elem->second;
}
if (timeIndexInSequence != std::numeric_limits<size_t>::max()) {
function(rowIndex, sequenceId, timeIndexInSequence);
}
}
}
private:
std::vector<std::map<double, size_t>> _observationTimes;
casacore::MeasurementSet& _ms;
ProgressListener& _progress;
};
#endif
|
