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
|
#ifndef WSCLEAN_STRUCTURES_IN_MEMORY_REORDERING_H_
#define WSCLEAN_STRUCTURES_IN_MEMORY_REORDERING_H_
#include "inmemorypart.h"
#include <cassert>
#include <map>
#include <string>
#include <vector>
#include <aocommon/vectormap.h>
#include <aocommon/polarization.h>
#include <schaapcommon/reordering/channelrange.h>
namespace wsclean {
struct InMemoryMs;
struct InMemoryPart;
class Settings;
class InMemoryHandle {
public:
/**
* Get a part that does or doest not exist. This function is typically
* used during reordering. Use @ref GetExistingPart() after reordering.
*/
InMemoryPart& GetPart(aocommon::PolarizationEnum polarization,
size_t out_channel) {
return parts_[std::pair(polarization, out_channel)];
}
/**
* Retrieve a part which which is known to exist. This function is typically
* used by the MsProvider to obtain its already reordered data, and is
* constant.
*/
const InMemoryPart& GetExistingPart(aocommon::PolarizationEnum polarization,
size_t out_channel) const {
const auto iterator = parts_.find(std::pair(polarization, out_channel));
assert(iterator != parts_.end());
return iterator->second;
}
private:
std::map<std::pair<aocommon::PolarizationEnum, size_t>, InMemoryPart> parts_;
};
/**
* @param channels has one element for every output channel, where each element
* has all data desc ids for the corresponding output channel.
*/
InMemoryHandle ReorderInMemory(
InMemoryMs&& data,
const std::vector<
aocommon::VectorMap<schaapcommon::reordering::ChannelRange>>& channels,
const schaapcommon::reordering::MSSelection& selection,
const std::string& data_column_name, const std::string& model_column_name,
bool include_model, const Settings& settings);
} // namespace wsclean
#endif
|
