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
85
86
87
88
89
90
91
92
93
94
|
import copy
import numpy
import aoflagger
def flag(input):
# Values below can be tweaked
flag_polarizations = input.polarizations()
flag_representations = [aoflagger.ComplexRepresentation.AmplitudePart]
iteration_count = 3
threshold_factor_step = 2.0
base_threshold = 1.4
# Use above values to calculate thresholds in each iteration
r = range((iteration_count - 1), 0, -1)
threshold_factors = numpy.power(threshold_factor_step, r)
inpPolarizations = input.polarizations()
input.clear_mask()
for polarization in flag_polarizations:
data = input.convert_to_polarization(polarization)
for representation in flag_representations:
data = data.convert_to_complex(representation)
original_image = copy.copy(data)
for threshold_factor in threshold_factors:
# print('Flagging polarization ' + str(polarization) + ' (' + str(representation) + ', ' + str(threshold_factor) + ')')
thr = threshold_factor * base_threshold
aoflagger.sumthreshold(data, thr, thr, True, True)
chdata = copy.copy(data)
aoflagger.threshold_timestep_rms(data, 3.5)
aoflagger.threshold_channel_rms(
chdata, 3.0 * threshold_factor, True
)
data.join_mask(chdata)
data.set_image(original_image)
resized_data = aoflagger.shrink(data, 3, 3)
aoflagger.low_pass_filter(resized_data, 21, 31, 1.6, 2.2)
aoflagger.enlarge(resized_data, data, 3, 3)
data = original_image - data
aoflagger.sumthreshold(
data, base_threshold, base_threshold, True, True
)
if polarization in inpPolarizations:
data = data.make_complex()
input.set_polarization_data(polarization, data)
else:
input.join_mask(polarization, data)
aoflagger.scale_invariant_rank_operator(input, 0.2, 0.2)
aoflagger.threshold_timestep_rms(input, 4.0)
def test_sumthreshold(input):
# Values below can be tweaked
flag_polarizations = input.polarizations()
flag_representations = [aoflagger.ComplexRepresentation.AmplitudePart]
iteration_count = 3
threshold_factor_step = 2.0
base_threshold = 1.4
# Use above values to calculate thresholds in iteration
r = range((iteration_count - 1), 0, -1)
threshold_factors = numpy.power(threshold_factor_step, r)
inpPolarizations = input.polarizations()
input.clear_mask()
for polarization in flag_polarizations:
data = input.convert_to_polarization(polarization)
for representation in flag_representations:
data = data.convert_to_complex(representation)
aoflagger.sumthreshold(data, base_threshold, True, True)
if polarization in inpPolarizations:
data = data.make_complex()
input.set_polarization_data(polarization, data)
else:
input.join_mask(polarization, data)
aoflagger.set_flag_function(flag)
print("strategy.py parsed")
|
