from global_var import global_var_init
cycle, fs_ecg, fs_ppg, record_name, record_name_csv = global_var_init()

from processing import processing
(signal_dwt , waves_dwt , rpeaks , ecg_signal,
 signal, signal.v, signal.fs, rows_to_extract, t_ecg, t_ppg,
 on_column_list, on_column_list_toArea, on_values, on_values_toArea,
 sp_column_list, sp_values, dn_column_list, dn_values, dp_column_list,
 dp_values, u_column_list, u_column_list_toArea, u_values, u_values_toArea,
 v_column_list, v_values, w_column_list, w_values, a_column_list, a_column_list_toArea,
 a_values, a_values_toArea, b_column_list, b_values, c_column_list, c_values,
 e_column_list, e_values, f_column_list, f_values,
 signal.ppg, signal.vpg, signal.apg, signal.jpg) = processing()


directory_PPG="D://python_study//big_boss//doc//points_coordinates_update//PPG"
directory_VPG="D://python_study//big_boss//doc//points_coordinates_update//VPG"
directory_APG="D://python_study//big_boss//doc//points_coordinates_update//APG"

data_update_PPG = {
    'on_column_list': on_column_list,
    'on_column_list_toArea': on_column_list_toArea,
    'sp_column_list': sp_column_list,
    'dn_column_list': dn_column_list,
    'dp_column_list': dp_column_list
}

data_update_VPG = {
    'u_column_list': u_column_list,
    'u_column_list_toArea': u_column_list_toArea,
    'v_column_list': v_column_list,
    'w_column_list': w_column_list
}

data_update_APG = {
    'a_column_list': a_column_list,
    'a_column_list_toArea': a_column_list_toArea,
    'b_column_list': b_column_list,
    'c_column_list': c_column_list,
    'e_column_list': e_column_list,
    'f_column_list': f_column_list
}


from txt_update import update_dict_from_files
data_update_PPG = update_dict_from_files(data_update_PPG, directory_PPG)
data_update_VPG = update_dict_from_files(data_update_VPG, directory_VPG)
data_update_APG = update_dict_from_files(data_update_APG, directory_APG)

#--------------------------------------------------------------------------------------------------------
#--------------------------------------------------------------------------------------------------------
#--------------------------------------------------------------------------------------------------------
#--------------------------------------------------------------------------------------------------------
#------------------------               得到点的横纵坐标              ------------------------------------
#--------------------------------------------------------------------------------------------------------
#--------------------------------------------------------------------------------------------------------
#--------------------------------------------------------------------------------------------------------
#--------------------------------------------------------------------------------------------------------

def ppg_peaks_val():

    # PPG 数据处理
    on = data_update_PPG['on_column_list']
    on_toArea = data_update_PPG['on_column_list_toArea']
    sp = data_update_PPG['sp_column_list']
    dn = data_update_PPG['dn_column_list']
    dp = data_update_PPG['dp_column_list']

    on_values = [signal.ppg[i] for i in on]
    on_toArea_values = [signal.ppg[i] for i in on_toArea]
    sp_values = [signal.ppg[i] for i in sp]
    dn_values = [signal.ppg[i] for i in dn]
    dp_values = [signal.ppg[i] for i in dp]

    # VPG 数据处理
    u = data_update_VPG['u_column_list']
    u_toArea = data_update_VPG['u_column_list_toArea']
    v = data_update_VPG['v_column_list']
    w = data_update_VPG['w_column_list']

    u_values = [signal.vpg[i] for i in u]
    u_toArea_values = [signal.vpg[i] for i in u_toArea]
    v_values = [signal.vpg[i] for i in v]
    w_values = [signal.vpg[i] for i in w]

    # APG 数据处理
    a = data_update_APG['a_column_list']
    a_toArea = data_update_APG['a_column_list_toArea']
    b = data_update_APG['b_column_list']
    c = data_update_APG['c_column_list']
    e = data_update_APG['e_column_list']
    f = data_update_APG['f_column_list']

    a_values = [signal.apg[i] for i in a]
    a_toArea_values = [signal.apg[i] for i in a_toArea]
    b_values = [signal.apg[i] for i in b]
    c_values = [signal.apg[i] for i in c]
    e_values = [signal.apg[i] for i in e]
    f_values = [signal.apg[i] for i in f]


    return (
        on, on_toArea, sp, dn, dp,
        on_values, on_toArea_values, sp_values, dn_values, dp_values,
        u, u_toArea, v, w,
        u_values, u_toArea_values, v_values, w_values,
        a, a_toArea, b, c, e, f,
        a_values, a_toArea_values, b_values, c_values, e_values, f_values
    )

if __name__ == "__main__":
    
    ppg_peaks_val()