#!/usr/bin/env python
import numpy as np
from astropy.io import fits
from astropy.wcs import WCS
from astropy.coordinates import SkyCoord
from astropy import units as u
import os
import csv

def find_by_bestObjID(bestObjID):
  hdulist = fits.open('imbh_so.fits')
  data=hdulist[1].data
  f_lso = open('list_of_sup_obj.txt', 'w')

  for row in data:
     if row['bestObjID'] == bestObjID:
       return [row['ra'],row['dec']]
  f_lso.close()
  hdulist.close()


with open('best_fits','r') as csvfile:
    rows = csv.reader(csvfile, delimiter=',')
    for row in rows:
      if 'HST' in row[5]:
	coords = find_by_bestObjID(int(row[0]))
	os.chdir(row[2])      
	hdulist = fits.open(row[3])
	i=0  
	w=WCS(hdulist['SCI',1].header)
	x, y = w.wcs_world2pix(coords[0], coords[1], 0)
	VAL=hdulist['CTX',1].data[int(y),int(x)]
	if VAL>0:
	  print x,y
	  print VAL
	  f_assoc = open('assoc_galaxy.txt', 'w')
	  f_assoc.write(str(x)+'	'+str(y) + '\n')
	  f_assoc.close()
	else:
	  print "ALARM",x,"-",y,"-",VAL,"...",row[0],"...",row[2]
	os.chdir('../../../')