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




def plot_max_pixels(files): 
    flux_sc=[]
    flux_ft=[]
    
    hdulist = pyfits.open(files[0])
    mode =  hdulist['PRIMARY'].header["HIERARCH ESO INS POLA MODE"] + " " + hdulist['PRIMARY'].header["HIERARCH ESO INS SPEC RES"]
    
    
    figure("max pixel %s" % mode)
    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
        
        image_SC = hdulist['IMAGING_DATA_SC'].data
        im_mean = image_SC.mean(axis=0)
        im_filtered=signal.medfilt(im_mean, kernel_size = (1,5))
        plot(im_filtered.max(axis=0), label = file)
        
    legend()


# new files

# MED SPLIT
files= [
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0018.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0019.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0020.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0021.fits"]

plot_max_pixels(files)

# HIGH SPLIT
files= [
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0044.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0045.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0046.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0047.fits"]

plot_max_pixels(files)

# MED COMB
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"]

plot_max_pixels(files)

# HIGH COMB
files= [
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0057.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0058.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0059.fits",
 "/data/1/2019-10-12/GRAVITY_PARALLEL_FLAT286_0060.fits"]

plot_max_pixels(files)


# ref files

# HIGH COMB
files= [
 "/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0093.fits",
 "/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0094.fits",
 "/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0095.fits",
 "/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0096.fits"]

plot_max_pixels(files)

# MED SPLIT
files= [
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0065.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0066.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0067.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0068.fits"]

plot_max_pixels(files)

# MED COMB
files= [
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT225_0054.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT225_0055.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT225_0056.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT225_0057.fits"]

plot_max_pixels(files)

# HIGH SPLIT
files= [
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0078.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0079.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0080.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_FLAT237_0081.fits"]
plot_max_pixels(files)