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



def plot_max_pixels(file): 
    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)
        
        image_SC = hdulist['IMAGING_DATA_SC'].data
        im_mean = image_SC.max(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-17.1/GRAVITY_PARALLEL_WAVE290_0023.fits",
#"/data/1/2019-10-15/GRAVITY_PARALLEL_WAVE288_0029.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_WAVE237_0069.fits"]

plot_max_pixels(files)

# HIGH SPLIT
files= ["/data/1/2019-10-17.1/GRAVITY_PARALLEL_WAVE290_0036.fits",
#"/data/1/2019-10-15/GRAVITY_PARALLEL_WAVE288_0042.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_WAVE237_0082.fits"]

plot_max_pixels(files)

# MED COMB
files= ["/data/1/2019-10-17.1/GRAVITY_PARALLEL_WAVE290_0088.fits",
#"/data/1/2019-10-15/GRAVITY_PARALLEL_WAVE288_0068.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_WAVE225_0058.fits"]

plot_max_pixels(files)

# HIGH COMB
files= ["/data/1/2019-10-17.1/GRAVITY_PARALLEL_WAVE290_0075.fits",
#"/data/1/2019-10-15/GRAVITY_PARALLEL_WAVE288_0055.fits",
"/data/1/ref_data/GRAVITY_PARALLEL_WAVE237_0097.fits"]

plot_max_pixels(files)