00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014 import time
00015 import os
00016 import pickle
00017
00018 pathname=os.environ.get('CASAPATH').split()[0]
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029 scriptmode = True
00030
00031
00032 benchmarking = True
00033
00034
00035 prefix = 'polcal_20080224.cband.regression'
00036
00037
00038
00039 regressdir = './'
00040 scriptprefix = 'polcal_20080224_cband_regression'
00041
00042
00043
00044
00045 os.system('rm -rf '+prefix+'*')
00046
00047 print 'Regression Script for VLA POLCAL C-Band Data'
00048 print 'Will do: import, flagging, calibration, imaging, analysis'
00049 print ''
00050
00051
00052
00053
00054 importmode = 'vla'
00055
00056
00057
00058
00059
00060
00061
00062
00063 datafile = ['POLCA_20080224_1']
00064 datafile = pathname + '/data/regression/polcal/20080224_cband/POLCA_20080224_1'
00065
00066
00067
00068 exportproject = 'POLCA'
00069 exportband = 'C'
00070
00071
00072 usespw = ''
00073 usespwlist = ['0','1']
00074
00075
00076 msfile = prefix + '.ms'
00077
00078
00079 gtable = prefix + '.gcal'
00080 ftable = prefix + '.fluxscale'
00081 ptable = prefix + '.pcal'
00082 xtable = prefix + '.polx'
00083
00084
00085
00086 myquackinterval = 20.0
00087
00088
00089 flagants = 'EA04'
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101
00102
00103
00104
00105
00106
00107 gaincalfield = ['0137+331','2202+422','1743-038','1924-292','2136+006','2253+161','2355+498','0319+415','0359+509']
00108
00109
00110 targets = []
00111
00112
00113 fieldgain = ','.join(gaincalfield)
00114 fieldtargets = ','.join(targets)
00115
00116
00117
00118 srclist = gaincalfield + targets
00119
00120
00121
00122
00123
00124 fluxcaldir = pathname + '/data/nrao/VLA/CalModels/'
00125
00126
00127
00128
00129 fluxcalfield = '0137+331'
00130 fluxcalmodel = '3C48_C.im'
00131 gaincalfield = ''
00132 usegaincurve = False
00133 gainopacity = 0.0
00134 calrefant = 'VA15'
00135 gainsolint = 20.0
00136 polcalfield = '2202+422'
00137 polcalmode = 'D+QU'
00138 polduvrange = ''
00139 setpolmodel = True
00140 polxfield = '0137+331'
00141 polxuvrange = ''
00142
00143 setjymode = 'set'
00144
00145
00146 srcsplitms = prefix + '.split.ms'
00147
00148
00149
00150
00151
00152
00153
00154
00155
00156
00157
00158
00159
00160
00161
00162
00163
00164
00165 clnalg = 'hogbom'
00166 usecsclean = False
00167 clnimsize = 288
00168 clncell = 0.4
00169
00170 clniter = 200
00171
00172
00173
00174
00175 clthreshold = 1.3
00176
00177
00178 clncenter = clnimsize/2
00179 clnblc = clncenter - clnimsize/8
00180 clntrc = clncenter + clnimsize/8
00181
00182 clnblc = clncenter - 10
00183 clntrc = clncenter + 10
00184 centerbox = [clnblc,clnblc,clntrc,clntrc]
00185
00186 myclnbox = centerbox
00187
00188
00189
00190
00191
00192
00193
00194
00195
00196
00197
00198 fcalmodel = {}
00199
00200
00201 fcalfield = {}
00202
00203
00204 fcalfield['0'] = [5.405,0,0,0]
00205 fcalfield['1'] = [5.458,0,0,0]
00206 fcalmodel['0137+331'] = fcalfield
00207
00208
00209 fcalfield = {}
00210 fcalfield['0'] = [2.465,0,0,0]
00211 fcalfield['1'] = [2.461,0,0,0]
00212 fcalmodel['2202+422'] = fcalfield
00213
00214
00215
00216 pcalmodel = {}
00217
00218
00219 pcalfield = {}
00220
00221
00222
00223 pcalfield['0'] = [5.405,0.041,-148.0]
00224 pcalfield['1'] = [5.458,0.041,-148.0]
00225 pcalmodel['0137+331'] = pcalfield
00226
00227
00228 pcalfield = {}
00229 pcalfield['0'] = [1.0,0.072,-55.00]
00230 pcalfield['1'] = [1.0,0.072,-55.00]
00231 pcalmodel['2202+422'] = pcalfield
00232
00233
00234
00235 print '--Setting up Polarization models--'
00236
00237 polmodel = {}
00238 for field in pcalmodel.keys() :
00239 spwmodel = {}
00240
00241 for spw in usespwlist:
00242 ipol = pcalmodel[field][spw][0]
00243 fpol = pcalmodel[field][spw][1]
00244 rlpd_deg = pcalmodel[field][spw][2]
00245 rlpd = rlpd_deg*pl.pi/180.0
00246 ppol = ipol*fpol
00247 qpol = ppol*cos(rlpd)
00248 upol = ppol*sin(rlpd)
00249 fluxdensity=[ipol,qpol,upol,0.0]
00250
00251 pmodel = {}
00252 pmodel['rlpd_deg'] = rlpd_deg
00253 pmodel['rlpd'] = rlpd
00254 pmodel['fpol'] = fpol
00255
00256 fmodel = {}
00257 fmodel['flux'] = fluxdensity
00258 fmodel['poln'] = pmodel
00259 spwmodel[spw] = fmodel
00260
00261 polmodel[field] = spwmodel
00262
00263 print "Created polmodel dictionary"
00264 print polmodel
00265
00266
00267
00268
00269
00270 if benchmarking:
00271 startTime=time.time()
00272 startProc=time.clock()
00273
00274
00275
00276
00277
00278
00279 if ( importmode == 'vla' ):
00280
00281
00282
00283 print '--ImportVLA--'
00284 default('importvla')
00285
00286 print "Use importvla to read VLA Export and make an MS"
00287
00288 archivefiles = datafile
00289 vis = msfile
00290 bandname = exportband
00291 autocorr = False
00292 antnamescheme = 'new'
00293 project = exportproject
00294
00295 saveinputs('importvla',prefix+'.importvla.saved')
00296 importvla()
00297 elif ( importmode == 'fits' ):
00298
00299
00300
00301 print '--ImportUVFITS--'
00302 default('importuvfits')
00303
00304 print "Use importuvfits to read UVFITS and make an MS"
00305
00306 fitsfile = datafile
00307 vis = msfile
00308 async = False
00309
00310 saveinputs('importuvfits',prefix+'.importuvfits.saved')
00311 importuvfits()
00312 else:
00313
00314
00315
00316 print '--MS Copy--'
00317 print "Copying "+datafile+" to "+msfile
00318
00319 os.system('cp -r '+datafile+' '+msfile)
00320 vis = msfile
00321
00322 if benchmarking:
00323 import2time=time.time()
00324
00325
00326
00327
00328 print '--Listobs--'
00329
00330 print "List summary of MS"
00331
00332 listobs()
00333
00334
00335
00336
00337
00338
00339
00340
00341
00342
00343
00344
00345
00346
00347
00348
00349
00350
00351
00352
00353
00354
00355
00356
00357
00358
00359
00360
00361
00362
00363
00364
00365
00366
00367
00368
00369
00370
00371
00372
00373
00374
00375
00376
00377
00378
00379
00380
00381
00382
00383
00384
00385
00386
00387
00388
00389
00390
00391
00392
00393
00394
00395
00396
00397
00398
00399
00400
00401
00402
00403
00404
00405
00406
00407
00408
00409
00410
00411
00412
00413
00414
00415
00416
00417
00418
00419
00420
00421
00422
00423
00424
00425
00426
00427
00428
00429
00430
00431
00432
00433
00434 if benchmarking:
00435 list2time=time.time()
00436
00437
00438
00439
00440
00441
00442 if ( myquackinterval > 0.0 ):
00443
00444
00445
00446 print '--Flagdata (scan starts)--'
00447 default('tflagdata')
00448
00449 print "Quacking scan beginnings using interval "+str(myquackinterval)
00450
00451 vis = msfile
00452 correlation = ''
00453 field = ''
00454 antenna = ''
00455 spw = usespw
00456 mode = 'quack'
00457 quackinterval = myquackinterval
00458
00459 saveinputs('tflagdata',prefix+'.tflagdata.quack.saved')
00460 tflagdata()
00461
00462
00463
00464
00465 default('flagmanager')
00466
00467 print "Now will use flagmanager to save the flags"
00468
00469 vis = msfile
00470 mode = 'save'
00471 versionname = 'quack'
00472 comment = 'Quack '+str(myquackinterval)
00473 merge = 'replace'
00474
00475 saveinputs('flagmanager',prefix+'.flagmanager.quack.saved')
00476 flagmanager()
00477
00478
00479 if (flagants != '' and not flagants.isspace() ):
00480 print '--Flagdata (antennas)--'
00481 default('tflagdata')
00482
00483 print "Flag all data to AN "+flagants
00484
00485 vis = msfile
00486 correlation = ''
00487 field = ''
00488 spw = usespw
00489 mode = 'manual'
00490 antenna = flagants
00491
00492 saveinputs('tflagdata',prefix+'.tflagdata.ants.saved')
00493 tflagdata()
00494
00495
00496
00497
00498 default('flagmanager')
00499
00500 print "Now will use flagmanager to save the flags"
00501
00502 vis = msfile
00503 mode = 'save'
00504 versionname = 'antflags'
00505 comment = 'flag AN '+flagants
00506 merge = 'replace'
00507
00508 saveinputs('flagmanager',prefix+'.flagmanager.ants.saved')
00509 flagmanager()
00510
00511 flagtimes = '19:06:50~19:06:57,19:21:17~19:21:20'
00512
00513 if (flagtimes != '' and not flagtimes.isspace() ):
00514 print '--Flagdata (timerange)--'
00515 default('tflagdata')
00516
00517 print "Flag timeranges "+flagtimes
00518
00519 vis = msfile
00520 correlation = ''
00521 field = ''
00522 spw = usespw
00523 mode = 'manual'
00524 antenna = ''
00525 timerange = flagtimes
00526
00527 saveinputs('tflagdata',prefix+'.tflagdata.times.saved')
00528 tflagdata()
00529
00530
00531
00532
00533 default('flagmanager')
00534
00535 print "Now will use flagmanager to save the flags"
00536
00537 vis = msfile
00538 mode = 'save'
00539 versionname = 'timeflags'
00540 comment = 'flag '+flagtimes
00541 merge = 'replace'
00542
00543 saveinputs('flagmanager',prefix+'.flagmanager.times.saved')
00544 flagmanager()
00545
00546 if benchmarking:
00547 flag2time=time.time()
00548
00549
00550
00551
00552
00553
00554
00555
00556 if ( setjymode == 'flux' ):
00557 print '--Setjy--'
00558 default('setjy')
00559
00560 vis = msfile
00561
00562 print "Use setjy to set flux of "+fluxcalfield+" to point model"
00563 field = fluxcalfield
00564 spw = usespw
00565
00566
00567 modimage = fluxcaldir + fluxcalmodel
00568
00569
00570 standard='Perley-Taylor 99'
00571
00572
00573 scalebychan=False
00574
00575
00576 for spw in usespwlist:
00577 fluxdensity = fcalmodel[fluxcalfield][spw]
00578 print "Setting SPW "+spw+" to "+str(fluxdensity)
00579 saveinputs('setjy',prefix+'.setjy.'+spw+'.saved')
00580 setjy()
00581
00582 elif ( setjymode == 'ft' ):
00583 print '--FT--'
00584
00585 default('ft')
00586 vis = msfile
00587 field = fluxcalfield
00588
00589 for spw in usespwlist:
00590 model = fluxcaldir + fluxcalmodel+'_'+spw+'_IQUV.model'
00591 print "Use FT to set model"+model
00592 saveinputs('ft',prefix+'.ft.0.saved')
00593 ft()
00594
00595 else:
00596 print '--Setjy--'
00597 default('setjy')
00598
00599 vis = msfile
00600
00601 print "Use setjy to set flux of "+fluxcalfield
00602 field = fluxcalfield
00603 spw = usespw
00604
00605
00606 modimage = fluxcaldir + fluxcalmodel
00607
00608
00609 standard='Perley-Taylor 99'
00610
00611 scalebychan=False
00612
00613 saveinputs('setjy',prefix+'.setjy.saved')
00614 setjy()
00615
00616
00617
00618
00619
00620
00621
00622
00623
00624
00625 print "Look in logger for the fluxes (should be 5.405 and 5.458 Jy)"
00626
00627 if benchmarking:
00628 setjy2time=time.time()
00629
00630
00631
00632
00633
00634 print '--Gaincal--'
00635 default('gaincal')
00636
00637 print "Solve for antenna gains on sources "+gaincalfield
00638 print "We have 2 single-channel continuum spw"
00639
00640 vis = msfile
00641
00642
00643 print "Output gain table name is "+gtable
00644 caltable = gtable
00645
00646
00647
00648
00649
00650 field = fieldgain
00651 print "Calibrating using fields "+field
00652
00653
00654 spw = usespw
00655
00656
00657 gaincurve = usegaincurve
00658 opacity = gainopacity
00659
00660
00661 parang = False
00662
00663
00664 gaintype = 'G'
00665 solint = gainsolint
00666 calmode = 'ap'
00667
00668
00669 refant = calrefant
00670
00671
00672 minsnr = 3
00673
00674 saveinputs('gaincal',prefix+'.gaincal.saved')
00675 gaincal()
00676
00677
00678
00679 if benchmarking:
00680 gaincal2time=time.time()
00681
00682
00683
00684
00685
00686 print '--Listcal--'
00687
00688 listfile = caltable + '.list'
00689
00690 print "Listing calibration to file "+listfile
00691
00692 listcal()
00693
00694 if benchmarking:
00695 listgcal2time=time.time()
00696
00697
00698
00699
00700
00701
00702 print '--Fluxscale--'
00703 default('fluxscale')
00704
00705 print "Use fluxscale to rescale gain table to make new one"
00706
00707 vis = msfile
00708
00709
00710 ftable = prefix + '.fluxscale'
00711 fluxtable = ftable
00712
00713 print "Output scaled gain cal table is "+ftable
00714
00715
00716 caltable = gtable
00717
00718
00719 reference = fluxcalfield
00720
00721
00722
00723 transfer = fieldgain
00724
00725 saveinputs('fluxscale',prefix+'.fluxscale.saved')
00726 fluxscale()
00727
00728
00729
00730
00731
00732
00733
00734
00735
00736
00737
00738
00739
00740
00741
00742
00743
00744
00745
00746
00747
00748
00749
00750 if benchmarking:
00751 fluxscale2time=time.time()
00752
00753
00754
00755
00756
00757 print '--Listcal--'
00758
00759 caltable = ftable
00760 listfile = caltable + '.list'
00761
00762 print "Listing calibration to file "+listfile
00763
00764 listcal()
00765
00766 if benchmarking:
00767 listfcal2time=time.time()
00768
00769
00770
00771
00772
00773 print '--Plotcal--'
00774
00775 iteration = ''
00776 showgui = False
00777
00778 xaxis = 'time'
00779 yaxis = 'amp'
00780 figfile = caltable + '.plot.amp.png'
00781 print "Plotting calibration to file "+figfile
00782 saveinputs('plotcal',prefix+'.plotcal.fluxscale.amp.saved')
00783 plotcal()
00784
00785 xaxis = 'time'
00786 yaxis = 'phase'
00787 figfile = caltable + '.plot.phase.png'
00788 print "Plotting calibration to file "+figfile
00789 saveinputs('plotcal',prefix+'.plotcal.fluxscale.phase.saved')
00790 plotcal()
00791
00792 xaxis = 'antenna'
00793 yaxis = 'amp'
00794 figfile = caltable + '.plot.antamp.png'
00795 print "Plotting calibration to file "+figfile
00796 saveinputs('plotcal',prefix+'.plotcal.fluxscale.antamp.saved')
00797 plotcal()
00798
00799 if benchmarking:
00800 plotcal2time=time.time()
00801
00802 dosetpoljy=False
00803 if ( setpolmodel and polcalmode.count('X') > 0 ):
00804 dosetpoljy=True
00805
00806
00807
00808
00809
00810 print '--Setjy--'
00811 default('setjy')
00812
00813 vis = msfile
00814
00815 print "Use setjy to set IQU fluxes of "+polxfield
00816 field = polxfield
00817
00818 scalebychan=False
00819
00820 for spw in usespwlist:
00821 fluxdensity = polmodel[field][spw]['flux']
00822
00823 saveinputs('setjy',prefix+'.setjy.polspw.'+spw+'.saved')
00824 setjy()
00825
00826 if benchmarking:
00827 setpoljy2time=time.time()
00828
00829
00830
00831
00832
00833 print '--PolCal--'
00834 default('polcal')
00835
00836 print "Polarization D-term Calibration (linear approx) on "+polcalfield
00837
00838 vis = msfile
00839
00840
00841 gaintable = gtable
00842
00843
00844 gaincurve = usegaincurve
00845 opacity = gainopacity
00846
00847
00848 ptable = prefix + '.pcal'
00849 caltable = ptable
00850
00851
00852 field = polcalfield
00853 spw = usespw
00854
00855 selectdata=True
00856 uvrange = polduvrange
00857
00858
00859 poltype = polcalmode
00860
00861
00862 solint = 86400.
00863
00864
00865 refant = calrefant
00866
00867
00868 minsnr = 3
00869
00870 saveinputs('polcal',prefix+'.polcal.saved')
00871 polcal()
00872
00873
00874
00875
00876
00877
00878
00879 if benchmarking:
00880 polcal2time=time.time()
00881
00882
00883
00884
00885
00886 print '--Listcal--'
00887
00888 listfile = caltable + '.list'
00889
00890 print "Listing calibration to file "+listfile
00891
00892 listcal()
00893
00894 if benchmarking:
00895 listpcal2time=time.time()
00896
00897
00898
00899
00900
00901 print '--Plotcal--'
00902
00903 iteration = ''
00904 showgui = False
00905
00906 xaxis = 'real'
00907 yaxis = 'imag'
00908 figfile = caltable + '.plot.reim.png'
00909 print "Plotting calibration to file "+figfile
00910 saveinputs('plotcal',prefix+'.plotcal.polcal.d.reim.saved')
00911 plotcal()
00912
00913 xaxis = 'antenna'
00914 yaxis = 'amp'
00915 figfile = caltable + '.plot.antamp.png'
00916 print "Plotting calibration to file "+figfile
00917 saveinputs('plotcal',prefix+'.plotcal.polcal.d.antamp.saved')
00918 plotcal()
00919
00920 xaxis = 'antenna'
00921 yaxis = 'phase'
00922 figfile = caltable + '.plot.antphase.png'
00923 print "Plotting calibration to file "+figfile
00924 saveinputs('plotcal',prefix+'.plotcal.polcal.d.antphase.saved')
00925 plotcal()
00926
00927 xaxis = 'antenna'
00928 yaxis = 'snr'
00929 figfile = caltable + '.plot.antsnr.png'
00930 print "Plotting calibration to file "+figfile
00931 saveinputs('plotcal',prefix+'.plotcal.polcal.d.antsnr.saved')
00932 plotcal()
00933
00934 if benchmarking:
00935 plotpcal2time=time.time()
00936
00937
00938
00939
00940
00941 dopolx = False
00942 if ( pcalmodel.has_key(polxfield) ):
00943 dopolx = True
00944
00945 if ( setpolmodel and not polcalmode.count('X') > 0 ):
00946
00947
00948
00949
00950
00951
00952 print '--Setjy--'
00953 default('setjy')
00954
00955 vis = msfile
00956
00957 print "Use setjy to set IQU fluxes of "+polxfield
00958 field = polxfield
00959
00960 scalebychan=False
00961
00962 for spw in usespwlist:
00963 fluxdensity = polmodel[field][spw]['flux']
00964
00965 saveinputs('setjy',prefix+'.setjy.polspw.'+spw+'.saved')
00966 setjy()
00967
00968
00969 if benchmarking:
00970 setxjy2time=time.time()
00971
00972
00973
00974
00975
00976
00977 print '--PolCal--'
00978 default('polcal')
00979
00980 print "Polarization R-L Phase Calibration (linear approx)"
00981
00982 vis = msfile
00983
00984
00985 gaintable = [gtable,ptable]
00986
00987
00988 gaincurve = usegaincurve
00989 opacity = gainopacity
00990
00991
00992 xtable = prefix + '.polx'
00993 caltable = xtable
00994
00995
00996 field = polxfield
00997 spw = usespw
00998
00999 selectdata=True
01000 uvrange = polxuvrange
01001
01002
01003 poltype = 'X'
01004 solint = 86400.
01005
01006
01007 refant = calrefant
01008
01009
01010 minsnr = 3
01011
01012 saveinputs('polcal',prefix+'.polcal.X.saved')
01013 polcal()
01014
01015
01016
01017
01018
01019 if benchmarking:
01020 xpolcal2time=time.time()
01021
01022
01023
01024
01025
01026
01027
01028
01029
01030
01031
01032
01033
01034
01035
01036
01037
01038
01039
01040 print '--Plotcal--'
01041
01042 xaxis = 'antenna'
01043 yaxis = 'phase'
01044 iteration = ''
01045
01046 showgui = False
01047 figfile = caltable + '.plot.png'
01048
01049 print "Plotting calibration to file "+figfile
01050 saveinputs('plotcal',prefix+'.plotcal.polcal.x.antphase.saved')
01051 plotcal()
01052
01053 if benchmarking:
01054 plotxcal2time=time.time()
01055
01056 else:
01057 if (polxfield != '' and not polxfield.isspace() ):
01058 print "DO NOT HAVE PCALMODEL FOR "+polxfield
01059 print "PCALMODEL = ",pcalmodel
01060
01061 if benchmarking:
01062 setxjy2time=time.time()
01063 xpolcal2time=time.time()
01064 plotxcal2time=time.time()
01065
01066
01067
01068
01069
01070
01071
01072
01073
01074 print '--ApplyCal--'
01075 default('applycal')
01076
01077 print "This will apply the calibration to the DATA"
01078 print "Fills CORRECTED_DATA"
01079
01080 vis = msfile
01081
01082
01083 if (dopolx):
01084 gaintable = [ftable,ptable,xtable]
01085 else:
01086 gaintable = [ftable,ptable]
01087
01088
01089 gaincurve = usegaincurve
01090 opacity = gainopacity
01091
01092
01093 spw = usespw
01094 selectdata = False
01095
01096
01097 parang = True
01098
01099
01100 field = fieldgain
01101 gainselect = field
01102 print "Applying calibration to gain calibrators "+field
01103
01104 saveinputs('applycal',prefix+'.applycal.saved')
01105 applycal()
01106
01107 if ( len(targets) > 0 ):
01108
01109
01110
01111
01112 field = fieldtargets
01113 print "Applying calibration to targets "+field
01114
01115 saveinputs('applycal',prefix+'.applycal.targets.saved')
01116 applycal()
01117
01118 if benchmarking:
01119 correct2time=time.time()
01120
01121
01122
01123
01124
01125
01126 print '--Split--'
01127 default('split')
01128
01129 vis = msfile
01130
01131
01132
01133
01134 srcsplitms = prefix + '.split.ms'
01135 outputvis = srcsplitms
01136
01137
01138 field = ''
01139
01140
01141 spw = ''
01142
01143
01144 datacolumn = 'corrected'
01145
01146 print "Split CORRECTED_DATA into DATA in new ms "+srcsplitms
01147
01148 saveinputs('split',prefix+'.split.saved')
01149 split()
01150
01151 if benchmarking:
01152 split2time=time.time()
01153
01154
01155
01156
01157
01158
01159 print '--Plotxy--'
01160 default('plotxy')
01161
01162 vis = srcsplitms
01163
01164 field = fluxcalfield
01165 spw = ''
01166
01167 selectdata=True
01168
01169 interactive=False
01170
01171 correlation='RR LL'
01172 xaxis = 'time'
01173 yaxis = 'amp'
01174 figfile = prefix+'.split.'+field+'.plotxy.amptime.png'
01175 saveinputs('plotxy',prefix+'.plotxy.'+field+'.amptime.saved')
01176 plotxy()
01177
01178 xaxis = 'uvdist'
01179 figfile = prefix+'.split.'+field+'.plotxy.ampuv.png'
01180 saveinputs('plotxy',prefix+'.plotxy.'+field+'.ampuv.saved')
01181 plotxy()
01182
01183 correlation='RL LR'
01184 yaxis = 'phase'
01185 figfile = prefix+'.split.'+field+'.plotxy.rlphaseuv.png'
01186 saveinputs('plotxy',prefix+'.plotxy.'+field+'.rlphase.saved')
01187 plotxy()
01188
01189 if ( polcalfield != fluxcalfield ):
01190
01191 field = polcalfield
01192
01193 correlation='RR LL'
01194 xaxis = 'time'
01195 yaxis = 'amp'
01196 figfile = prefix+'.split.'+field+'.plotxy.amptime.png'
01197 saveinputs('plotxy',prefix+'.plotxy.'+field+'.amptime.saved')
01198 plotxy()
01199
01200 xaxis = 'uvdist'
01201 figfile = prefix+'.split.'+field+'.plotxy.ampuv.png'
01202 saveinputs('plotxy',prefix+'.plotxy.'+field+'.ampuv.saved')
01203 plotxy()
01204
01205 correlation='RL LR'
01206 yaxis = 'phase'
01207 figfile = prefix+'.split.'+field+'.plotxy.rlphaseuv.png'
01208 saveinputs('plotxy',prefix+'.plotxy.'+field+'.rlphase.saved')
01209 plotxy()
01210
01211 if benchmarking:
01212 plotxyfinal2time=time.time()
01213
01214
01215
01216
01217
01218
01219 clnmodel = {}
01220
01221
01222
01223
01224
01225 for src in srclist:
01226
01227 srcmodel = {}
01228
01229 for spwid in usespwlist:
01230
01231 print '-- Clean '+src+' spw '+spwid+' --'
01232 default('clean')
01233
01234 field = src
01235 spw = spwid
01236
01237
01238 vis = srcsplitms
01239
01240
01241 imname1 = prefix + '.' + src + '.' + spwid + '.clean'
01242 imagename = imname1
01243
01244 print " Output images will be prefixed with "+imname1
01245
01246
01247 mode = 'mfs'
01248
01249
01250 stokes = 'IQUV'
01251
01252
01253 psfmode = clnalg
01254 csclean = usecsclean
01255 imagermode=''
01256 if csclean:
01257 imagermode='csclean'
01258
01259
01260 imsize = [clnimsize,clnimsize]
01261 cell = [clncell,clncell]
01262
01263
01264 gain = 0.1
01265
01266 niter = clniter
01267
01268 threshold = clthreshold
01269
01270
01271
01272 weighting = 'briggs'
01273 robust = 0.5
01274
01275 weighting = 'natural'
01276
01277
01278 mask = myclnbox
01279
01280 saveinputs('clean',prefix+'.clean.'+src+'.'+spwid+'.saved')
01281 clean()
01282
01283
01284 clnimage1 = imname1+'.image'
01285 clnmodel1 = imname1+'.model'
01286 clnresid1 = imname1+'.residual'
01287 clnmask1 = imname1+'.mask'
01288 clnpsf1 = imname1+'.psf'
01289 clnflux1 = imname1+'.flux'
01290
01291
01292
01293
01294
01295
01296 default('imstat')
01297
01298 field = src
01299 spw = spwid
01300
01301
01302 mybox = str(clnblc)+','+str(clnblc)+','+str(clntrc)+','+str(clntrc)
01303
01304 spwmodel = {}
01305
01306 spwstats = {}
01307 spwfluxes = {}
01308 spwsum = {}
01309 spwmod = {}
01310
01311 for stokes in ['I','Q','U','V']:
01312
01313
01314 imagename = clnimage1
01315 box = mybox
01316
01317 saveinputs('imstat',prefix+'.imstat.'+src+'.'+spwid+'.'+stokes+'.saved')
01318 xstat = imstat()
01319
01320 spwstats[stokes] = xstat
01321
01322
01323 xmax = xstat['max'][0]
01324 xmin = xstat['min'][0]
01325 if( abs(xmin) > abs(xmax) ):
01326 xpol = xmin
01327 else:
01328 xpol = xmax
01329
01330 spwfluxes[stokes]= xpol
01331
01332
01333 xsum = xstat['flux'][0]
01334 spwsum[stokes]= xsum
01335
01336
01337 imagename = clnmodel1
01338 box = ''
01339
01340 saveinputs('imstat',prefix+'.imstat.'+src+'.'+spwid+'.'+stokes+'.model.saved')
01341 xstat = imstat()
01342
01343 xmod = xstat['sum'][0]
01344 spwmod[stokes]= xmod
01345
01346
01347
01348 spwmodel['stat'] = spwstats
01349 spwmodel['flux'] = spwfluxes
01350 spwmodel['integ'] = spwsum
01351 spwmodel['model'] = spwmod
01352
01353
01354 imagename = clnimage1
01355
01356 spwref = {}
01357
01358
01359
01360
01361 myhead = imhead(imagename,'list')
01362 xref = int(myhead['crpix1'])
01363 yref = int(myhead['crpix2'])
01364
01365
01366 refbox = str(xref)+','+str(yref)+','+str(xref)+','+str(yref)
01367 ipix = imval(imagename,box=refbox,stokes='I')
01368 iflx = ipix['data'][0]
01369 spwref['I'] = iflx
01370
01371
01372
01373
01374 qpix = imval(imagename,box=refbox,stokes='Q')
01375 qflx = qpix['data'][0]
01376 spwref['Q'] = qflx
01377
01378
01379
01380
01381 upix = imval(imagename,box=refbox,stokes='U')
01382 uflx = upix['data'][0]
01383 spwref['U'] = uflx
01384
01385
01386
01387
01388 vpix = imval(imagename,box=refbox,stokes='V')
01389 vflx = vpix['data'][0]
01390 spwref['V'] = vflx
01391
01392
01393 pflx = sqrt( qflx**2 + uflx**2 )
01394 fflx = pflx/iflx
01395 xflx = atan2(uflx,qflx)*180.0/pi
01396 spwref['P'] = pflx
01397 spwref['F'] = fflx
01398 spwref['X'] = xflx
01399 spwref['xref'] = xref
01400 spwref['yref'] = yref
01401
01402
01403
01404 spwmax = {}
01405
01406
01407 xref = spwstats['I']['maxpos'][0]
01408 yref = spwstats['I']['maxpos'][1]
01409 refbox = str(xref)+','+str(yref)+','+str(xref)+','+str(yref)
01410
01411
01412 iflx = spwstats['I']['max'][0]
01413 spwmax['I'] = iflx
01414
01415
01416
01417
01418 qpix = imval(imagename,box=refbox,stokes='Q')
01419 qflx = qpix['data'][0]
01420 spwmax['Q'] = qflx
01421
01422
01423
01424
01425 upix = imval(imagename,box=refbox,stokes='U')
01426 uflx = upix['data'][0]
01427 spwmax['U'] = uflx
01428
01429
01430
01431
01432 vpix = imval(imagename,box=refbox,stokes='V')
01433 vflx = vpix['data'][0]
01434 spwmax['V'] = vflx
01435
01436 spwmax['xref'] = xref
01437 spwmax['yref'] = yref
01438
01439
01440
01441 spwmodel['refval'] = spwref
01442 spwmodel['maxval'] = spwmax
01443
01444 srcmodel[spwid] = spwmodel
01445
01446
01447
01448 clnmodel[src] = srcmodel
01449
01450 if benchmarking:
01451 clean2time=time.time()
01452
01453
01454
01455 if benchmarking:
01456 endProc=time.clock()
01457 endTime=time.time()
01458
01459
01460
01461
01462
01463 regression = {}
01464 regressmodel = {}
01465 regressfile = regressdir + scriptprefix + '.pickle'
01466
01467 try:
01468 fr = open(regressfile,'r')
01469 except:
01470 print "No previous regression results file "+regressfile
01471 else:
01472 u = pickle.Unpickler(fr)
01473 regression = u.load()
01474 fr.close()
01475 print "Previous regression results filled from "+regressfile
01476
01477 if regression.has_key('results'):
01478 print " on "+regression['host']+" for "+regression['version']+" at "+regression['date']
01479 regressmodel = regression['results']
01480 else:
01481
01482 regressmodel = regression
01483
01484
01485
01486
01487
01488 print 'Results for '+prefix+' :'
01489 print ""
01490
01491 import datetime
01492 datestring=datetime.datetime.isoformat(datetime.datetime.today())
01493
01494 outfile = 'out.'+prefix+'.'+datestring+'.log'
01495 logfile=open(outfile,'w')
01496
01497
01498 myvers = casalog.version()
01499 myuser = os.getenv('USER')
01500 myhost = str( os.getenv('HOST') )
01501 mycwd = os.getcwd()
01502 myos = os.uname()
01503
01504
01505 print >>logfile,'Running '+myvers+' on host '+myhost
01506 print >>logfile,'at '+datestring
01507 print >>logfile,''
01508
01509 print >>logfile,'Results for '+prefix+' :'
01510 print >>logfile,""
01511
01512 if ( polmodel.has_key(polxfield) ):
01513
01514 print "R-L phase residual from image of "+polxfield
01515 print ""
01516 print >>logfile,"R-L phase residual from image of "+polxfield+" :"
01517 print >>logfile,""
01518
01519 src = polxfield
01520 rlcor = {}
01521
01522 for spwid in usespwlist:
01523 ipol = clnmodel[src][spwid]['flux']['I']
01524 qpol = clnmodel[src][spwid]['flux']['Q']
01525 upol = clnmodel[src][spwid]['flux']['U']
01526 vpol = clnmodel[src][spwid]['flux']['V']
01527 rlpd = atan2(upol,qpol)
01528 rlpdcal = polmodel[src][spwid]['poln']['rlpd']
01529 rlpcor = rlpdcal - rlpd
01530 scor = sin(rlpcor); ccor = cos(rlpcor); rlpcor = atan2(scor,ccor)
01531 rlcor[spwid] = rlpcor
01532 rlpcor_deg = rlpcor*180.0/pl.pi
01533
01534 print "R-L Phase Correction SPW "+spwid+" = %7.2f deg" % rlpcor_deg
01535 print >>logfile,"R-L Phase Correction SPW "+spwid+" = %7.2f deg" % rlpcor_deg
01536
01537
01538 if regression.has_key('results'):
01539 print >>logfile,""
01540 print >>logfile,"Regression versus "+regression['version']+" on host "+regression['host']+" at "+regression['date']
01541
01542
01543
01544
01545
01546
01547 print ""
01548 print "Final Stats:"
01549 print ""
01550
01551 print >>logfile,""
01552 print >>logfile,"Final Stats:"
01553 print >>logfile,""
01554
01555 new_regression = {}
01556
01557
01558 new_regression['date'] = datestring
01559 new_regression['version'] = myvers
01560 new_regression['user'] = myuser
01561 new_regression['host'] = myhost
01562 new_regression['cwd'] = mycwd
01563 new_regression['os'] = myos
01564
01565 new_regression['dataset'] = 'VLA POLCAL 20080224'
01566
01567 outpolmodel = {}
01568
01569 passfail = True
01570 for src in srclist:
01571
01572 print "Source "+src+" :"
01573 print >>logfile,"Source "+src+" :"
01574
01575 outpolsrc = {}
01576 for spwid in usespwlist:
01577
01578 field = src
01579 spw = spwid
01580
01581
01582
01583 ipol = clnmodel[src][spwid]['flux']['I']
01584 qpol = clnmodel[src][spwid]['flux']['Q']
01585 upol = clnmodel[src][spwid]['flux']['U']
01586 vpol = clnmodel[src][spwid]['flux']['V']
01587
01588
01589
01590 ppol = sqrt(qpol**2 + upol**2)
01591 fpol = ppol/ipol
01592 rlpd = atan2(upol,qpol)
01593 rlpd_deg = rlpd*180.0/pl.pi
01594
01595 outpolspw = {}
01596 outpolspw['ipol']=ipol
01597 outpolspw['ppol']=ppol
01598 outpolspw['fpol']=fpol
01599 outpolspw['rlpd']=rlpd
01600 outpolspw['rlpd_deg']=rlpd_deg
01601
01602 outpolsrc[spwid] = outpolspw
01603
01604
01605
01606 print ' spw %s CASA I = %7.3f P = %7.3f F = %7.4f X = %7.2f deg' %\
01607 (spwid,ipol,ppol,fpol,rlpd_deg)
01608 print >>logfile,' spw %s CASA I = %7.3f P = %7.3f F = %7.4f X = %7.2f deg' %\
01609 (spwid,ipol,ppol,fpol,rlpd_deg)
01610
01611 if (regressmodel.has_key(src)):
01612 iflx = regressmodel[src][spwid]['ipol']
01613 fflx = regressmodel[src][spwid]['fpol']
01614 rlreg = regressmodel[src][spwid]['rlpd']
01615 rlreg_deg = regressmodel[src][spwid]['rlpd_deg']
01616
01617 pflx = iflx*fflx
01618 qflx = pflx*cos(rlreg)
01619 uflx = pflx*sin(rlreg)
01620 vflx = 0.0
01621
01622 print ' spw %s PREV I = %7.3f P = %7.3f F = %7.4f X = %7.2f deg' %\
01623 (spwid,iflx,pflx,fflx,rlreg_deg)
01624 print >>logfile,' spw %s PREV I = %7.3f P = %7.3f F = %7.4f X = %7.2f deg' %\
01625 (spwid,iflx,pflx,fflx,rlreg_deg)
01626
01627 ipol_diff = ipol - iflx
01628 ppol_diff = ppol - pflx
01629 fpol_diff = fpol - fflx
01630 rldiff = rlpd - rlreg
01631 rlpd_diff = atan2( sin(rldiff), cos(rldiff) )
01632 rlpd_diff_deg = rlpd_diff*180.0/pl.pi
01633
01634 test = (abs(ipol_diff/ipol) < 0.08) & (abs(fpol_diff) < 0.008) & (abs(rlpd_diff < 0.08))
01635 if test:
01636 teststr = 'PASS'
01637 else:
01638 teststr = 'FAIL'
01639
01640 passfail = passfail & test
01641
01642 print ' spw %s DIFF I = %7.3f P = %7.3f F = %7.4f X = %7.2f deg %s ' %\
01643 (spwid,ipol_diff,ppol_diff,fpol_diff,rlpd_diff_deg,teststr)
01644 print >>logfile,' spw %s PREV I = %7.3f P = %7.3f F = %7.4f X = %7.2f deg %s ' %\
01645 (spwid,ipol_diff,ppol_diff,fpol_diff,rlpd_diff_deg,teststr)
01646 print ''
01647 print >>logfile,""
01648
01649
01650 outpolmodel[src] = outpolsrc
01651
01652 if (regressmodel.has_key(src)):
01653 pass
01654 else:
01655 print ""
01656 print >>logfile,""
01657
01658
01659 if (regressmodel.has_key(src)):
01660 if passfail:
01661 passfailstr = 'PASSED'
01662 else:
01663 passfailstr = 'FAILED'
01664
01665 print 'POLCAL Regression '+passfailstr
01666 print >>logfile,'POLCAL Regression '+passfailstr
01667
01668 print ''
01669 print 'Regression '+passfailstr
01670 print ''
01671
01672 new_regression['results'] = outpolmodel
01673
01674
01675
01676
01677
01678
01679
01680
01681
01682
01683
01684
01685
01686
01687
01688
01689
01690
01691
01692
01693
01694
01695
01696
01697
01698
01699
01700
01701
01702
01703
01704
01705
01706
01707
01708
01709
01710
01711
01712
01713
01714
01715
01716
01717
01718
01719
01720
01721
01722
01723
01724
01725 if benchmarking:
01726 print >>logfile,''
01727 print >>logfile,'********* Benchmarking *****************'
01728 print >>logfile,'* *'
01729 print >>logfile,'Total wall clock time was: '+str(endTime - startTime)
01730 print >>logfile,'Total CPU time was: '+str(endProc - startProc)
01731 print >>logfile,'Processing rate MB/s was: '+str(300./(endTime - startTime))
01732 print >>logfile,'* Breakdown: *'
01733 print >>logfile,'* import time was: '+str(import2time-startTime)
01734 print >>logfile,'* listobs time was: '+str(list2time-import2time)
01735 print >>logfile,'* tflagdata time was: '+str(flag2time-list2time)
01736 print >>logfile,'* setjy time was: '+str(setjy2time-flag2time)
01737 print >>logfile,'* gaincal time was: '+str(gaincal2time-setjy2time)
01738 print >>logfile,'* listcal(G) time was: '+str(listgcal2time-gaincal2time)
01739 print >>logfile,'* fluxscale time was: '+str(fluxscale2time-listgcal2time)
01740 print >>logfile,'* listcal(F) time was: '+str(listfcal2time-fluxscale2time)
01741 print >>logfile,'* plotcal(F) time was: '+str(plotcal2time-listfcal2time)
01742
01743 if dosetpoljy:
01744 print >>logfile,'* setjy(D) time was: '+str(setpoljy2time-plotcal2time)
01745 print >>logfile,'* polcal(D) time was: '+str(polcal2time-setpoljy2time)
01746 else:
01747 print >>logfile,'* polcal(D) time was: '+str(polcal2time-plotcal2time)
01748
01749 print >>logfile,'* listcal(D) time was: '+str(listpcal2time-polcal2time)
01750 print >>logfile,'* plotcal(D) time was: '+str(plotpcal2time-listpcal2time)
01751
01752 if dopolx:
01753 print >>logfile,'* setjy(X) time was: '+str(setxjy2time-plotpcal2time)
01754 print >>logfile,'* polcal(X) time was: '+str(xpolcal2time-setxjy2time)
01755 print >>logfile,'* plotcal(X) time was: '+str(plotxcal2time-xpolcal2time)
01756 print >>logfile,'* applycal time was: '+str(correct2time-plotxcal2time)
01757 else:
01758 print >>logfile,'* applycal time was: '+str(correct2time-plotpcal2time)
01759
01760 print >>logfile,'* split time was: '+str(split2time-correct2time)
01761 print >>logfile,'* plotxy time was: '+str(plotxyfinal2time-split2time)
01762 print >>logfile,'* clean/stat time was: '+str(clean2time-plotxyfinal2time)
01763 print >>logfile,'*****************************************'
01764 print >>logfile,'sandrock (2008-06-17) wall time was: 255 seconds'
01765 print >>logfile,'sandrock (2008-06-17) CPU time was: 233 seconds'
01766
01767 logfile.close()
01768
01769 print ''
01770 if benchmarking:
01771 print 'Total wall clock time was: '+str(endTime - startTime)
01772 print 'Total CPU time was: '+str(endProc - startProc)
01773 print 'Processing rate MB/s was: '+str(300./(endTime - startTime))
01774 print ''
01775
01776 print "Done with NGC4826 Tutorial Regression"
01777
01778
01779
01780 logfile.close()
01781 print "Results are in "+outfile
01782
01783
01784
01785
01786
01787
01788 pickfile = 'out.'+prefix + '.polmodel.'+datestring+'.pickle'
01789 f = open(pickfile,'w')
01790 p = pickle.Pickler(f)
01791
01792 p.dump(new_regression)
01793
01794 p.dump(clnmodel)
01795 p.dump(polmodel)
01796 f.close()
01797 print ""
01798 print "Dictionaries new_regression,clnmodel,polmodel saved in "+pickfile
01799 print ""
01800 print "Use Pickle to retrieve these"
01801 print ""
01802
01803
01804
01805
01806
01807
01808
01809
01810
01811 print ""
01812 print "Completed POLCAL Regression"