try:
   import pyfits
except:
   from astropy.io import fits as pyfits
from numpy import *
from matplotlib.pyplot import *
import os


def smooth(y,n):
   w = ones(n)
   return convolve (w/w.sum(),y,mode='same')

def wrap(y):
   return angle (np.exp (1.j * y))


# LOW SPLIT # OK
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T05-40-41_debug.fits" # OK
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T05-33-33_debug.fits" # OK

# MED SPLIT OK
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T06-20-18_debug.fits" # OK
file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T06-13-09_debug.fits" # OK

# HIGH SPLIT 
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T06-55-41_debug.fits" # ~BAD 
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T06-50-34_debug.fits" # ~BAD

# HIGH COMB BAD 
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T07-31-35_debug.fits" # BAD
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T07-26-27_debug.fits" # BAD

# MED COMB BAD BAD
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T08-08-41_debug.fits" # BAD
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T08-02-39_debug.fits" # BAD

# LOW COMB 
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T08-42-16_debug.fits" # OK
#file= "/data/1/2016-12-11/reduced/GRAVITY.2016-12-11T08-36-14_debug.fits" # OK

# MED COMB
files= [
 "/data/1/2016-12-11/GRAVITY.2016-12-11T07-55-22.fits",
 "/data/1/2016-12-11/GRAVITY.2016-12-11T07-56-06.fits",
 "/data/1/2016-12-11/GRAVITY.2016-12-11T07-56-47.fits",
 "/data/1/2016-12-11/GRAVITY.2016-12-11T07-57-22.fits"]
files= [
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0083.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0084.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0085.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0086.fits"]
 
 
flux_sc=[]
flux_ft=[]
figure("max pixel values")
clf()

for file in files :
    hdulist = pyfits.open(file)
    print(file)
    
    tel = [[0,1],[0,2],[0,3],[1,2],[1,3],[2,3]]
    region_sc=array([[8,9,10,11],[20,21,22,23],[4,5,6,7],[16,17,18,19],[0,1,2,3],[12,13,14,15]])
    region_ft=array([[20,21,22,23],[12,13,14,15],[16,17,18,19],[4,5,6,7],[8,9,10,11],[0,1,2,3]])
    
    if hdulist['IMAGING_DETECTOR_SC'].data.size == 48:
        region_sc=region_sc*2
    
    if hdulist['IMAGING_DETECTOR_FT'].data.size == 48:
        region_ft=region_ft*2
    
    
    lbd_met = 1.908
    lbd_sc  = 2.24
    lbd_ft  = 2.2
    tpi = np.pi * 2
    
    image_SC = hdulist['IMAGING_DATA_SC'].data
    flux_sc.append(image_SC.sum(axis=1).sum(axis=1))
    image_FT = hdulist['IMAGING_DATA_FT'].data["PIX"]
    flux_ft.append(image_FT.sum(axis=1).sum(axis=1))
    hdulist.close()

    plot(image_SC.mean(axis=0).max(axis=0), label = file)
    
legend()

print(flux_ft)

flux_sc=array(flux_sc)
flux_ft=array(flux_ft)

figure("flux vs time")
clf()
plot(flux_sc.transpose(), label="FLUX SC")

#figure(11)
#clf()
#plot(flux_ft[:,::10].transpose())

legend()