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