#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import xmltodict
from biosppy import utils

def read_ref(file,id_file):
    
    """ Preprocesses an ECG REF signal.

    Parameters
    ----------
    file : array
        Raw ECG signal.
    
    Returns
    -------
    acquisitionDateTime : numpy.ndarray
        The date and time of the acquisition.
    name_dev : numpy.ndarray
        The identification of each lead acquired (e.g. I, II, III, AVR, AVL, AVF, V1, V2, V3, V4, V5, V6).
    meanTemplate : numpy.ndarray
        Average of the heartbeat models extracted for each lead.
    dev : numpy.ndarray
        Time series signal extracted for each lead over a period of 10s.
    unit : Unit of the extracted signal (uV).
    """

    name_dev=[]
    meanTemplate=[]
    dev=[]
    with open(file,"r") as xml_obj:
        
        obj = xmltodict.parse(xml_obj.read())
        xml_obj.close()
        
    acquisitionDateTime=list(obj['sapphire']['dcarRecord']['patientInfo']['visit']['order']['testInfo']['acquisitionDateTime'].values())
    aux=obj['sapphire']['dcarRecord']['patientInfo']['visit']['order']['ecgResting']['params']['ecg']
    auxT=aux['var']['medianTemplate']['ecgWaveformMXG']['ecgWaveform']
    auxD=aux['wav']['ecgWaveformMXG']['ecgWaveform']
    
    for x in range(0,12):
        
        name_dev.append(list(auxT[x].values())[0])   
        aux_=list(auxT[x].values())[4].split(' ')
        meanTemplate.append(list(map(float,aux_)))
        aux_=list(auxD[x].values())[4].split(' ')
        dev.append(list(map(float,aux_)))
        
    unit='uV'
        
    args = (
        id_file,acquisitionDateTime, name_dev, meanTemplate, dev, unit)
    names = (
        "id",
        "acquisitionDateTime",
        "name_dev",
        "meanTemplate",
        "dev",
        "unit"
    )


    return utils.ReturnTuple(args, names)

if __name__ == "__main__":
    
    data_ref=[]
    folder='ECG_REF'
    ind_files=utils.walktree(r''+folder+'', r'\.XML$');
    for i in ind_files:
        
        data_ref.append(read_ref(folder+'/'+utils.fileparts(i)[1]+'.XML',utils.fileparts(i)[1]))