hdu_in[0].header['BMIN'] *= psf_ratio
print "Writing to " + outfits
hdu_in.writeto(outfits, clobber=True)
if options.average:
if options.scaleint:
print "Scaling the integrated fluxes in the catalogue downward by a direction-independent factor of " + str(
psf_ratio)
data['int_flux'] /= psf_ratio
else:
print "Scaling the peak fluxes in the catalogue upward by a direction-independent factor of " + str(
psf_ratio)
data['peak_flux'] *= psf_ratio
vot = Table(data)
# description of this votable
vot.description = "Corrected for position-independent PSF variation"
print "Writing to " + output
writetoVO(vot, output)
else:
print "Performing direction-dependent PSF correction."
psf_in = fits.open(psfimage)
w = wcs.WCS(psf_in[0].header, naxis=2)
unmara, unmadec = np.array(data['ra']).tolist(), np.array(
data['dec']).tolist()
rapix, decpix = w.wcs_world2pix(unmara, unmadec, 1)
raintpix, decintpix = np.array(np.round(rapix),
dtype=int), np.array(np.round(decpix),
dtype=int)
raclip, decclip = np.clip(raintpix, 0,
psf_in[0].data.shape[1] - 1), np.clip(
def main():
"""
"""
parser = ArgumentParser()
parser.add_argument('--input',
dest="input",
default=None,
help="Input FITS table to downselect.")
parser.add_argument(
'--minsnr',
dest="minsnr",
default=10.0,
type=float,
help=
"Minimum S/N ratio for sources used to fit PSF (default = 10 sigma)")
parser.add_argument(
'--nofilter',
action="store_false",
dest="usefilter",
default=True,
help=
"Don't use NVSS (and SUMSS South of Dec -40) to select unresolved sources? (default = use NVSS and SUMSS)"
)
parser.add_argument(
'--noisolate',
action="store_false",
dest="isolate",
default=True,
help=
"Don't exclude sources near other sources? (will exclude by default)")
parser.add_argument(
'--output',
dest="output",
default=None,
help="Output psf FITS file -- default is input_psfcat.fits")
# allow manual specification of NVSS/SUMSS catalogue (e.g. if running locally...)
parser.add_argument("--nscat", dest="nscat", type=str,
default="/group/mwasci/$USER/GLEAM-X-pipeline/models/NVSS_SUMSS_psfcal.fits",
help="NVSS-SUMSS catalogue filtering extended sources. Default is located at " \
"'/group/mwasci/$USER/GLEAM-X-pipeline/models/NVSS_SUMSS_psfcal.fits'")
options = parser.parse_args()
# Parse the input options
if not os.path.exists(options.input):
# raise RuntimeError("An input file must be specified!")
print("Error! Must specify an input file.")
sys.exit(1)
else:
inputfile = options.input
ext = options.input.split(".")[-1] # make sure .vot can be used as input
if options.output:
outputfile = options.output
else:
outputfile = inputfile.replace("." + ext, "_psfcat.fits")
# Read the VO table and start processing
if options.isolate is True:
sparse = re.sub("." + ext, "_isolated.fits", inputfile)
os.system('stilts tmatch1 matcher=sky values="ra dec" params=600 \
action=keep0 in=' + inputfile + ' out=' + sparse)
else:
sparse = inputfile
if options.usefilter is True:
nscat = options.nscat
# Snapshot: Get rid of crazy-bright sources, really super-extended sources, and sources with high residuals after fit
os.system(
'stilts tpipe in=' + sparse +
' cmd=\'select ((local_rms<1.0)&&((int_flux/peak_flux)<3)&&((residual_std/peak_flux)<0.1))\' out=temp_crop.fits'
)
# Match GLEAM with NVSS/SUMSS
os.system(
'stilts tmatch2 matcher=sky params=30 in1=' + nscat +
' in2=temp_crop.fits out=temp_ns_match.fits values1="RAJ2000 DEJ2000" values2="ra dec" fixcols="dups" suffix1="_ns" suffix2=""'
)
# Keep only basic aegean headings
os.system(
'stilts tpipe in=temp_ns_match.fits cmd=\'keepcols "ra dec peak_flux err_peak_flux int_flux err_int_flux local_rms a err_a b err_b pa err_pa psf_a psf_b psf_pa residual_std flags"\' out='
+ outputfile)
os.remove('temp_crop.fits')
os.remove('temp_ns_match.fits')
hdu = fits.open(outputfile)
data = hdu[1].data
else:
hdu = fits.open(sparse)
data = hdu[1].data
x = data['ra']
if max(x) > 360.:
# raise ValueError("RA goes higher than 360 degrees ({}: {})! Panic!".format(np.where(x == max(x)),
# max(x)))
print "RA goes higher than 360 degrees! Panic!"
sys.exit(1)
y = data['dec']
# Downselect to unresolved sources
mask = []
# Filter out any sources where Aegean's flags weren't zero
indices = np.where((data['flags'] == 0) & (
(data['peak_flux'] / data['local_rms']) >= options.minsnr))
mask.extend(indices[0])
mask = np.array(mask)
tab = Table(data[mask])
tab.description = "Sources selected for PSF calculation."
tab.write(outputfile, overwrite=True)
os.remove('vlssr_temp.vot')
os.remove('vlssr_crop.vot')
os.remove('temp_vlssr_match.vot')
os.remove(Mmatchvot)
os.remove(Vmatchvot)
table = parse_single_table(outputfile)
data = table.array
else:
table = parse_single_table(sparse)
data = table.array
x = data['ra']
if max(x) > 360.:
print "RA goes higher than 360 degrees! Panic!"
sys.exit(1)
y = data['dec']
# Downselect to unresolved sources
mask = []
# Filter out any sources where Aegean's flags weren't zero
indices = np.where((data['flags'] == 0) & (
(data['peak_flux'] / data['local_rms']) >= options.minsnr))
mask.extend(indices[0])
mask = np.array(mask)
vot = Table(data[mask])
vot.description = "Sources selected for PSF calculation."
writetoVO(vot, outputfile)
