From c6bd99cc74316feae30f6c09beb9d72d10b4ceef Mon Sep 17 00:00:00 2001 From: Yorusora <2023025086@m.scnu.edu.cn> Date: Sun, 8 Jun 2025 12:45:04 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E4=BA=86<=E4=BD=93=E5=8A=A8?= =?UTF-8?q?=E6=A0=87=E6=B3=A8>=E7=9A=84=E4=BD=93=E5=8A=A8=E5=90=88?= =?UTF-8?q?=E5=B9=B6=E8=A7=84=E5=88=99?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 2 + func/Module_artifact_label.py | 251 +++++++++++++++++++++++----------- func/utils/Constants.py | 14 +- 3 files changed, 176 insertions(+), 91 deletions(-) diff --git a/README.md b/README.md index 7357f8d..5fc67c3 100644 --- a/README.md +++ b/README.md @@ -21,6 +21,8 @@ ~~7、各个模块中的检测父级文件夹是否存在的功能仍存在问题，无法正确创建文件夹~~ +8、将导入设置中的采样率修改为自动获取数据文件名中的采样率 + ## 1、主菜单 ![](/img/0.png) diff --git a/func/Module_artifact_label.py b/func/Module_artifact_label.py index 8021924..3a5b63a 100644 --- a/func/Module_artifact_label.py +++ b/func/Module_artifact_label.py @@ -711,117 +711,202 @@ class MainWindow_artifact_label(QMainWindow): (0 < (int(self.ui.lineEdit_start_time.text()) - value_endTime) <= (2 * Config["InputConfig"]["UseFreq"])) or (0 < (value_startTime - int(self.ui.lineEdit_end_time.text())) <= (2 * Config["InputConfig"]["UseFreq"]))): select_row.append(row.to_dict()) - if row['type'] == 1: + if value_type == 1: select_type[0] = select_type[0] + 1 - elif row['type'] == 2: + elif value_type == 2: select_type[1] = select_type[1] + 1 - elif row['type'] == 3: + elif value_type == 3: select_type[2] = select_type[2] + 1 - elif row['type'] == 4: + elif value_type == 4: select_type[3] = select_type[3] + 1 - elif row['type'] == 5: + elif value_type == 5: select_type[4] = select_type[4] + 1 - if (0 < (int(self.ui.lineEdit_start_time.text()) - value_endTime) <= (2 * Config["InputConfig"]["UseFreq"])) and (value_type == type): + if (0 < (int(self.ui.lineEdit_start_time.text()) - value_endTime) <= (2 * Config["InputConfig"]["UseFreq"])): flagf = True - if (0 < (value_startTime - int(self.ui.lineEdit_end_time.text())) <= (2 * Config["InputConfig"]["UseFreq"])) and (value_type == type): + if (0 < (value_startTime - int(self.ui.lineEdit_end_time.text())) <= (2 * Config["InputConfig"]["UseFreq"])): flagb = True - count = np_sum(select_type >= 1) - if count >= 2 or (count == 1 and nonzero(select_type)[0][0] + 1 == type): - if count >= 2: - reply = QMessageBox.question(self, '确认', - '{}{}'.format(Constants.ARTIFACT_LABEL_MULTIPLE_ARTIFACT_COVER_OR_DELETE, [d["number"] for d in select_row]), - QMessageBox.Yes | QMessageBox.No, - QMessageBox.No) - if reply == QMessageBox.No: - return - + # 按照括注的体动的数量来进行条件判断 + if len(select_row) >= 1: if flagf == True and flagb == False: - if type == 1 or type == 2 or type == 3: - PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_TWO_SECONDS_MERGE, - Constants.MSGBOX_TYPE_INFO) - start_time = select_row[0]['startTime'] - end_time = int(self.ui.lineEdit_end_time.text()) - elif type == 4 or type == 5: + start_time = select_row[0]['startTime'] + end_time = int(self.ui.lineEdit_end_time.text()) + typef = select_row[0]['type'] + if typef == 1 or typef == 2 or typef == 3: + if type == 1: + type = 1 + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_TWO_SECONDS_MERGE, + Constants.MSGBOX_TYPE_INFO) + elif type == 2: + if typef == 1: + type = 1 + elif typef == 2: + type = 3 + elif typef == 3: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_TWO_SECONDS_MERGE, + Constants.MSGBOX_TYPE_INFO) + elif type == 3: + if typef == 1: + type = 1 + elif typef == 2: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + elif typef == 3: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_TWO_SECONDS_MERGE, + Constants.MSGBOX_TYPE_INFO) + else: + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_TWO_SECONDS_WARNING, + Constants.MSGBOX_TYPE_INFO) + select_row = select_row[1:] + start_time = int(self.ui.lineEdit_start_time.text()) + end_time = int(self.ui.lineEdit_end_time.text()) + elif typef == 4 or typef == 5: PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_TWO_SECONDS_WARNING, Constants.MSGBOX_TYPE_INFO) select_row = select_row[1:] start_time = int(self.ui.lineEdit_start_time.text()) end_time = int(self.ui.lineEdit_end_time.text()) - if flagf == False and flagb == True: - if type == 1 or type == 2 or type == 3: - PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_BACK_TWO_SECONDS_MERGE, - Constants.MSGBOX_TYPE_INFO) - start_time = int(self.ui.lineEdit_start_time.text()) - end_time = select_row[-1]['endTime'] - elif type == 4 or type == 5: + elif flagf == False and flagb == True: + start_time = int(self.ui.lineEdit_start_time.text()) + end_time = select_row[-1]['endTime'] + typeb = select_row[-1]['type'] + if typeb == 1 or typeb == 2 or typeb == 3: + if type == 1: + type = 1 + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_BACK_TWO_SECONDS_MERGE, + Constants.MSGBOX_TYPE_INFO) + elif type == 2: + if typeb == 1: + type = 1 + elif typeb == 2: + type = 3 + elif typeb == 3: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_BACK_TWO_SECONDS_MERGE, + Constants.MSGBOX_TYPE_INFO) + elif type == 3: + if typeb == 1: + type = 1 + elif typeb == 2: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + elif typeb == 3: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_BACK_TWO_SECONDS_MERGE, + Constants.MSGBOX_TYPE_INFO) + else: + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_BACK_TWO_SECONDS_WARNING, + Constants.MSGBOX_TYPE_INFO) + select_row = select_row[:-1] + start_time = int(self.ui.lineEdit_start_time.text()) + end_time = int(self.ui.lineEdit_end_time.text()) + elif typeb == 4 or typeb == 5: PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_BACK_TWO_SECONDS_WARNING, Constants.MSGBOX_TYPE_INFO) select_row = select_row[:-1] start_time = int(self.ui.lineEdit_start_time.text()) end_time = int(self.ui.lineEdit_end_time.text()) - if flagf == True and flagb == True: - if type == 1 or type == 2 or type == 3: - PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_MERGE, - Constants.MSGBOX_TYPE_INFO) - start_time = select_row[0]['startTime'] - end_time = select_row[-1]['endTime'] - elif type == 4 or type == 5: + elif flagf == True and flagb == True: + start_time = select_row[0]['startTime'] + end_time = select_row[-1]['endTime'] + typef = select_row[0]['type'] + typeb = select_row[-1]['type'] + if typef == 1 or typef == 2 or typef == 3: + if type == 1: + type = 1 + elif type == 2: + if typef == 1: + type = 1 + elif typef == 2: + type = 3 + elif typef == 3: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + elif type == 3: + if typef == 1: + type = 1 + elif typef == 2: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + elif typef == 3: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + if type == 1: + type = 1 + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_MERGE, + Constants.MSGBOX_TYPE_INFO) + elif type == 2: + if typeb == 1: + type = 1 + elif typeb == 2: + type = 3 + elif typeb == 3: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_MERGE, + Constants.MSGBOX_TYPE_INFO) + elif type == 3: + if typeb == 1: + type = 1 + elif typeb == 2: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + elif typeb == 3: + if end_time - start_time >= (5 * Config["InputConfig"]["UseFreq"]): + type = 1 + else: + type = 3 + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_MERGE, + Constants.MSGBOX_TYPE_INFO) + else: + PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_WARNING, + Constants.MSGBOX_TYPE_INFO) + select_row = select_row[1:-1] + start_time = int(self.ui.lineEdit_start_time.text()) + end_time = int(self.ui.lineEdit_end_time.text()) + elif typef == 4 or typef == 5: PublicFunc.msgbox_output(self, Constants.ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_WARNING, Constants.MSGBOX_TYPE_INFO) select_row = select_row[1:-1] start_time = int(self.ui.lineEdit_start_time.text()) end_time = int(self.ui.lineEdit_end_time.text()) - if flagf == False and flagb == False: + elif flagf == False and flagb == False: start_time = int(self.ui.lineEdit_start_time.text()) end_time = int(self.ui.lineEdit_end_time.text()) - - for select_each_row in select_row: - self.data.df_Artifact_a = self.data.df_Artifact_a[self.data.df_Artifact_a['number'] != select_each_row["number"]] else: - select_row = [] - select_type = array([0, 0, 0, 0, 0]) - for index, row in self.data.df_Artifact_a.iterrows(): - value_startTime = row['startTime'] - value_endTime = row['endTime'] - value_type = row['type'] - if ((value_startTime <= int(self.ui.lineEdit_end_time.text()) <= value_endTime) or - (value_startTime <= int(self.ui.lineEdit_start_time.text()) <= value_endTime) or - (value_startTime >= int(self.ui.lineEdit_start_time.text()) and value_endTime <= int( - self.ui.lineEdit_end_time.text()))): - select_row.append(row.to_dict()) - if row['type'] == 1: - select_type[0] = select_type[0] + 1 - elif row['type'] == 2: - select_type[1] = select_type[1] + 1 - elif row['type'] == 3: - select_type[2] = select_type[2] + 1 - elif row['type'] == 4: - select_type[3] = select_type[3] + 1 - elif row['type'] == 5: - select_type[4] = select_type[4] + 1 + start_time = int(self.ui.lineEdit_start_time.text()) + end_time = int(self.ui.lineEdit_end_time.text()) - count = np_sum(select_type >= 1) - - if (count == 1 and nonzero(select_type)[0][0] + 1 != type): - reply = QMessageBox.question(self, '确认', - '{}{}'.format(Constants.ARTIFACT_LABEL_SINGLE_TYPE_NOT_EQUAL, - [d["number"] for d in select_row]), - QMessageBox.Yes | QMessageBox.No, - QMessageBox.No) - if reply == QMessageBox.No: - return - - start_time = int(self.ui.lineEdit_start_time.text()) - end_time = int(self.ui.lineEdit_end_time.text()) - - for select_each_row in select_row: - self.data.df_Artifact_a = self.data.df_Artifact_a[self.data.df_Artifact_a['number'] != select_each_row["number"]] - elif count == 0: - start_time = int(self.ui.lineEdit_start_time.text()) - end_time = int(self.ui.lineEdit_end_time.text()) - else: - raise ValueError("count值不存在") + for select_each_row in select_row: + self.data.df_Artifact_a = self.data.df_Artifact_a[self.data.df_Artifact_a['number'] != select_each_row["number"]] new_row = { 'number': 1, diff --git a/func/utils/Constants.py b/func/utils/Constants.py index 6a1362c..bbc2c04 100644 --- a/func/utils/Constants.py +++ b/func/utils/Constants.py @@ -347,14 +347,12 @@ class Constants: ARTIFACT_LABEL_JUMP_ARTIFACT: str = "跳转到体动" ARTIFACT_LABEL_RECOVER_SCALE: str = "尺度恢复" ARTIFACT_LABEL_MISS_ARGS: str = "打标参数未填写" - ARTIFACT_LABEL_MULTIPLE_ARTIFACT_COVER_OR_DELETE: str = "所选区域包含多种类型的体动，是否需要删除它们并使用新标注进行覆盖？所选区域中含有的体动序号：" - ARTIFACT_LABEL_FRONT_TWO_SECONDS_MERGE: str = "所选区域前2秒内有与当前标注相同类型的体动，将执行合并" - ARTIFACT_LABEL_BACK_TWO_SECONDS_MERGE: str = "所选区域后2秒内有与当前标注相同类型的体动，将执行合并" - ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_MERGE: str = "所选区域前后2秒内都有与当前标注相同类型的体动，将执行合并" - ARTIFACT_LABEL_FRONT_TWO_SECONDS_WARNING: str = "所选区域前2秒内有与当前标注相同类型的体动，仅进行提示" - ARTIFACT_LABEL_BACK_TWO_SECONDS_WARNING: str = "所选区域后2秒内有与当前标注相同类型的体动，仅进行提示" - ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_WARNING: str = "所选区域前后2秒内都有与当前标注相同类型的体动，仅进行提示" - ARTIFACT_LABEL_SINGLE_TYPE_NOT_EQUAL: str = "所选区域仅包含一种类型的体动，但其类型与当前标注的类型不匹配，是否需要删除它们并使用新标注覆盖？所选区域中含有的体动序号：" + ARTIFACT_LABEL_FRONT_TWO_SECONDS_MERGE: str = "所选区域前2秒内有体动，将根据规则执行合并" + ARTIFACT_LABEL_BACK_TWO_SECONDS_MERGE: str = "所选区域后2秒内有体动，将根据规则执行合并" + ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_MERGE: str = "所选区域前后2秒内都有体动，将根据规则执行合并" + ARTIFACT_LABEL_FRONT_TWO_SECONDS_WARNING: str = "所选区域前2秒内有体动，仅进行提示" + ARTIFACT_LABEL_BACK_TWO_SECONDS_WARNING: str = "所选区域后2秒内有体动，仅进行提示" + ARTIFACT_LABEL_FRONT_AND_BACK_TWO_SECONDS_WARNING: str = "所选区域前后2秒内都有体动，仅进行提示" ARTIFACT_LABEL_DELETE_ARTIFACT_SUCCESSFULLY: str = "体动被删除" ARTIFACT_LABEL_DELETE_ARTIFACT_FAILURE: str = "需要被删除的体动不存在" ARTIFACT_LABEL_ACTION_LABEL: str = f"标注体动({Params.ARTIFACT_LABEL_ACTION_LABEL_ARTIFACT_SHORTCUT_KEY})"