Coverage for /wheeldirectory/casa-6.7.0-12-py3.10.el8/lib/py/lib/python3.10/site-packages/casatasks/uvcontsub.py: 100%

27 statements  

« prev     ^ index     » next       coverage.py v7.6.4, created at 2024-11-01 07:19 +0000

1##################### generated by xml-casa (v2) from uvcontsub.xml ################# 

2##################### 608b29d7e73cf5fff26de07ecd06eac3 ############################## 

3from __future__ import absolute_import 

4import numpy 

5from casatools.typecheck import CasaValidator as _val_ctor 

6_pc = _val_ctor( ) 

7from casatools.coercetype import coerce as _coerce 

8from casatools.errors import create_error_string 

9from .private.task_uvcontsub import uvcontsub as _uvcontsub_t 

10from casatasks.private.task_logging import start_log as _start_log 

11from casatasks.private.task_logging import end_log as _end_log 

12from casatasks.private.task_logging import except_log as _except_log 

13 

14class _uvcontsub: 

15 """ 

16 uvcontsub ---- continuum subtraction in the uv domain 

17 

18 --------- parameter descriptions --------------------------------------------- 

19 

20 vis Name of input visibility file. 

21 Default: '', must be specified 

22  

23 Example: vis='ngc5921.ms' 

24 outputvis Name of output visibility file 

25 Default: '', must be specified 

26  

27 Example: outputvis='ngc5921_contsub.ms' 

28  

29 If an MS with the output name already exists, it 

30 will not be overwritten and the task stops with 

31 an error. 

32 field Select field using field id(s) or field name(s) 

33 Default: '' (all fields) 

34  

35 Use 'listobs' to obtain the list of id's or 

36 names. If field string is a non-negative integer, 

37 it is assumed a field index, otherwise, it is 

38 assumed a field name. 

39  

40 Examples: 

41 field='0~2'; field ids 0,1,2 

42 field='0,4,5~7'; field ids 0,4,5,6,7 

43 field='3C286,3C295'; field named 3C286 and 

44 3C295 

45 field = '3,4C*'; field id 3, all names 

46 starting with 4C 

47 spw Select spectral window/channels 

48 Default: '' (meaning all spectral windows and channels) 

49  

50 Examples: 

51 spw='0~2,4'; spectral windows 0,1,2,4 (all channels) 

52 spw='<2'; spectral windows less than 2 (i.e. 0,1) 

53 spw='0:5~61'; spw 0, channels 5 to 61 

54 spw='0,10,3:3~45'; spw 0,10 all channels, spw 

55 3 - chans 3 to 45. 

56 spw='0~2:2~6'; spw 0,1,2 with channels 2 

57 through 6 in each. 

58 spw = '*:3~64' channels 3 through 64 for all sp id's 

59 spw = ' :3~64' will NOT work. 

60 scan Select scans by scan numbers 

61 Default: '' = all 

62  

63 Examples: 

64 scan='3' 

65 scan='2~23' 

66 scan=''4,7,9,12' 

67 intent Select observing intent 

68 Default: '' (no selection by intent) 

69  

70 Example: intent='*BANDPASS*' (selects data 

71 labelled with BANDPASS intent) 

72 array Select (sub)array(s) by array ID number 

73 Default: '' (all) 

74 observation Select by observation ID(s) 

75 Default: '' = all 

76  

77 Example: observation='0~2,4' 

78 datacolumn Which data column to use for processing 

79 (case-insensitive). 

80 Default: 'data' 

81 Options: 'data', 'model', 'corrected', 

82 'float_data', 'lag_data', 

83  

84 Example: datacolumn='corrected' 

85 fitspec Specification of fit, which can be general or 

86 specific for every field and SPW. The continuum will be fitted 

87 using a polynomial shape and using as input data points the 

88 channels given, which can be considered the line-free 

89 channels. fitspec can be a simple string or a dictionary. If 

90 it is a simple string, it gives an spw:channel specification in 

91 MSSelection syntax, and that same spw:channel specification is 

92 used for all fields. In that case, the polynomial order for 

93 all fields and SPWs is defined by the fitorder parameter. 

94  

95 If it is a dictionary, its entries can define a different 

96 specification of fit order and channels for every field and 

97 for every SPW. The dictionary can have up to as many entries 

98 as fields are present in the MeasurementSet, and within each 

99 field entry, as many entries as SPWs are present. Each SPW 

100 entry specifies the polynomial order (in a 'fitorder' entry) 

101 and the line-free channels (in a 'chan' entry). 

102  

103 If a field or SPW is not included, the default is assumed (all 

104 channels, and polynomial order given in the global fitorder 

105 parameter). 

106  

107 Default: '' (all channels for all SPWs for all fields) 

108 Examples: 

109 '17:100~500;600~910;1215~1678;1810~1903,19:7~100' 

110 (The ranges of channels given for SPW 17 and 19 

111 are used for all the fields present in the 

112 MeasurementSet.) 

113  

114 Dictionary to use different channels and polynomial 

115 orders for different fields and SPWs, assuming: 

116 4 SPWs, with IDs 17, 19, 21, 23 

117 6 fields, with IDs 0, 1, 4, 5, 6, 7 

118 {'0': {'17,19,21,23': {'chan': '', 'fitorder': 1}}, 

119 '1': {'17': {'chan': '8~120', 'fitorder': 0}, 

120 '19': {'chan': '7~100', 'fitorder': 1}, 

121 '21': {'chan': '21:2~119', 'fitorder': 1}, 

122 '23': {'chan': '', 'fitorder': 0}}, 

123 '4,5,6': {'19': {'chan': '7~115', 'fitorder': 1}, 

124 '21': {'chan': '', 'fitorder': 1}, 

125 '23': {'chan': '', 'fitorder': 2}}, 

126 '7': 'NONE'} 

127  

128 (Field 0: fit all SPWs and channels (same as not giving it), 

129 but using polynomial order 1 for all SPWs. 

130 Field 1: fit only some channel ranges (all for SPW 23), with 

131 different polynomial orders. 

132 Fields 4,5,6: fit only some channel ranges in SPW 19, all 

133 channels for SPWs 21 and 23, and don't fit SPWs 17), 

134 also using different polynomial orders. 

135 Field 7: do not subtract continuum) 

136 (The dictionary fields 'chan' can be given in channel 

137 numbers, as in the example above, or in frequencies, using 

138 the MSSelection syntax) 

139  

140 In the example, the empty string given for field 0 and all its 

141 SPWs means that all the channels are fitted (and subtracted) 

142 in all those SPWs. For the corner case where for a given field 

143 no channels should be fitted in any SPW, the field entry can 

144 be given as a string set to 'NONE'. 

145  

146 This specification of SPWs and channels uses a syntax similar 

147 to the spw parameter. But in contrast to the spw parameter, 

148 the SPWs and channels given in fitspec are not used to select 

149 data but as a mask applied on the input data only for fitting 

150 purposes. The channels that are not included in the 

151 specification are masked out and not used as input by the 

152 polynomial fitting algorithm. This is different from the spw 

153 parameter, which selects the data that will be included in the 

154 output. All channels present in the output are 

155 continuum-subtracted, regardless of whether they were 

156 specified as inputs to the fitting in fitspec 

157 fitmethod Choose fitting methods 

158 Default: 'gsl' 

159 Options: 'gsl', 'casacore' 

160  

161 Example: fitmethod='gsl' 

162 fitorder Polynomial order for the fits 

163 Default: 0 

164  

165 Values of fitorder > 1 should be used with 

166 care. Higher order polynomials are more flexible, 

167 and may overfit and absorb line emission. They 

168 also tend to go wild at the edges of the range of 

169 channels used for fitting. 

170 writemodel Write fitted model into the MODEL column of the 

171 output MeasurementSet. 

172  

173 Default: False 

174 Options: True|False 

175  

176 By default the task creates an MS with the 

177 continuum subtracted data in the DATA column. If 

178 this option is enabled, the task will also write 

179 the fitted model data into the MODEL column of the 

180 output MS. 

181 

182 --------- examples ----------------------------------------------------------- 

183 

184  

185 For more information, see the task pages of uvcontsub in CASA Docs: 

186 https://casadocs.readthedocs.io 

187  

188 

189 

190 """ 

191 

192 _info_group_ = """manipulation""" 

193 _info_desc_ = """continuum subtraction in the uv domain""" 

194 

195 def __call__( self, vis='', outputvis='', field='', spw='', scan='', intent='', array='', observation='', datacolumn='data', fitspec='', fitmethod='gsl', fitorder=int(0), writemodel=False ): 

196 schema = {'vis': {'type': 'cReqPath', 'coerce': _coerce.expand_path}, 'outputvis': {'type': 'cStr', 'coerce': _coerce.to_str}, 'field': {'type': 'cStr', 'coerce': _coerce.to_str}, 'spw': {'type': 'cStr', 'coerce': _coerce.to_str}, 'scan': {'type': 'cStr', 'coerce': _coerce.to_str}, 'intent': {'type': 'cStr', 'coerce': _coerce.to_str}, 'array': {'type': 'cStr', 'coerce': _coerce.to_str}, 'observation': {'type': 'cStr', 'coerce': _coerce.to_str}, 'datacolumn': {'type': 'cStr', 'coerce': _coerce.to_str, 'allowed': [ 'DATA', 'model', 'corrected', 'LAG_DATA', 'lag_data', 'FLOAT_DATA', 'CORRECTED', 'float_data', 'MODEL', 'data' ]}, 'fitspec': {'anyof': [{'type': 'cStr', 'coerce': _coerce.to_str}, {'type': 'cDict'}]}, 'fitmethod': {'type': 'cStr', 'coerce': _coerce.to_str, 'allowed': [ 'gsl', 'casacore', 'GSL', 'CASACORE' ]}, 'fitorder': {'type': 'cInt'}, 'writemodel': {'type': 'cBool'}} 

197 doc = {'vis': vis, 'outputvis': outputvis, 'field': field, 'spw': spw, 'scan': scan, 'intent': intent, 'array': array, 'observation': observation, 'datacolumn': datacolumn, 'fitspec': fitspec, 'fitmethod': fitmethod, 'fitorder': fitorder, 'writemodel': writemodel} 

198 assert _pc.validate(doc,schema), create_error_string(_pc.errors) 

199 _logging_state_ = _start_log( 'uvcontsub', [ 'vis=' + repr(_pc.document['vis']), 'outputvis=' + repr(_pc.document['outputvis']), 'field=' + repr(_pc.document['field']), 'spw=' + repr(_pc.document['spw']), 'scan=' + repr(_pc.document['scan']), 'intent=' + repr(_pc.document['intent']), 'array=' + repr(_pc.document['array']), 'observation=' + repr(_pc.document['observation']), 'datacolumn=' + repr(_pc.document['datacolumn']), 'fitspec=' + repr(_pc.document['fitspec']), 'fitmethod=' + repr(_pc.document['fitmethod']), 'fitorder=' + repr(_pc.document['fitorder']), 'writemodel=' + repr(_pc.document['writemodel']) ] ) 

200 task_result = None 

201 try: 

202 task_result = _uvcontsub_t( _pc.document['vis'], _pc.document['outputvis'], _pc.document['field'], _pc.document['spw'], _pc.document['scan'], _pc.document['intent'], _pc.document['array'], _pc.document['observation'], _pc.document['datacolumn'], _pc.document['fitspec'], _pc.document['fitmethod'], _pc.document['fitorder'], _pc.document['writemodel'] ) 

203 except Exception as exc: 

204 _except_log('uvcontsub', exc) 

205 raise 

206 finally: 

207 task_result = _end_log( _logging_state_, 'uvcontsub', task_result ) 

208 return task_result 

209 

210uvcontsub = _uvcontsub( ) 

211