import h5py
import matplotlib.pyplot as plt
import numpy as np
import sys
import re
import ast
from scipy.optimize import curve_fit
import os
from scipy import interpolate

# Solo levanto algunos experimentos
ALL_FILES = """000003130-IR_Scan.h5""" 

def SeeKeys(files):
    for i, fname in enumerate(files.split()):
        data = h5py.File(fname, 'r') # Leo el h5: Recordar que nuestros datos estan en 'datasets'
        print(fname)
        print(list(data['datasets'].keys()))

print(SeeKeys(ALL_FILES))
#carpeta pc nico labo escritorio:
#C:\Users\Usuario\Documents\artiq\artiq_experiments\artiq_master\results\2021-07-14\16

for i, fname in enumerate(ALL_FILES.split()):
    print(i)
    print(fname)
    data = h5py.File(fname, 'r') # Leo el h5: Recordar que nuestros datos estan en 'datasets'

    # Aca hago algo repugnante para poder levantar los strings que dejamos
    # que además tenian un error de tipeo al final. Esto no deberá ser necesario
    # cuando se solucione el error este del guardado.
    freqs = np.array(data['datasets']['IR_frequencies'])
    measurements = np.array(data['datasets']['measurements'])

    
y = np.reshape(measurements, (len(freqs),-1)).sum(axis=1)

plt.figure()
plt.plot([f*1e-6 for f in freqs], y, 'o-')
plt.xlabel('Frecuencia AOM (MHz)')
plt.ylabel('counts')
plt.grid()