forked from typpo/astrokit
/
photometry_test.py
executable file
·45 lines (33 loc) · 1.09 KB
/
photometry_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#!/usr/bin/env python2.7
'''
Usage: python photometry_test.py <corr.fits path> <image.fits path>
'''
import sys
import numpy
import astropy.units as u
from astropy.io import fits
from photutils.psf import psf_photometry, GaussianPSF
CORR_PATH = sys.argv[1]
IMAGE_PATH = sys.argv[2]
OUTPUT_PATH = sys.argv[3] if len(sys.argv) > 3 else 'flux_comparison.png'
corr = fits.open(CORR_PATH)
coords = zip(corr[1].data['field_x'], corr[1].data['field_y'])
catalog_fluxes = corr[1].data['FLUX']
# Convert
factor = (u.MJy / u.sr * (0.402 * u.arcsec) ** 2 / u.pixel).to(u.mJy / u.pixel)
catalog_fluxes *= factor.value
im = fits.open(IMAGE_PATH)
data = im[0].data[2]
psf_gaussian = GaussianPSF(1)
computed_fluxes = psf_photometry(data, coords, psf_gaussian)
print len(catalog_fluxes), 'sources present in metadata'
# Check
import matplotlib.pyplot as plt
plt.scatter(catalog_fluxes, computed_fluxes)
plt.xlabel('Fluxes catalog')
plt.ylabel('Fluxes photutils')
#plt.plot(numpy.sort(computed_fluxes)/numpy.sort(catalog_fluxes))
#plt.xlabel('star #')
#plt.ylabel('computed/catalog')
plt.savefig(OUTPUT_PATH)
print 'Done.'