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
|
#include "rfibaselineset.h"
#include "../msio/singlebaselinefile.h"
#include <fstream>
#include <sstream>
#include "../util/progress/dummyprogresslistener.h"
namespace imagesets {
RFIBaselineSet::RFIBaselineSet(const std::string& path)
: SingleImageSet(), _path(path) {}
std::unique_ptr<BaselineData> RFIBaselineSet::Read(
class ProgressListener& progress) {
if (_data.IsEmpty()) {
progress.OnStartTask("Reading file");
SingleBaselineFile file;
std::ifstream str(_path);
file.Read(str, progress);
_data = std::move(file.data);
_metaData = std::move(file.metaData);
_telescopeName = std::move(file.telescopeName);
}
progress.OnFinish();
return std::unique_ptr<BaselineData>(new BaselineData(
_data, TimeFrequencyMetaDataPtr(new TimeFrequencyMetaData(_metaData))));
}
std::string RFIBaselineSet::BaselineDescription() {
if (_metaData.HasAntenna1() && _metaData.HasAntenna2()) {
std::ostringstream sstream;
sstream << _metaData.Antenna1().station << ' ' << _metaData.Antenna1().name
<< " x " << _metaData.Antenna2().station << ' '
<< _metaData.Antenna2().name;
return sstream.str();
} else {
return Files().front();
}
}
void RFIBaselineSet::Write(const std::vector<Mask2DCPtr>& masks) {
SingleBaselineFile file;
{
std::ifstream str(_path);
DummyProgressListener dummy;
file.Read(str, dummy);
}
if (file.data.MaskCount() != masks.size())
throw std::runtime_error(
"Number of masks in flag writing action don't match rfibl file");
for (size_t i = 0; i != masks.size(); ++i) file.data.SetMask(i, masks[i]);
std::ofstream outstr(_path);
file.Write(outstr);
}
} // namespace imagesets
|
