From fada84e1155ec6974ee950a2eac32e437ce592a4 Mon Sep 17 00:00:00 2001 From: Yorusora <2944763079@qq.com> Date: Tue, 20 May 2025 16:30:17 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96<=E6=95=B0=E6=8D=AE=E9=A2=84?= =?UTF-8?q?=E5=A4=84=E7=90=86>=E7=9A=84=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- func/Module_preprocess.py | 77 ++++++++++++++++++++++++++------------ func/utils/ConfigParams.py | 10 +++++ func/utils/Constants.py | 20 ++++++++-- 3 files changed, 80 insertions(+), 27 deletions(-) diff --git a/func/Module_preprocess.py b/func/Module_preprocess.py index a15c11d..4b6d655 100644 --- a/func/Module_preprocess.py +++ b/func/Module_preprocess.py @@ -75,13 +75,9 @@ class SettingWindow(QMainWindow): Config.update({ "Path": { "Input": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT / - Path(str(self.sampID)) / Path(ConfigParams.PREPROCESS_INPUT_BCG_FILENAME + - str(Config["InputConfig"]["Freq"]) + - ConfigParams.ENDSWITH_TXT))), + Path(str(self.sampID)))), "Save": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT / - Path(str(self.sampID)) / Path(ConfigParams.PREPROCESS_SAVE_BCG_FILENAME + - str(Config["OutputConfig"]["Freq"]) + - ConfigParams.ENDSWITH_TXT))) + Path(str(self.sampID)))) }, "Mode": self.mode }) @@ -89,13 +85,9 @@ class SettingWindow(QMainWindow): Config.update({ "Path": { "Input": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_TEXT / - Path(str(self.sampID)) / Path(ConfigParams.PREPROCESS_INPUT_ECG_FILENAME + - str(Config["InputConfig"]["Freq"]) + - ConfigParams.ENDSWITH_TXT))), + Path(str(self.sampID)))), "Save": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_TEXT / - Path(str(self.sampID)) / Path(ConfigParams.PREPROCESS_SAVE_ECG_FILENAME + - str(Config["OutputConfig"]["Freq"]) + - ConfigParams.ENDSWITH_TXT))) + Path(str(self.sampID)))) }, "Mode": self.mode }) @@ -133,14 +125,14 @@ class SettingWindow(QMainWindow): str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT / Path(str(self.sampID)) / - Path(ConfigParams.PREPROCESS_INPUT_BCG_FILENAME + + Path(ConfigParams.ORGBCG_RAW + str(self.ui.spinBox_input_freq.value()) + ConfigParams.ENDSWITH_TXT)))) self.ui.plainTextEdit_file_path_save.setPlainText( str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT / Path(str(self.sampID)) / - Path(ConfigParams.PREPROCESS_SAVE_BCG_FILENAME + + Path(ConfigParams.BCG_FILTER + str(self.ui.spinBox_output_freq.value()) + ConfigParams.ENDSWITH_TXT)))) elif self.mode == "ECG": @@ -148,14 +140,14 @@ class SettingWindow(QMainWindow): str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_TEXT / Path(str(self.sampID)) / - Path(ConfigParams.PREPROCESS_INPUT_ECG_FILENAME + + Path(ConfigParams.ECG_RAW + str(self.ui.spinBox_input_freq.value()) + ConfigParams.ENDSWITH_TXT)))) self.ui.plainTextEdit_file_path_save.setPlainText( str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_TEXT / Path(str(self.sampID)) / - Path(ConfigParams.PREPROCESS_SAVE_ECG_FILENAME + + Path(ConfigParams.ECG_FILTER + str(self.ui.spinBox_output_freq.value()) + ConfigParams.ENDSWITH_TXT)))) else: @@ -394,21 +386,57 @@ class Data: self.processed_data = None def open_file(self): + if Config["Mode"] == "BCG": + signal = ConfigParams.ORGBCG_RAW + elif Config["Mode"] == "ECG": + signal = ConfigParams.ECG_RAW + else: + raise ValueError("模式不存在") + if Path(Config["Path"]["Input"]).is_file(): + Config["Path"]["Input"] = str(Path(Config["Path"]["Input"]).parent) + if not Path(Config["Path"]["Input"]).exists(): - return Result().failure(info=Constants.INPUT_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Data_Path_Not_Exist"]) + return Result().failure(info=Constants.INPUT_FAILURE + Constants.FAILURE_REASON["Path_Not_Exist"]) + + temp_path = list( + Path(Config["Path"]["Input"]).glob(f"{signal}*")) + if len(temp_path) == 0: + return Result().failure( + info=Constants.INPUT_FAILURE + "\n" + + signal + "：" + + Config["Path"]["Input"] + + Constants.FAILURE_REASON["File_Not_Exist"]) + elif len(temp_path) > 1: + return Result().failure( + info=Constants.INPUT_FAILURE + "\n" + + signal + "：" + + Config["Path"]["Input"] + + Constants.FAILURE_REASON["Data_File_More_Than_One"]) + else: + path = temp_path[0] + Config["Path"]["Input"] = str(path) + freq = path.stem.split("_")[-1] + # 验证是否为整数或是否存在 + if not freq.isdigit(): + return Result().failure( + info=Constants.INPUT_FAILURE + "\n" + + signal + "：" + + Constants.FAILURE_REASON["Data_Frequency_Not_In_Filename"]) + Config["InputConfig"]["Freq"] = int(freq) try: self.raw_data = read_csv(Config["Path"]["Input"], encoding=ConfigParams.UTF8_ENCODING, header=None).to_numpy().reshape(-1) except Exception as e: - return Result().failure(info=Constants.INPUT_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Read_Data_Exception"] + "\n" + format_exc()) + return Result().failure(info=Constants.INPUT_FAILURE + + Constants.FAILURE_REASON["Open_Data_Exception"] + "\n" + format_exc()) return Result().success(info=Constants.INPUT_FINISHED) def preprocess(self): if self.raw_data is None: - Result().failure(info=Constants.PREPROCESS_PROCESS_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Raw_Data_Not_Exist"]) + Result().failure(info=Constants.PREPROCESS_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"]) try: if Config["InputConfig"]["Freq"] != Config["OutputConfig"]["Freq"]: @@ -430,18 +458,19 @@ class Data: else: raise ValueError("模式不存在") except Exception as e: - Result().failure(info=Constants.PREPROCESS_PROCESS_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Filter_Exception"] + "\n" + format_exc()) - - return Result().success(info=Constants.PREPROCESS_PROCESS_FINISHED) + Result().failure(info=Constants.PREPROCESS_FAILURE + + Constants.FAILURE_REASON["Preprocess_Exception"] + "\n" + format_exc()) + return Result().success(info=Constants.PREPROCESS_FINISHED) def save(self, chunk): if self.processed_data is None: - return Result().failure(info=Constants.SAVING_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Processed_Data_Not_Exist"]) + return Result().failure(info=Constants.SAVING_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"]) try: chunk.to_csv(Config["Path"]["Save"], mode='a', index=False, header=False, float_format='%.4f') except Exception as e: - return Result().failure(info=Constants.SAVING_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Save_Exception"] + "\n" + format_exc()) + return Result().failure(info=Constants.SAVING_FAILURE + + Constants.FAILURE_REASON["Save_Exception"] + "\n" + format_exc()) return Result().success(Constants.SAVING_FINISHED) \ No newline at end of file diff --git a/func/utils/ConfigParams.py b/func/utils/ConfigParams.py index 9cc94dd..a5c7f6c 100644 --- a/func/utils/ConfigParams.py +++ b/func/utils/ConfigParams.py @@ -29,6 +29,16 @@ class ConfigParams: ACTION_ZOOM_SHORTCUT_KEY: str = "C" FONT: str = "Microsoft YaHei UI" + ORGBCG_RAW: str = "OrgBCG_Raw_" + ECG_RAW: str = "ECG II_Raw_" + THO_RAW: str = "Effort Tho_Raw_" + ABD_RAW: str = "Effort Abd_Raw_" + + BCG_FILTER: str = "BCG_Filter_" + ECG_FILTER: str = "ECG_Filter_" + + + # 数据粗同步 APPROXIMATELY_ALIGN_CONFIG_FILE_PATH: str = "./config/Config_approximately_align.yaml" APPROXIMATELY_ALIGN_CONFIG_NEW_CONTENT: dict = { diff --git a/func/utils/Constants.py b/func/utils/Constants.py index 94dc40b..88c58be 100644 --- a/func/utils/Constants.py +++ b/func/utils/Constants.py @@ -19,6 +19,10 @@ class Constants: INPUT_FINISHED: str = "导入完成" INPUT_FAILURE: str = "导入失败" + PREPROCESSING_DATA: str = "正在预处理数据" + PREPROCESS_FINISHED: str = "预处理完成" + PREPROCESS_FAILURE: str = "预处理失败" + RESAMPLING_DATA: str = "正在数据重采样" RESAMPLE_FINISHED: str = "重采样完成" RESAMPLE_FAILURE: str = "重采样失败" @@ -79,6 +83,18 @@ class Constants: } """ + FAILURE_REASON: dict = { + "Path_Not_Exist": "（路径不存在）", + "File_Not_Exist": "（数据文件不存在）", + "File_More_Than_One": "（数据文件超过一个）", + "Frequency_Not_In_Filename": "（数据频率不在文件名中）", + "Data_Not_Exist": "（数据不存在）", + + "Open_Data_Exception": "（打开数据异常）", + "Preprocess_Exception": "（预处理异常）", + "Save_Exception": "（保存异常）" + } + # 数据粗同步 APPROXIMATELY_ONLY_ALIGN_RESAMPLING: str = "正在仅重采样" APPROXIMATELY_ONLY_ALIGN_RESAMPLE_FINISHED: str = "仅重采样完成" @@ -164,9 +180,7 @@ class Constants: DETECT_JPEAK_LOAD_FINISHED: str = "读取完成" DETECT_JPEAK_LOAD_FAILURE: str = "读取失败" - DETECT_JPEAK_PROCESSING_DATA: str = "正在处理数据" - DETECT_JPEAK_PROCESS_FINISHED: str = "处理完成" - DETECT_JPEAK_PROCESS_FAILURE: str = "处理失败" + DETECT_JPEAK_PREDICTING_PEAK: str = "正在预测峰值" DETECT_JPEAK_PREDICT_FINISHED: str = "预测完成"