photometry.py

# -*- coding: utf-8 -*-
"""
Created on Thu Dec 28 23:19:28 2017

@author: iwong
"""
import pdb
import os
import time
import numpy as np

import auxiliary as aux

def do_photometry(files,stardir,sourcedir,astrometrydir,photometrydir,
                oldphotometrydir=None,rephot=False,flipxy=False,
                FILTlabel='FILTER',TIMElabel='JD',OBJlabel='OBJECT'):
    '''
    Calibrate photometry using solved astrometry OR rematch sources given
    a different aperture extraction based on a previous do_photometry run
    '''

    nfiles = len(files)
    files = sorted(files)
    filters = np.array(["g","r","i","z","B","V","R","I"])

    for i,file in enumerate(sorted(files)):
        phot = aux.photometry(file,stardir,sourcedir,astrometrydir,photometrydir,flipxy=flipxy,
                          FILTlabel=FILTlabel,TIMElabel=TIMElabel,OBJlabel=OBJlabel)
        done_phot = os.path.exists(phot.photometrydir+phot.shortname+'.phot')
        if rephot is True or done_phot is False:
            print "Working on "+phot.shortname
            if oldphotometrydir is None:
                phot.transform()
                phot.matching()
                phot.zeropoint(filters)
                phot.autotarget()
            else:
                phot.rematching(oldphotometrydir,filters)
            delattr(phot,"sources")
            if phot.found:
                aux.savepickle(phot,phot.photometrydir+phot.shortname+'.phot')
                print "mag = "+str(round(phot.mag,3))+", magerr = "+str(round(phot.magerr,3))