from make_coadd_ukidss import make_coadd_ukidss
from lupton_rgb import YHKfits2RGBjpeg
'''
ra = 166.88013
dec = 13.78691
size = 60.0
working_dir = 'images/'
out_rgb_dir = 'out_color_images'
'''

def run_ukidss_rgb(ra, dec, size, working_dir, out_rgb_dir):
   import astropy.units as u
   import astropy.coordinates as coord
   import os
   import shutil
   from astropy.io import fits
   from astropy.coordinates import SkyCoord
   c = SkyCoord(ra*u.degree, dec*u.degree)
   out_fname = 'UKIDSS_J{0}{1}'.format(c.ra.to_string(unit=u.hourangle, sep='', precision=2, pad=True), 
                                       c.dec.to_string(sep='', precision=2, alwayssign=True, pad=True))
   plate_dir = os.path.join(working_dir, out_fname)
   make_coadd_ukidss(ra, dec, size, plate_dir)
   YHKfits2RGBjpeg(os.path.join(plate_dir, 'swarp'), os.path.join(out_rgb_dir, out_fname))
   shutil.rmtree(plate_dir, ignore_errors=True)

if __name__ == '__main__':
  run_ukidss_rgb(ra, dec, size, working_dir, out_rgb_dir)