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()
|
||||
Reference in New Issue
Block a user