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
|
#include "msrowproviderbase.h"
#include "bdamsrowprovider.h"
#include "directmsrowprovider.h"
#include "msprovider.h"
#include <aocommon/throwruntimeerror.h>
#include <casacore/tables/Tables/TableRecord.h>
#include <memory>
namespace wsclean {
MsRowProviderBase::MsRowProviderBase(const casacore::MeasurementSet& ms,
const MSSelection& selection,
const std::string& data_column_name,
const std::string& model_column_name)
: ms_(ms),
selection_(selection),
columns_(ms_, data_column_name),
model_column_name_(model_column_name) {
MSProvider::GetRowRange(ms_, selection_, begin_row_, end_row_);
}
std::unique_ptr<MsRowProviderBase> MakeMsRowProvider(
const std::string& ms_name, const MSSelection& selection,
const std::map<size_t, size_t>& selected_data_description_ids,
const std::string& data_column_name, const std::string& model_column_name,
bool require_model) {
if (!casacore::Table::isReadable(ms_name)) {
aocommon::ThrowRuntimeError("The measurement set ", ms_name,
" can't be opened for reading.");
}
casacore::MeasurementSet ms(ms_name);
if (MsHasBdaData(ms))
return std::make_unique<BdaMsRowProvider>(
ms, selection, selected_data_description_ids, data_column_name,
model_column_name, require_model);
return std::make_unique<DirectMSRowProvider>(
ms, selection, selected_data_description_ids, data_column_name,
model_column_name, require_model);
}
bool MsHasBdaData(const casacore::MeasurementSet& ms) {
return ms.keywordSet().isDefined(BdaMsRowProvider::kBDAFactorsTable) &&
ms.keywordSet().asTable(BdaMsRowProvider::kBDAFactorsTable).nrow() !=
0;
}
} // namespace wsclean
|
