def ptclean3(vis, imageprefix, imagesuffix, ncpu, twidth, doreg, usephacenter,
reftime, toTb, overwrite, selectdata, field, spw, timerange,
uvrange, antenna, scan, observation, intent, datacolumn, imsize,
cell, phasecenter, stokes, projection, startmodel, specmode,
reffreq, nchan, start, width, outframe, veltype, restfreq,
interpolation, gridder, facets, chanchunks, wprojplanes, vptable,
usepointing, mosweight, aterm, psterm, wbawp, conjbeams, cfcache,
computepastep, rotatepastep, pblimit, normtype, deconvolver,
scales, nterms, smallscalebias, restoration, restoringbeam, pbcor,
outlierfile, weighting, robust, npixels, uvtaper, niter, gain,
threshold, nsigma, cycleniter, cyclefactor, minpsffraction,
maxpsffraction, interactive, usemask, mask, pbmask,
sidelobethreshold, noisethreshold, lownoisethreshold,
negativethreshold, smoothfactor, minbeamfrac, cutthreshold,
growiterations, dogrowprune, minpercentchange, verbose, restart,
savemodel, calcres, calcpsf, parallel, subregion):
if not (type(ncpu) is int):
casalog.post('ncpu should be an integer')
ncpu = 8
if doreg:
# check if ephem and msinfo exist. If not, generate one on the fly
try:
ephem = hf.read_horizons(vis=vis)
except ValueError:
print("error in obtaining ephemeris")
try:
msinfo = hf.read_msinfo(vis)
except ValueError:
print("error in getting ms info")
else:
ephem = None
msinfo = None
if imageprefix:
workdir = os.path.dirname(imageprefix)
else:
workdir = './'
tmpdir = workdir + '/tmp/'
if not os.path.exists(tmpdir):
os.makedirs(tmpdir)
# get number of time pixels
ms.open(vis)
ms.selectinit()
timfreq = ms.getdata(['time', 'axis_info'], ifraxis=True)
tim = timfreq['time']
ms.close()
if twidth < 1:
casalog.post('twidth less than 1. Change to 1')
twidth = 1
if twidth > len(tim):
casalog.post(
'twidth greater than # of time pixels in the dataset. Change to the timerange of the entire dateset'
)
twidth = len(tim)
# find out the start and end time index according to the parameter timerange
# if not defined (empty string), use start and end from the entire time of the ms
if not timerange:
btidx = 0
etidx = len(tim) - 1
else:
try:
(tstart, tend) = timerange.split('~')
bt_s = qa.convert(qa.quantity(tstart, 's'), 's')['value']
et_s = qa.convert(qa.quantity(tend, 's'), 's')['value']
# only time is given but not date, add the date (at 0 UT) from the first record
if bt_s < 86400. or et_s < 86400.:
bt_s += np.fix(
qa.convert(qa.quantity(tim[0], 's'),
'd')['value']) * 86400.
et_s += np.fix(
qa.convert(qa.quantity(tim[0], 's'),
'd')['value']) * 86400.
btidx = np.argmin(np.abs(tim - bt_s))
etidx = np.argmin(np.abs(tim - et_s))
# make the indice back to those bracket by the timerange
if tim[btidx] < bt_s:
btidx += 1
if tim[etidx] > et_s:
etidx -= 1
if etidx <= btidx:
print("ending time must be greater than starting time")
print("reinitiating to the entire time range")
btidx = 0
etidx = len(tim) - 1
except ValueError:
print("keyword 'timerange' has a wrong format")
btstr = qa.time(qa.quantity(tim[btidx], 's'), prec=9, form='fits')[0]
etstr = qa.time(qa.quantity(tim[etidx], 's'), prec=9, form='fits')[0]
iterable = range(btidx, etidx + 1, twidth)
print('First time pixel: ' + btstr)
print('Last time pixel: ' + etstr)
print(str(len(iterable)) + ' images to clean...')
res = []
# partition
clnpart = partial(
clean_iter, tim, vis, imageprefix, imagesuffix, twidth, doreg,
usephacenter, reftime, ephem, msinfo, toTb, overwrite, selectdata,
field, spw, uvrange, antenna, scan, observation, intent, datacolumn,
imsize, cell, phasecenter, stokes, projection, startmodel, specmode,
reffreq, nchan, start, width, outframe, veltype, restfreq,
interpolation, gridder, facets, chanchunks, wprojplanes, vptable,
usepointing, mosweight, aterm, psterm, wbawp, conjbeams, cfcache,
computepastep, rotatepastep, pblimit, normtype, deconvolver, scales,
nterms, smallscalebias, restoration, restoringbeam, pbcor, outlierfile,
weighting, robust, npixels, uvtaper, niter, gain, threshold, nsigma,
cycleniter, cyclefactor, minpsffraction, maxpsffraction, interactive,
usemask, mask, pbmask, sidelobethreshold, noisethreshold,
lownoisethreshold, negativethreshold, smoothfactor, minbeamfrac,
cutthreshold, growiterations, dogrowprune, minpercentchange, verbose,
restart, savemodel, calcres, calcpsf, parallel, subregion, tmpdir)
timelapse = 0
t0 = time()
# parallelization
if ncpu > 1:
import multiprocessing as mprocs
casalog.post('Perform clean in parallel ...')
print('Perform clean in parallel ...')
pool = mprocs.Pool(ncpu)
res = pool.map(clnpart, iterable)
pool.close()
pool.join()
else:
casalog.post('Perform clean in single process ...')
print('Perform clean in single process ...')
for i in iterable:
res.append(clnpart(i))
t1 = time()
timelapse = t1 - t0
print('It took %f secs to complete' % timelapse)
# repackage this into a single dictionary
results = {
'Succeeded': [],
'BeginTime': [],
'EndTime': [],
'ImageName': []
}
for r in res:
results['Succeeded'].append(r[0])
results['BeginTime'].append(r[1])
results['EndTime'].append(r[2])
results['ImageName'].append(r[3])
if os.path.exists(tmpdir):
os.system('rm -rf ' + tmpdir)
return results
def clean_iter(
tim, vis, imageprefix, imagesuffix, twidth, doreg, usephacenter,
reftime, ephem, msinfo, toTb, overwrite, selectdata, field, spw,
uvrange, antenna, scan, observation, intent, datacolumn, imsize, cell,
phasecenter, stokes, projection, startmodel, specmode, reffreq, nchan,
start, width, outframe, veltype, restfreq, interpolation, gridder,
facets, chanchunks, wprojplanes, vptable, usepointing, mosweight,
aterm, psterm, wbawp, conjbeams, cfcache, computepastep, rotatepastep,
pblimit, normtype, deconvolver, scales, nterms, smallscalebias,
restoration, restoringbeam, pbcor, outlierfile, weighting, robust,
npixels, uvtaper, niter, gain, threshold, nsigma, cycleniter,
cyclefactor, minpsffraction, maxpsffraction, interactive, usemask,
mask, pbmask, sidelobethreshold, noisethreshold, lownoisethreshold,
negativethreshold, smoothfactor, minbeamfrac, cutthreshold,
growiterations, dogrowprune, minpercentchange, verbose, restart,
savemodel, calcres, calcpsf, parallel, subregion, tmpdir, btidx):
from tclean_cli import tclean_cli as tclean
from split_cli import split_cli as split
bt = btidx # 0
if bt + twidth < len(tim) - 1:
et = btidx + twidth - 1
else:
et = len(tim) - 1
if bt == 0:
bt_d = tim[bt] - ((tim[bt + 1] - tim[bt]) / 2)
else:
bt_d = tim[bt] - ((tim[bt] - tim[bt - 1]) / 2)
if et == (len(tim) - 1) or et == -1:
et_d = tim[et] + ((tim[et] - tim[et - 1]) / 2)
else:
et_d = tim[et] + ((tim[et + 1] - tim[et]) / 2)
timerange = qa.time(qa.quantity(bt_d, 's'), prec=9, form='ymd')[0] + '~' + \
qa.time(qa.quantity(et_d, 's'), prec=9, form='ymd')[0]
btstr = qa.time(qa.quantity(bt_d, 's'), prec=9, form='fits')[0]
etstr = qa.time(qa.quantity(et_d, 's'), prec=9, form='fits')[0]
print('cleaning timerange: ' + timerange)
image0 = btstr.replace(':', '').replace('-', '')
imname = imageprefix + image0 + imagesuffix
# ms_tmp = tmpdir + image0 + '.ms'
# print('checkpoint 1')
# # split(vis=vis, outputvis=ms_tmp, field=field, scan=scan, antenna=antenna, timerange=timerange,
# # datacolumn=datacolumn)
# ms.open(vis)
# print('checkpoint 1-1')
# ms.split(ms_tmp,field=field, scan=scan, baseline=antenna, time=timerange,whichcol=datacolumn)
# print('checkpoint 1-2')
# ms.close()
# print('checkpoint 2')
if overwrite or (len(glob.glob(imname + '*')) == 0):
os.system('rm -rf {}*'.format(imname))
try:
tclean(vis=vis,
selectdata=selectdata,
field=field,
spw=spw,
timerange=timerange,
uvrange=uvrange,
antenna=antenna,
scan=scan,
observation=observation,
intent=intent,
datacolumn=datacolumn,
imagename=imname,
imsize=imsize,
cell=cell,
phasecenter=phasecenter,
stokes=stokes,
projection=projection,
startmodel=startmodel,
specmode=specmode,
reffreq=reffreq,
nchan=nchan,
start=start,
width=width,
outframe=outframe,
veltype=veltype,
restfreq=restfreq,
interpolation=interpolation,
gridder=gridder,
facets=facets,
chanchunks=chanchunks,
wprojplanes=wprojplanes,
vptable=vptable,
usepointing=usepointing,
mosweight=mosweight,
aterm=aterm,
psterm=psterm,
wbawp=wbawp,
conjbeams=conjbeams,
cfcache=cfcache,
computepastep=computepastep,
rotatepastep=rotatepastep,
pblimit=pblimit,
normtype=normtype,
deconvolver=deconvolver,
scales=scales,
nterms=nterms,
smallscalebias=smallscalebias,
restoration=restoration,
restoringbeam=restoringbeam,
pbcor=pbcor,
outlierfile=outlierfile,
weighting=weighting,
robust=robust,
npixels=npixels,
uvtaper=uvtaper,
niter=niter,
gain=gain,
threshold=threshold,
nsigma=nsigma,
cycleniter=cycleniter,
cyclefactor=cyclefactor,
minpsffraction=minpsffraction,
maxpsffraction=maxpsffraction,
interactive=interactive,
usemask=usemask,
mask=mask,
pbmask=pbmask,
sidelobethreshold=sidelobethreshold,
noisethreshold=noisethreshold,
lownoisethreshold=lownoisethreshold,
negativethreshold=negativethreshold,
smoothfactor=smoothfactor,
minbeamfrac=minbeamfrac,
cutthreshold=cutthreshold,
growiterations=growiterations,
dogrowprune=dogrowprune,
minpercentchange=minpercentchange,
verbose=verbose,
restart=restart,
savemodel=savemodel,
calcres=calcres,
calcpsf=calcpsf,
parallel=parallel)
# print('checkpoint 3')
if pbcor:
clnjunks = [
'.flux', '.mask', '.model', '.psf', '.residual', '.pb',
'.sumwt', '.image'
]
else:
clnjunks = [
'.flux', '.mask', '.model', '.psf', '.residual', '.pb',
'.sumwt', '.image.pbcor'
]
for clnjunk in clnjunks:
if os.path.exists(imname + clnjunk):
shutil.rmtree(imname + clnjunk)
if pbcor:
os.system('mv {} {}'.format(imname + '.image.pbcor',
imname + '.image'))
except:
print('error in cleaning image: ' + btstr)
return [False, btstr, etstr, '']
else:
print(imname + ' exists. Clean task aborted.')
if doreg and not os.path.isfile(imname + '.fits'):
# ephem.keys()
# msinfo.keys()
try:
# check if ephemfile and msinfofile exist
if not ephem:
print(
"ephemeris info does not exist, querying from JPL Horizons on the fly"
)
ephem = hf.read_horizons(vis=vis)
if not msinfo:
print("ms info not provided, generating one on the fly")
msinfo = hf.read_msinfo(vis)
hf.imreg(vis=vis,
ephem=ephem,
msinfo=msinfo,
timerange=timerange,
reftime=reftime,
imagefile=imname + '.image',
fitsfile=imname + '.fits',
toTb=toTb,
scl100=False,
usephacenter=usephacenter,
subregion=subregion)
if os.path.exists(imname + '.fits'):
shutil.rmtree(imname + '.image')
return [True, btstr, etstr, imname + '.fits']
else:
return [False, btstr, etstr, '']
except:
print('error in registering image: ' + btstr)
return [False, btstr, etstr, imname + '.image']
else:
if os.path.exists(imname + '.image'):
return [True, btstr, etstr, imname + '.image']
else:
return [False, btstr, etstr, '']
ttotle = 90
spws = [str(s + 1) for s in range(30)]
antennas = ''
pol = 'XX'
slfcalms_ = 'aa'
spwrans = ['1~5', '6~12', '13~20', '21~30']
#for tint in range(ttotle):
for tint in range(24, 40):
#define timerange:
init_t = Time(init_time, format='isot')
timed1 = TimeDelta((tint * time_interval) * 1.0, format='sec')
timed2 = TimeDelta(time_interval, format='sec')
start_time = init_t + timed1
end_time = start_time + timed2
midtime_mjd = (start_time.mjd + end_time.mjd) / 2.
eph = hf.read_horizons(t0=Time(midtime_mjd, format='mjd'))
trange = '{0}~{1}'.format(
start_time.iso.replace('-', '/').replace(' ', '/'),
end_time.iso.replace('-', '/').replace(' ', '/'))
slfcalms = workdir + 'slfcal/No' + str(
tint) + 'IDB20170906T' + trange.split(':')[3] + trange.split(
':')[4].split('.')[0] + '.ms.corrected.xx.slfcal'
slfcaledms = workdir + 'slfcal/No' + str(
tint) + 'IDB20170906T' + trange.split(':')[3] + trange.split(
':')[4].split('.')[0] + '.ms.corrected.xx.slfcaled'
calprefix = workdir + 'slfcal/caltbs/No' + str(
tint) + 'slf_19' + trange.split(':')[3] + trange.split(':')[4].split(
'.')[0]
#imgprefix=workdir+'slfcal/images/No'+str(tint)+'slf_19'+trange.split(':')[3]+trange.split(':')[4].split('.')[0]
imgprefix = workdir + 'slfcal/images/sbs/No' + str(
tint) + 'slf_19' + trange.split(':')[3] + trange.split(':')[4].split(
def ptclean(vis, imageprefix, imagesuffix, ncpu, twidth, doreg, usephacenter,
reftime, toTb, overwrite, outlierfile, field, spw, selectdata,
timerange, uvrange, antenna, scan, observation, intent, mode,
resmooth, gridmode, wprojplanes, facets, cfcache, rotpainc, painc,
aterm, psterm, mterm, wbawp, conjbeams, epjtable, interpolation,
niter, gain, threshold, psfmode, imagermode, ftmachine, mosweight,
scaletype, multiscale, negcomponent, smallscalebias, interactive,
mask, nchan, start, width, outframe, veltype, imsize, cell,
phasecenter, restfreq, stokes, weighting, robust, uvtaper,
outertaper, innertaper, modelimage, restoringbeam, pbcor, minpb,
usescratch, noise, npixels, npercycle, cyclefactor, cyclespeedup,
nterms, reffreq, chaniter, flatnoise, allowchunk):
if not (type(ncpu) is int):
casalog.post('ncpu should be an integer')
ncpu = 8
if doreg:
# check if ephem and msinfo exist. If not, generate one on the fly
try:
ephem = hf.read_horizons(vis)
except ValueError:
print("error in obtaining ephemeris")
try:
msinfo = hf.read_msinfo(vis)
except ValueError:
print("error in getting ms info")
else:
ephem = None
# get number of time pixels
ms.open(vis)
ms.selectinit()
timfreq = ms.getdata(['time', 'axis_info'], ifraxis=True)
tim = timfreq['time']
# dt = tim[1]-tim[0] #need to change to median of all time intervals
dt = np.median(np.diff(tim))
freq = timfreq['axis_info']['freq_axis']['chan_freq'].flatten()
ms.close()
if twidth < 1:
casalog.post('twidth less than 1. Change to 1')
twidth = 1
if twidth > len(tim):
casalog.post(
'twidth greater than # of time pixels in the dataset. Change to the timerange of the entire dateset'
)
twidth = len(tim)
# find out the start and end time index according to the parameter timerange
# if not defined (empty string), use start and end from the entire time of the ms
if not timerange:
btidx = 0
etidx = len(tim) - 1
else:
try:
(tstart, tend) = timerange.split('~')
bt_s = qa.convert(qa.quantity(tstart, 's'), 's')['value']
et_s = qa.convert(qa.quantity(tend, 's'), 's')['value']
# only time is given but not date, add the date (at 0 UT) from the first record
if bt_s < 86400. or et_s < 86400.:
bt_s += np.fix(
qa.convert(qa.quantity(tim[0], 's'),
'd')['value']) * 86400.
et_s += np.fix(
qa.convert(qa.quantity(tim[0], 's'),
'd')['value']) * 86400.
btidx = np.argmin(np.abs(tim - bt_s))
etidx = np.argmin(np.abs(tim - et_s))
# make the indice back to those bracket by the timerange
if tim[btidx] < bt_s:
btidx += 1
if tim[etidx] > et_s:
etidx -= 1
if etidx <= btidx:
print "ending time must be greater than starting time"
print "reinitiating to the entire time range"
btidx = 0
etidx = len(tim) - 1
except ValueError:
print "keyword 'timerange' has a wrong format"
btstr = qa.time(qa.quantity(tim[btidx], 's'), prec=9, form='fits')[0]
etstr = qa.time(qa.quantity(tim[etidx], 's'), prec=9, form='fits')[0]
iterable = range(btidx, etidx + 1, twidth)
print 'First time pixel: ' + btstr
print 'Last time pixel: ' + etstr
print str(len(iterable)) + ' images to clean...'
res = []
# partition
clnpart = partial(
clean_iter, tim, freq, vis, imageprefix, imagesuffix, ncpu, twidth,
doreg, usephacenter, reftime, ephem, msinfo, toTb, overwrite,
outlierfile, field, spw, selectdata, uvrange, antenna, scan,
observation, intent, mode, resmooth, gridmode, wprojplanes, facets,
cfcache, rotpainc, painc, aterm, psterm, mterm, wbawp, conjbeams,
epjtable, interpolation, niter, gain, threshold, psfmode, imagermode,
ftmachine, mosweight, scaletype, multiscale, negcomponent,
smallscalebias, interactive, mask, nchan, start, width, outframe,
veltype, imsize, cell, phasecenter, restfreq, stokes, weighting,
robust, uvtaper, outertaper, innertaper, modelimage, restoringbeam,
pbcor, minpb, usescratch, noise, npixels, npercycle, cyclefactor,
cyclespeedup, nterms, reffreq, chaniter, flatnoise, allowchunk)
timelapse = 0
t0 = time()
# parallelization
para = 1
if para:
casalog.post('Perform clean in parallel ...')
pool = mprocs.Pool(ncpu)
# res = pool.map_async(clnpart, iterable)
res = pool.map(clnpart, iterable)
pool.close()
pool.join()
else:
for i in iterable:
res.append(clnpart(i))
t1 = time()
timelapse = t1 - t0
print 'It took %f secs to complete' % timelapse
# repackage this into a single dictionary
results = {
'Succeeded': [],
'BeginTime': [],
'EndTime': [],
'ImageName': []
}
for r in res:
results['Succeeded'].append(r[0])
results['BeginTime'].append(r[1])
results['EndTime'].append(r[2])
results['ImageName'].append(r[3])
return results
def clean_iter(
tim, freq, vis, imageprefix, imagesuffix, ncpu, twidth, doreg,
usephacenter, reftime, ephem, msinfo, toTb, overwrite, outlierfile,
field, spw, selectdata, uvrange, antenna, scan, observation, intent,
mode, resmooth, gridmode, wprojplanes, facets, cfcache, rotpainc,
painc, aterm, psterm, mterm, wbawp, conjbeams, epjtable, interpolation,
niter, gain, threshold, psfmode, imagermode, ftmachine, mosweight,
scaletype, multiscale, negcomponent, smallscalebias, interactive, mask,
nchan, start, width, outframe, veltype, imsize, cell, phasecenter,
restfreq, stokes, weighting, robust, uvtaper, outertaper, innertaper,
modelimage, restoringbeam, pbcor, minpb, usescratch, noise, npixels,
npercycle, cyclefactor, cyclespeedup, nterms, reffreq, chaniter,
flatnoise, allowchunk, btidx):
from taskinit import ms
from taskinit import qa
# from __casac__.quanta import quanta as qa
from __main__ import default, inp
#from clean import clean
from clean_cli import clean_cli as clean
bt = btidx # 0
if bt + twidth < len(tim) - 1:
et = btidx + twidth - 1
else:
et = len(tim) - 1
# tim_d = tim/3600./24.-np.fix(tim/3600./24.)
if bt == 0:
bt_d = tim[bt] - ((tim[bt + 1] - tim[bt]) / 2)
else:
bt_d = tim[bt] - ((tim[bt] - tim[bt - 1]) / 2)
if et == (len(tim) - 1) or et == -1:
et_d = tim[et] + ((tim[et] - tim[et - 1]) / 2)
else:
et_d = tim[et] + ((tim[et + 1] - tim[et]) / 2)
#
# bt_d=tim[bt]
# et_d=tim[et]+0.005
timerange = qa.time(qa.quantity(bt_d, 's'), prec=9, form='ymd')[0] + '~' + \
qa.time(qa.quantity(et_d, 's'), prec=9, form='ymd')[0]
tmid = (bt_d + et_d) / 2.
btstr = qa.time(qa.quantity(bt_d, 's'), prec=9, form='fits')[0]
etstr = qa.time(qa.quantity(et_d, 's'), prec=9, form='fits')[0]
print 'cleaning timerange: ' + timerange
image0 = btstr.replace(':', '').replace('-', '')
imname = imageprefix + image0 + imagesuffix
if overwrite or (len(glob.glob(imname + '*')) == 0):
# inp(taskname = 'clean')
os.system('rm -rf {}*'.format(imname))
try:
clean(vis=vis,
imagename=imname,
outlierfile=outlierfile,
field=field,
spw=spw,
selectdata=selectdata,
timerange=timerange,
uvrange=uvrange,
antenna=antenna,
scan=scan,
observation=str(observation),
intent=intent,
mode=mode,
resmooth=resmooth,
gridmode=gridmode,
wprojplanes=wprojplanes,
facets=facets,
cfcache=cfcache,
rotpainc=rotpainc,
painc=painc,
psterm=psterm,
aterm=aterm,
mterm=mterm,
wbawp=wbawp,
conjbeams=conjbeams,
epjtable=epjtable,
interpolation=interpolation,
niter=niter,
gain=gain,
threshold=threshold,
psfmode=psfmode,
imagermode=imagermode,
ftmachine=ftmachine,
mosweight=mosweight,
scaletype=scaletype,
multiscale=multiscale,
negcomponent=negcomponent,
smallscalebias=smallscalebias,
interactive=interactive,
mask=mask,
nchan=nchan,
start=start,
width=width,
outframe=outframe,
veltype=veltype,
imsize=imsize,
cell=cell,
phasecenter=phasecenter,
restfreq=restfreq,
stokes=stokes,
weighting=weighting,
robust=robust,
uvtaper=uvtaper,
outertaper=outertaper,
innertaper=innertaper,
modelimage=modelimage,
restoringbeam=restoringbeam,
pbcor=pbcor,
minpb=minpb,
usescratch=usescratch,
noise=noise,
npixels=npixels,
npercycle=npercycle,
cyclefactor=cyclefactor,
cyclespeedup=cyclespeedup,
nterms=nterms,
reffreq=reffreq,
chaniter=chaniter,
flatnoise=flatnoise,
allowchunk=False)
clnjunks = ['.flux', '.mask', '.model', '.psf', '.residual']
for clnjunk in clnjunks:
if os.path.exists(imname + clnjunk):
shutil.rmtree(imname + clnjunk)
except:
print('error in cleaning image: ' + btstr)
return [False, btstr, etstr, '']
else:
print imname + ' exists. Clean task aborted.'
if doreg and not os.path.isfile(imname + '.fits'):
#ephem.keys()
#msinfo.keys()
try:
# check if ephemfile and msinfofile exist
if not ephem:
print(
"ephemeris info does not exist, querying from JPL Horizons on the fly"
)
ephem = hf.read_horizons(vis)
if not msinfo:
print("ms info not provided, generating one on the fly")
msinfo = hf.read_msinfo(vis)
hf.imreg(vis=vis,
ephem=ephem,
msinfo=msinfo,
timerange=timerange,
reftime=reftime,
imagefile=imname + '.image',
fitsfile=imname + '.fits',
toTb=toTb,
scl100=False,
usephacenter=usephacenter)
if os.path.exists(imname + '.fits'):
shutil.rmtree(imname + '.image')
return [True, btstr, etstr, imname + '.fits']
else:
return [False, btstr, etstr, '']
except:
print('error in registering image: ' + btstr)
return [False, btstr, etstr, imname + '.image']
else:
if os.path.exists(imname + '.image'):
return [True, btstr, etstr, imname + '.image']
else:
return [False, btstr, etstr, '']
def svplot(vis, timerange=None, spw='', workdir='./', specfile=None, bl=None, uvrange=None,
stokes='RR,LL', dmin=None, dmax=None,
goestime=None, reftime=None,
xycen=None, fov=[500.,500.], xyrange=None, restoringbeam=[''], robust=0.0,
niter=500, imsize=[512], cell=['5.0arcsec'],interactive=False,
usemsphacenter=True, imagefile=None, fitsfile=None, plotaia=True,
aiawave=171, aiafits=None, savefig=False, mkmovie=False, overwrite=True, ncpu=10, twidth=1, verbose=True):
'''
Required inputs:
vis: calibrated CASA measurement set
Important optional inputs:
timerange: timerange for clean. Standard CASA time selection format.
If not provided, use the entire range (*BE CAREFUL, COULD BE VERY SLOW*)
spw: spectral window selection following the CASA syntax.
Examples: spw='1:2~60' (spw id 1, channel range 2-60); spw='*:1.2~1.3GHz' (selects all channels within 1.2-1.3 GHz; note the *)
specfile: supply dynamic spectrum save file (from suncasa.utils.dspec2.get_dspec()). Otherwise
generate a median dynamic spectrum on the fly
Optional inputs:
bl: baseline to generate dynamic spectrum
uvrange: uvrange to select baselines for generating dynamic spectrum
stokes: polarization of the clean image, can be 'RR,LL' or 'I,V'
dmin,dmax: color bar parameter
goestime: goes plot time, example ['2016/02/18 18:00:00','2016/02/18 23:00:00']
rhessisav: rhessi savefile
reftime: reftime for the image
xycen: center of the image in helioprojective coordinates (HPLN/HPLT), in arcseconds. Example: [900, -150.]
fov: field of view in arcsecs. Example: [500., 500.]
xyrange: field of view in solar XY coordinates. Format: [[x1,x2],[y1,y2]]. Example: [[900., 1200.],[0,300]]
***NOTE: THIS PARAMETER OVERWRITES XYCEN AND FOV***
aiawave: wave length of aia file in a
imagefile: if imagefile provided, use it. Otherwise do clean and generate a new one.
fitsfile: if fitsfile provided, use it. Otherwise generate a new one
savefig: whether to save the figure
Example:
'''
if xycen:
xc, yc = xycen
xlen, ylen = fov
if parse_version(sunpy.__version__)>parse_version('0.8.0'):
xyrange = [[xc - xlen / 2.0, yc - ylen / 2.0], [xc + xlen / 2.0, yc + ylen / 2.0]]
else:
xyrange = [[xc - xlen / 2.0, xc + xlen / 2.0], [yc - ylen / 2.0, yc + ylen / 2.0]]
stokes_allowed = ['RR,LL', 'I,V', 'RRLL', 'IV']
if not stokes in stokes_allowed:
print 'wrong stokes parameter ' + str(stokes) + '. Allowed values are ' + ', '.join(stokes_allowed)
return -1
if stokes == 'RRLL':
stokes = 'RR,LL'
if stokes == 'IV':
stokes = 'I,V'
if vis[-1] == '/':
vis = vis[:-1]
if not os.path.exists(vis):
print 'input measurement not exist'
return -1
if aiafits is None:
aiafits = ''
# split the data
# generating dynamic spectrum
if not os.path.exists(workdir):
os.makedirs(workdir)
if specfile:
try:
specdata = np.load(specfile)
except:
print('Provided dynamic spectrum file not numpy npz. Generating one from the visibility data')
specfile = os.path.join(workdir, os.path.basename(vis) + '.dspec.npz')
dspec_external(vis, workdir=workdir, specfile=specfile)
specdata = np.load(specfile) # specdata = ds.get_dspec(vis, domedian=True, verbose=True)
else:
print('Dynamic spectrum file not provided; Generating one from the visibility data')
# specdata = ds.get_dspec(vis, domedian=True, verbose=True)
specfile = os.path.join(workdir, os.path.basename(vis) + '.dspec.npz')
dspec_external(vis, workdir=workdir, specfile=specfile)
specdata = np.load(specfile)
tb.open(vis)
starttim = Time(tb.getcell('TIME', 0) / 24. / 3600., format='mjd')
endtim = Time(tb.getcell('TIME', tb.nrows() - 1) / 24. / 3600., format='mjd')
tb.close()
datstr = starttim.iso[:10]
if timerange is None or timerange == '':
starttim1 = starttim
endtim1 = endtim
timerange = '{0}~{1}'.format(starttim.iso.replace('-', '/').replace(' ', '/'), endtim.iso.replace('-', '/').replace(' ', '/'))
else:
try:
(tstart, tend) = timerange.split('~')
if tstart[2] == ':':
starttim1 = Time(datstr + 'T' + tstart)
endtim1 = Time(datstr + 'T' + tend)
timerange = '{0}/{1}~{0}/{2}'.format(datstr.replace('-', '/'), tstart, tend)
else:
starttim1 = Time(qa.quantity(tstart, 'd')['value'], format='mjd')
endtim1 = Time(qa.quantity(tend, 'd')['value'], format='mjd')
except ValueError:
print "keyword 'timerange' in wrong format"
midtime_mjd = (starttim1.mjd + endtim1.mjd) / 2.
if vis.endswith('/'):
vis = vis[:-1]
visname = os.path.basename(vis)
bt = starttim1.plot_date
et = endtim1.plot_date
# find out min and max frequency for plotting in dynamic spectrum
ms.open(vis)
metadata = ms.metadata()
observatory = metadata.observatorynames()[0]
spwInfo = ms.getspectralwindowinfo()
nspw = len(spwInfo)
if not spw:
spw = '0~' + str(nspw - 1)
staql = {'timerange': timerange, 'spw': spw}
if ms.msselect(staql, onlyparse=True):
ndx = ms.msselectedindices()
chan_sel = ndx['channel']
nspw = chan_sel.shape[0]
bspw = chan_sel[0, 0]
bchan = chan_sel[0, 1]
espw = chan_sel[-1, 0]
echan = chan_sel[-1, 2]
bfreq = spwInfo[str(bspw)]['Chan1Freq'] + spwInfo[str(bspw)]['ChanWidth'] * bchan
efreq = spwInfo[str(espw)]['Chan1Freq'] + spwInfo[str(espw)]['ChanWidth'] * echan
bfreqghz = bfreq / 1e9
efreqghz = efreq / 1e9
if verbose:
print 'selected timerange {}'.format(timerange)
print 'selected frequency range {0:6.3f} to {1:6.3f} GHz'.format(bfreqghz, efreqghz)
else:
print "spw or timerange selection failed. Aborting..."
ms.close()
return -1
ms.close()
if observatory == 'EOVSA':
print 'Provide stokes: ' + str(stokes) + '. However EOVSA has linear feeds. Force stokes to be IV'
stokes = 'I,V'
if mkmovie:
plt.ioff()
# fig = plt.figure(figsize=(12, 7.5), dpi=100)
if fitsfile:
pass
else:
if not imagefile:
# from ptclean_cli import ptclean_cli as ptclean
eph = hf.read_horizons(t0=Time(midtime_mjd, format='mjd'))
if observatory == 'EOVSA' or (not usemsphacenter):
phasecenter = ''
else:
phasecenter = 'J2000 ' + str(eph['ra'][0])[:15] + 'rad ' + str(eph['dec'][0])[:15] + 'rad'
print 'use phasecenter: ' + phasecenter
qlookfitsdir = os.path.join(workdir, 'qlookfits/')
qlookfigdir = os.path.join(workdir, 'qlookimgs/')
imresfile = os.path.join(qlookfitsdir, '{}.imres.npz'.format(os.path.basename(vis)))
if overwrite:
imres = mk_qlook_image(vis, twidth=twidth, ncpu=ncpu, imagedir=qlookfitsdir, phasecenter=phasecenter, stokes=stokes,
c_external=True)
else:
if os.path.exists(imresfile):
imres = np.load(imresfile)
imres = imres['imres'].item()
else:
print('Image results file not found; Creating new images.')
imres = mk_qlook_image(vis, twidth=twidth, ncpu=ncpu, imagedir=qlookfitsdir, phasecenter=phasecenter, stokes=stokes,
c_external=True)
if not os.path.exists(qlookfigdir):
os.makedirs(qlookfigdir)
plt_qlook_image(imres, figdir=qlookfigdir, specdata=specdata, verbose=True, stokes=stokes, fov=xyrange)
else:
spec = specdata['spec']
(npol, nbl, nfreq, ntim) = spec.shape
tidx = range(ntim)
fidx = range(nfreq)
tim = specdata['tim']
freq = specdata['freq']
freqghz = freq / 1e9
spec_tim = Time(specdata['tim'] / 3600. / 24., format='mjd')
timstrr = spec_tim.plot_date
plt.ion()
fig = plt.figure(figsize=(12, 7), dpi=100)
gs1 = gridspec.GridSpec(3, 1)
gs1.update(left=0.08, right=0.32, wspace=0.05)
gs2 = gridspec.GridSpec(2, 2)
gs2.update(left=0.38, right=0.98, hspace=0.02, wspace=0.02)
spec_1 = np.absolute(spec[0, 0, :, :])
spec_2 = np.absolute(spec[1, 0, :, :])
if observatory == 'EVLA':
# circular feeds
polstr = ['RR', 'LL']
if observatory == 'EOVSA' or observatory == 'ALMA':
# linear feeds
polstr = ['XX', 'YY']
print 'plot the dynamic spectrum in pol ' + ' & '.join(polstr)
ax1 = plt.subplot(gs1[0])
ax1.pcolormesh(timstrr, freqghz, spec_1, cmap='jet', vmin=dmin, vmax=dmax)
ax1.set_xlim(timstrr[tidx[0]], timstrr[tidx[-1]])
ax1.xaxis_date()
ax1.xaxis.set_major_formatter(DateFormatter("%H:%M:%S"))
# ax1.set_xticklabels(['']*10)
ax1.set_ylim(freqghz[fidx[0]], freqghz[fidx[-1]])
ax1.set_ylabel('Frequency (GHz)', fontsize=10)
ax1.set_title(observatory + ' ' + datstr + ' ' + polstr[0] + ' & ' + polstr[1], fontsize=12)
ax1.set_autoscale_on(False)
ax1.add_patch(patches.Rectangle((bt, bfreqghz), et - bt, efreqghz - bfreqghz, ec='w', fill=False))
ax1.plot([(bt + et) / 2.], [(bfreqghz + efreqghz) / 2.], '*w', ms=12)
for tick in ax1.get_xticklabels():
tick.set_fontsize(8)
for tick in ax1.get_yticklabels():
tick.set_fontsize(8)
ax2 = plt.subplot(gs1[1])
ax2.pcolormesh(timstrr, freqghz, spec_2, cmap='jet', vmin=dmin, vmax=dmax)
ax2.set_xlim(timstrr[tidx[0]], timstrr[tidx[-1]])
ax2.xaxis_date()
ax2.xaxis.set_major_formatter(DateFormatter("%H:%M:%S"))
ax2.set_ylim(freqghz[fidx[0]], freqghz[fidx[-1]])
ax2.set_ylabel('Frequency (GHz)', fontsize=10)
for tick in ax2.get_xticklabels():
tick.set_fontsize(8)
for tick in ax2.get_yticklabels():
tick.set_fontsize(8)
ax2.set_autoscale_on(False)
ax2.add_patch(patches.Rectangle((bt, bfreqghz), et - bt, efreqghz - bfreqghz, ec='w', fill=False))
ax2.plot([(bt + et) / 2.], [(bfreqghz + efreqghz) / 2.], '*w', ms=12)
# Second part: GOES plot
if goestime:
btgoes = goestime[0]
etgoes = goestime[1]
else:
datstrg = datstr.replace('-', '/')
btgoes = datstrg + ' ' + qa.time(qa.quantity(tim[0] - 1800, 's'), form='clean', prec=9)[0]
etgoes = datstrg + ' ' + qa.time(qa.quantity(tim[tidx[-1] - 1] + 1800, 's'), form='clean', prec=9)[0]
if verbose:
print 'Acquire GOES soft X-ray data in from ' + btgoes + ' to ' + etgoes
ax3 = plt.subplot(gs1[2])
try:
from sunpy import lightcurve as lc
from sunpy.time import TimeRange
goest = lc.GOESLightCurve.create(TimeRange(btgoes, etgoes))
except:
goesscript = os.path.join(workdir, 'goes.py')
goesdatafile = os.path.join(workdir, 'goes.dat')
os.system('rm -rf {}'.format(goesscript))
fi = open(goesscript, 'wb')
fi.write('import os \n')
fi.write('from sunpy.time import TimeRange \n')
fi.write('from sunpy import lightcurve as lc \n')
fi.write('import pickle \n')
fi.write('goesplottim = TimeRange("{0}", "{1}") \n'.format(btgoes, etgoes))
fi.write('goes = lc.GOESLightCurve.create(goesplottim) \n')
fi.write('fi2 = open("{}", "wb") \n'.format(goesdatafile))
fi.write('pickle.dump(goes, fi2) \n')
fi.write('fi2.close()')
fi.close()
try:
os.system('python {}'.format(goesscript))
os.system('rm -rf {}'.format(goesscript))
except NameError:
print "Bad input names"
except ValueError:
print "Bad input values"
except:
print "Unexpected error:", sys.exc_info()[0]
print "Error in generating GOES light curves. Proceed without GOES..."
if os.path.exists(goesdatafile):
fi1 = file(goesdatafile, 'rb')
goest = pickle.load(fi1)
fi1.close()
try:
dates = mpl.dates.date2num(parse_time(goest.data.index))
goesdif = np.diff(goest.data['xrsb'])
gmax = np.nanmax(goesdif)
gmin = np.nanmin(goesdif)
ran = gmax - gmin
db = 2.8 / ran
goesdifp = goesdif * db + gmin + (-6)
ax3.plot_date(dates, np.log10(goest.data['xrsb']), '-', label='1.0--8.0 $\AA$', color='red', lw=2)
ax3.plot_date(dates[0:-1], goesdifp, '-', label='derivate', color='blue', lw=0.4)
ax3.set_ylim([-7, -3])
ax3.set_yticks([-7, -6, -5, -4, -3])
ax3.set_yticklabels([r'$10^{-7}$', r'$10^{-6}$', r'$10^{-5}$', r'$10^{-4}$', r'$10^{-3}$'])
ax3.set_title('Goes Soft X-ray', fontsize=12)
ax3.set_ylabel('Watts m$^{-2}$')
ax3.set_xlabel(datetime.datetime.isoformat(goest.data.index[0])[0:10])
ax3.axvspan(dates[899], dates[dates.size - 899], alpha=0.2)
ax2 = ax3.twinx()
# ax2.set_yscale("log")
ax2.set_ylim([-7, -3])
ax2.set_yticks([-7, -6, -5, -4, -3])
ax2.set_yticklabels(['B', 'C', 'M', 'X', ''])
ax3.yaxis.grid(True, 'major')
ax3.xaxis.grid(False, 'major')
ax3.legend(prop={'size': 6})
formatter = mpl.dates.DateFormatter('%H:%M')
ax3.xaxis.set_major_formatter(formatter)
ax3.fmt_xdata = mpl.dates.DateFormatter('%H:%M')
except:
print 'Error in downloading GOES soft X-ray data. Proceeding with out soft X-ray plot.'
# third part
# start to download the fits files
if plotaia:
if not aiafits:
newlist = []
items = glob.glob('*.fits')
for names in items:
str1 = starttim1.iso[:4] + '_' + starttim1.iso[5:7] + '_' + starttim1.iso[8:10] + 't' + starttim1.iso[
11:13] + '_' + starttim1.iso[14:16]
str2 = str(aiawave)
if names.endswith(".fits"):
if names.find(str1) != -1 and names.find(str2) != -1:
newlist.append(names)
newlist.append('0')
if newlist and os.path.exists(newlist[0]):
aiafits = newlist[0]
else:
print 'downloading the aiafits file'
wave1 = aiawave - 3
wave2 = aiawave + 3
t1 = Time(starttim1.mjd - 0.02 / 24., format='mjd')
t2 = Time(endtim1.mjd + 0.02 / 24., format='mjd')
try:
from sunpy.net import vso
client = vso.VSOClient()
qr = client.query(vso.attrs.Time(t1.iso, t2.iso), vso.attrs.Instrument('aia'), vso.attrs.Wave(wave1 * u.AA, wave2 * u.AA))
res = client.get(qr, path='{file}')
except:
SdoDownloadscript = os.path.join(workdir, 'SdoDownload.py')
os.system('rm -rf {}'.format(SdoDownloadscript))
fi = open(SdoDownloadscript, 'wb')
fi.write('from sunpy.net import vso \n')
fi.write('from astropy import units as u \n')
fi.write('client = vso.VSOClient() \n')
fi.write(
"qr = client.query(vso.attrs.Time('{0}', '{1}'), vso.attrs.Instrument('aia'), vso.attrs.Wave({2} * u.AA, {3} * u.AA)) \n".format(
t1.iso, t2.iso, wave1, wave2))
fi.write("res = client.get(qr, path='{file}') \n")
fi.close()
try:
os.system('python {}'.format(SdoDownloadscript))
except NameError:
print "Bad input names"
except ValueError:
print "Bad input values"
except:
print "Unexpected error:", sys.exc_info()[0]
print "Error in Downloading AIA fits files. Proceed without AIA..."
# Here something is needed to check whether it has finished downloading the fits files or not
if not aiafits:
newlist = []
items = glob.glob('*.fits')
for nm in items:
str1 = starttim1.iso[:4] + '_' + starttim1.iso[5:7] + '_' + starttim1.iso[8:10] + 't' + starttim1.iso[
11:13] + '_' + starttim1.iso[14:16]
str2 = str(aiawave)
if nm.find(str1) != -1 and nm.find(str2) != -1:
newlist.append(nm)
if newlist:
aiafits = newlist[0]
print 'AIA fits ' + aiafits + ' selected'
else:
print 'no AIA fits files found. Proceed without AIA'
try:
aiamap = smap.Map(aiafits)
except:
print 'error in reading aiafits. Proceed without AIA'
# RCP or I
ax4 = plt.subplot(gs2[0, 0])
ax5 = plt.subplot(gs2[1, 0])
# LCP or V
ax6 = plt.subplot(gs2[0, 1])
ax7 = plt.subplot(gs2[1, 1])
if fitsfile:
pass
else:
if not imagefile:
eph = hf.read_horizons(t0=Time(midtime_mjd, format='mjd'))
if observatory == 'EOVSA' or (not usemsphacenter):
print 'This is EOVSA data'
# use RA and DEC from FIELD ID 0
tb.open(vis+'/FIELD')
phadir = tb.getcol('PHASE_DIR').flatten()
tb.close()
ra0 = phadir[0]
dec0 = phadir[1]
if stokes == 'RRLL' or stokes == 'RR,LL':
print 'Provide stokes: ' + str(stokes) + '. However EOVSA has linear feeds. Force stokes to be IV'
stokes = 'I,V'
else:
ra0 = eph['ra'][0]
dec0 = eph['dec'][0]
if not xycen:
# use solar disk center as default
phasecenter = 'J2000 ' + str(ra0) + 'rad ' + str(dec0) + 'rad'
else:
x0 = np.radians(xycen[0]/3600.)
y0 = np.radians(xycen[1]/3600.)
p0 = np.radians(eph['p0'][0]) # p angle in radians
raoff = -((x0) * np.cos(p0) - y0 * np.sin(p0))/np.cos(eph['dec'][0])
decoff = (x0) * np.sin(p0) + y0 * np.cos(p0)
newra = ra0 + raoff
newdec = dec0 + decoff
phasecenter = 'J2000 ' + str(newra) + 'rad ' + str(newdec) + 'rad'
imagename = os.path.join(workdir, visname + '.outim')
if os.path.exists(imagename + '.image') or os.path.exists(imagename + '.flux'):
os.system('rm -rf ' + imagename + '.*')
sto = stokes.replace(',', '')
print 'do clean for ' + timerange + ' in spw ' + spw + ' stokes ' + sto
print 'Original phasecenter: '+ str(ra0) + str(dec0)
print 'use phasecenter: ' + phasecenter
clean(vis=vis, imagename=imagename, selectdata=True, spw=spw, timerange=timerange, stokes=sto,
niter=niter, interactive=interactive, npercycle=50, imsize=imsize, cell=cell, restoringbeam=restoringbeam,
weighting='briggs', robust=robust, phasecenter=phasecenter)
os.system('rm -rf ' + imagename + '.psf')
os.system('rm -rf ' + imagename + '.flux')
os.system('rm -rf ' + imagename + '.model')
os.system('rm -rf ' + imagename + '.mask')
os.system('rm -rf ' + imagename + '.residual')
imagefile = imagename + '.image'
fitsfile = imagefile + '.fits'
hf.imreg(vis=vis, ephem=eph, imagefile=imagefile, timerange=timerange, reftime=reftime, fitsfile=fitsfile, verbose=True, overwrite=True)
print 'fits file ' + fitsfile + ' selected'
ax4.cla()
ax5.cla()
ax6.cla()
ax7.cla()
rfits = fitsfile
try:
hdulist = fits.open(rfits)
hdu = hdulist[0]
(npol, nf, nx, ny) = hdu.data.shape
rmap = smap.Map(hdu.data[0, 0, :, :], hdu.header)
except:
print 'radio fits file not recognized by sunpy.map. Aborting...'
return -1
if npol > 1:
rmap1 = smap.Map(hdu.data[0, 0, :, :], hdu.header)
rmap2 = smap.Map(hdu.data[1, 0, :, :], hdu.header)
XX, YY = np.meshgrid(np.arange(rmap.data.shape[1]), np.arange(rmap.data.shape[0]))
try:
rmapx, rmapy = rmap.pixel_to_data(XX * u.pix, YY * u.pix)
except:
rmapxy = rmap.pixel_to_data(XX * u.pix, YY * u.pix)
rmapx = rmapxy.Tx
rmapy = rmapxy.Ty
if not xyrange:
if xycen:
x0 = xycen[0] * u.arcsec
y0 = xycen[1] * u.arcsec
if not xycen:
row, col = rmap1.data.shape
positon = np.nanargmax(rmap1.data)
m, n = divmod(positon, col)
x0 = rmap1.xrange[0] + rmap1.scale[1] * (n + 0.5) * u.pix
y0 = rmap1.yrange[0] + rmap1.scale[0] * (m + 0.5) * u.pix
if len(fov) == 1:
fov=[fov]*2
sz_x = fov[0] * u.arcsec
sz_y = fov[1] * u.arcsec
x1 = x0 - sz_x/2.
x2 = x0 + sz_x/2.
y1 = y0 - sz_y/2.
y2 = y0 + sz_y/2.
xyrange = [[x1.value, x2.value], [y1.value, y2.value]]
else:
sz_x = xyrange[0][1] - xyrange[0][0]
sz_y = xyrange[1][1] - xyrange[1][0]
clevels1 = np.linspace(0.2, 0.9, 5)
if stokes.split(',')[1] == 'V':
clevels2 = np.array([0.8, -0.6, -0.4, -0.2, 0.2, 0.4, 0.6, 0.8])
else:
clevels2 = np.linspace(0.2, 0.9, 5)
if 'aiamap' in vars():
aiamap.plot_settings['cmap'] = plt.get_cmap('binary')
if rmap:
title = 'AIA {0:.0f} + {1} {2:6.3f} GHz'.format(aiamap.wavelength.value, observatory, (bfreqghz + efreqghz) / 2.0)
else:
title = 'AIA {0:.0f}'.format(aiamap.wavelength.value)
aiamap.plot(axes=ax4)
ax4.set_title(title + ' ' + stokes.split(',')[0], fontsize=12)
aiamap.draw_limb()
aiamap.draw_grid()
aiamap.draw_rectangle((xyrange[0][0], xyrange[1][0]) * u.arcsec, sz_x, sz_y)
aiamap.plot(axes=ax6)
ax6.set_title(title + ' ' + stokes.split(',')[1], fontsize=12)
aiamap.draw_limb()
aiamap.draw_grid()
aiamap.draw_rectangle((xyrange[0][0], xyrange[1][0]) * u.arcsec, sz_x, sz_y)
if rmap:
ax4.contour(rmapx.value, rmapy.value, rmap1.data, levels=clevels1 * np.nanmax(rmap1.data), cmap=cm.jet)
ax6.contour(rmapx.value, rmapy.value, rmap2.data, levels=clevels2 * np.nanmax(rmap2.data), cmap=cm.RdBu)
ax4.text(0.02, 0.02, 'AIA {0:.0f} '.format(aiamap.wavelength.value) + aiamap.date.strftime('%H:%M:%S'),
verticalalignment='bottom', horizontalalignment='left', transform=ax4.transAxes, color='k',
fontsize=10)
ax6.text(0.02, 0.02, 'AIA {0:.0f} '.format(aiamap.wavelength.value) + aiamap.date.strftime('%H:%M:%S'),
verticalalignment='bottom', horizontalalignment='left', transform=ax6.transAxes, color='k',
fontsize=10)
else:
title = '{0} {1:6.3f} GHz'.format(observatory, (bfreqghz + efreqghz) / 2.0)
rmap1.plot(axes=ax4, cmap=cm.jet)
ax4.set_title(title + ' ' + stokes.split(',')[0], fontsize=12)
rmap1.draw_limb()
rmap1.draw_grid()
rmap1.draw_rectangle((xyrange[0][0], xyrange[1][0]) * u.arcsec, sz_x, sz_y)
rmap2.plot(axes=ax6, cmap=cm.RdBu)
ax6.set_title(title + ' ' + stokes.split(',')[1], fontsize=12)
rmap2.draw_limb()
rmap2.draw_grid()
# ax4.contour(rmapx.value, rmapy.value, rmap1.data, levels=np.linspace(0.2, 0.9, 5) * np.nanmax(rmap1.data),
# cmap=cm.gray)
# ax6.contour(rmapx.value, rmapy.value, rmap2.data, levels=np.linspace(0.2, 0.9, 5) * np.nanmax(rmap2.data),
# cmap=cm.gray)
rmap2.draw_rectangle((xyrange[0][0], xyrange[1][0]) * u.arcsec, sz_x, sz_y)
ax4.set_xlim(-1200, 1200)
ax4.set_ylim(-1200, 1200)
ax6.set_xlim(-1200, 1200)
ax6.set_ylim(-1200, 1200)
try:
subrmap1 = rmap1.submap(xyrange[0] * u.arcsec, xyrange[1] * u.arcsec)
subrmap2 = rmap2.submap(xyrange[0] * u.arcsec, xyrange[1] * u.arcsec)
except:
bl = SkyCoord(xyrange[0][0] * u.arcsec, xyrange[1][0] * u.arcsec, frame=rmap1.coordinate_frame)
tr = SkyCoord(xyrange[0][1] * u.arcsec, xyrange[1][1] * u.arcsec, frame=rmap1.coordinate_frame)
subrmap1 = rmap1.submap(bl, tr)
subrmap2 = rmap2.submap(bl, tr)
XX, YY = np.meshgrid(np.arange(subrmap1.data.shape[1]), np.arange(subrmap1.data.shape[0]))
try:
subrmapx, subrmapy = subrmap1.pixel_to_data(XX * u.pix, YY * u.pix)
except:
subrmapxy = subrmap1.pixel_to_data(XX * u.pix, YY * u.pix)
subrmapx = subrmapxy.Tx
subrmapy = subrmapxy.Ty
if 'aiamap' in vars():
try:
subaiamap = aiamap.submap(xyrange[0] * u.arcsec, xyrange[1] * u.arcsec)
except:
bl = SkyCoord(xyrange[0][0] * u.arcsec, xyrange[1][0] * u.arcsec, frame=aiamap.coordinate_frame)
tr = SkyCoord(xyrange[0][1] * u.arcsec, xyrange[1][1] * u.arcsec, frame=aiamap.coordinate_frame)
subaiamap = aiamap.submap(bl, tr)
subaiamap.plot(axes=ax5, title='')
subaiamap.draw_limb()
subaiamap.draw_grid()
subaiamap.plot(axes=ax7, title='')
subaiamap.draw_limb()
subaiamap.draw_grid()
ax5.contour(subrmapx.value, subrmapy.value, subrmap1.data, levels=clevels1 * np.nanmax(subrmap1.data), cmap=cm.jet)
ax7.contour(subrmapx.value, subrmapy.value, subrmap2.data, levels=clevels2 * np.nanmax(subrmap2.data),
cmap=cm.RdBu) # subaiamap.draw_rectangle((fov[0][0], fov[1][0]) * u.arcsec, 400 * u.arcsec, 400 * u.arcsec)
else:
subrmap1.plot(axes=ax5, cmap=cm.jet, title='')
subrmap1.draw_limb()
subrmap1.draw_grid()
subrmap2.plot(axes=ax7, cmap=cm.RdBu, title='')
subrmap2.draw_limb()
subrmap2.draw_grid() # ax5.contour(subrmapx.value, subrmapy.value, subrmap1.data, # levels=clevels1 * np.nanmax(subrmap1.data), cmap=cm.gray) # ax7.contour(subrmapx.value, subrmapy.value, subrmap2.data, # levels=clevels2 * np.nanmax(subrmap2.data), cmap=cm.gray) # subrmap1.draw_rectangle((fov[0][0], fov[1][0]) * u.arcsec, 400 * u.arcsec, 400 * u.arcsec) # subrmap2.draw_rectangle((fov[0][0], fov[1][0]) * u.arcsec, 400 * u.arcsec, 400 * u.arcsec)
ax5.set_xlim(xyrange[0])
ax5.set_ylim(xyrange[1])
ax5.text(0.02, 0.02, observatory + ' ' + rmap.date.strftime('%H:%M:%S.%f')[:-3], verticalalignment='bottom',
horizontalalignment='left', transform=ax5.transAxes, color='k', fontsize=10)
ax7.set_xlim(xyrange[0])
ax7.set_ylim(xyrange[1])
ax7.text(0.02, 0.02, observatory + ' ' + rmap.date.strftime('%H:%M:%S.%f')[:-3], verticalalignment='bottom',
horizontalalignment='left', transform=ax7.transAxes, color='k', fontsize=10)
fig.show()