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
|
#ifndef MSPROVIDERS_TIMESTEP_BUFFER_H
#define MSPROVIDERS_TIMESTEP_BUFFER_H
#include "msprovider.h"
#include <aocommon/uvector.h>
namespace wsclean {
class TimestepBufferReader;
/**
* This class wraps any MSProvider to make it read whole blocks of rows
* at once that correspond to the same timestep.
*
* This is used in IDGMSGridder to be able to get the UVWs for calculating the
* a-terms.
*/
class TimestepBuffer final : public MSProvider {
friend class TimestepBufferReader;
public:
TimestepBuffer(MSProvider* msProvider, bool readModel)
: _msProvider(msProvider), _readModel(readModel) {
_msProvider->ResetWritePosition();
}
virtual ~TimestepBuffer(){};
SynchronizedMS MS() override { return _msProvider->MS(); }
std::unique_ptr<MSReader> MakeReader() final override;
const std::string& DataColumnName() override {
return _msProvider->DataColumnName();
}
void NextOutputRow() override { _msProvider->NextOutputRow(); }
void ResetWritePosition() override { _msProvider->ResetWritePosition(); }
virtual void WriteModel(const std::complex<float>* buffer,
bool addToMS) override {
_msProvider->WriteModel(buffer, addToMS);
}
void ReopenRW() override { _msProvider->ReopenRW(); }
double StartTime() override { return _msProvider->StartTime(); }
size_t DataDescId() override { return _msProvider->DataDescId(); }
void MakeIdToMSRowMapping(std::vector<size_t>& idToMSRow) override {
_msProvider->MakeIdToMSRowMapping(idToMSRow);
}
aocommon::PolarizationEnum Polarization() override {
return _msProvider->Polarization();
}
size_t NChannels() override { return _msProvider->NChannels(); }
size_t NAntennas() override { return _msProvider->NAntennas(); }
size_t NPolarizations() override { return _msProvider->NPolarizations(); }
const aocommon::BandData& Band() override { return _msProvider->Band(); }
private:
struct RowData {
std::vector<std::complex<float>> data, model;
std::vector<float> weights;
MSProvider::MetaData metaData;
size_t rowId;
};
MSProvider* _msProvider;
bool _readModel;
};
} // namespace wsclean
#endif
|
