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
|
#include "imagingtableentry.h"
#include <memory>
#include <radler/work_table_entry.h>
#include "../io/cachedimageaccessor.h"
namespace wsclean {
ImagingTableEntry::ImagingTableEntry()
: index(0),
lowestFrequency(0.0),
highestFrequency(0.0),
bandStartFrequency(0.0),
bandEndFrequency(0.0),
inputChannelCount(0),
polarization(aocommon::Polarization::StokesI),
facetGroupIndex(0),
facetIndex(0),
facet(nullptr),
isDdPsf(false),
centreShiftX(0),
centreShiftY(0),
outputChannelIndex(0),
outputIntervalIndex(0),
msData(),
squaredDeconvolutionIndex(0),
joinedGroupIndex(0),
imageCount(0),
tmpFilePrefix(),
imageWeight(0.0) {}
std::unique_ptr<radler::WorkTableEntry>
ImagingTableEntry::CreateDeconvolutionEntry(
size_t channel_index_offset, CachedImageSet* psf_images,
CachedImageSet& model_images, CachedImageSet& residual_images,
const std::vector<std::shared_ptr<schaapcommon::facets::Facet>>& psf_facets,
bool is_imaginary) const {
auto entry = std::make_unique<radler::WorkTableEntry>();
entry->index = index;
entry->band_start_frequency = bandStartFrequency;
entry->band_end_frequency = bandEndFrequency;
entry->polarization = polarization;
entry->mask_channel_index = outputChannelIndex;
entry->original_channel_index = outputChannelIndex - channel_index_offset;
entry->original_interval_index = outputIntervalIndex;
entry->image_weight = imageWeight;
// A PSF accessor is only needed for the first entry of a squared group.
if (psf_images) {
if (psf_facets.empty()) {
entry->psf_accessors.emplace_back(std::make_unique<CachedImageAccessor>(
*psf_images, polarization, outputChannelIndex, is_imaginary));
} else {
for (size_t facet_id = 0; facet_id < psf_facets.size(); ++facet_id) {
entry->psf_accessors.emplace_back(std::make_unique<CachedImageAccessor>(
*psf_images, polarization, outputChannelIndex, facet_id,
psf_facets[facet_id], is_imaginary));
}
}
}
entry->model_accessor = std::make_unique<CachedImageAccessor>(
model_images, polarization, outputChannelIndex, is_imaginary);
entry->residual_accessor = std::make_unique<CachedImageAccessor>(
residual_images, polarization, outputChannelIndex, is_imaginary);
return entry;
}
} // namespace wsclean
|
