find_boxes.py

#!/usr/bin/env python

import sys
import os
import numpy as np
import scipy.ndimage as nd
from astropy.io import fits as pyfits


def rfits(file):
    f = pyfits.open(file, memmap=True)
    hdu = f[0]
    return hdu


def daofind(im):
    mean = np.mean(im)
    sig = np.std(im)
    smooth = nd.gaussian_filter(im, 2.0)
    clip = smooth >= (mean + 1.0)
    labels, num = nd.label(clip)
    pos = nd.center_of_mass(im, labels, range(num + 1))
    return num, pos[1:]



def find_boxes(infile):
    hdu = rfits(infile)
    image = hdu.data
    hdr = hdu.header
    n, stars = daofind(image)

    if n == 2:
        print "Found boxes and writing positions."
        output = open("init_cen_all", 'w')
        xsum = 0.0
        ysum = 0.0
        for star in stars:
            (y, x) = star
            xsum += x
            ysum += y
            output.write("%8.3f   %8.3f\n" % (x, y))
        output.close()
        x = xsum / 2
        y = ysum / 2
        if np.abs(x - 160.0) < 20 and \
               np.abs(y - 120.0) < 20 and \
               os.path.exists("SYNCME"):
            print "\033[0;32mSYNCING TARGET!\033[0;39m"
            os.system("./pygto900.py sync")
            os.system("rm SYNCME")
            os.system("sleep 2")
    else:
        print "No or not enough stars found."
        output = open("init_cen_all", 'w')
        output.write("160.0   120.0\n")
        output.write("198.0   120.0\n")
        output.close()

    return n

if __name__ == '__main__':
    infile = sys.argv[1]
    find_boxes(infile)