#!/usr/bin/python
# -*- coding: UTF-8 -*-
"""
@author:Marques
@file:Main_Quality_Relabel.py
@email:admin@marques22.com
@email:2021022362@m.scnu.edu.cn
@time:2022/03/28
"""
from pathlib import Path
from utils.Quality_Relabel import Quality_Relabel

# start-----一般不用修改------
# 绘图时的采样率
frequency = 100
# 心晓数据采样率
bcg_frequency = 1000
# PSG_Data_Path = Path("Data/PSG/")
# PSG_Label_Path = Path("Data/PSG_label/")
# BCG_Data_Path = Path("Data/BCG/")
# BCG_Label_Path = Path("Data/BCG_label/")
PSG_Data_Path = Path("I:/CXH/Quality_Relabel/Data/PSG/")
PSG_Label_Path = Path("I:/CXH/Quality_Relabel/Data/PSG_label/")
BCG_Data_Path = Path("I:/CXH/Quality_Relabel/Data/BCG/")
BCG_Label_Path = Path("I:/CXH/Quality_Relabel/Data/BCG_label/")
all_path = [PSG_Data_Path, PSG_Label_Path, BCG_Data_Path, BCG_Label_Path]
# end-----一般不用修改------

# 要遍历的事件
# 可选一个或多个 "Hypopnea" "Central apnea" "Obstructive apnea" "Mixed apnea"
focus_event_list = ["Obstructive apnea"]
# 信号显示事件前多少秒
front_add_second = 60
# 信号显示事件后多少秒
back_add_second = 60

# 样本编号
sampNo = 670
# 从第几个心晓事件数量开始
start_bcg_index = 0

if __name__ == '__main__':
    qualityRelabel = Quality_Relabel(all_path=all_path, sampNo=sampNo, frequency=frequency, bcg_frequency=bcg_frequency,
                                     focus_event_list=focus_event_list)

    qualityRelabel.show_all_event(start_bcg_index=start_bcg_index,
                                  front_add_second=front_add_second,
                                  back_add_second=back_add_second)