import os, numpy, h5py
from shutil import copyfile

def isNumber(str):
    try:
        float(str)
        return True
    except:
        return False

def getfirstFilefromPath(path, meta, ext):
    validFilelist = []
    fileList = os.listdir(path)
    if len(fileList) < 1:
     return None
    # meta    1234 567 8-18 BCDE
    # H,D,PE  YYYY DDD EPOC .ext

    for thisFile in fileList:
        if meta == "PE":
            try:
                number = int(thisFile[len(meta) + 7:len(meta) + 17])
            except:
                 print("There is a file or folder with different format")
        if meta == "D":
            try:
                number = int(thisFile[8:11])
            except:
                print("There is a file or folder with different format")

        if not isNumber(str=number):
            continue
        if (os.path.splitext(thisFile)[-1].lower() != ext.lower()):
            continue
        validFilelist.sort()
        validFilelist.append(thisFile)
    if len(validFilelist) > 0:
        validFilelist = sorted(validFilelist, key=str.lower)
        return validFilelist
    return None

def gettimeutcfromDirFilename(path, file):
    dir_file = path + "/" + file
    fp = h5py.File(dir_file, 'r')
    epoc = fp['Metadata'].get('utctimeInit')[()]
    fp.close()
    return epoc

def getDatavaluefromDirFilename(path, file, value):
    dir_file = path + "/" + file
    fp = h5py.File(dir_file, 'r')
    array = fp['Data'].get(value)[()]
    fp.close()
    return array


# ·········· Velocidad de Pedestal·················
w = input ("Ingresa velocidad de Pedestal:   ")
w = 4
w = float(w)
# ·········· Resolucion minimo en grados···········
alfa = input ("Ingresa resolucion minima en grados:  ")
alfa = 1
alfa = float(alfa)
# ·········· IPP del Experimento ··················
IPP = input ("Ingresa el IPP del experimento:   ")
IPP = 0.0004
IPP = float(IPP)
# ·········· MODE ··················
mode = input ("Ingresa el MODO del experimento T or F:   ")
mode = "T"
mode = str(mode)

# ·········· Tiempo en generar la resolucion min···
# ············ MCU ·· var_ang = w * (var_tiempo)···
var_tiempo = alfa / w
# ·········· Tiempo Equivalente en perfiles········
# ··········  var_tiempo =  IPP * ( num_perfiles )·
num_perfiles = int(var_tiempo / IPP)

# ··········DATA PEDESTAL··························
dir_pedestal = "/home/alex/Downloads/pedestal"
# ·········· DATA ADQ······························
if mode == "T":
    dir_adq = "/home/alex/Downloads/hdf5_testPP/d2020194"  # Time domain
else:
    dir_adq = "/home/alex/Downloads/hdf5_test/d2020194"  # Frequency domain

print("Velocidad angular             :", w)
print("Resolucion minima en grados   :", alfa)
print("Numero de perfiles equivalente:", num_perfiles)
print("Mode                          :", mode)

# ············ First File·············
list_pedestal = getfirstFilefromPath(path=dir_pedestal, meta="PE", ext=".hdf5")
list_adq = getfirstFilefromPath(path=dir_adq , meta="D", ext=".hdf5")

# ············ utc time ··············
utc_pedestal = gettimeutcfromDirFilename(path=dir_pedestal, file=list_pedestal[0])
utc_adq = gettimeutcfromDirFilename(path=dir_adq     , file=list_adq[0])

print("utc_pedestal                  :", utc_pedestal)
print("utc_adq                       :", utc_adq)
# ·············Relacion:  utc_adq (+/-) var_tiempo*nro_file= utc_pedestal
time_Interval_p = 0.01
n_perfiles_p = 100
if utc_adq > utc_pedestal:
    nro_file = int((int(utc_adq) - int(utc_pedestal)) / (time_Interval_p * n_perfiles_p))
    ff_pedestal = list_pedestal[nro_file]
    utc_pedestal = gettimeutcfromDirFilename(path=dir_pedestal, file=ff_pedestal)
    nro_key_p = int((utc_adq - utc_pedestal) / time_Interval_p)
    if utc_adq > utc_pedestal:
        ff_pedestal = ff_pedestal
    else:
        nro_file = nro_file - 1
        ff_pedestal = list_pedestal[nro_file]
    angulo = getDatavaluefromDirFilename(path=dir_pedestal, file=ff_pedestal, value="azimuth")
    nro_key_p = int((utc_adq - utc_pedestal) / time_Interval_p)
    print("nro_file                      :", nro_file)
    print("name_file                     :", ff_pedestal)
    print("utc_pedestal_file             :", utc_pedestal)
    print("nro_key_p                     :", nro_key_p)
    print("utc_pedestal_init             :", utc_pedestal + nro_key_p * time_Interval_p)
    print("angulo_array                  :", angulo[nro_key_p])
# 4+25+25+25+21
# while True:
list_pedestal = getfirstFilefromPath(path=dir_pedestal, meta="PE", ext=".hdf5")
list_adq = getfirstFilefromPath(path=dir_adq , meta="D", ext=".hdf5")

nro_file = nro_file  # 10
nro_key_perfil = nro_key_p
blocksPerFile = 100
wr_path = "/home/alex/Downloads/hdf5_wr/"
# Lectura  de archivos de adquisicion para adicion de azimuth
for thisFile in range(len(list_adq)):
    print("thisFileAdq", thisFile)
    angulo_adq = numpy.zeros(blocksPerFile)
    tmp = 0
    for j in range(blocksPerFile):
        iterador = nro_key_perfil + 25 * (j - tmp)
        if iterador < n_perfiles_p:
            nro_file = nro_file
        else:
            nro_file = nro_file + 1
            tmp = j
            iterador = nro_key_perfil
        ff_pedestal = list_pedestal[nro_file]
        angulo = getDatavaluefromDirFilename(path=dir_pedestal, file=ff_pedestal, value="azimuth")
        angulo_adq[j] = angulo[iterador]
    copyfile(dir_adq + "/" + list_adq[thisFile], wr_path + list_adq[thisFile])
    fp = h5py.File(wr_path + list_adq[thisFile], 'a')
    grp = fp.create_group("Pedestal")
    dset = grp.create_dataset("azimuth"  , data=angulo_adq)
    fp.close()
    print("Angulo", angulo_adq)
    print("Angulo", len(angulo_adq))
    nro_key_perfil = iterador + 25
    if nro_key_perfil < n_perfiles_p:
        nro_file = nro_file
    else:
        nro_file = nro_file + 1
        nro_key_perfil = nro_key_p