#!/usr/bin/env python
"""
Generic python script.
"""
__author__ = "Alex Drlica-Wagner"
import glob
import numpy as np
import pylab as plt
import fitsio
from ugali.utils.projector import angsep

STARS = [
    ['V1', 189.570323, -40.939879], 
    ['V2', 189.633635, -40.878072], 
    ['V3', 189.516808, -40.744898], 
    ['V4', 189.584351, -41.101214], 
    ['V5', 189.776219, -40.977123], 
]

for filename in glob.glob('hpx*.fits'):
    print('Reading %s...'%filename)
    data = fitsio.read(filename)

    out = []
    for s in STARS:
        sep = angsep(s[1],s[2],data['RA'],data['DEC'])
        idx = sep < 0.5/3600.
        print("  %s: number of matches = %s"%(s[0],idx.sum()))
        print("  %s: median separation = %.2f arcsec."%(s[0],np.median(sep[idx])*3600.))
        out.append(data[idx])

    out = np.concatenate(out)
    print("Found %s matches."%len(out))

    outfile = filename.replace('10243','rrl')
    print('Writing %s...'%outfile)
    fitsio.write(outfile,out,clobber=True)
    print("")