1、完成了<数据精对齐>的所有代码，修复了部分内容
2、新增命名规范
This commit is contained in:
@ -358,7 +358,7 @@ class MainWindow_detect_Jpeak(QMainWindow):
|
||||
|
||||
# 保存
|
||||
PublicFunc.progressbar_update(self, 1, 1, Constants.SAVING_DATA, 0)
|
||||
# status, info = self.data.save()
|
||||
|
||||
total_rows = len(DataFrame(self.data.peak.reshape(-1)))
|
||||
chunk_size = ConfigParams.DETECT_JPEAK_SAVE_CHUNK_SIZE
|
||||
with open(Config["Path"]["Save"], 'w') as f:
|
||||
@ -463,9 +463,6 @@ class Data:
|
||||
return False, Constants.SAVING_FAILURE + Constants.DETECT_JPEAK_FAILURE_REASON["Peak_Not_Exist"]
|
||||
|
||||
try:
|
||||
# DataFrame(self.processed_data.reshape(-1)).to_csv(self.file_path_save,
|
||||
# index=False,
|
||||
# header=False)
|
||||
chunk.to_csv(self.file_path_save, mode='a', index=False, header=False)
|
||||
except Exception:
|
||||
return False, Constants.SAVING_FAILURE + Constants.DETECT_JPEAK_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
@ -357,7 +357,7 @@ class MainWindow_detect_Rpeak(QMainWindow):
|
||||
|
||||
# 保存
|
||||
PublicFunc.progressbar_update(self, 1, 1, Constants.SAVING_DATA, 0)
|
||||
# status, info = self.data.save()
|
||||
|
||||
total_rows = len(DataFrame(self.data.peak.reshape(-1)))
|
||||
chunk_size = ConfigParams.DETECT_RPEAK_SAVE_CHUNK_SIZE
|
||||
with open(Config["Path"]["Save"], 'w') as f:
|
||||
@ -458,9 +458,6 @@ class Data:
|
||||
return False, Constants.SAVING_FAILURE + Constants.DETECT_RPEAK_FAILURE_REASON["Peak_Not_Exist"]
|
||||
|
||||
try:
|
||||
# DataFrame(self.processed_data.reshape(-1)).to_csv(self.file_path_save,
|
||||
# index=False,
|
||||
# header=False)
|
||||
chunk.to_csv(self.file_path_save, mode='a', index=False, header=False)
|
||||
except Exception:
|
||||
return False, Constants.SAVING_FAILURE + Constants.DETECT_RPEAK_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
@ -536,7 +536,7 @@ class MainWindow_label_check(QMainWindow):
|
||||
|
||||
# 保存
|
||||
PublicFunc.progressbar_update(self, 1, 1, Constants.SAVING_DATA, 0)
|
||||
# status, info = self.data.save()
|
||||
|
||||
total_rows = len(DataFrame(self.data.corrected_peak.reshape(-1)))
|
||||
chunk_size = ConfigParams.LABEL_CHECK_SAVE_CHUNK_SIZE
|
||||
with open(Config["Path"]["Save"], 'w') as f:
|
||||
@ -968,9 +968,6 @@ class Data:
|
||||
return False, Constants.SAVING_FAILURE + Constants.LABEL_CHECK_FAILURE_REASON["Peak_Not_Exist"]
|
||||
|
||||
try:
|
||||
# DataFrame(self.corrected_peak.reshape(-1)).to_csv(self.file_path_save,
|
||||
# index=False,
|
||||
# header=False)
|
||||
chunk.to_csv(self.file_path_save, mode='a', index=False, header=False)
|
||||
except Exception:
|
||||
return False, Constants.SAVING_FAILURE + Constants.LABEL_CHECK_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
@ -111,26 +111,48 @@ class SettingWindow(QMainWindow):
|
||||
|
||||
Config.update({
|
||||
"Path": {
|
||||
"Input_orgBcg": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_INPUT_ORGBCG_FILENAME +
|
||||
str(Config["InputConfig"]["ECGFreq"]) +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Input_BCG": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_INPUT_BCG_FILENAME +
|
||||
str(Config["InputConfig"]["ECGFreq"]) +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Input_Jpeak": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_INPUT_JPEAK_FILENAME +
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_INPUT_BCG_FILENAME +
|
||||
str(Config["InputConfig"]["ECGFreq"]) +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Input_Jpeak": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_INPUT_JPEAK_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Input_ECG": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_TEXT /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_INPUT_ECG_FILENAME +
|
||||
str(Config["InputConfig"]["ECGFreq"]) +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_INPUT_ECG_FILENAME +
|
||||
str(Config["InputConfig"]["ECGFreq"]) +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Input_Rpeak": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_TEXT /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_INPUT_RPEAK_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Save_BCG_AlignInfo": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_ALIGNED /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_SAVE_BCG_ALIGNINFO_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Save_ECG_AlignInfo": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_ALIGNED /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_SAVE_ECG_ALIGNINFO_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Save_orgBcg": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_ALIGNED /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_SAVE_ORGBCG_FILENAME +
|
||||
str(Config["InputConfig"]["ECGFreq"]) +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Save_BCG": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_ALIGNED /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_SAVE_BCG_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_SAVE_BCG_FILENAME +
|
||||
str(Config["InputConfig"]["ECGFreq"]) +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Save_ECG": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_ALIGNED /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_SAVE_ECG_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT)))
|
||||
str(Config["InputConfig"]["ECGFreq"]) +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Save_Jpeak": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_ALIGNED /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_SAVE_JPEAK_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT))),
|
||||
"Save_Rpeak": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_ALIGNED /
|
||||
Path(str(self.sampID)) / Path(ConfigParams.PRECISELY_ALIGN_SAVE_RPEAK_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT)))
|
||||
},
|
||||
"Coordinate": {
|
||||
"BCG_front_1": 0,
|
||||
@ -177,23 +199,35 @@ class SettingWindow(QMainWindow):
|
||||
|
||||
# 数据回显
|
||||
self.ui.spinBox_input_freq_ECG.setValue(Config["InputConfig"]["ECGFreq"])
|
||||
self.ui.plainTextEdit_file_path_input_orgBcg.setPlainText(Config["Path"]["Input_orgBcg"])
|
||||
self.ui.plainTextEdit_file_path_input_BCG.setPlainText(Config["Path"]["Input_BCG"])
|
||||
self.ui.plainTextEdit_file_path_input_Jpeak.setPlainText(Config["Path"]["Input_Jpeak"])
|
||||
self.ui.plainTextEdit_file_path_input_ECG.setPlainText(Config["Path"]["Input_ECG"])
|
||||
self.ui.plainTextEdit_file_path_input_Rpeak.setPlainText(Config["Path"]["Input_Rpeak"])
|
||||
self.ui.plainTextEdit_file_path_save_BCG_AlignInfo.setPlainText(Config["Path"]["Save_BCG_AlignInfo"])
|
||||
self.ui.plainTextEdit_file_path_save_ECG_AlignInfo.setPlainText(Config["Path"]["Save_ECG_AlignInfo"])
|
||||
self.ui.plainTextEdit_file_path_save_orgBcg.setPlainText(Config["Path"]["Save_orgBcg"])
|
||||
self.ui.plainTextEdit_file_path_save_BCG.setPlainText(Config["Path"]["Save_BCG"])
|
||||
self.ui.plainTextEdit_file_path_save_ECG.setPlainText(Config["Path"]["Save_ECG"])
|
||||
self.ui.plainTextEdit_file_path_save_Jpeak.setPlainText(Config["Path"]["Save_Jpeak"])
|
||||
self.ui.plainTextEdit_file_path_save_Rpeak.setPlainText(Config["Path"]["Save_Rpeak"])
|
||||
|
||||
def __write_config__(self):
|
||||
|
||||
# 从界面写入配置
|
||||
Config["InputConfig"]["ECGFreq"] = self.ui.spinBox_input_freq_ECG.value()
|
||||
Config["Path"]["Input_orgBcg"] = self.ui.plainTextEdit_file_path_input_orgBcg.toPlainText()
|
||||
Config["Path"]["Input_BCG"] = self.ui.plainTextEdit_file_path_input_BCG.toPlainText()
|
||||
Config["Path"]["Input_Jpeak"] = self.ui.plainTextEdit_file_path_input_Jpeak.toPlainText()
|
||||
Config["Path"]["Input_ECG"] = self.ui.plainTextEdit_file_path_input_ECG.toPlainText()
|
||||
Config["Path"]["Input_Rpeak"] = self.ui.plainTextEdit_file_path_input_Rpeak.toPlainText()
|
||||
Config["Path"]["Save_BCG_AlignInfo"] = self.ui.plainTextEdit_file_path_save_BCG_AlignInfo.toPlainText()
|
||||
Config["Path"]["Save_ECG_AlignInfo"] = self.ui.plainTextEdit_file_path_save_ECG_AlignInfo.toPlainText()
|
||||
Config["Path"]["Save_orgBcg"] = self.ui.plainTextEdit_file_path_save_orgBcg.toPlainText()
|
||||
Config["Path"]["Save_BCG"] = self.ui.plainTextEdit_file_path_save_BCG.toPlainText()
|
||||
Config["Path"]["Save_ECG"] = self.ui.plainTextEdit_file_path_save_ECG.toPlainText()
|
||||
Config["Path"]["Save_Jpeak"] = self.ui.plainTextEdit_file_path_save_Jpeak.toPlainText()
|
||||
Config["Path"]["Save_Rpeak"] = self.ui.plainTextEdit_file_path_save_Rpeak.toPlainText()
|
||||
|
||||
# 保存配置到文件
|
||||
self.config["InputConfig"]["ECGFreq"] = self.ui.spinBox_input_freq_ECG.value()
|
||||
@ -209,6 +243,13 @@ class SettingWindow(QMainWindow):
|
||||
|
||||
def __update_ui__(self):
|
||||
|
||||
self.ui.plainTextEdit_file_path_input_orgBcg.setPlainText(
|
||||
str((Path(self.root_path) /
|
||||
ConfigParams.PUBLIC_PATH_ORGBCG_TEXT /
|
||||
Path(str(self.sampID)) /
|
||||
Path(ConfigParams.PRECISELY_ALIGN_INPUT_ORGBCG_FILENAME +
|
||||
str(self.ui.spinBox_input_freq_ECG.value()) +
|
||||
ConfigParams.ENDSWITH_TXT))))
|
||||
self.ui.plainTextEdit_file_path_input_BCG.setPlainText(
|
||||
str((Path(self.root_path) /
|
||||
ConfigParams.PUBLIC_PATH_ORGBCG_TEXT /
|
||||
@ -223,6 +264,39 @@ class SettingWindow(QMainWindow):
|
||||
Path(ConfigParams.PRECISELY_ALIGN_INPUT_ECG_FILENAME +
|
||||
str(self.ui.spinBox_input_freq_ECG.value()) +
|
||||
ConfigParams.ENDSWITH_TXT))))
|
||||
self.ui.plainTextEdit_file_path_save_orgBcg.setPlainText(
|
||||
str((Path(self.root_path) /
|
||||
ConfigParams.PUBLIC_PATH_ORGBCG_ALIGNED /
|
||||
Path(str(self.sampID)) /
|
||||
Path(ConfigParams.PRECISELY_ALIGN_SAVE_ORGBCG_FILENAME +
|
||||
str(self.ui.spinBox_input_freq_ECG.value()) +
|
||||
ConfigParams.ENDSWITH_TXT))))
|
||||
self.ui.plainTextEdit_file_path_save_BCG.setPlainText(
|
||||
str((Path(self.root_path) /
|
||||
ConfigParams.PUBLIC_PATH_ORGBCG_ALIGNED /
|
||||
Path(str(self.sampID)) /
|
||||
Path(ConfigParams.PRECISELY_ALIGN_SAVE_BCG_FILENAME +
|
||||
str(self.ui.spinBox_input_freq_ECG.value()) +
|
||||
ConfigParams.ENDSWITH_TXT))))
|
||||
self.ui.plainTextEdit_file_path_save_ECG.setPlainText(
|
||||
str((Path(self.root_path) /
|
||||
ConfigParams.PUBLIC_PATH_PSG_ALIGNED /
|
||||
Path(str(self.sampID)) /
|
||||
Path(ConfigParams.PRECISELY_ALIGN_SAVE_ECG_FILENAME +
|
||||
str(self.ui.spinBox_input_freq_ECG.value()) +
|
||||
ConfigParams.ENDSWITH_TXT))))
|
||||
self.ui.plainTextEdit_file_path_save_Jpeak.setPlainText(
|
||||
str((Path(self.root_path) /
|
||||
ConfigParams.PUBLIC_PATH_ORGBCG_ALIGNED /
|
||||
Path(str(self.sampID)) /
|
||||
Path(ConfigParams.PRECISELY_ALIGN_SAVE_JPEAK_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT))))
|
||||
self.ui.plainTextEdit_file_path_save_Rpeak.setPlainText(
|
||||
str((Path(self.root_path) /
|
||||
ConfigParams.PUBLIC_PATH_PSG_ALIGNED /
|
||||
Path(str(self.sampID)) /
|
||||
Path(ConfigParams.PRECISELY_ALIGN_SAVE_RPEAK_FILENAME +
|
||||
ConfigParams.ENDSWITH_TXT))))
|
||||
|
||||
|
||||
class MainWindow_precisely_align(QMainWindow):
|
||||
@ -745,77 +819,190 @@ class MainWindow_precisely_align(QMainWindow):
|
||||
if reply == QMessageBox.Yes:
|
||||
PublicFunc.__disableAllButton__(self, ButtonState)
|
||||
|
||||
# 保存
|
||||
PublicFunc.progressbar_update(self, 1, 1, Constants.SAVING_DATA, 0)
|
||||
status, info = self.data.save()
|
||||
# 保存对齐信息
|
||||
PublicFunc.progressbar_update(self, 1, 6, Constants.PRECISELY_ALIGN_SAVING_ALIGNINFO, 0)
|
||||
status, info = self.data.save_alignInfo()
|
||||
if not status:
|
||||
PublicFunc.text_output(self.ui, "(1/1)" + info, Constants.TIPS_TYPE_ERROR)
|
||||
PublicFunc.text_output(self.ui, "(1/6)" + info, Constants.TIPS_TYPE_ERROR)
|
||||
PublicFunc.msgbox_output(self, info, Constants.MSGBOX_TYPE_ERROR)
|
||||
PublicFunc.finish_operation(self, ButtonState)
|
||||
return
|
||||
else:
|
||||
PublicFunc.text_output(self.ui, "(1/1)" + info, Constants.TIPS_TYPE_INFO)
|
||||
PublicFunc.msgbox_output(self, info, Constants.TIPS_TYPE_INFO)
|
||||
PublicFunc.text_output(self.ui, "(1/6)" + info, Constants.TIPS_TYPE_INFO)
|
||||
|
||||
# 保存切割后orgBcg
|
||||
PublicFunc.progressbar_update(self, 2, 6, Constants.PRECISELY_ALIGN_SAVING_RES_ORGBCG, 0)
|
||||
|
||||
total_rows = len(DataFrame(self.data.res_orgBcg.reshape(-1)))
|
||||
chunk_size = ConfigParams.PRECISELY_ALIGN_SAVE_CHUNK_SIZE
|
||||
with open(Config["Path"]["Save_orgBcg"], 'w') as f:
|
||||
for start in range(0, total_rows, chunk_size):
|
||||
end = min(start + chunk_size, total_rows)
|
||||
chunk = DataFrame(self.data.res_orgBcg.reshape(-1)).iloc[start:end]
|
||||
status, info = self.data.save_res_orgBcg(chunk)
|
||||
progress = int((end / total_rows) * 100)
|
||||
self.progressbar.setValue(progress)
|
||||
QApplication.processEvents()
|
||||
|
||||
if not status:
|
||||
PublicFunc.text_output(self.ui, "(2/6)" + info, Constants.TIPS_TYPE_ERROR)
|
||||
PublicFunc.msgbox_output(self, info, Constants.MSGBOX_TYPE_ERROR)
|
||||
PublicFunc.finish_operation(self, ButtonState)
|
||||
return
|
||||
else:
|
||||
PublicFunc.text_output(self.ui, "(2/6)" + info, Constants.TIPS_TYPE_INFO)
|
||||
|
||||
# 保存切割后BCG
|
||||
PublicFunc.progressbar_update(self, 3, 6, Constants.PRECISELY_ALIGN_SAVING_RES_BCG, 0)
|
||||
|
||||
total_rows = len(DataFrame(self.data.res_BCG.reshape(-1)))
|
||||
chunk_size = ConfigParams.PRECISELY_ALIGN_SAVE_CHUNK_SIZE
|
||||
with open(Config["Path"]["Save_orgBcg"], 'w') as f:
|
||||
for start in range(0, total_rows, chunk_size):
|
||||
end = min(start + chunk_size, total_rows)
|
||||
chunk = DataFrame(self.data.res_BCG.reshape(-1)).iloc[start:end]
|
||||
status, info = self.data.save_res_BCG(chunk)
|
||||
progress = int((end / total_rows) * 100)
|
||||
self.progressbar.setValue(progress)
|
||||
QApplication.processEvents()
|
||||
|
||||
if not status:
|
||||
PublicFunc.text_output(self.ui, "(3/6)" + info, Constants.TIPS_TYPE_ERROR)
|
||||
PublicFunc.msgbox_output(self, info, Constants.MSGBOX_TYPE_ERROR)
|
||||
PublicFunc.finish_operation(self, ButtonState)
|
||||
return
|
||||
else:
|
||||
PublicFunc.text_output(self.ui, "(3/6)" + info, Constants.TIPS_TYPE_INFO)
|
||||
|
||||
# 保存切割后ECG
|
||||
PublicFunc.progressbar_update(self, 4, 6, Constants.PRECISELY_ALIGN_SAVING_CUT_ECG, 0)
|
||||
|
||||
total_rows = len(DataFrame(self.data.cut_ECG.reshape(-1)))
|
||||
chunk_size = ConfigParams.PRECISELY_ALIGN_SAVE_CHUNK_SIZE
|
||||
with open(Config["Path"]["Save_ECG"], 'w') as f:
|
||||
for start in range(0, total_rows, chunk_size):
|
||||
end = min(start + chunk_size, total_rows)
|
||||
chunk = DataFrame(self.data.cut_ECG.reshape(-1)).iloc[start:end]
|
||||
status, info = self.data.save_cut_ECG(chunk)
|
||||
progress = int((end / total_rows) * 100)
|
||||
self.progressbar.setValue(progress)
|
||||
QApplication.processEvents()
|
||||
|
||||
if not status:
|
||||
PublicFunc.text_output(self.ui, "(4/6)" + info, Constants.TIPS_TYPE_ERROR)
|
||||
PublicFunc.msgbox_output(self, info, Constants.MSGBOX_TYPE_ERROR)
|
||||
PublicFunc.finish_operation(self, ButtonState)
|
||||
return
|
||||
else:
|
||||
PublicFunc.text_output(self.ui, "(4/6)" + info, Constants.TIPS_TYPE_INFO)
|
||||
|
||||
# 保存切割后J峰
|
||||
PublicFunc.progressbar_update(self, 5, 6, Constants.PRECISELY_ALIGN_SAVING_CUT_JPEAK, 0)
|
||||
|
||||
total_rows = len(DataFrame(self.data.cut_Jpeak.reshape(-1)))
|
||||
chunk_size = ConfigParams.PRECISELY_ALIGN_SAVE_PEAK_CHUNK_SIZE
|
||||
with open(Config["Path"]["Save_Jpeak"], 'w') as f:
|
||||
for start in range(0, total_rows, chunk_size):
|
||||
end = min(start + chunk_size, total_rows)
|
||||
chunk = DataFrame(self.data.cut_Jpeak.reshape(-1)).iloc[start:end]
|
||||
status, info = self.data.save_Jpeak(chunk)
|
||||
progress = int((end / total_rows) * 100)
|
||||
self.progressbar.setValue(progress)
|
||||
QApplication.processEvents()
|
||||
|
||||
if not status:
|
||||
PublicFunc.text_output(self.ui, "(5/6)" + info, Constants.TIPS_TYPE_ERROR)
|
||||
PublicFunc.msgbox_output(self, info, Constants.MSGBOX_TYPE_ERROR)
|
||||
PublicFunc.finish_operation(self, ButtonState)
|
||||
return
|
||||
else:
|
||||
PublicFunc.text_output(self.ui, "(5/6)" + info, Constants.TIPS_TYPE_INFO)
|
||||
|
||||
# 保存切割后R峰
|
||||
PublicFunc.progressbar_update(self, 6, 6, Constants.PRECISELY_ALIGN_SAVING_CUT_RPEAK, 0)
|
||||
|
||||
total_rows = len(DataFrame(self.data.cut_Rpeak.reshape(-1)))
|
||||
chunk_size = ConfigParams.PRECISELY_ALIGN_SAVE_PEAK_CHUNK_SIZE
|
||||
with open(Config["Path"]["Save_Rpeak"], 'w') as f:
|
||||
for start in range(0, total_rows, chunk_size):
|
||||
end = min(start + chunk_size, total_rows)
|
||||
chunk = DataFrame(self.data.cut_Rpeak.reshape(-1)).iloc[start:end]
|
||||
status, info = self.data.save_Rpeak(chunk)
|
||||
progress = int((end / total_rows) * 100)
|
||||
self.progressbar.setValue(progress)
|
||||
QApplication.processEvents()
|
||||
|
||||
if not status:
|
||||
PublicFunc.text_output(self.ui, "(6/6)" + info, Constants.TIPS_TYPE_ERROR)
|
||||
PublicFunc.msgbox_output(self, info, Constants.MSGBOX_TYPE_ERROR)
|
||||
PublicFunc.finish_operation(self, ButtonState)
|
||||
return
|
||||
else:
|
||||
PublicFunc.text_output(self.ui, "(6/6)" + info, Constants.TIPS_TYPE_INFO)
|
||||
PublicFunc.msgbox_output(self, Constants.SAVING_FINISHED, Constants.TIPS_TYPE_INFO)
|
||||
|
||||
PublicFunc.finish_operation(self, ButtonState)
|
||||
|
||||
def __update_coordinate__(self):
|
||||
|
||||
if self.data is not None:
|
||||
if self.data.Jpeak is None or self.data.Rpeak is None:
|
||||
PublicFunc.msgbox_output(self, Constants.PRECISELY_ALIGN_FAILURE_REASON["Data_Not_Exist"], Constants.MSGBOX_TYPE_ERROR)
|
||||
return
|
||||
try:
|
||||
if self.data is not None:
|
||||
if self.data.Jpeak is None or self.data.Rpeak is None:
|
||||
PublicFunc.msgbox_output(self, Constants.PRECISELY_ALIGN_FAILURE_REASON["Data_Not_Exist"], Constants.MSGBOX_TYPE_ERROR)
|
||||
return
|
||||
|
||||
sender = self.sender()
|
||||
sender = self.sender()
|
||||
|
||||
if sender == self.ui.spinBox_BCG_front_JJIV_1:
|
||||
if self.ui.spinBox_BCG_front_JJIV_1.value() >= len(self.data.Jpeak[:-2]):
|
||||
self.ui.spinBox_BCG_front_JJIV_1.setValue(len(self.data.Jpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["BCG_front_1"] = self.ui.spinBox_BCG_front_JJIV_1.value()
|
||||
Config["Coordinate"]["BCG_front_1"] = self.data.Jpeak[:-2][self.ui.spinBox_BCG_front_JJIV_1.value()]
|
||||
self.ui.spinBox_BCG_front_Signal_1.setValue(Config["Coordinate"]["BCG_front_1"])
|
||||
elif sender == self.ui.spinBox_BCG_front_JJIV_2:
|
||||
if self.ui.spinBox_BCG_front_JJIV_2.value() >= len(self.data.Jpeak[:-2]):
|
||||
self.ui.spinBox_BCG_front_JJIV_2.setValue(len(self.data.Jpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["BCG_front_2"] = self.ui.spinBox_BCG_front_JJIV_2.value()
|
||||
Config["Coordinate"]["BCG_front_2"] = self.data.Jpeak[:-2][self.ui.spinBox_BCG_front_JJIV_2.value()]
|
||||
self.ui.spinBox_BCG_front_Signal_2.setValue(Config["Coordinate"]["BCG_front_2"])
|
||||
elif sender == self.ui.spinBox_BCG_back_JJIV_1:
|
||||
if self.ui.spinBox_BCG_back_JJIV_1.value() >= len(self.data.Jpeak[:-2]):
|
||||
self.ui.spinBox_BCG_back_JJIV_1.setValue(len(self.data.Jpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["BCG_back_1"] = self.ui.spinBox_BCG_back_JJIV_1.value()
|
||||
Config["Coordinate"]["BCG_back_1"] = self.data.Jpeak[:-2][self.ui.spinBox_BCG_back_JJIV_1.value()]
|
||||
self.ui.spinBox_BCG_back_Signal_1.setValue(Config["Coordinate"]["BCG_back_1"])
|
||||
elif sender == self.ui.spinBox_BCG_back_JJIV_2:
|
||||
if self.ui.spinBox_BCG_back_JJIV_2.value() >= len(self.data.Jpeak[:-2]):
|
||||
self.ui.spinBox_BCG_back_JJIV_2.setValue(len(self.data.Jpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["BCG_back_2"] = self.ui.spinBox_BCG_back_JJIV_2.value()
|
||||
Config["Coordinate"]["BCG_back_2"] = self.data.Jpeak[:-2][self.ui.spinBox_BCG_back_JJIV_2.value()]
|
||||
self.ui.spinBox_BCG_back_Signal_2.setValue(Config["Coordinate"]["BCG_back_2"])
|
||||
elif sender == self.ui.spinBox_ECG_front_RRIV_1:
|
||||
if self.ui.spinBox_ECG_front_RRIV_1.value() >= len(self.data.Rpeak[:-2]):
|
||||
self.ui.spinBox_ECG_front_RRIV_1.setValue(len(self.data.Rpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["ECG_front_1"] = self.ui.spinBox_ECG_front_RRIV_1.value()
|
||||
Config["Coordinate"]["ECG_front_1"] = self.data.Rpeak[:-2][self.ui.spinBox_ECG_front_RRIV_1.value()]
|
||||
self.ui.spinBox_ECG_front_Signal_1.setValue(Config["Coordinate"]["ECG_front_1"])
|
||||
elif sender == self.ui.spinBox_ECG_front_RRIV_2:
|
||||
if self.ui.spinBox_ECG_front_RRIV_2.value() >= len(self.data.Rpeak[:-2]):
|
||||
self.ui.spinBox_ECG_front_RRIV_2.setValue(len(self.data.Rpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["ECG_front_2"] = self.ui.spinBox_ECG_front_RRIV_2.value()
|
||||
Config["Coordinate"]["ECG_front_2"] = self.data.Rpeak[:-2][self.ui.spinBox_ECG_front_RRIV_2.value()]
|
||||
self.ui.spinBox_ECG_front_Signal_2.setValue(Config["Coordinate"]["ECG_front_2"])
|
||||
elif sender == self.ui.spinBox_ECG_back_RRIV_1:
|
||||
if self.ui.spinBox_ECG_back_RRIV_1.value() >= len(self.data.Rpeak[:-2]):
|
||||
self.ui.spinBox_ECG_back_RRIV_1.setValue(len(self.data.Rpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["ECG_back_1"] = self.ui.spinBox_ECG_back_RRIV_1.value()
|
||||
Config["Coordinate"]["ECG_back_1"] = self.data.Rpeak[:-2][self.ui.spinBox_ECG_back_RRIV_1.value()]
|
||||
self.ui.spinBox_ECG_back_Signal_1.setValue(Config["Coordinate"]["ECG_back_1"])
|
||||
elif sender == self.ui.spinBox_ECG_back_RRIV_2:
|
||||
if self.ui.spinBox_ECG_back_RRIV_2.value() >= len(self.data.Rpeak[:-2]):
|
||||
self.ui.spinBox_ECG_back_RRIV_2.setValue(len(self.data.Rpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["ECG_back_2"] = self.ui.spinBox_ECG_back_RRIV_2.value()
|
||||
Config["Coordinate"]["ECG_back_2"] = self.data.Rpeak[:-2][self.ui.spinBox_ECG_back_RRIV_2.value()]
|
||||
self.ui.spinBox_ECG_back_Signal_2.setValue(Config["Coordinate"]["ECG_back_2"])
|
||||
if sender == self.ui.spinBox_BCG_front_JJIV_1:
|
||||
if self.ui.spinBox_BCG_front_JJIV_1.value() >= len(self.data.Jpeak[:-2]):
|
||||
self.ui.spinBox_BCG_front_JJIV_1.setValue(len(self.data.Jpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["BCG_front_1"] = self.ui.spinBox_BCG_front_JJIV_1.value()
|
||||
Config["Coordinate"]["BCG_front_1"] = self.data.Jpeak[:-2][self.ui.spinBox_BCG_front_JJIV_1.value()]
|
||||
self.ui.spinBox_BCG_front_Signal_1.setValue(Config["Coordinate"]["BCG_front_1"])
|
||||
elif sender == self.ui.spinBox_BCG_front_JJIV_2:
|
||||
if self.ui.spinBox_BCG_front_JJIV_2.value() >= len(self.data.Jpeak[:-2]):
|
||||
self.ui.spinBox_BCG_front_JJIV_2.setValue(len(self.data.Jpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["BCG_front_2"] = self.ui.spinBox_BCG_front_JJIV_2.value()
|
||||
Config["Coordinate"]["BCG_front_2"] = self.data.Jpeak[:-2][self.ui.spinBox_BCG_front_JJIV_2.value()]
|
||||
self.ui.spinBox_BCG_front_Signal_2.setValue(Config["Coordinate"]["BCG_front_2"])
|
||||
elif sender == self.ui.spinBox_BCG_back_JJIV_1:
|
||||
if self.ui.spinBox_BCG_back_JJIV_1.value() >= len(self.data.Jpeak[:-2]):
|
||||
self.ui.spinBox_BCG_back_JJIV_1.setValue(len(self.data.Jpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["BCG_back_1"] = self.ui.spinBox_BCG_back_JJIV_1.value()
|
||||
Config["Coordinate"]["BCG_back_1"] = self.data.Jpeak[:-2][self.ui.spinBox_BCG_back_JJIV_1.value()]
|
||||
self.ui.spinBox_BCG_back_Signal_1.setValue(Config["Coordinate"]["BCG_back_1"])
|
||||
elif sender == self.ui.spinBox_BCG_back_JJIV_2:
|
||||
if self.ui.spinBox_BCG_back_JJIV_2.value() >= len(self.data.Jpeak[:-2]):
|
||||
self.ui.spinBox_BCG_back_JJIV_2.setValue(len(self.data.Jpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["BCG_back_2"] = self.ui.spinBox_BCG_back_JJIV_2.value()
|
||||
Config["Coordinate"]["BCG_back_2"] = self.data.Jpeak[:-2][self.ui.spinBox_BCG_back_JJIV_2.value()]
|
||||
self.ui.spinBox_BCG_back_Signal_2.setValue(Config["Coordinate"]["BCG_back_2"])
|
||||
elif sender == self.ui.spinBox_ECG_front_RRIV_1:
|
||||
if self.ui.spinBox_ECG_front_RRIV_1.value() >= len(self.data.Rpeak[:-2]):
|
||||
self.ui.spinBox_ECG_front_RRIV_1.setValue(len(self.data.Rpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["ECG_front_1"] = self.ui.spinBox_ECG_front_RRIV_1.value()
|
||||
Config["Coordinate"]["ECG_front_1"] = self.data.Rpeak[:-2][self.ui.spinBox_ECG_front_RRIV_1.value()]
|
||||
self.ui.spinBox_ECG_front_Signal_1.setValue(Config["Coordinate"]["ECG_front_1"])
|
||||
elif sender == self.ui.spinBox_ECG_front_RRIV_2:
|
||||
if self.ui.spinBox_ECG_front_RRIV_2.value() >= len(self.data.Rpeak[:-2]):
|
||||
self.ui.spinBox_ECG_front_RRIV_2.setValue(len(self.data.Rpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["ECG_front_2"] = self.ui.spinBox_ECG_front_RRIV_2.value()
|
||||
Config["Coordinate"]["ECG_front_2"] = self.data.Rpeak[:-2][self.ui.spinBox_ECG_front_RRIV_2.value()]
|
||||
self.ui.spinBox_ECG_front_Signal_2.setValue(Config["Coordinate"]["ECG_front_2"])
|
||||
elif sender == self.ui.spinBox_ECG_back_RRIV_1:
|
||||
if self.ui.spinBox_ECG_back_RRIV_1.value() >= len(self.data.Rpeak[:-2]):
|
||||
self.ui.spinBox_ECG_back_RRIV_1.setValue(len(self.data.Rpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["ECG_back_1"] = self.ui.spinBox_ECG_back_RRIV_1.value()
|
||||
Config["Coordinate"]["ECG_back_1"] = self.data.Rpeak[:-2][self.ui.spinBox_ECG_back_RRIV_1.value()]
|
||||
self.ui.spinBox_ECG_back_Signal_1.setValue(Config["Coordinate"]["ECG_back_1"])
|
||||
elif sender == self.ui.spinBox_ECG_back_RRIV_2:
|
||||
if self.ui.spinBox_ECG_back_RRIV_2.value() >= len(self.data.Rpeak[:-2]):
|
||||
self.ui.spinBox_ECG_back_RRIV_2.setValue(len(self.data.Rpeak[:-2]) - 1)
|
||||
Config["IV_Coordinate"]["ECG_back_2"] = self.ui.spinBox_ECG_back_RRIV_2.value()
|
||||
Config["Coordinate"]["ECG_back_2"] = self.data.Rpeak[:-2][self.ui.spinBox_ECG_back_RRIV_2.value()]
|
||||
self.ui.spinBox_ECG_back_Signal_2.setValue(Config["Coordinate"]["ECG_back_2"])
|
||||
except AttributeError:
|
||||
pass
|
||||
|
||||
def reset_axes(self):
|
||||
|
||||
@ -900,6 +1087,8 @@ class MainWindow_precisely_align(QMainWindow):
|
||||
|
||||
return True, Constants.DRAWING_FINISHED
|
||||
|
||||
return False, Constants.DRAWING_FAILURE
|
||||
|
||||
def redraw_correlation_align(self, plot_element=None):
|
||||
|
||||
if plot_element is not None and plot_element["mode"] == "select":
|
||||
@ -937,6 +1126,8 @@ class MainWindow_precisely_align(QMainWindow):
|
||||
|
||||
return True, Constants.DRAWING_FINISHED
|
||||
|
||||
return False, Constants.DRAWING_FAILURE
|
||||
|
||||
def toggle_home(self):
|
||||
|
||||
if self.ax0 is not None:
|
||||
@ -1217,16 +1408,25 @@ class MainWindow_precisely_align(QMainWindow):
|
||||
raise ValueError("this_line不存在")
|
||||
self.__slot_btn_correlation_align__()
|
||||
|
||||
|
||||
class Data:
|
||||
|
||||
def __init__(self):
|
||||
|
||||
self.file_path_input_orgBcg = Config["Path"]["Input_orgBcg"]
|
||||
self.file_path_input_BCG = Config["Path"]["Input_BCG"]
|
||||
self.file_path_input_Jpeak = Config["Path"]["Input_Jpeak"]
|
||||
self.file_path_input_ECG = Config["Path"]["Input_ECG"]
|
||||
self.file_path_input_Rpeak = Config["Path"]["Input_Rpeak"]
|
||||
self.file_path_save_BCG_AlignInfo = Config["Path"]["Save_BCG_AlignInfo"]
|
||||
self.file_path_save_ECG_AlignInfo = Config["Path"]["Save_ECG_AlignInfo"]
|
||||
self.file_path_save_orgBcg = Config["Path"]["Save_orgBcg"]
|
||||
self.file_path_save_BCG = Config["Path"]["Save_BCG"]
|
||||
self.file_path_save_ECG = Config["Path"]["Save_ECG"]
|
||||
self.file_path_save_Jpeak = Config["Path"]["Save_Jpeak"]
|
||||
self.file_path_save_Rpeak = Config["Path"]["Save_Rpeak"]
|
||||
|
||||
self.raw_orgBcg = None
|
||||
self.raw_BCG = None
|
||||
self.Jpeak = None
|
||||
self.Jpeak_y = None
|
||||
@ -1234,10 +1434,11 @@ class Data:
|
||||
self.Rpeak = None
|
||||
self.Rpeak_y = None
|
||||
|
||||
self.cut_ECG = None
|
||||
self.res_orgBcg = None
|
||||
self.res_BCG = None
|
||||
self.cut_Rpeak = None
|
||||
self.cut_ECG = None
|
||||
self.cut_Jpeak = None
|
||||
self.cut_Rpeak = None
|
||||
|
||||
self.RRIs = None
|
||||
self.JJIs = None
|
||||
@ -1268,6 +1469,9 @@ class Data:
|
||||
return False, Constants.INPUT_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["Data_Path_Not_Exist"]
|
||||
|
||||
try:
|
||||
self.raw_orgBcg = read_csv(self.file_path_input_orgBcg,
|
||||
encoding=ConfigParams.UTF8_ENCODING,
|
||||
header=None).to_numpy().reshape(-1)
|
||||
self.raw_BCG = read_csv(self.file_path_input_BCG,
|
||||
encoding=ConfigParams.UTF8_ENCODING,
|
||||
header=None).to_numpy().reshape(-1)
|
||||
@ -1457,8 +1661,9 @@ class Data:
|
||||
orgfs = Config["orgfs"]
|
||||
off = Config["offset_anchor"]
|
||||
|
||||
self.cut_ECG = self.raw_ECG
|
||||
self.res_orgBcg = self.raw_orgBcg
|
||||
self.res_BCG = self.raw_BCG
|
||||
self.cut_ECG = self.raw_ECG
|
||||
self.cut_Rpeak = self.Rpeak
|
||||
|
||||
if off > 0:
|
||||
@ -1469,10 +1674,12 @@ class Data:
|
||||
self.cut_Rpeak = self.cut_Rpeak[idxs] - off
|
||||
else:
|
||||
self.res_BCG = self.res_BCG[-off:]
|
||||
self.res_orgBcg = self.res_orgBcg[-off:]
|
||||
anchor0[1] = anchor0[1] + off
|
||||
anchor1[1] = anchor1[1] + off
|
||||
|
||||
self.res_BCG = resample(self.res_BCG, orgfs, Config["InputConfig"]["ECGFreq"])
|
||||
self.res_orgBcg = resample(self.res_orgBcg, orgfs, Config["InputConfig"]["ECGFreq"])
|
||||
|
||||
anchor0[1] = round(int(anchor0[1]) * Config["InputConfig"]["ECGFreq"] / orgfs)
|
||||
anchor1[1] = round(int(anchor1[1]) * Config["InputConfig"]["ECGFreq"] / orgfs)
|
||||
@ -1486,12 +1693,14 @@ class Data:
|
||||
self.cut_Rpeak = self.cut_Rpeak[idxs] - off
|
||||
else:
|
||||
self.res_BCG = self.res_BCG[-off:]
|
||||
self.res_orgBcg = self.res_orgBcg[-off:]
|
||||
anchor0[1] = anchor0[1] + off
|
||||
anchor1[1] = anchor1[1] + off
|
||||
|
||||
datalen = np_min([len(self.cut_ECG), len(self.res_BCG)])
|
||||
self.cut_ECG = self.cut_ECG[:datalen]
|
||||
self.res_BCG = self.res_BCG[:datalen]
|
||||
self.res_orgBcg = self.res_orgBcg[:datalen]
|
||||
a = np_max([np_max(self.cut_ECG), np_max(self.res_BCG)])
|
||||
b = np_min([np_min(self.cut_ECG), np_min(self.res_BCG)])
|
||||
peak_ECG, _ = find_peaks(self.cut_ECG)
|
||||
@ -1544,12 +1753,14 @@ class Data:
|
||||
self.cut_Rpeak = self.cut_Rpeak[where(self.cut_Rpeak > off)[0]] - off
|
||||
else:
|
||||
self.res_BCG = self.res_BCG[-off:]
|
||||
self.res_orgBcg = self.res_orgBcg[-off:]
|
||||
anchor0[1] = anchor0[1] + off
|
||||
anchor1[1] = anchor1[1] + off
|
||||
|
||||
datalen = np_min([len(self.cut_ECG), len(self.res_BCG)])
|
||||
self.cut_ECG = self.cut_ECG[:datalen]
|
||||
self.res_BCG = self.res_BCG[:datalen]
|
||||
self.res_orgBcg = self.res_orgBcg[:datalen]
|
||||
|
||||
idxs = where(self.cut_Rpeak < datalen)[0]
|
||||
self.cut_Rpeak = self.cut_Rpeak[idxs]
|
||||
@ -1573,13 +1784,12 @@ class Data:
|
||||
Config["backcut_index_BCG"] = backcut_index_BCG
|
||||
Config["frontcut_index_ECG"] = frontcut_index_ECG
|
||||
Config["backcut_index_ECG"] = backcut_index_ECG
|
||||
|
||||
except Exception:
|
||||
return False, Constants.PRECISELY_ALIGN_POSTPROCESS_VIEW_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["PostProcess_Align_Exception"]
|
||||
|
||||
return True, f"{Constants.PRECISELY_ALIGN_POSTPROCESS_VIEW_FINISHED}，BCG前后段被切割的坐标值为[{frontcut_index_BCG}, {backcut_index_BCG}]，ECG前后段被切割的坐标值为[{frontcut_index_ECG}, {backcut_index_ECG}]"
|
||||
|
||||
def save(self):
|
||||
def save_alignInfo(self):
|
||||
|
||||
try:
|
||||
save_data = {
|
||||
@ -1613,13 +1823,73 @@ class Data:
|
||||
}
|
||||
}
|
||||
save_data = [str(save_data)]
|
||||
DataFrame(save_data).to_csv(self.file_path_save_BCG, index=False, header=False)
|
||||
DataFrame(save_data).to_csv(self.file_path_save_ECG, index=False, header=False)
|
||||
DataFrame(save_data).to_csv(self.file_path_save_BCG_AlignInfo, index=False, header=False)
|
||||
DataFrame(save_data).to_csv(self.file_path_save_ECG_AlignInfo, index=False, header=False)
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return False, Constants.SAVING_FAILURE + Constants.DETECT_RPEAK_FAILURE_REASON["Save_Exception"]
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_ALIGNINFO_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
return True, Constants.SAVING_FINISHED
|
||||
return True, Constants.PRECISELY_ALIGN_SAVING_ALIGNINFO_FINISHED
|
||||
|
||||
def save_res_orgBcg(self, chunk):
|
||||
|
||||
if self.res_orgBcg is None:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_RES_ORGBCG_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["res_orgBcg_Not_Exist"]
|
||||
|
||||
try:
|
||||
chunk.to_csv(self.file_path_save_orgBcg, mode='a', index=False, header=False)
|
||||
except Exception:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_RES_ORGBCG_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
return True, Constants.PRECISELY_ALIGN_SAVING_RES_ORGBCG_FINISHED
|
||||
|
||||
def save_res_BCG(self, chunk):
|
||||
|
||||
if self.res_BCG is None:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_RES_BCG_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["res_BCG_Not_Exist"]
|
||||
|
||||
try:
|
||||
chunk.to_csv(self.file_path_save_BCG, mode='a', index=False, header=False)
|
||||
except Exception:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_RES_BCG_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
return True, Constants.PRECISELY_ALIGN_SAVING_RES_BCG_FINISHED
|
||||
|
||||
def save_cut_ECG(self, chunk):
|
||||
|
||||
if self.cut_ECG is None:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_CUT_ECG_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["cut_ECG_Not_Exist"]
|
||||
|
||||
try:
|
||||
chunk.to_csv(self.file_path_save_ECG, mode='a', index=False, header=False)
|
||||
except Exception:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_CUT_ECG_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
return True, Constants.PRECISELY_ALIGN_SAVING_CUT_ECG_FINISHED
|
||||
|
||||
def save_Jpeak(self, chunk):
|
||||
|
||||
if self.cut_Jpeak is None:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_CUT_JPEAK_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["cut_Jpeak_Not_Exist"]
|
||||
|
||||
try:
|
||||
chunk.to_csv(self.file_path_save_Jpeak, mode='a', index=False, header=False)
|
||||
except Exception:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_CUT_JPEAK_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
return True, Constants.PRECISELY_ALIGN_SAVING_CUT_JPEAK_FINISHED
|
||||
|
||||
def save_Rpeak(self, chunk):
|
||||
|
||||
if self.cut_Rpeak is None:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_CUT_RPEAK_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["cut_Rpeak_Not_Exist"]
|
||||
|
||||
try:
|
||||
chunk.to_csv(self.file_path_save_Rpeak, mode='a', index=False, header=False)
|
||||
except Exception:
|
||||
return False, Constants.PRECISELY_ALIGN_SAVING_CUT_RPEAK_FAILURE + Constants.PRECISELY_ALIGN_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
return True, Constants.PRECISELY_ALIGN_SAVING_CUT_RPEAK_FINISHED
|
||||
|
||||
|
||||
class CustomNavigationToolbar(NavigationToolbar2QT):
|
||||
|
||||
@ -370,7 +370,7 @@ class MainWindow_preprocess(QMainWindow):
|
||||
|
||||
# 保存
|
||||
PublicFunc.progressbar_update(self, 1, 1, Constants.SAVING_DATA, 0)
|
||||
# status, info = self.data.save()
|
||||
|
||||
total_rows = len(DataFrame(self.data.processed_data.reshape(-1)))
|
||||
chunk_size = ConfigParams.PREPROCESS_SAVE_CHUNK_SIZE
|
||||
with open(Config["Path"]["Save"], 'w') as f:
|
||||
@ -459,10 +459,6 @@ class Data:
|
||||
return False, Constants.SAVING_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Processed_Data_Not_Exist"]
|
||||
|
||||
try:
|
||||
# DataFrame(self.processed_data.reshape(-1)).to_csv(self.file_path_save,
|
||||
# index=False,
|
||||
# header=False,
|
||||
# float_format='%.4f')
|
||||
chunk.to_csv(self.file_path_save, mode='a', index=False, header=False, float_format='%.4f')
|
||||
except Exception:
|
||||
return False, Constants.SAVING_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Save_Exception"]
|
||||
|
||||
@ -51,8 +51,8 @@ class ConfigParams:
|
||||
}
|
||||
PREPROCESS_INPUT_BCG_FILENAME: str = "orgBcg_Raw_"
|
||||
PREPROCESS_INPUT_ECG_FILENAME: str = "ECG I_Raw_"
|
||||
PREPROCESS_SAVE_BCG_FILENAME: str = "DSbcg_sig_"
|
||||
PREPROCESS_SAVE_ECG_FILENAME: str = "ECG_filter_"
|
||||
PREPROCESS_SAVE_BCG_FILENAME: str = "BCG_Raw_"
|
||||
PREPROCESS_SAVE_ECG_FILENAME: str = "ECG_Raw_"
|
||||
PREPROCESS_SAVE_CHUNK_SIZE: int = 1000000
|
||||
|
||||
# BCG的J峰算法定位
|
||||
@ -73,7 +73,7 @@ class ConfigParams:
|
||||
"UseCPU": False,
|
||||
"DetectMethod": ""
|
||||
}
|
||||
DETECT_JPEAK_INPUT_BCG_FILENAME: str = "DSbcg_sig_"
|
||||
DETECT_JPEAK_INPUT_BCG_FILENAME: str = "BCG_Raw_"
|
||||
DETECT_JPEAK_SAVE_FILENAME: str = "JPeak_revise"
|
||||
DETECT_JPEAK_SAVE_CHUNK_SIZE: int = 100
|
||||
|
||||
@ -90,8 +90,8 @@ class ConfigParams:
|
||||
"PeaksValue": 200,
|
||||
"DetectMethod": ""
|
||||
}
|
||||
DETECT_RPEAK_INPUT_ECG_FILENAME: str = "ECG_filter_"
|
||||
DETECT_RPEAK_SAVE_FILENAME: str = "final_Rpeak"
|
||||
DETECT_RPEAK_INPUT_ECG_FILENAME: str = "ECG_Raw_"
|
||||
DETECT_RPEAK_SAVE_FILENAME: str = "Rpeak_final"
|
||||
DETECT_RPEAK_SAVE_CHUNK_SIZE: int = 100
|
||||
|
||||
# 人工纠正
|
||||
@ -118,12 +118,12 @@ class ConfigParams:
|
||||
"MoveSpeed": 1000
|
||||
}
|
||||
}
|
||||
LABEL_CHECK_INPUT_BCG_FILENAME: str = "DSbcg_sig_"
|
||||
LABEL_CHECK_INPUT_BCG_FILENAME: str = "BCG_Raw_"
|
||||
LABEL_CHECK_INPUT_JPEAK_FILENAME: str = "JPeak_revise"
|
||||
LABEL_CHECK_SAVE_JPEAK_FILENAME: str = "JPeak_revise_corrected"
|
||||
LABEL_CHECK_INPUT_ECG_FILENAME: str = "ECG_filter_"
|
||||
LABEL_CHECK_INPUT_RPEAK_FILENAME: str = "final_Rpeak"
|
||||
LABEL_CHECK_SAVE_RPEAK_FILENAME: str = "final_Rpeak_corrected"
|
||||
LABEL_CHECK_INPUT_ECG_FILENAME: str = "ECG_Raw_"
|
||||
LABEL_CHECK_INPUT_RPEAK_FILENAME: str = "Rpeak_final"
|
||||
LABEL_CHECK_SAVE_RPEAK_FILENAME: str = "Rpeak_final_corrected"
|
||||
LABEL_CHECK_SAVE_CHUNK_SIZE: int = 100
|
||||
LABEL_CHECK_LABEL_TRANSPARENCY: float = 0.2
|
||||
LABEL_CHECK_ACTION_LABEL_MULTIPLE_SHORTCUT_KEY: str = "Z"
|
||||
@ -135,13 +135,21 @@ class ConfigParams:
|
||||
"ECGFreq": 1000
|
||||
}
|
||||
}
|
||||
PRECISELY_ALIGN_INPUT_BCG_FILENAME: str = "DSbcg_sig_"
|
||||
PRECISELY_ALIGN_INPUT_ORGBCG_FILENAME: str = "orgBcg_Raw_"
|
||||
PRECISELY_ALIGN_INPUT_BCG_FILENAME: str = "BCG_Raw_"
|
||||
PRECISELY_ALIGN_INPUT_JPEAK_FILENAME: str = "JPeak_revise_corrected"
|
||||
PRECISELY_ALIGN_SAVE_BCG_FILENAME: str = "Align_info"
|
||||
PRECISELY_ALIGN_INPUT_ECG_FILENAME: str = "ECG_filter_"
|
||||
PRECISELY_ALIGN_INPUT_RPEAK_FILENAME: str = "final_Rpeak_corrected"
|
||||
PRECISELY_ALIGN_SAVE_ECG_FILENAME: str = "Align_info"
|
||||
PRECISELY_ALIGN_SAVE_BCG_ALIGNINFO_FILENAME: str = "Align_info"
|
||||
PRECISELY_ALIGN_INPUT_ECG_FILENAME: str = "ECG_Raw_"
|
||||
PRECISELY_ALIGN_INPUT_RPEAK_FILENAME: str = "Rpeak_final_corrected"
|
||||
PRECISELY_ALIGN_SAVE_ECG_ALIGNINFO_FILENAME: str = "Align_info"
|
||||
PRECISELY_ALIGN_SAVE_ORGBCG_FILENAME: str = "orgBcg_Sync_"
|
||||
PRECISELY_ALIGN_SAVE_BCG_FILENAME: str = "BCG_Sync_"
|
||||
PRECISELY_ALIGN_SAVE_ECG_FILENAME: str = "ECG_Sync_"
|
||||
PRECISELY_ALIGN_SAVE_JPEAK_FILENAME: str = "JPeak_Sync"
|
||||
PRECISELY_ALIGN_SAVE_RPEAK_FILENAME: str = "Rpeak_Sync"
|
||||
PRECISELY_ALIGN_ACTION_GET_RANGE_SHORTCUT_KEY: str = "Z"
|
||||
PRECISELY_ALIGN_SAVE_CHUNK_SIZE: int = 1000000
|
||||
PRECISELY_ALIGN_SAVE_PEAK_CHUNK_SIZE: int = 100
|
||||
|
||||
# 体动标注
|
||||
|
||||
|
||||
@ -216,6 +216,30 @@ class Constants:
|
||||
PRECISELY_ALIGN_POSTPROCESS_VIEW_FINISHED: str = "数据后处理完成"
|
||||
PRECISELY_ALIGN_POSTPROCESS_VIEW_FAILURE: str = "数据后处理失败"
|
||||
|
||||
PRECISELY_ALIGN_SAVING_ALIGNINFO: str = "正在保存对齐信息"
|
||||
PRECISELY_ALIGN_SAVING_ALIGNINFO_FINISHED: str = "保存对齐信息完成"
|
||||
PRECISELY_ALIGN_SAVING_ALIGNINFO_FAILURE: str = "保存对齐信息失败"
|
||||
|
||||
PRECISELY_ALIGN_SAVING_RES_ORGBCG: str = "正在保存切割后orgBcg"
|
||||
PRECISELY_ALIGN_SAVING_RES_ORGBCG_FINISHED: str = "保存切割后orgBcg完成"
|
||||
PRECISELY_ALIGN_SAVING_RES_ORGBCG_FAILURE: str = "保存切割后orgBcg失败"
|
||||
|
||||
PRECISELY_ALIGN_SAVING_RES_BCG: str = "正在保存切割后BCG"
|
||||
PRECISELY_ALIGN_SAVING_RES_BCG_FINISHED: str = "保存切割后BCG完成"
|
||||
PRECISELY_ALIGN_SAVING_RES_BCG_FAILURE: str = "保存切割后BCG失败"
|
||||
|
||||
PRECISELY_ALIGN_SAVING_CUT_ECG: str = "正在保存切割后ECG"
|
||||
PRECISELY_ALIGN_SAVING_CUT_ECG_FINISHED: str = "保存切割后ECG完成"
|
||||
PRECISELY_ALIGN_SAVING_CUT_ECG_FAILURE: str = "保存切割后ECG失败"
|
||||
|
||||
PRECISELY_ALIGN_SAVING_CUT_JPEAK: str = "正在保存切割后J峰"
|
||||
PRECISELY_ALIGN_SAVING_CUT_JPEAK_FINISHED: str = "保存切割后J峰完成"
|
||||
PRECISELY_ALIGN_SAVING_CUT_JPEAK_FAILURE: str = "保存切割后J峰失败"
|
||||
|
||||
PRECISELY_ALIGN_SAVING_CUT_RPEAK: str = "正在保存切割后R峰"
|
||||
PRECISELY_ALIGN_SAVING_CUT_RPEAK_FINISHED: str = "保存切割后R峰完成"
|
||||
PRECISELY_ALIGN_SAVING_CUT_RPEAK_FAILURE: str = "保存切割后R峰失败"
|
||||
|
||||
PRECISELY_ALIGN_FAILURE_REASON = {
|
||||
"Data_Path_Not_Exist": "（数据路径不存在）",
|
||||
"Read_Data_Exception": "（读取数据异常）",
|
||||
@ -226,6 +250,11 @@ class Constants:
|
||||
"Calculate_Correlation_Exception": "（计算相关性异常）",
|
||||
"Correlation_Align_Exception": "（处理相关对齐异常）",
|
||||
"PostProcess_Align_Exception": "（数据后处理异常）",
|
||||
"res_orgBcg_Not_Exist": "（切割后orgBcg不存在）",
|
||||
"res_BCG_Not_Exist": "（切割后BCG不存在）",
|
||||
"cut_ECG_Not_Exist": "（切割后ECG不存在）",
|
||||
"cut_Jpeak_Not_Exist": "（切割后J峰不存在）",
|
||||
"cut_Rpeak_Not_Exist": "（切割后R峰不存在）",
|
||||
"Save_Exception": "（保存异常）"
|
||||
}
|
||||
|
||||
|
||||
@ -1,465 +0,0 @@
|
||||
import os
|
||||
import shutil
|
||||
import pandas as pd
|
||||
|
||||
from resampy import resample
|
||||
from scipy import signal
|
||||
|
||||
|
||||
import numpy as np
|
||||
import matplotlib.pyplot as plt
|
||||
|
||||
def read_data(root):
|
||||
orgBCGpath = os.path.join(root, 'raw_org.txt')
|
||||
BCGpath = os.path.join(root, 'DSbcg_sig_1000hz3.txt')
|
||||
ECGpath = os.path.join(root, 'filter_ecg.txt')
|
||||
Jpeakspath = os.path.join(root, 'Jpeak.txt')
|
||||
Rpeakspath = os.path.join(root, 'final_Rpeak.txt')
|
||||
|
||||
orgBCG = np.array(pd.read_csv(orgBCGpath, header=None)).reshape(-1)
|
||||
# B, A = signal.butter(4, np.array([2,8])*2/1000, 'bandpass')
|
||||
# BCG = signal.filtfilt(B, A, orgBCG)
|
||||
BCG = np.array(pd.read_csv(BCGpath)).reshape(-1)
|
||||
ECG = np.array(pd.read_csv(ECGpath, header=None)).reshape(-1)
|
||||
|
||||
|
||||
Jpeaks = np.loadtxt(Jpeakspath).astype(int)
|
||||
Rpeaks = np.loadtxt(Rpeakspath).astype(int)
|
||||
|
||||
return orgBCG, BCG, ECG, Jpeaks, Rpeaks
|
||||
|
||||
|
||||
|
||||
def Align_two_way(Jpeaks, Rpeaks, fs=1000):
|
||||
def get_info(Jpeaks, Rpeaks):
|
||||
"""
|
||||
|
||||
:param Jpeaks:
|
||||
:param Rpeaks:
|
||||
:return:
|
||||
"""
|
||||
RRIs = np.diff(Rpeaks)
|
||||
JJIs = np.diff(Jpeaks)
|
||||
RRIVs = np.diff(RRIs)
|
||||
JJIVs = np.diff(JJIs)
|
||||
|
||||
N_JJIV = get_area(Jpeaks, JJIVs, Rpeaks, RRIVs, title='JJIV')
|
||||
N_RRIV = get_area(Rpeaks, RRIVs, Jpeaks, JJIVs, N_JJIV=N_JJIV[0], title='RRIV')
|
||||
JJIs_cut = JJIs[N_JJIV[1, 0]: N_JJIV[1, 1]]
|
||||
RRIs_cut = RRIs[N_RRIV[1, 0]: N_RRIV[1, 1]]
|
||||
Rpeaks_cut = Rpeaks[N_RRIV[1, 0]:N_RRIV[1, 1] + 2]
|
||||
Jpeaks_cut = Jpeaks[N_JJIV[1, 0]:N_JJIV[1, 1] + 2]
|
||||
RRI = np.diff(Rpeaks_cut)
|
||||
# shift = get_IIV_shift(JJIs_cut, RRIs_cut)
|
||||
shift, correlation_IIV, correlation_II, total_time_ratio = get_IIV_shift(JJIs_cut, RRIs_cut)
|
||||
offset_interval = np.sum(RRI[:shift]) - (Jpeaks_cut[0] - Rpeaks_cut[0]) # BCG 相对于ECG的偏移量
|
||||
anchor_R = Rpeaks[N_RRIV[1, 0] + shift]
|
||||
anchor_J = Jpeaks[N_JJIV[1, 0]]
|
||||
|
||||
# info = {'N_JJIV':N_JJIV, 'N_RRIV':N_RRIV, 'shift':shift, 'offset_interval':offset_interval, 'anchors':[anchor_R, anchor_J]}
|
||||
info = {'N_JJIV': N_JJIV, 'N_RRIV': N_RRIV, 'correlation_IIV': correlation_IIV,
|
||||
'correlation_II': correlation_II, 'shift': shift, 'total_time_ratio': total_time_ratio,
|
||||
'offset_interval': offset_interval, 'anchors': [anchor_R, anchor_J]}
|
||||
return info
|
||||
|
||||
info1 = get_info(Jpeaks, Rpeaks)
|
||||
offset_interval1 = info1['offset_interval']
|
||||
anchor1 = info1['anchors']
|
||||
offset_interval1_T = offset_interval1 / fs # 前端偏移时间量，大于0
|
||||
print("前端对齐的偏移量为：{}s".format(offset_interval1_T))
|
||||
|
||||
# while True:
|
||||
info2 = get_info(Jpeaks, Rpeaks)
|
||||
anchor2 = info2['anchors']
|
||||
offset_interval2 = info2['offset_interval']
|
||||
offset_interval2_T = offset_interval2 / fs # 后端偏移时间量，大于0
|
||||
print("后端对齐的偏移量为：{}s".format(offset_interval2_T))
|
||||
|
||||
orgfs = (int(anchor2[1]) - int(anchor1[1])) * fs / (int(anchor2[0]) - int(anchor1[0]))
|
||||
|
||||
offset_anchor = anchor1[0] - anchor1[1]
|
||||
info = {'1': info1, '2': info2, 'offset_anchor': offset_anchor, 'orgfs': orgfs}
|
||||
|
||||
return info
|
||||
|
||||
|
||||
################################################ Align utils #############################################################
|
||||
def get_area(peaks1_, data1, peaks2_, data2, fs=1000, N_JJIV=None, title='JJIV'):
|
||||
global N, line0, line1
|
||||
|
||||
peaks1 = peaks1_[:-2] / fs
|
||||
peaks2 = peaks2_[:-2] / fs
|
||||
ylim = [-500, 500]
|
||||
|
||||
'''手动选择区域'''
|
||||
N = np.array([]).astype(int)
|
||||
fig, ax = plt.subplots(nrows=2, ncols=1, sharex=True, sharey=True)
|
||||
ax0, ax1 = ax[0], ax[1]
|
||||
ax0.set_title(title)
|
||||
ax0.stem(peaks1, data1, markerfmt='C0.')
|
||||
line0, = ax0.plot(peaks1, data1, 'o', c='b', markersize=2, picker=True, pickradius=5)
|
||||
line1, = ax0.plot(peaks1[N], data1[N] + 1, 'v', c='r', picker=True, pickradius=5)
|
||||
ax0.set_ylim(ylim[0], ylim[1])
|
||||
ax0.grid()
|
||||
ax1.stem(peaks2, data2, markerfmt='C0.')
|
||||
if N_JJIV is not None:
|
||||
N_JJIV2 = np.copy(N_JJIV)
|
||||
N_JJIV2 = N_JJIV2 / fs
|
||||
ax1.plot([N_JJIV2[0], N_JJIV2[0]], [ylim[0], ylim[1]], 'k--')
|
||||
ax1.plot([N_JJIV2[1], N_JJIV2[1]], [ylim[0], ylim[1]], 'k--')
|
||||
ax1.set_ylim(ylim[0], ylim[1])
|
||||
|
||||
def onpick(event):
|
||||
global line0, line1
|
||||
global N
|
||||
thisline = event.artist
|
||||
xdata = thisline.get_xdata()
|
||||
ind = event.ind[-1]
|
||||
MM = ind
|
||||
if thisline == line0:
|
||||
if len(N) < 2:
|
||||
N = np.append(N, MM)
|
||||
N.sort()
|
||||
elif thisline == line1:
|
||||
N = np.delete(N, int(ind))
|
||||
|
||||
xlime = ax0.get_xlim()
|
||||
ylime = ax0.get_ylim()
|
||||
ax0.clear()
|
||||
ax0.set_title(title)
|
||||
ax0.stem(peaks1, data1, markerfmt='C0.')
|
||||
line0, = ax0.plot(peaks1, data1, 'o', c='b', markersize=2, picker=True, pickradius=5)
|
||||
line1, = ax0.plot(peaks1[N], data1[N] + 1, 'v', c='r', picker=True, pickradius=5)
|
||||
ax0.set_xlim(xlime)
|
||||
ax0.set_ylim(ylime)
|
||||
ax0.grid()
|
||||
plt.draw()
|
||||
|
||||
fig.canvas.mpl_connect('pick_event', onpick)
|
||||
plt.show()
|
||||
|
||||
if len(N) == 0:
|
||||
N = np.array([0, len(data1) - 1]).astype(int)
|
||||
|
||||
N_peak = peaks1_[N]
|
||||
N_out = np.asarray([N_peak[:2], N[:2]]).astype(int)
|
||||
|
||||
return N_out
|
||||
|
||||
|
||||
def get_IIV_shift(JJIs, RRIs):
|
||||
global shift, line0
|
||||
|
||||
RRIVs = np.diff(RRIs)
|
||||
JJIVs = np.diff(JJIs)
|
||||
corre_ = np.correlate(RRIVs, JJIVs, 'valid')
|
||||
|
||||
'''手动选择偏移'''
|
||||
corre = np.copy(corre_)
|
||||
shift = np.argmax(corre)
|
||||
RRIVs_cut = RRIVs[shift:shift + len(JJIVs)]
|
||||
RRIs_cut = RRIs[shift:shift + len(JJIs)]
|
||||
|
||||
correlation = np.corrcoef(RRIVs_cut, JJIVs)
|
||||
correlation_IIV = (correlation[0, 1] + correlation[1, 0]) / 2
|
||||
|
||||
correlation = np.corrcoef(RRIs_cut, JJIs)
|
||||
correlation_II = (correlation[0, 1] + correlation[1, 0]) / 2
|
||||
|
||||
tmp = RRIVs_cut * JJIVs
|
||||
same_sign_rate = np.sum(tmp > 0) / len(JJIVs)
|
||||
total_time_ratio = np.sum(JJIs) / np.sum(RRIs[shift:shift + len(JJIs)])
|
||||
fig, ax = plt.subplots(2, 1)
|
||||
ax0, ax1 = ax[0], ax[1]
|
||||
ax0.set_title(
|
||||
"corre_IIV: {}, corre_II: {}\n same_sign_rate:{}, total_time_ratio: {}\n shift: {}".format(correlation_IIV,
|
||||
correlation_II,
|
||||
same_sign_rate,
|
||||
total_time_ratio,
|
||||
shift))
|
||||
ax0.stem(corre, markerfmt='C0.', label='corre(RRIV, JJIV)')
|
||||
line0, = ax0.plot(corre, 'o', c='b', markersize=2, picker=True, pickradius=5)
|
||||
ax0.plot(shift, corre[shift] + 1, 'v', c='r', picker=True, pickradius=5)
|
||||
ax0.legend()
|
||||
ax0.grid()
|
||||
ax1.stem(RRIVs, markerfmt='b.', label='RRIV')
|
||||
ax1.stem(np.arange(shift, shift + len(JJIVs)), JJIVs, markerfmt='ko', label='JJIV')
|
||||
ax1.legend()
|
||||
|
||||
def onpick(event):
|
||||
global line0
|
||||
global shift
|
||||
thisline = event.artist
|
||||
xdata = thisline.get_xdata()
|
||||
ind = event.ind[-1]
|
||||
shift = int(xdata[ind])
|
||||
RRIVs_cut = RRIVs[shift:shift + len(JJIVs)]
|
||||
RRIs_cut = RRIs[shift:shift + len(JJIs)]
|
||||
|
||||
correlation = np.corrcoef(RRIVs_cut, JJIVs)
|
||||
correlation_IIV = (correlation[0, 1] + correlation[1, 0]) / 2
|
||||
|
||||
correlation = np.corrcoef(RRIs_cut, JJIs)
|
||||
correlation_II = (correlation[0, 1] + correlation[1, 0]) / 2
|
||||
|
||||
tmp = RRIVs_cut * JJIVs
|
||||
same_sign_rate = np.sum(tmp > 0) / len(JJIVs)
|
||||
total_time_ratio = np.sum(JJIs) / np.sum(RRIs[shift:shift + len(JJIs)])
|
||||
|
||||
xlime = ax0.get_xlim()
|
||||
ylime = ax0.get_ylim()
|
||||
ax0.clear()
|
||||
ax0.set_title(
|
||||
"corre_IIV: {}, corre_II: {}\n same_sign_rate:{}, total_time_ratio: {}\n shift: {}".format(correlation_IIV,
|
||||
correlation_II,
|
||||
same_sign_rate,
|
||||
total_time_ratio,
|
||||
shift))
|
||||
ax0.stem(corre, markerfmt='C0.', label='corre(RRIV, JJIV)')
|
||||
line0, = ax0.plot(corre, 'o', c='b', markersize=2, picker=True, pickradius=5)
|
||||
ax0.plot(shift, corre[shift] + 1, 'v', c='r', picker=True, pickradius=5)
|
||||
ax0.legend()
|
||||
ax0.grid()
|
||||
ax0.set_xlim(xlime)
|
||||
ax0.set_ylim(ylime)
|
||||
xlime = ax1.get_xlim()
|
||||
ylime = ax1.get_ylim()
|
||||
ax1.clear()
|
||||
ax1.stem(RRIVs, markerfmt='b.', label='RRIV')
|
||||
ax1.stem(np.arange(shift, shift + len(JJIVs)), JJIVs, markerfmt='ko', label='JJIV')
|
||||
ax1.legend()
|
||||
ax1.set_xlim(xlime)
|
||||
ax1.set_ylim(ylime)
|
||||
plt.draw()
|
||||
|
||||
fig.canvas.mpl_connect('pick_event', onpick)
|
||||
plt.show()
|
||||
|
||||
# return shift
|
||||
return shift, correlation_IIV, correlation_II, total_time_ratio
|
||||
|
||||
|
||||
def main():
|
||||
def offset_correct(BCG, ECG, anchors):
|
||||
global line0, line1, N, M
|
||||
a = np.max([np.max(ECG), np.max(BCG)])
|
||||
b = np.min([np.min(ECG), np.min(BCG)])
|
||||
peaks_ECG, _ = signal.find_peaks(ECG)
|
||||
peaks_BCG, _ = signal.find_peaks(BCG)
|
||||
N, M = np.array([]).astype(int), np.array([]).astype(int)
|
||||
fig, ax = plt.subplots()
|
||||
ax.set_title("offset correct")
|
||||
ax.plot(ECG)
|
||||
ax.plot(BCG)
|
||||
ax.plot([anchors[0], anchors[0]], [b, a], 'k--')
|
||||
ax.plot([anchors[1], anchors[1]], [b, a], 'k--')
|
||||
ax.plot(N, ECG[N] + 1, 'rv')
|
||||
ax.plot(M, BCG[M] + 1, 'yv')
|
||||
line0, = ax.plot(peaks_ECG, ECG[peaks_ECG], 'o', markersize=2, c='g', picker=True, pickradius=5)
|
||||
line1, = ax.plot(peaks_BCG, BCG[peaks_BCG], 'o', markersize=2, c='g', picker=True, pickradius=5)
|
||||
ax.grid()
|
||||
|
||||
def onpick(event):
|
||||
global line0, line1, N, M
|
||||
thisline = event.artist
|
||||
xdata = thisline.get_xdata()
|
||||
ind = event.ind[-1]
|
||||
nm = int(xdata[ind])
|
||||
if thisline == line0:
|
||||
if nm in N:
|
||||
idx = np.where(N == nm)[0]
|
||||
N = np.delete(N, idx)
|
||||
elif len(N) < 2:
|
||||
N = np.append(N, nm)
|
||||
elif thisline == line1:
|
||||
if nm in M:
|
||||
idx = np.where(M == nm)[0]
|
||||
M = np.delete(M, idx)
|
||||
elif len(M) < 2:
|
||||
M = np.append(M, nm)
|
||||
|
||||
xlime = ax.get_xlim()
|
||||
ylime = ax.get_ylim()
|
||||
ax.clear()
|
||||
ax.set_title("offset correct")
|
||||
ax.plot(ECG)
|
||||
ax.plot(BCG)
|
||||
ax.plot([anchors[0], anchors[0]], [b, a], 'k--')
|
||||
ax.plot([anchors[1], anchors[1]], [b, a], 'k--')
|
||||
ax.plot(N, ECG[N] + 1, 'rv')
|
||||
ax.plot(M, BCG[M] + 1, 'yv')
|
||||
line0, = ax.plot(peaks_ECG, ECG[peaks_ECG], 'o', markersize=2, c='g', picker=True, pickradius=5)
|
||||
line1, = ax.plot(peaks_BCG, BCG[peaks_BCG], 'o', markersize=2, c='g', picker=True, pickradius=5)
|
||||
ax.grid()
|
||||
ax.set_xlim(xlime)
|
||||
ax.set_ylim(ylime)
|
||||
plt.draw()
|
||||
|
||||
fig.canvas.mpl_connect('pick_event', onpick)
|
||||
plt.show()
|
||||
|
||||
if len(N) == 0 or len(M) == 0:
|
||||
off = 0
|
||||
else:
|
||||
M.sort()
|
||||
N.sort()
|
||||
off = round(((int(N[1]) - int(M[1])) + (int(N[0]) - int(M[0]))) / 2)
|
||||
|
||||
return off
|
||||
|
||||
'''输入信息（如数据路径、采样频率）'''
|
||||
fs = 1000
|
||||
root = r"E:\data_annotation\pratice_data\282" # E:\BCG_data\703
|
||||
|
||||
saveroot = os.path.join(root, 'Align')
|
||||
if not os.path.exists(saveroot):
|
||||
os.mkdir(saveroot)
|
||||
else:
|
||||
shutil.rmtree(saveroot)
|
||||
os.mkdir(saveroot)
|
||||
|
||||
'''获取数据（BCG、ECG、J_peak、R_peak）'''
|
||||
raw_org_, DSbcg_sig_1000hz3_, filter_ecg_, Jpeak_, final_Rpeak_ = read_data(root)
|
||||
|
||||
orgBCG = raw_org_
|
||||
BCG = DSbcg_sig_1000hz3_
|
||||
ECG = filter_ecg_
|
||||
Rpeaks = final_Rpeak_
|
||||
Jpeaks = Jpeak_
|
||||
|
||||
orgbcg_len = len(BCG)
|
||||
argmax_bcg = np.argmax(BCG)
|
||||
orgecg_len = len(ECG)
|
||||
argmax_ecg = np.argmax(ECG)
|
||||
|
||||
# plt.figure()
|
||||
# plt.subplot(2,1,1)
|
||||
# plt.plot(BCG[:3600000])
|
||||
# plt.subplot(2,1,2)
|
||||
# plt.plot(ECG)
|
||||
|
||||
info = Align_two_way(Jpeaks, Rpeaks, fs=fs)
|
||||
|
||||
print(info)
|
||||
|
||||
anchor0 = info['1']['anchors']
|
||||
anchor1 = info['2']['anchors']
|
||||
|
||||
off = info['offset_anchor']
|
||||
orgfs = info['orgfs']
|
||||
|
||||
if off > 0:
|
||||
ECG = ECG[off:]
|
||||
anchor0[0] = anchor0[0] - off
|
||||
anchor1[0] = anchor1[0] - off
|
||||
idxs = np.where(Rpeaks > off)[0]
|
||||
Rpeaks = Rpeaks[idxs] - off
|
||||
else:
|
||||
BCG = BCG[-off:]
|
||||
orgBCG = orgBCG[-off:]
|
||||
anchor0[1] = anchor0[1] + off
|
||||
anchor1[1] = anchor1[1] + off
|
||||
|
||||
BCG_res = resample(BCG, orgfs, fs)
|
||||
orgBCG_res = resample(orgBCG, orgfs, fs)
|
||||
|
||||
# ————————————————————————————————————————————————————————————————
|
||||
anchor0[1] = round(int(anchor0[1]) * fs / orgfs)
|
||||
anchor1[1] = round(int(anchor1[1]) * fs / orgfs)
|
||||
off = anchor1[0] - anchor1[1]
|
||||
if off > 0:
|
||||
ECG = ECG[off:]
|
||||
anchor0[0] = anchor0[0] - off
|
||||
anchor1[0] = anchor1[0] - off
|
||||
idxs = np.where(Rpeaks > off)[0]
|
||||
Rpeaks = Rpeaks[idxs] - off
|
||||
else:
|
||||
BCG_res = BCG_res[-off:]
|
||||
orgBCG_res = orgBCG_res[-off:]
|
||||
anchor0[1] = anchor0[1] + off
|
||||
anchor1[1] = anchor1[1] + off
|
||||
|
||||
datalen = np.min([len(ECG), len(BCG_res)])
|
||||
ECG = ECG[:datalen]
|
||||
BCG_res = BCG_res[:datalen]
|
||||
orgBCG_res = orgBCG_res[:datalen]
|
||||
|
||||
idxs = np.where(Rpeaks < datalen)[0]
|
||||
Rpeaks = Rpeaks[idxs]
|
||||
|
||||
# ——————————————————————————————————————————————————————————————————
|
||||
off = offset_correct(BCG_res, ECG, [anchor0[0], anchor1[0]])
|
||||
if off > 0:
|
||||
ECG = ECG[off:]
|
||||
anchor0[0] = anchor0[0] - off
|
||||
anchor1[0] = anchor1[0] - off
|
||||
idxs = np.where(Rpeaks > off)[0]
|
||||
Rpeaks = Rpeaks[idxs] - off
|
||||
else:
|
||||
BCG_res = BCG_res[-off:]
|
||||
orgBCG_res = orgBCG_res[-off:]
|
||||
anchor0[1] = anchor0[1] + off
|
||||
anchor1[1] = anchor1[1] + off
|
||||
|
||||
info['offset_correct'] = off
|
||||
|
||||
datalen = np.min([len(ECG), len(BCG_res)])
|
||||
ECG = ECG[:datalen]
|
||||
BCG_res = BCG_res[:datalen]
|
||||
orgBCG_res = orgBCG_res[:datalen]
|
||||
|
||||
idxs = np.where(Rpeaks < datalen)[0]
|
||||
Rpeaks = Rpeaks[idxs]
|
||||
|
||||
Jpeaks = []
|
||||
peaks, _ = signal.find_peaks(BCG_res)
|
||||
for i in Rpeaks:
|
||||
tmp = np.abs(peaks - i)
|
||||
idx = np.argmin(tmp)
|
||||
Jpeaks.append(peaks[idx])
|
||||
Jpeaks = np.asarray(Jpeaks).astype(int)
|
||||
|
||||
# ----------------2024.4.17 CYS 新增计算BCG与ECG前后两端切割的横坐标----------------
|
||||
orgfs = info['orgfs'] # 原始数据采样频率
|
||||
frontcut_index_bcg = int((argmax_bcg - np.argmax(BCG_res) / 1000 * orgfs))
|
||||
backcut_index_bcg = int((len(BCG_res) / 1000 * orgfs + argmax_bcg - np.argmax(BCG_res) / 1000 * orgfs))
|
||||
|
||||
frontcut_index_ecg = argmax_ecg - np.argmax(ECG)
|
||||
backcut_index_ecg = len(ECG) + argmax_ecg - np.argmax(ECG)
|
||||
|
||||
print("1. BCG前段被切割的坐标值为：", frontcut_index_bcg)
|
||||
print("1. BCG后段被切割的坐标值为：", backcut_index_bcg)
|
||||
print("2. ECG前段被切割的坐标值为：", frontcut_index_ecg)
|
||||
print("2. ECG后段被切割的坐标值为：", backcut_index_ecg)
|
||||
info = {'0': info, 'frontcut_index_bcg': frontcut_index_bcg, 'backcut_index_bcg': backcut_index_bcg,
|
||||
'frontcut_index_ecg': frontcut_index_ecg, 'backcut_index_ecg': backcut_index_ecg}
|
||||
# ------------------------------------------------------------------------------
|
||||
|
||||
plt.figure()
|
||||
plt.plot(ECG, label='ECG')
|
||||
plt.plot(BCG_res, label='BCG')
|
||||
plt.plot(Rpeaks, ECG[Rpeaks], 'ko', label='Rpeaks')
|
||||
plt.plot(Jpeaks, BCG_res[Jpeaks], 'rv', label='Jpeaks')
|
||||
# plt.legend()
|
||||
plt.show()
|
||||
|
||||
savepath = os.path.join(saveroot, 'Align_info.npy')
|
||||
np.save(savepath, info)
|
||||
savepath = os.path.join(saveroot, 'Align_info.txt')
|
||||
with open(savepath, 'w') as file:
|
||||
file.write(str(info))
|
||||
|
||||
savepath = os.path.join(saveroot, 'orgData_sync.txt')
|
||||
pd.DataFrame(np.round(orgBCG_res, 4)).to_csv(savepath, index=None, header=None)
|
||||
savepath = os.path.join(saveroot, 'BCG_sync.txt')
|
||||
pd.DataFrame(np.round(BCG_res, 4)).to_csv(savepath, index=None, header=None)
|
||||
savepath = os.path.join(saveroot, 'ECG_sync.txt')
|
||||
pd.DataFrame(ECG).to_csv(savepath, index=None, header=None)
|
||||
savepath = os.path.join(saveroot, 'Jpeaks_sync.txt')
|
||||
pd.DataFrame(Jpeaks).to_csv(savepath, index=None, header=None)
|
||||
savepath = os.path.join(saveroot, 'Rpeaks_sync.txt')
|
||||
pd.DataFrame(Rpeaks).to_csv(savepath, index=None, header=None)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@ -24,7 +24,7 @@ class Ui_MainWindow_precisely_align_input_setting(object):
|
||||
def setupUi(self, MainWindow_precisely_align_input_setting):
|
||||
if not MainWindow_precisely_align_input_setting.objectName():
|
||||
MainWindow_precisely_align_input_setting.setObjectName(u"MainWindow_precisely_align_input_setting")
|
||||
MainWindow_precisely_align_input_setting.resize(540, 760)
|
||||
MainWindow_precisely_align_input_setting.resize(840, 1010)
|
||||
self.centralwidget = QWidget(MainWindow_precisely_align_input_setting)
|
||||
self.centralwidget.setObjectName(u"centralwidget")
|
||||
self.gridLayout = QGridLayout(self.centralwidget)
|
||||
@ -42,33 +42,76 @@ class Ui_MainWindow_precisely_align_input_setting(object):
|
||||
font1 = QFont()
|
||||
font1.setPointSize(10)
|
||||
self.groupBox.setFont(font1)
|
||||
self.verticalLayout_2 = QVBoxLayout(self.groupBox)
|
||||
self.verticalLayout_2.setObjectName(u"verticalLayout_2")
|
||||
self.groupBox_file_path_input_BCG = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_input_BCG.setObjectName(u"groupBox_file_path_input_BCG")
|
||||
self.verticalLayout_5 = QVBoxLayout(self.groupBox_file_path_input_BCG)
|
||||
self.verticalLayout_5.setObjectName(u"verticalLayout_5")
|
||||
self.plainTextEdit_file_path_input_BCG = QPlainTextEdit(self.groupBox_file_path_input_BCG)
|
||||
self.plainTextEdit_file_path_input_BCG.setObjectName(u"plainTextEdit_file_path_input_BCG")
|
||||
self.gridLayout_2 = QGridLayout(self.groupBox)
|
||||
self.gridLayout_2.setObjectName(u"gridLayout_2")
|
||||
self.groupBox_file_path_save_2 = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_save_2.setObjectName(u"groupBox_file_path_save_2")
|
||||
self.verticalLayout_9 = QVBoxLayout(self.groupBox_file_path_save_2)
|
||||
self.verticalLayout_9.setObjectName(u"verticalLayout_9")
|
||||
self.plainTextEdit_file_path_save_orgBcg = QPlainTextEdit(self.groupBox_file_path_save_2)
|
||||
self.plainTextEdit_file_path_save_orgBcg.setObjectName(u"plainTextEdit_file_path_save_orgBcg")
|
||||
|
||||
self.verticalLayout_5.addWidget(self.plainTextEdit_file_path_input_BCG)
|
||||
self.verticalLayout_9.addWidget(self.plainTextEdit_file_path_save_orgBcg)
|
||||
|
||||
self.verticalLayout_5.setStretch(0, 3)
|
||||
self.plainTextEdit_file_path_save_BCG = QPlainTextEdit(self.groupBox_file_path_save_2)
|
||||
self.plainTextEdit_file_path_save_BCG.setObjectName(u"plainTextEdit_file_path_save_BCG")
|
||||
|
||||
self.verticalLayout_2.addWidget(self.groupBox_file_path_input_BCG)
|
||||
self.verticalLayout_9.addWidget(self.plainTextEdit_file_path_save_BCG)
|
||||
|
||||
self.groupBox_file_path_input_Jpeak = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_input_Jpeak.setObjectName(u"groupBox_file_path_input_Jpeak")
|
||||
self.verticalLayout_3 = QVBoxLayout(self.groupBox_file_path_input_Jpeak)
|
||||
self.verticalLayout_3.setObjectName(u"verticalLayout_3")
|
||||
self.plainTextEdit_file_path_input_Jpeak = QPlainTextEdit(self.groupBox_file_path_input_Jpeak)
|
||||
self.plainTextEdit_file_path_input_Jpeak.setObjectName(u"plainTextEdit_file_path_input_Jpeak")
|
||||
self.plainTextEdit_file_path_save_ECG = QPlainTextEdit(self.groupBox_file_path_save_2)
|
||||
self.plainTextEdit_file_path_save_ECG.setObjectName(u"plainTextEdit_file_path_save_ECG")
|
||||
|
||||
self.verticalLayout_3.addWidget(self.plainTextEdit_file_path_input_Jpeak)
|
||||
self.verticalLayout_9.addWidget(self.plainTextEdit_file_path_save_ECG)
|
||||
|
||||
self.verticalLayout_3.setStretch(0, 2)
|
||||
|
||||
self.verticalLayout_2.addWidget(self.groupBox_file_path_input_Jpeak)
|
||||
self.gridLayout_2.addWidget(self.groupBox_file_path_save_2, 6, 0, 1, 1)
|
||||
|
||||
self.groupBox_file_path_save_3 = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_save_3.setObjectName(u"groupBox_file_path_save_3")
|
||||
self.verticalLayout_10 = QVBoxLayout(self.groupBox_file_path_save_3)
|
||||
self.verticalLayout_10.setObjectName(u"verticalLayout_10")
|
||||
self.plainTextEdit_file_path_save_Jpeak = QPlainTextEdit(self.groupBox_file_path_save_3)
|
||||
self.plainTextEdit_file_path_save_Jpeak.setObjectName(u"plainTextEdit_file_path_save_Jpeak")
|
||||
|
||||
self.verticalLayout_10.addWidget(self.plainTextEdit_file_path_save_Jpeak)
|
||||
|
||||
self.plainTextEdit_file_path_save_Rpeak = QPlainTextEdit(self.groupBox_file_path_save_3)
|
||||
self.plainTextEdit_file_path_save_Rpeak.setObjectName(u"plainTextEdit_file_path_save_Rpeak")
|
||||
|
||||
self.verticalLayout_10.addWidget(self.plainTextEdit_file_path_save_Rpeak)
|
||||
|
||||
|
||||
self.gridLayout_2.addWidget(self.groupBox_file_path_save_3, 6, 1, 1, 1)
|
||||
|
||||
self.groupBox_file_path_save = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_save.setObjectName(u"groupBox_file_path_save")
|
||||
self.verticalLayout_4 = QVBoxLayout(self.groupBox_file_path_save)
|
||||
self.verticalLayout_4.setObjectName(u"verticalLayout_4")
|
||||
self.plainTextEdit_file_path_save_BCG_AlignInfo = QPlainTextEdit(self.groupBox_file_path_save)
|
||||
self.plainTextEdit_file_path_save_BCG_AlignInfo.setObjectName(u"plainTextEdit_file_path_save_BCG_AlignInfo")
|
||||
|
||||
self.verticalLayout_4.addWidget(self.plainTextEdit_file_path_save_BCG_AlignInfo)
|
||||
|
||||
self.plainTextEdit_file_path_save_ECG_AlignInfo = QPlainTextEdit(self.groupBox_file_path_save)
|
||||
self.plainTextEdit_file_path_save_ECG_AlignInfo.setObjectName(u"plainTextEdit_file_path_save_ECG_AlignInfo")
|
||||
|
||||
self.verticalLayout_4.addWidget(self.plainTextEdit_file_path_save_ECG_AlignInfo)
|
||||
|
||||
|
||||
self.gridLayout_2.addWidget(self.groupBox_file_path_save, 5, 0, 1, 2)
|
||||
|
||||
self.groupBox_file_path_input_Rpeak = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_input_Rpeak.setObjectName(u"groupBox_file_path_input_Rpeak")
|
||||
self.verticalLayout_7 = QVBoxLayout(self.groupBox_file_path_input_Rpeak)
|
||||
self.verticalLayout_7.setObjectName(u"verticalLayout_7")
|
||||
self.plainTextEdit_file_path_input_Rpeak = QPlainTextEdit(self.groupBox_file_path_input_Rpeak)
|
||||
self.plainTextEdit_file_path_input_Rpeak.setObjectName(u"plainTextEdit_file_path_input_Rpeak")
|
||||
|
||||
self.verticalLayout_7.addWidget(self.plainTextEdit_file_path_input_Rpeak)
|
||||
|
||||
self.verticalLayout_7.setStretch(0, 2)
|
||||
|
||||
self.gridLayout_2.addWidget(self.groupBox_file_path_input_Rpeak, 4, 0, 1, 2)
|
||||
|
||||
self.groupBox_file_path_input_ECG = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_input_ECG.setObjectName(u"groupBox_file_path_input_ECG")
|
||||
@ -101,43 +144,54 @@ class Ui_MainWindow_precisely_align_input_setting(object):
|
||||
self.verticalLayout_6.setStretch(0, 2)
|
||||
self.verticalLayout_6.setStretch(1, 3)
|
||||
|
||||
self.verticalLayout_2.addWidget(self.groupBox_file_path_input_ECG)
|
||||
self.gridLayout_2.addWidget(self.groupBox_file_path_input_ECG, 3, 0, 1, 2)
|
||||
|
||||
self.groupBox_file_path_input_Rpeak = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_input_Rpeak.setObjectName(u"groupBox_file_path_input_Rpeak")
|
||||
self.verticalLayout_7 = QVBoxLayout(self.groupBox_file_path_input_Rpeak)
|
||||
self.verticalLayout_7.setObjectName(u"verticalLayout_7")
|
||||
self.plainTextEdit_file_path_input_Rpeak = QPlainTextEdit(self.groupBox_file_path_input_Rpeak)
|
||||
self.plainTextEdit_file_path_input_Rpeak.setObjectName(u"plainTextEdit_file_path_input_Rpeak")
|
||||
self.groupBox_file_path_input_Jpeak = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_input_Jpeak.setObjectName(u"groupBox_file_path_input_Jpeak")
|
||||
self.verticalLayout_3 = QVBoxLayout(self.groupBox_file_path_input_Jpeak)
|
||||
self.verticalLayout_3.setObjectName(u"verticalLayout_3")
|
||||
self.plainTextEdit_file_path_input_Jpeak = QPlainTextEdit(self.groupBox_file_path_input_Jpeak)
|
||||
self.plainTextEdit_file_path_input_Jpeak.setObjectName(u"plainTextEdit_file_path_input_Jpeak")
|
||||
|
||||
self.verticalLayout_7.addWidget(self.plainTextEdit_file_path_input_Rpeak)
|
||||
self.verticalLayout_3.addWidget(self.plainTextEdit_file_path_input_Jpeak)
|
||||
|
||||
self.verticalLayout_7.setStretch(0, 2)
|
||||
self.verticalLayout_3.setStretch(0, 2)
|
||||
|
||||
self.verticalLayout_2.addWidget(self.groupBox_file_path_input_Rpeak)
|
||||
self.gridLayout_2.addWidget(self.groupBox_file_path_input_Jpeak, 2, 0, 1, 2)
|
||||
|
||||
self.groupBox_file_path_save = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_save.setObjectName(u"groupBox_file_path_save")
|
||||
self.verticalLayout_4 = QVBoxLayout(self.groupBox_file_path_save)
|
||||
self.verticalLayout_4.setObjectName(u"verticalLayout_4")
|
||||
self.plainTextEdit_file_path_save_BCG = QPlainTextEdit(self.groupBox_file_path_save)
|
||||
self.plainTextEdit_file_path_save_BCG.setObjectName(u"plainTextEdit_file_path_save_BCG")
|
||||
self.groupBox_file_path_input_BCG = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_input_BCG.setObjectName(u"groupBox_file_path_input_BCG")
|
||||
self.verticalLayout_5 = QVBoxLayout(self.groupBox_file_path_input_BCG)
|
||||
self.verticalLayout_5.setObjectName(u"verticalLayout_5")
|
||||
self.plainTextEdit_file_path_input_BCG = QPlainTextEdit(self.groupBox_file_path_input_BCG)
|
||||
self.plainTextEdit_file_path_input_BCG.setObjectName(u"plainTextEdit_file_path_input_BCG")
|
||||
|
||||
self.verticalLayout_4.addWidget(self.plainTextEdit_file_path_save_BCG)
|
||||
self.verticalLayout_5.addWidget(self.plainTextEdit_file_path_input_BCG)
|
||||
|
||||
self.plainTextEdit_file_path_save_ECG = QPlainTextEdit(self.groupBox_file_path_save)
|
||||
self.plainTextEdit_file_path_save_ECG.setObjectName(u"plainTextEdit_file_path_save_ECG")
|
||||
self.verticalLayout_5.setStretch(0, 3)
|
||||
|
||||
self.verticalLayout_4.addWidget(self.plainTextEdit_file_path_save_ECG)
|
||||
self.gridLayout_2.addWidget(self.groupBox_file_path_input_BCG, 1, 0, 1, 2)
|
||||
|
||||
self.groupBox_file_path_input_orgBcg = QGroupBox(self.groupBox)
|
||||
self.groupBox_file_path_input_orgBcg.setObjectName(u"groupBox_file_path_input_orgBcg")
|
||||
self.verticalLayout_8 = QVBoxLayout(self.groupBox_file_path_input_orgBcg)
|
||||
self.verticalLayout_8.setObjectName(u"verticalLayout_8")
|
||||
self.plainTextEdit_file_path_input_orgBcg = QPlainTextEdit(self.groupBox_file_path_input_orgBcg)
|
||||
self.plainTextEdit_file_path_input_orgBcg.setObjectName(u"plainTextEdit_file_path_input_orgBcg")
|
||||
|
||||
self.verticalLayout_2.addWidget(self.groupBox_file_path_save)
|
||||
self.verticalLayout_8.addWidget(self.plainTextEdit_file_path_input_orgBcg)
|
||||
|
||||
self.verticalLayout_2.setStretch(0, 5)
|
||||
self.verticalLayout_2.setStretch(1, 4)
|
||||
self.verticalLayout_2.setStretch(2, 5)
|
||||
self.verticalLayout_2.setStretch(3, 4)
|
||||
self.verticalLayout_2.setStretch(4, 8)
|
||||
self.verticalLayout_8.setStretch(0, 3)
|
||||
|
||||
self.gridLayout_2.addWidget(self.groupBox_file_path_input_orgBcg, 0, 0, 1, 2)
|
||||
|
||||
self.gridLayout_2.setRowStretch(0, 3)
|
||||
self.gridLayout_2.setRowStretch(1, 3)
|
||||
self.gridLayout_2.setRowStretch(2, 3)
|
||||
self.gridLayout_2.setRowStretch(3, 4)
|
||||
self.gridLayout_2.setRowStretch(4, 3)
|
||||
self.gridLayout_2.setRowStretch(5, 6)
|
||||
self.gridLayout_2.setRowStretch(6, 3)
|
||||
|
||||
self.gridLayout.addWidget(self.groupBox, 0, 0, 1, 4)
|
||||
|
||||
@ -161,22 +215,32 @@ class Ui_MainWindow_precisely_align_input_setting(object):
|
||||
MainWindow_precisely_align_input_setting.setWindowTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u5bfc\u5165\u8bbe\u7f6e", None))
|
||||
self.pushButton_cancel.setText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u53d6\u6d88", None))
|
||||
self.groupBox.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
|
||||
self.groupBox_file_path_input_BCG.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u9884\u5904\u7406\u540e\u7684BCG\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_input_BCG.setPlainText("")
|
||||
self.plainTextEdit_file_path_input_BCG.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
|
||||
self.groupBox_file_path_input_Jpeak.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4eba\u5de5\u7ea0\u6b63\u540e\u7684J\u5cf0\u5cf0\u503c\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_input_Jpeak.setPlainText("")
|
||||
self.plainTextEdit_file_path_input_Jpeak.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
|
||||
self.groupBox_file_path_save_2.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u5bf9\u9f50orgBcg\u3001BCG\u548cECG\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_save_orgBcg.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_save_BCG.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_save_ECG.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.groupBox_file_path_save_3.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u5bf9\u9f50J\u5cf0\u548cR\u5cf0\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_save_Jpeak.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_save_Rpeak.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.groupBox_file_path_save.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u5bf9\u9f50\u4fe1\u606f\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_save_BCG_AlignInfo.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_save_ECG_AlignInfo.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.groupBox_file_path_input_Rpeak.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4eba\u5de5\u7ea0\u6b63\u540e\u7684R\u5cf0\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_input_Rpeak.setPlainText("")
|
||||
self.plainTextEdit_file_path_input_Rpeak.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
|
||||
self.groupBox_file_path_input_ECG.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u9884\u5904\u7406\u540e\u7684ECG\u8def\u5f84", None))
|
||||
self.label_3.setText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u91c7\u6837\u7387(Hz)\uff1a", None))
|
||||
self.plainTextEdit_file_path_input_ECG.setPlainText("")
|
||||
self.plainTextEdit_file_path_input_ECG.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
|
||||
self.groupBox_file_path_input_Rpeak.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4eba\u5de5\u7ea0\u6b63\u540e\u7684R\u5cf0\u5cf0\u503c\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_input_Rpeak.setPlainText("")
|
||||
self.plainTextEdit_file_path_input_Rpeak.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
|
||||
self.groupBox_file_path_save.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u5bf9\u9f50\u4fe1\u606f\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_save_BCG.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_save_ECG.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
|
||||
self.groupBox_file_path_input_Jpeak.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u4eba\u5de5\u7ea0\u6b63\u540e\u7684J\u5cf0\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_input_Jpeak.setPlainText("")
|
||||
self.plainTextEdit_file_path_input_Jpeak.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
|
||||
self.groupBox_file_path_input_BCG.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u9884\u5904\u7406\u540e\u7684BCG\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_input_BCG.setPlainText("")
|
||||
self.plainTextEdit_file_path_input_BCG.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
|
||||
self.groupBox_file_path_input_orgBcg.setTitle(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u539f\u59cborgBcg\u8def\u5f84", None))
|
||||
self.plainTextEdit_file_path_input_orgBcg.setPlainText("")
|
||||
self.plainTextEdit_file_path_input_orgBcg.setPlaceholderText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
|
||||
self.pushButton_confirm.setText(QCoreApplication.translate("MainWindow_precisely_align_input_setting", u"\u786e\u5b9a", None))
|
||||
# retranslateUi
|
||||
|
||||
|
||||
@ -6,8 +6,8 @@
|
||||
<rect>
|
||||
<x>0</x>
|
||||
<y>0</y>
|
||||
<width>540</width>
|
||||
<height>760</height>
|
||||
<width>840</width>
|
||||
<height>1010</height>
|
||||
</rect>
|
||||
</property>
|
||||
<property name="windowTitle">
|
||||
@ -37,15 +37,91 @@
|
||||
<property name="title">
|
||||
<string>文件路径</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_2" stretch="5,4,5,4,8">
|
||||
<item>
|
||||
<widget class="QGroupBox" name="groupBox_file_path_input_BCG">
|
||||
<layout class="QGridLayout" name="gridLayout_2" rowstretch="3,3,3,4,3,6,3,0">
|
||||
<item row="6" column="0">
|
||||
<widget class="QGroupBox" name="groupBox_file_path_save_2">
|
||||
<property name="title">
|
||||
<string>预处理后的BCG路径</string>
|
||||
<string>对齐orgBcg、BCG和ECG保存路径</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_5" stretch="3">
|
||||
<layout class="QVBoxLayout" name="verticalLayout_9">
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_input_BCG">
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save_orgBcg">
|
||||
<property name="placeholderText">
|
||||
<string>保存路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save_BCG">
|
||||
<property name="placeholderText">
|
||||
<string>保存路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save_ECG">
|
||||
<property name="placeholderText">
|
||||
<string>保存路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
</layout>
|
||||
</widget>
|
||||
</item>
|
||||
<item row="6" column="1">
|
||||
<widget class="QGroupBox" name="groupBox_file_path_save_3">
|
||||
<property name="title">
|
||||
<string>对齐J峰和R峰保存路径</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_10">
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save_Jpeak">
|
||||
<property name="placeholderText">
|
||||
<string>保存路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save_Rpeak">
|
||||
<property name="placeholderText">
|
||||
<string>保存路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
</layout>
|
||||
</widget>
|
||||
</item>
|
||||
<item row="5" column="0" colspan="2">
|
||||
<widget class="QGroupBox" name="groupBox_file_path_save">
|
||||
<property name="title">
|
||||
<string>对齐信息保存路径</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_4">
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save_BCG_AlignInfo">
|
||||
<property name="placeholderText">
|
||||
<string>保存路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save_ECG_AlignInfo">
|
||||
<property name="placeholderText">
|
||||
<string>保存路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
</layout>
|
||||
</widget>
|
||||
</item>
|
||||
<item row="4" column="0" colspan="2">
|
||||
<widget class="QGroupBox" name="groupBox_file_path_input_Rpeak">
|
||||
<property name="title">
|
||||
<string>人工纠正后的R峰路径</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_7" stretch="2">
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_input_Rpeak">
|
||||
<property name="plainText">
|
||||
<string/>
|
||||
</property>
|
||||
@ -57,26 +133,7 @@
|
||||
</layout>
|
||||
</widget>
|
||||
</item>
|
||||
<item>
|
||||
<widget class="QGroupBox" name="groupBox_file_path_input_Jpeak">
|
||||
<property name="title">
|
||||
<string>人工纠正后的J峰峰值路径</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_3" stretch="2">
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_input_Jpeak">
|
||||
<property name="plainText">
|
||||
<string/>
|
||||
</property>
|
||||
<property name="placeholderText">
|
||||
<string>文件路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
</layout>
|
||||
</widget>
|
||||
</item>
|
||||
<item>
|
||||
<item row="3" column="0" colspan="2">
|
||||
<widget class="QGroupBox" name="groupBox_file_path_input_ECG">
|
||||
<property name="title">
|
||||
<string>预处理后的ECG路径</string>
|
||||
@ -126,14 +183,14 @@
|
||||
</layout>
|
||||
</widget>
|
||||
</item>
|
||||
<item>
|
||||
<widget class="QGroupBox" name="groupBox_file_path_input_Rpeak">
|
||||
<item row="2" column="0" colspan="2">
|
||||
<widget class="QGroupBox" name="groupBox_file_path_input_Jpeak">
|
||||
<property name="title">
|
||||
<string>人工纠正后的R峰峰值路径</string>
|
||||
<string>人工纠正后的J峰路径</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_7" stretch="2">
|
||||
<layout class="QVBoxLayout" name="verticalLayout_3" stretch="2">
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_input_Rpeak">
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_input_Jpeak">
|
||||
<property name="plainText">
|
||||
<string/>
|
||||
</property>
|
||||
@ -145,23 +202,38 @@
|
||||
</layout>
|
||||
</widget>
|
||||
</item>
|
||||
<item>
|
||||
<widget class="QGroupBox" name="groupBox_file_path_save">
|
||||
<item row="1" column="0" colspan="2">
|
||||
<widget class="QGroupBox" name="groupBox_file_path_input_BCG">
|
||||
<property name="title">
|
||||
<string>对齐信息保存路径</string>
|
||||
<string>预处理后的BCG路径</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_4">
|
||||
<layout class="QVBoxLayout" name="verticalLayout_5" stretch="3">
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save_BCG">
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_input_BCG">
|
||||
<property name="plainText">
|
||||
<string/>
|
||||
</property>
|
||||
<property name="placeholderText">
|
||||
<string>保存路径</string>
|
||||
<string>文件路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
</layout>
|
||||
</widget>
|
||||
</item>
|
||||
<item row="0" column="0" colspan="2">
|
||||
<widget class="QGroupBox" name="groupBox_file_path_input_orgBcg">
|
||||
<property name="title">
|
||||
<string>原始orgBcg路径</string>
|
||||
</property>
|
||||
<layout class="QVBoxLayout" name="verticalLayout_8" stretch="3">
|
||||
<item>
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save_ECG">
|
||||
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_input_orgBcg">
|
||||
<property name="plainText">
|
||||
<string/>
|
||||
</property>
|
||||
<property name="placeholderText">
|
||||
<string>保存路径</string>
|
||||
<string>文件路径</string>
|
||||
</property>
|
||||
</widget>
|
||||
</item>
|
||||
|
||||
187
数据结构化输入和输出命名规范.md
Normal file
187
数据结构化输入和输出命名规范.md
Normal file
@ -0,0 +1,187 @@
|
||||
# 文件命名规范
|
||||
|
||||
### 1 数据粗同步
|
||||
|
||||
输入：
|
||||
|
||||
原始orgBcg信号：`./OrgBCG_Text/<sampID>/orgBcg_Raw_采样率.txt`
|
||||
原始PSG信号：`./PSG_Text/<sampID>/Axxxxxxx.edf`
|
||||
|
||||
输出：
|
||||
|
||||
粗同步后的位置索引：`./Sync_Index/<sampID>/Approximately_Align_Info.csv`
|
||||
|
||||
### 2 预处理
|
||||
|
||||
输入：
|
||||
|
||||
原始orgBcg信号：`./OrgBCG_Text/<sampID>/orgBcg_Raw_采样率.txt`
|
||||
原始ECG信号：`./PSG_Text/<sampID>/ECG I_采样率.txt`
|
||||
|
||||
输出：
|
||||
|
||||
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Raw_采样率.txt`
|
||||
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Raw_采样率.txt`
|
||||
|
||||
### 3 数据精同步
|
||||
|
||||
输入：
|
||||
|
||||
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Raw_采样率.txt`
|
||||
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Raw_采样率.txt`
|
||||
|
||||
输出：
|
||||
|
||||
PSG的对齐信息：`./PSG_Aligned/<sampID>/Align_info.txt`
|
||||
同步后的ECG信号：`./PSG_Aligned/<sampID>/ECG_Sync_采样率.txt`
|
||||
同步后的R峰坐标：`./PSG_Aligned/<sampID>/Rpeaks_Sync.txt`
|
||||
同步后的其他PSG通道信号：`./PSG_Aligned/<sampID>/通道名_Sync_采样率.txt`
|
||||
BCG的对齐信息：`./OrgBCG_Aligned/<sampID>/Align_info.txt`
|
||||
同步后的BCG信号：`./OrgBCG_Aligned/<sampID>/BCG_Sync_采样率.txt`
|
||||
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/orgBcg_Sync_采样率.txt`
|
||||
同步后的J峰坐标：`./OrgBCG_Aligned/<sampID>/Jpeaks_Sync.txt`
|
||||
|
||||
#### 3.1 算法定位
|
||||
|
||||
#### 3.1.1 R峰算法定位
|
||||
|
||||
输入：
|
||||
|
||||
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Raw_采样率.txt`
|
||||
|
||||
输出：
|
||||
|
||||
算法定位的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final.txt`
|
||||
|
||||
#### 3.1.2 J峰算法定位
|
||||
|
||||
输入：
|
||||
|
||||
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Raw_采样率.txt`
|
||||
|
||||
输出：
|
||||
|
||||
算法定位的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise.txt`
|
||||
|
||||
#### 3.2 人工纠正
|
||||
|
||||
#### 3.2.1 R峰人工纠正
|
||||
|
||||
输入：
|
||||
|
||||
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_filter_采样率.txt`
|
||||
算法定位的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final.txt`
|
||||
|
||||
输出：
|
||||
|
||||
人工纠正后的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final_corrected.txt`
|
||||
|
||||
#### 3.2.2 J峰人工纠正*
|
||||
|
||||
输入：
|
||||
|
||||
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/DSbcg_sig_采样率.txt`
|
||||
算法定位的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise.txt`
|
||||
|
||||
输出：
|
||||
|
||||
人工纠正后的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise_corrected.txt`
|
||||
|
||||
#### 3.3 数据片段起止对齐、数据采样率同步
|
||||
|
||||
输入：
|
||||
|
||||
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Raw_采样率.txt`
|
||||
人工纠正后的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final_corrected.txt`
|
||||
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Raw_采样率.txt`
|
||||
人工纠正后的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise_corrected.txt`
|
||||
|
||||
输出：
|
||||
|
||||
PSG的对齐信息：`./PSG_Aligned/<sampID>/Align_info.txt`
|
||||
同步后的ECG信号：`./PSG_Aligned/<sampID>/ECG_Sync_采样率.txt`
|
||||
同步后的R峰坐标：`./PSG_Aligned/<sampID>/Rpeak_Sync.txt`
|
||||
BCG的对齐信息：`./OrgBCG_Aligned/<sampID>/Align_info.txt`
|
||||
同步后的BCG信号：`./OrgBCG_Aligned/<sampID>/BCG_Sync_采样率.txt`
|
||||
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/orgBcg_Sync_采样率.txt`
|
||||
同步后的J峰坐标：`./OrgBCG_Aligned/<sampID>/Jpeak_Sync.txt`
|
||||
|
||||
#### 3.4、冗余数据切割、标签映射
|
||||
|
||||
输入：
|
||||
|
||||
PSG的对齐信息：`./PSG_Aligned/<sampID>/Align_info.txt`
|
||||
BCG的对齐信息：`./OrgBCG_Aligned/<sampID>/Align_info.txt`
|
||||
|
||||
输出：
|
||||
|
||||
同步后的其他PSG通道信号：`./PSG_Aligned/<sampID>/通道名_Sync_采样率.txt`
|
||||
|
||||
### 4 体动标记
|
||||
|
||||
输入：
|
||||
|
||||
同步后的BCG信号：`./OrgBCG_Aligned/<sampID>/BCG_Sync_采样率.txt`
|
||||
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/orgBcg_Sync_采样率.txt`
|
||||
|
||||
输出：
|
||||
|
||||
txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
|
||||
体动标签类型数量统计：`./Label/<sampID>/Artifact_b.txt`
|
||||
csv格式的体动标签：`./Label/<sampID>/Artifact_c.csv`
|
||||
|
||||
### 5 质量评估
|
||||
|
||||
输入：
|
||||
|
||||
同步后的BCG信号：`./OrgBCG_Aligned/<sampID>/BCG_Sync_采样率.txt`
|
||||
txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
|
||||
|
||||
输出：
|
||||
|
||||
质量标签：`./Label/<sampID>/SQ_label_10s.csv`或`./Label/<sampID>/SQ_label_30s.csv`
|
||||
|
||||
### 6 心搏定位数据标注
|
||||
|
||||
输入：
|
||||
|
||||
同步后的ECG信号：`./PSG_Aligned/<sampID>/ECG_Sync_采样率.txt`
|
||||
同步后的BCG信号：`./OrgBCG_Aligned/<sampID>/BCG_Sync_采样率.txt`
|
||||
同步后的R峰坐标：`./PSG_Aligned/<sampID>/Rpeak_Sync.txt`
|
||||
同步后的J峰坐标：`./OrgBCG_Aligned/<sampID>/Jpeak_Sync.txt`
|
||||
txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
|
||||
质量标签：`./Label/<sampID>/SQ_label_10s.csv`或`./Label/<sampID>/SQ_label_30s.csv`
|
||||
|
||||
输出：
|
||||
|
||||
？
|
||||
|
||||
### 7 呼吸提取
|
||||
|
||||
输入：
|
||||
|
||||
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/orgBcg_Sync_采样率.txt`
|
||||
同步后的THO信号：`./PSG_Aligned/<sampID>/Effort_Tho_Sync_采样率.txt`
|
||||
txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
|
||||
|
||||
输出：
|
||||
|
||||
呼吸可用性标签：`./Label/<sampID>/Resp_quality_label.txt`
|
||||
THO信号呼吸间期标签：`./Label/<sampID>/Tho_peak.txt`
|
||||
|
||||
### 8 呼吸暂停事件标注
|
||||
|
||||
输入：
|
||||
|
||||
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/orgBcg_Sync_采样率.txt`
|
||||
txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
|
||||
同步后的FlowT信号：`./PSG_Aligned/<sampID>/Effort_Tho_Sync_采样率.txt`
|
||||
同步后的FlowP信号：`./PSG_Aligned/<sampID>/Effort_Tho_Sync_采样率.txt`
|
||||
同步后的THO信号：`./PSG_Aligned/<sampID>/Effort_Tho_Sync_采样率.txt`
|
||||
同步后的ABD信号：`./PSG_Aligned/<sampID>/Effort_Abd_Sync_采样率.txt`
|
||||
同步后的SpO2信号：`./PSG_Aligned/<sampID>/SpO2_Sync_采样率.txt`
|
||||
原始呼吸暂停标签：`./PSG_`？
|
||||
|
||||
输出：
|
||||
|
||||
？
|
||||
Reference in New Issue
Block a user