ecg_label_check/ecg_label_check.py

"""
@author:Yosa
@file:ecg_label_check.py
@email:2023025086@m.scnu.edu.cn
@time:2025/2/5
"""

import sys
from logging import NOTSET, getLogger, FileHandler, Formatter, StreamHandler, info, error, debug
from time import time, strftime, localtime

import pandas as pd
import numpy as np
from PyQt5 import QtGui
from PyQt5.QtGui import QFont, QDoubleValidator, QIntValidator
from matplotlib.ticker import FuncFormatter
from numpy import append
from matplotlib import use
from matplotlib import pyplot as plt, gridspec
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg, NavigationToolbar2QT
from datetime import datetime
from pathlib import Path
from PyQt5.QtCore import QCoreApplication, QTimer
from PyQt5.QtWidgets import QFileDialog, QMainWindow, QMessageBox, QButtonGroup, QApplication, QTableWidgetItem, \
    QLineEdit, QAction, QTableWidget, QDialog, QVBoxLayout, QLabel, QDialogButtonBox
from scipy.signal import butter, filtfilt, find_peaks

import append
from MainWindow import Ui_MainWindow

use("Qt5Agg")  # 声明使用QT5

# 设置日志
logger = getLogger()
logger.setLevel(NOTSET)
realtime = strftime('%Y%m%d', localtime(time()))
if not Path("logs").exists():
    Path("logs").mkdir(exist_ok=True)
fh = FileHandler(Path("logs") / (realtime + ".log"), mode='a')
fh.setLevel(NOTSET)
fh.setFormatter(Formatter("%(asctime)s: %(message)s"))
logger.addHandler(fh)

ch = StreamHandler()
ch.setLevel(NOTSET)
ch.setFormatter(Formatter("%(asctime)s: %(message)s"))
logger.addHandler(ch)
getLogger('matplotlib.font_manager').disabled = True
info("------------------------------------")
info("---------ecg_label_check.py---------")

class MainWindow(QMainWindow, Ui_MainWindow):

    # 全局变量初始化
    data1 = np.array([])
    data1 = data1.astype(np.float64)
    data2 = np.array([])
    data2 = data2.astype(np.float64)
    label1 = np.array([])
    label1 = label1.astype(np.int64)
    label2 = np.array([])
    label2 = label2.astype(np.int64)

    # 程序初始化操作
    def __init__(self):
        super(MainWindow, self).__init__()
        self.setupUi(self)

        # 设置画框
        self.figure = plt.figure(figsize=(12, 6), dpi=150)
        self.canvas = FigureCanvasQTAgg(self.figure)
        self.figToolbar = CustomNavigationToolbar(self.canvas, self)
        self.verticalLayout_canvas.addWidget(self.canvas)
        self.verticalLayout_canvas.addWidget(self.figToolbar)
        self.figToolbar._actions['home'].triggered.connect(self.toggle_home)
        self.figToolbar.action_Label_Single.triggered.connect(self.toggle_changeLabel_Single_mode)
        self.figToolbar.action_Label_Multiple.triggered.connect(self.toggle_changeLabel_Multiple_mode)

        # 画框子图初始化
        self.gs = gridspec.GridSpec(2, 1, height_ratios=[1, 1])
        self.figure.subplots_adjust(top=1, bottom=0, right=1, left=0, hspace=0, wspace=0)
        plt.margins(0, 0)
        plt.tight_layout()
        plt.xticks([])
        plt.yticks([])

        # 单选框组件分组
        self.buttonGroup_inputMode = QButtonGroup()
        self.buttonGroup_inputMode.addButton(self.radioButton_inputMode_auto)
        self.buttonGroup_inputMode.addButton(self.radioButton_inputMode_manual)
        self.buttonGroup_data1_preprocess = QButtonGroup()
        self.buttonGroup_data1_preprocess.addButton(self.radioButton_data1_fillterMode)
        self.buttonGroup_data2_preprocess = QButtonGroup()
        self.buttonGroup_data2_preprocess.addButton(self.radioButton_data2_fillterMode)
        self.buttonGroup_move_preset = QButtonGroup()
        self.buttonGroup_move_preset.addButton(self.radioButton_move_preset_1)
        self.buttonGroup_move_preset.addButton(self.radioButton_move_preset_2)
        self.buttonGroup_move_preset.addButton(self.radioButton_move_preset_3)
        self.buttonGroup_move_preset.addButton(self.radioButton_move_custom)

        # 设置表格属性
        self.tableWidget_label1.setHorizontalHeaderLabels(['label1'])
        self.tableWidget_label2.setHorizontalHeaderLabels(['label2'])
        self.tableWidget_label1.setEditTriggers(QTableWidget.NoEditTriggers)
        self.tableWidget_label2.setEditTriggers(QTableWidget.NoEditTriggers)
        self.tableWidget_label1.horizontalHeader().setStretchLastSection(True)
        self.tableWidget_label1.horizontalHeader().setSectionResizeMode(1)
        self.tableWidget_label2.horizontalHeader().setStretchLastSection(True)
        self.tableWidget_label2.horizontalHeader().setSectionResizeMode(1)

        # 槽函数连接控件初始化
        self.pushButton_savepath_open.clicked.connect(self.slot_btn_selectPath)
        self.pushButton_rootpath_open.clicked.connect(self.slot_btn_selectPath)
        self.pushButton_data1path_open.clicked.connect(self.slot_btn_selectPath)
        self.pushButton_data2path_open.clicked.connect(self.slot_btn_selectPath)
        self.pushButton_label1path_open.clicked.connect(self.slot_btn_selectPath)
        self.pushButton_label2path_open.clicked.connect(self.slot_btn_selectPath)
        self.pushButton_dataInput.clicked.connect(self.slot_btn_dataInput)
        self.pushButton_append.clicked.connect(self.slot_btn_append)
        self.pushButton_outputLabel.clicked.connect(self.slot_btn_outputLabel)
        self.comboBox_choose.currentTextChanged.connect(self.slot_comboBox_textChanged)
        self.radioButton_inputMode_auto.clicked.connect(self.slot_radioBtn_inputMode_auto)
        self.radioButton_inputMode_manual.clicked.connect(self.slot_radioBtn_inputMode_manual)
        self.pushButton_left_move.clicked.connect(self.slot_btn_left_move)
        self.pushButton_pause.clicked.connect(self.slot_btn_pause)
        self.pushButton_right_move.clicked.connect(self.slot_btn_right_move)
        self.radioButton_move_preset_1.clicked.connect(self.slot_radioBtn_move_preset_1)
        self.radioButton_move_preset_2.clicked.connect(self.slot_radioBtn_move_preset_2)
        self.radioButton_move_preset_3.clicked.connect(self.slot_radioBtn_move_preset_3)
        self.radioButton_move_custom.clicked.connect(self.slot_radioBtn_move_custom)
        self.tableWidget_label1.cellDoubleClicked.connect(self.slot_tableWidget_1_on_cell_double_clicked)
        self.tableWidget_label2.cellDoubleClicked.connect(self.slot_tableWidget_2_on_cell_double_clicked)

        # 初始化按钮快捷键
        self.pushButton_left_move.setShortcut(QCoreApplication.translate("MainWindow", "A"))
        self.pushButton_pause.setShortcut(QCoreApplication.translate("MainWindow", "S"))
        self.pushButton_right_move.setShortcut(QCoreApplication.translate("MainWindow", "D"))

        # 界面控件状态初始化
        self.lineEdit_rootpath.setEnabled(False)
        self.lineEdit_savepath.setEnabled(False)
        self.lineEdit_data1path.setEnabled(False)
        self.lineEdit_data2path.setEnabled(False)
        self.lineEdit_label1path.setEnabled(False)
        self.lineEdit_label2path.setEnabled(False)
        self.label_choose.setEnabled(False)
        self.comboBox_choose.setEnabled(False)
        self.groupBox_save.setEnabled(False)
        self.groupBox_data1.setEnabled(False)
        self.groupBox_data2.setEnabled(False)
        self.groupBox_label1.setEnabled(False)
        self.groupBox_label2.setEnabled(False)
        self.groupBox_autoplay.setEnabled(False)
        self.groupBox_labelDisplay.setEnabled(False)
        self.radioButton_inputMode_auto.setChecked(True)
        self.radioButton_data1_fillterMode.setChecked(True)
        self.radioButton_data2_fillterMode.setChecked(True)
        self.radioButton_move_preset_1.setChecked(True)
        self.pushButton_outputLabel.setEnabled(False)
        self.figToolbar.action_Label_Single.setEnabled(False)
        self.figToolbar.action_Label_Multiple.setEnabled(False)
        for action in self.figToolbar._actions.values():
            action.setEnabled(False)

        # 消息弹窗初始化
        self.msgBox = QMessageBox()
        self.msgBox.setWindowTitle("消息")

        # 自动播放参数初始化
        self.autoplay_mode = "pause"
        self.autoplay_moveLength = int(self.label_moveLength_preset_1.text())
        self.autoplay_maxRange = int(self.label_maxRange_preset_1.text())
        self.autoplay_moveSpeed = int(self.label_moveSpeed_preset_1.text())

        # 初始化自动播放定时器
        self.timer_autoplay = QTimer()
        self.timer_autoplay.timeout.connect(self.autoplay_move_xlim)

        # 绘图工具初始化
        self.points_y = np.array([])
        self.temp_point = None
        self.is_left_button_pressed = False
        self.is_right_button_pressed = False

        # 定义验证器，用于规范lineEdit的输入内容
        validator_double = QDoubleValidator(-1e100, 1e100, 10)
        validator_integer = QIntValidator(-2**31, 2**31 - 1)
        self.lineEdit_data1_fillterNum.setValidator(validator_integer)
        self.lineEdit_data2_fillterNum.setValidator(validator_integer)
        self.lineEdit_data1_fillterFrequency_min.setValidator(validator_double)
        self.lineEdit_data1_fillterFrequency_max.setValidator(validator_double)
        self.lineEdit_data2_fillterFrequency_min.setValidator(validator_double)
        self.lineEdit_data2_fillterFrequency_max.setValidator(validator_double)
        self.lineEdit_frequency.setValidator(validator_double)
        self.lineEdit_moveLength.setValidator(validator_integer)
        self.lineEdit_maxRange.setValidator(validator_double)
        self.lineEdit_moveSpeed.setValidator(validator_integer)
        self.lineEdit_findpeaks_min_interval.setValidator(validator_integer)
        self.lineEdit_findpeaks_min_height.setValidator(validator_double)

        self.textBrowser_update("程序启动成功，导入数据以开始任务")
        info("Program Started.")

    def slot_btn_selectPath(self):
        fileDialog = QFileDialog()
        if self.sender() == self.pushButton_rootpath_open:
            fileDialog.setFileMode(QFileDialog.Directory)
            fileDialog.setOption(QFileDialog.ShowDirsOnly, True)
            if fileDialog.exec_() == QFileDialog.Accepted:
                file_path = fileDialog.selectedFiles()[0]
                if not file_path:
                    error("Root Path not Exist...")
                    self.textBrowser_update("操作：根目录路径输入错误")
                    self.msgBox.setText("根目录路径输入错误")
                    self.msgBox.setIcon(QMessageBox.Critical)
                    self.msgBox.exec()
                    return

                self.lineEdit_rootpath.setText(file_path)
                file_path = Path(file_path)
                info("Loading Root Path...")
                self.data_and_label_Path = file_path / "label"

                if self.data_and_label_Path.exists():
                    # 检查路径下的hecg.txt和hRpeak.txt文件是否一一对应
                    data_files = sorted(self.data_and_label_Path.glob('*ecg.txt'))
                    label_files = sorted(self.data_and_label_Path.glob('*Rpeak.txt'))
                    if len(data_files) != len(label_files):
                        error("Data Quantity not Match...")
                        self.textBrowser_update(f"操作：数据导入错误，ecg.txt和Rpeak.txt文件数量不匹配，分别是{str(len(data_files))}和{str(len(label_files))}")
                        self.msgBox.setText(f"数据导入错误，ecg.txt和Rpeak.txt文件数量不匹配，分别是{str(len(data_files))}和{str(len(label_files))}")
                        self.msgBox.setIcon(QMessageBox.Critical)
                        self.msgBox.exec()
                        return
                    for data_file, label_file in zip(data_files, label_files):
                        if data_file.stem.replace('ecg', '') != label_file.stem.replace('Rpeak', ''):
                            self.textBrowser_update(f"操作：数据导入错误，文件名称存在不匹配，分别是{str(data_file.stem)}和{str(label_file.stem)}")
                            self.msgBox.setText(f"操作：数据导入错误，文件名称存在不匹配，分别是{str(data_file.stem)}和{str(label_file.stem)}")
                            self.msgBox.setIcon(QMessageBox.Critical)
                            self.msgBox.exec()
                            return
                    nums_for_comboBox = np.array([])
                    for data_file in data_files:
                        prefix = data_file.stem.replace('ecg', '')
                        nums_for_comboBox = np.append(nums_for_comboBox, prefix)
                    nums_for_comboBox = nums_for_comboBox.astype(np.int8)
                    nums_for_comboBox.sort()
                    nums_for_comboBox = nums_for_comboBox.astype(str)
                    self.comboBox_choose.clear()
                    self.comboBox_choose.addItems(nums_for_comboBox)
                    self.label_choose.setEnabled(True)
                    self.comboBox_choose.setEnabled(True)
                    self.groupBox_save.setEnabled(True)
                    self.groupBox_data1.setEnabled(True)
                    self.groupBox_data2.setEnabled(True)
                    self.groupBox_label1.setEnabled(True)
                    self.groupBox_label2.setEnabled(True)

                    # 更新数据路径文本框
                    self.lineEdit_savepath.setText(
                        str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "location_J.txt")
                    self.lineEdit_data1path.setText(
                        str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "ecg.txt")
                    self.lineEdit_data2path.setText(
                        str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "ecg.txt")
                    self.lineEdit_label1path.setText(
                        str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "Rpeak.txt")
                    self.lineEdit_label2path.setText(
                        str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "Rpeak.txt")

                    info("Successfully Loaded Root Path.")
                    self.textBrowser_update("操作：根目录路径选择成功")
                else:
                    info("Failed to Load Root Path.")
                    self.textBrowser_update("操作：根目录路径选择错误，缺乏必要数据文件夹")
                    self.msgBox.setText("根目录路径选择错误，缺乏必要数据文件夹")
                    self.msgBox.setIcon(QMessageBox.Critical)
                    self.msgBox.exec()
                    return
            else:
                info("Canceled Loading Root Path.")
                self.textBrowser_update("提示：根目录路径选择取消")
                self.msgBox.setText("根目录路径选择取消")
                self.msgBox.setIcon(QMessageBox.Warning)
                self.msgBox.exec()
        elif self.sender() == self.pushButton_savepath_open:
            fileDialog.setFileMode(QFileDialog.Directory)
            fileDialog.setOption(QFileDialog.ShowDirsOnly, True)
            if fileDialog.exec_() == QFileDialog.Accepted:
                file_path = fileDialog.selectedFiles()[0]
                self.lineEdit_savepath.setText(str(Path(file_path) / "newlocation_J.txt"))
                info("Successfully Loaded Save Path.")
                self.textBrowser_update("操作：保存路径选择成功")
            else:
                info("Canceled Loading Save Path.")
                self.textBrowser_update("提示：保存路径选择取消")
                self.msgBox.setText("保存路径选择取消")
                self.msgBox.setIcon(QMessageBox.Warning)
                self.msgBox.exec()
        else:
            fileDialog.setFileMode(QFileDialog.ExistingFile)
            fileDialog.setOption(QFileDialog.ReadOnly, True)
            fileDialog.setNameFilter("Text Files (*.txt)")
            if fileDialog.exec_() == QFileDialog.Accepted:
                file_path = fileDialog.selectedFiles()[0]
                if self.sender() == self.pushButton_data1path_open:
                    self.lineEdit_data1path.setText(file_path)
                elif self.sender() == self.pushButton_data2path_open:
                    self.lineEdit_data2path.setText(file_path)
                elif self.sender() == self.pushButton_label1path_open:
                    self.lineEdit_label1path.setText(file_path)
                elif self.sender() == self.pushButton_label2path_open:
                    self.lineEdit_label2path.setText(file_path)
                info("Successfully Loaded Data Path.")
                self.textBrowser_update("操作：数据路径选择成功")
            else:
                info("Canceled Loading Data Path.")
                self.textBrowser_update("提示：数据路径选择取消")
                self.msgBox.setText("数据路径选择取消")
                self.msgBox.setIcon(QMessageBox.Warning)
                self.msgBox.exec()

    def slot_btn_dataInput(self):
        if self.lineEdit_data1path.text() != "" or self.lineEdit_data2path.text() != "" or self.lineEdit_label1path.text() != "" or self.lineEdit_label2path.text() != "" or self.lineEdit_savepath.text() != "":
            info("Inputing Data...")
            self.textBrowser_update("提示：开始导入数据")

            # 导入数据
            self.label1 = np.array([])
            self.label2 = np.array([])
            file = open(str(self.lineEdit_data1path.text()), 'r')
            self.data1 = file.readlines()
            self.data1 = list(map(float, self.data1))
            self.data2 = self.data1
            self.data1 = self.data_preprocess(self.data1, int(self.lineEdit_data1_fillterNum.text()),
                                              float(self.lineEdit_data1_fillterFrequency_min.text()),
                                              float(self.lineEdit_data1_fillterFrequency_max.text()),
                                              float(self.lineEdit_frequency.text()))
            self.data2 = self.data_preprocess(self.data2, int(self.lineEdit_data2_fillterNum.text()),
                                              float(self.lineEdit_data2_fillterFrequency_min.text()),
                                              float(self.lineEdit_data2_fillterFrequency_max.text()),
                                              float(self.lineEdit_frequency.text()))
            file = open(str(self.lineEdit_label1path.text()), 'r')
            self.label1 = np.array(file.readlines())
            self.label2 = self.label1
            self.data1 = self.data1.astype(np.float64)
            self.data2 = self.data2.astype(np.float64)
            self.label1 = self.label1.astype(np.int64)
            self.label2 = self.label2.astype(np.int64)
            self.points_y_1 = [self.data1[x] for x in self.label1]
            self.points_y_2 = [self.data2[x] for x in self.label2]

            # 更新tableWidget
            self.tableWidget_label1.setRowCount(len(self.label1))
            for row, value in enumerate(self.label1):
                item = QTableWidgetItem(str(value).strip())
                self.tableWidget_label1.setItem(row, 0, item)
            self.tableWidget_label2.setRowCount(self.label2.__len__())
            for row, value in enumerate(self.label2):
                item = QTableWidgetItem(str(value).strip())
                self.tableWidget_label2.setItem(row, 0, item)
            self.groupBox_labelDisplay.setEnabled(True)

            # 更新界面
            self.groupBox_inputSetting.setEnabled(False)
            self.groupBox_autoplay.setEnabled(True)
            self.pushButton_outputLabel.setEnabled(True)
            self.figToolbar.action_Label_Single.setEnabled(True)
            self.figToolbar.action_Label_Multiple.setEnabled(True)
            MainWindow.setWindowTitle(self, QCoreApplication.translate("MainWindow",
                                                                       "ECG_Label_Check - Data Path: " + self.lineEdit_rootpath.text()))
            for action in self.figToolbar._actions.values():
                action.setEnabled(True)

            # 更新信息
            self.label_data1_length.setText(str(len(self.data1)))
            self.label_data2_length.setText(str(len(self.data2)))
            self.label_label1_length.setText(str(len(self.label1)))
            self.label_label2_length.setText(str(len(self.label2)))

            # 更新画框
            self.figure.clear()
            self.plot_data_and_label()

            # 连接画框中的槽函数
            # Connect mouse events
            self.canvas.mpl_connect('motion_notify_event', self.on_motion)

            # 保存路径文件是否存在的检查
            if not Path(self.lineEdit_savepath.text()).exists():
                Path(self.lineEdit_savepath.text()).touch()
            np.savetxt(Path(self.lineEdit_savepath.text()), self.label2, fmt='%d', newline='\n')

            info("Finished Input Data.")
            self.textBrowser_update("提示：导入数据完成")
        else:
            info("Failed to Input Data!")
            self.textBrowser_update("操作：导入数据失败")
            self.msgBox.setText("导入失败，请正确输入路径")
            self.msgBox.setIcon(QMessageBox.Critical)
            self.msgBox.exec()

    def slot_radioBtn_inputMode_auto(self):
        self.pushButton_rootpath_open.setEnabled(True)
        self.label_choose.setEnabled(False)
        self.comboBox_choose.setEnabled(False)
        self.groupBox_save.setEnabled(False)
        self.groupBox_data1.setEnabled(False)
        self.groupBox_data2.setEnabled(False)
        self.groupBox_label1.setEnabled(False)
        self.groupBox_label2.setEnabled(False)
        self.lineEdit_savepath.setText("")
        self.lineEdit_rootpath.setText("")
        self.lineEdit_data1path.setText("")
        self.lineEdit_data2path.setText("")
        self.lineEdit_label1path.setText("")
        self.lineEdit_label2path.setText("")
        self.comboBox_choose.clear()
        info("Switched to inputMode_auto.")
        self.textBrowser_update("操作：切换到输入模式-自动识别")

    def slot_radioBtn_inputMode_manual(self):
        self.pushButton_rootpath_open.setEnabled(False)
        self.label_choose.setEnabled(False)
        self.comboBox_choose.setEnabled(False)
        self.groupBox_save.setEnabled(True)
        self.groupBox_data1.setEnabled(True)
        self.groupBox_data2.setEnabled(True)
        self.groupBox_label1.setEnabled(True)
        self.groupBox_label2.setEnabled(True)
        self.lineEdit_rootpath.setText("")
        self.comboBox_choose.clear()
        info("Switched to inputMode_manual.")
        self.textBrowser_update("操作：切换到输入模式-手动输入")

    def slot_comboBox_textChanged(self):
        # 更新数据路径文本框
        self.lineEdit_savepath.setText(
            str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "location_J.txt")
        self.lineEdit_data1path.setText(
            str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "ecg.txt")
        self.lineEdit_data2path.setText(
            str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "ecg.txt")
        self.lineEdit_label1path.setText(
            str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "Rpeak.txt")
        self.lineEdit_label2path.setText(
            str(self.data_and_label_Path) + "\\" + self.comboBox_choose.currentText() + "Rpeak.txt")

    def slot_btn_left_move(self):
        self.autoplay_mode = "left"
        self.autoplay_xlim_start = int(self.ax0.get_xlim()[1] - self.autoplay_maxRange)
        self.autoplay_xlim_end = int(self.ax0.get_xlim()[1])
        if self.autoplay_xlim_end > self.data2.__len__():
            self.autoplay_xlim_start = int(self.data2.__len__() - self.autoplay_maxRange)
            self.autoplay_xlim_end = int(self.data2.__len__())
        self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
        self.canvas.draw()
        self.timer_autoplay.start(self.autoplay_moveSpeed)
        info("Started Autoplay left_mode.")
        self.textBrowser_update("操作：开始自动播放-向左")

    def slot_btn_pause(self):
        self.autoplay_mode = "pause"
        self.timer_autoplay.stop()
        info("Paused Autoplay.")
        self.textBrowser_update("操作：暂停自动播放")

    def slot_btn_right_move(self):
        self.autoplay_mode = "right"
        self.autoplay_xlim_start = int(self.ax0.get_xlim()[0])
        self.autoplay_xlim_end = int(self.ax0.get_xlim()[0] + self.autoplay_maxRange)
        if self.autoplay_xlim_start < 0:
            self.autoplay_xlim_start = 0
            self.autoplay_xlim_end = 0 + self.autoplay_maxRange
        self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
        self.canvas.draw()
        self.timer_autoplay.start(self.autoplay_moveSpeed)
        info("Started Autoplay right_mode.")
        self.textBrowser_update("操作：开始自动播放-向右")

    def slot_radioBtn_move_preset_1(self):
        self.autoplay_moveLength = int(self.label_moveLength_preset_1.text())
        self.autoplay_maxRange = int(self.label_maxRange_preset_1.text())
        self.autoplay_moveSpeed = int(self.label_moveSpeed_preset_1.text())
        if self.autoplay_mode != "pause":
            if self.autoplay_mode == "right":
                self.autoplay_xlim_start = int(self.ax0.get_xlim()[0])
                self.autoplay_xlim_end = int(self.ax0.get_xlim()[0] + self.autoplay_maxRange)
                self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
                self.canvas.draw()
            elif self.autoplay_mode == "left":
                self.autoplay_xlim_start = int(self.ax0.get_xlim()[1] - self.autoplay_maxRange)
                self.autoplay_xlim_end = int(self.ax0.get_xlim()[1])
                self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
                self.canvas.draw()
            self.timer_autoplay.start(self.autoplay_moveSpeed)
        info("Switched to Autoplay preset_1.")
        self.textBrowser_update("操作：切换到自动播放-预设1")

    def slot_radioBtn_move_preset_2(self):
        self.autoplay_moveLength = int(self.label_moveLength_preset_2.text())
        self.autoplay_maxRange = int(self.label_maxRange_preset_2.text())
        self.autoplay_moveSpeed = int(self.label_moveSpeed_preset_2.text())
        if self.autoplay_mode != "pause":
            if self.autoplay_mode == "right":
                self.autoplay_xlim_start = int(self.ax0.get_xlim()[0])
                self.autoplay_xlim_end = int(self.ax0.get_xlim()[0] + self.autoplay_maxRange)
                self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
                self.canvas.draw()
            elif self.autoplay_mode == "left":
                self.autoplay_xlim_start = int(self.ax0.get_xlim()[1] - self.autoplay_maxRange)
                self.autoplay_xlim_end = int(self.ax0.get_xlim()[1])
                self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
                self.canvas.draw()
            self.timer_autoplay.start(self.autoplay_moveSpeed)
        info("Switched to Autoplay preset_2.")
        self.textBrowser_update("操作：切换到自动播放-预设2")

    def slot_radioBtn_move_preset_3(self):
        self.autoplay_moveLength = int(self.label_moveLength_preset_3.text())
        self.autoplay_maxRange = int(self.label_maxRange_preset_3.text())
        self.autoplay_moveSpeed = int(self.label_moveSpeed_preset_3.text())
        if self.autoplay_mode != "pause":
            if self.autoplay_mode == "right":
                self.autoplay_xlim_start = int(self.ax0.get_xlim()[0])
                self.autoplay_xlim_end = int(self.ax0.get_xlim()[0] + self.autoplay_maxRange)
                self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
                self.canvas.draw()
            elif self.autoplay_mode == "left":
                self.autoplay_xlim_start = int(self.ax0.get_xlim()[1] - self.autoplay_maxRange)
                self.autoplay_xlim_end = int(self.ax0.get_xlim()[1])
                self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
                self.canvas.draw()
            self.timer_autoplay.start(self.autoplay_moveSpeed)
        info("Switched to Autoplay preset_3.")
        self.textBrowser_update("操作：切换到自动播放-预设3")

    def slot_radioBtn_move_custom(self):
        self.autoplay_moveLength = int(self.lineEdit_moveLength.text())
        self.autoplay_maxRange = int(self.lineEdit_maxRange.text())
        self.autoplay_moveSpeed = int(self.lineEdit_moveSpeed.text())
        if self.autoplay_mode != "pause":
            if self.autoplay_mode == "right":
                self.autoplay_xlim_start = int(self.ax0.get_xlim()[0])
                self.autoplay_xlim_end = int(self.ax0.get_xlim()[0] + self.autoplay_maxRange)
                self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
                self.canvas.draw()
            elif self.autoplay_mode == "left":
                self.autoplay_xlim_start = int(self.ax0.get_xlim()[1] - self.autoplay_maxRange)
                self.autoplay_xlim_end = int(self.ax0.get_xlim()[1])
                self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
                self.canvas.draw()
            self.timer_autoplay.start(self.autoplay_moveSpeed)
        info("Switched to Autoplay custom.")
        self.textBrowser_update("操作：切换到自动播放-自定义")
        self.msgBox.setText("自定义的输入参数未做任何检查，请斟酌输入参数，否则可能会导致程序异常")
        self.msgBox.setIcon(QMessageBox.Warning)
        self.msgBox.exec()

    def slot_tableWidget_1_on_cell_double_clicked(self, row, column):
        x = float(self.tableWidget_label1.item(row, column).text())
        self.ax0.set_xlim(x - 5000, x + 5000)
        self.annotation_tableWidget = self.ax0.annotate(f'x={int(x)}', xy=(int(x), self.ax0.get_ylim()[0]), xytext=(int(x), self.ax0.get_ylim()[0] + (self.ax0.get_ylim()[1] - self.ax0.get_ylim()[0]) * 0.1),
                         arrowprops=dict(facecolor='black', shrink=0.1))
        self.canvas.draw()
        info(f"Jumped to x_axis: {str(int(x))}.")
        self.textBrowser_update(f"操作：跳转到x坐标: {str(int(x))}")

    def slot_tableWidget_2_on_cell_double_clicked(self, row, column):
        x = float(self.tableWidget_label2.item(row, column).text())
        self.ax0.set_xlim(x - 5000, x + 5000)
        self.annotation_tableWidget = self.ax0.annotate(f'x={int(x)}', xy=(int(x), self.ax0.get_ylim()[0]), xytext=(int(x), self.ax0.get_ylim()[0] + (self.ax0.get_ylim()[1] - self.ax0.get_ylim()[0]) * 0.1),
                         arrowprops=dict(facecolor='black', shrink=0.1))
        self.canvas.draw()
        info(f"Jumped to x_axis: {str(int(x))}.")
        self.textBrowser_update(f"操作：跳转到x坐标: {str(int(x))}")

    def slot_btn_append(self):
        dialog = CustomMessageBox(self)
        reply = dialog.exec_()
        if reply == QDialog.Accepted:
            if self.lineEdit_rootpath.text() == "" or len(list((Path(self.lineEdit_rootpath.text()) / "label").rglob("*location_J.txt"))) == 0:
                info(f"*location_J.txt Files not Exist in the Directory.")
                self.textBrowser_update(f"错误：无法进行<片段合并>")
                self.msgBox.setText(f"目录下不存在*location_J.txt文件")
                self.msgBox.setIcon(QMessageBox.Warning)
                self.msgBox.exec()
                return
            final_Rpeak = append.Rpeak_Detection(self.lineEdit_rootpath.text(), int(dialog.lineEdit_fs.text()), int(dialog.lineEdit_th1.text()))
            pd.DataFrame(final_Rpeak).to_csv(Path(self.lineEdit_rootpath.text()) / "final_Rpeak.txt", index=False, header=None)
            info(f"Saved final_Rpeak to Directory {self.lineEdit_rootpath.text()}.")
            self.textBrowser_update(f"提示：保存final_Rpeak.txt成功至文件夹{self.lineEdit_rootpath.text()}")
            self.msgBox.setText(f"保存成功至{self.lineEdit_rootpath.text()}")
            self.msgBox.setIcon(QMessageBox.Information)
            self.msgBox.exec()
        else:
            self.textBrowser_update("提示：操作取消")

    def slot_btn_outputLabel(self):
        np.savetxt(Path(self.lineEdit_savepath.text()), self.label2, fmt='%d', newline='\n')
        info(f"Manually Saved Data to: {str(Path(self.lineEdit_savepath.text()))}.")
        self.textBrowser_update(f"操作：手动保存数据至{str(Path(self.lineEdit_savepath.text()))}")

    def data_preprocess(self, data, n, f1, f2, fs):
        f1 = f1 / (fs / 2.0)
        f2 = f2 / (fs / 2.0)
        b, a = butter(n, [f1, f2], btype = 'bandpass')
        data = np.array(filtfilt(b, a, data))
        return data

    def update_tableWidget_and_info(self):
        self.tableWidget_label2.setRowCount(self.label2.__len__())
        for row, value in enumerate(self.label2):
            item = QTableWidgetItem(str(value).strip())
            self.tableWidget_label2.setItem(row, 0, item)
        self.groupBox_labelDisplay.setEnabled(True)
        self.label_label2_length.setText(str(len(self.label2)))

    def textBrowser_update(self, message):
        self.textBrowser_infoOutput.append(str(datetime.now().strftime("%H:%M:%S")) + ": " + message)
        self.textBrowser_infoOutput.verticalScrollBar().setValue(self.textBrowser_infoOutput.verticalScrollBar().maximum())

    def autoplay_move_xlim(self):
        if self.autoplay_mode == "left" and self.autoplay_xlim_start < 0:
            self.autoplay_mode = "pause"
            self.timer_autoplay.stop()
            info("Paused Autoplay.")
            self.textBrowser_update("操作：暂停自动播放")
        elif self.autoplay_mode == "right" and self.autoplay_xlim_end > self.data2.__len__():
            self.autoplay_mode = "pause"
            self.timer_autoplay.stop()
            info("Paused Autoplay.")
            self.textBrowser_update("操作：暂停自动播放")
        else:
            if self.autoplay_mode == "right":
                self.autoplay_xlim_start += self.autoplay_moveLength
                self.autoplay_xlim_end += self.autoplay_moveLength
            elif self.autoplay_mode == "left":
                self.autoplay_xlim_start -= self.autoplay_moveLength
                self.autoplay_xlim_end -= self.autoplay_moveLength
            self.ax0.set_xlim(self.autoplay_xlim_start, self.autoplay_xlim_end)
            self.canvas.draw()

    def plot_data_and_label(self, ax_top=True, ax_bottom=True, ax_left=True, ax_right=True):
        self.ax0 = self.figure.add_subplot(self.gs[0])
        self.ax0.plot(self.data1, label="data1", color='b')
        self.ax0.plot(self.label1, self.points_y_1, 'ro', label='label1')
        self.ax0 = plt.gca()
        self.ax0.grid(True)
        self.ax0.spines["top"].set_visible(ax_top)
        self.ax0.spines["right"].set_visible(ax_right)
        self.ax0.spines["bottom"].set_visible(ax_bottom)
        self.ax0.spines["left"].set_visible(ax_left)
        self.ax0.tick_params(axis='x', colors='white')
        self.ax0.xaxis.set_major_formatter(FuncFormatter(lambda x, p: f"{x:.0f}"))

        self.ax1 = self.figure.add_subplot(self.gs[1], sharex=self.ax0, sharey=self.ax0)
        self.line_data2, = self.ax1.plot(self.data2, label="data2", color='b')
        self.point_label2, = self.ax1.plot(self.label2, self.points_y_2, 'ro', label='label2')
        self.ax1 = plt.gca()
        self.ax1.grid(True)
        self.ax1.spines["top"].set_visible(ax_top)
        self.ax1.spines["right"].set_visible(ax_right)
        self.ax1.spines["bottom"].set_visible(ax_bottom)
        self.ax1.spines["left"].set_visible(ax_left)

        self.ax0.autoscale(False)
        self.ax0.callbacks.connect('xlim_changed', lambda ax: self.on_xlim_change(ax))

    def toggle_home(self):
        if self.timer_autoplay.isActive() == True:
            self.autoplay_mode = "pause"
            self.timer_autoplay.stop()
            info("Paused Autoplay.")
            self.textBrowser_update("操作：暂停自动播放")
        self.ax0.autoscale(True)
        self.ax0.relim()
        self.ax0.autoscale_view()
        self.canvas.draw()
        self.ax0.autoscale(False)
        self.textBrowser_update("操作：尺度恢复")

    def toggle_changeLabel_Single_mode(self, state):
        if state:
            self.deactivate_figToolbar_buttons()
            self.figToolbar.action_Label_Single.setChecked(True)
            self.figToolbar.action_Label_Multiple.setChecked(False)
            self.figToolbar.cid_mouse_press = self.canvas.mpl_connect('button_press_event', self.on_click)
            self.figToolbar.cid_mouse_release = self.canvas.mpl_connect('button_release_event', self.on_release)
        else:
            if self.figToolbar.cid_mouse_press is not None:
                self.canvas.mpl_disconnect(self.figToolbar.cid_mouse_press)
                self.figToolbar.cid_mouse_press = None
            if self.figToolbar.cid_mouse_release is not None:
                self.canvas.mpl_disconnect(self.figToolbar.cid_mouse_release)
                self.figToolbar.cid_mouse_release = None

    def toggle_changeLabel_Multiple_mode(self, state):
        if state:
            self.deactivate_figToolbar_buttons()
            self.figToolbar.action_Label_Single.setChecked(False)
            self.figToolbar.action_Label_Multiple.setChecked(True)
            self.figToolbar.cid_mouse_press = self.canvas.mpl_connect('button_press_event', self.on_click)
            self.figToolbar.cid_mouse_release = self.canvas.mpl_connect('button_release_event', self.on_release)
            self.figToolbar.cid_mouse_hold = self.canvas.mpl_connect('motion_notify_event', self.on_hold)
        else:
            if self.figToolbar.cid_mouse_press is not None:
                self.canvas.mpl_disconnect(self.figToolbar.cid_mouse_press)
                self.figToolbar.cid_mouse_press = None
            if self.figToolbar.cid_mouse_release is not None:
                self.canvas.mpl_disconnect(self.figToolbar.cid_mouse_release)
                self.figToolbar.cid_mouse_release = None
            if self.figToolbar.cid_mouse_hold:
                self.canvas.mpl_disconnect(self.figToolbar.cid_mouse_hold)
                self.figToolbar.cid_mouse_hold = None

    def deactivate_figToolbar_buttons(self):
        for action in self.figToolbar._actions.values():
            if action.isChecked() == True:
                if action == self.figToolbar._actions['pan']:
                    self.figToolbar.pan()
                if action == self.figToolbar._actions['zoom']:
                    self.figToolbar.zoom()

    def on_click(self, event):
        if self.figToolbar.action_Label_Single.isChecked():
            if event.button == 1:
                self.is_left_button_pressed = True
                self.add_temp_point(event)
            elif event.button == 3:
                self.is_right_button_pressed = True
                self.remove_temp_point(event)
        elif self.figToolbar.action_Label_Multiple.isChecked():
            if event.button == 1 or event.button == 3:  # 左键或右键
                if event.button == 1:
                    self.is_left_button_pressed = True
                elif event.button == 3:
                    self.is_right_button_pressed = True
                self.figToolbar.rect_start_x = event.xdata
                # 如果矩形patch已存在，先移除
                if self.figToolbar.rect_patch is not None:
                    self.figToolbar.rect_patch.remove()
                    self.figToolbar.rect_patch = None
                self.canvas.draw()

    def on_release(self, event):
        if self.figToolbar.action_Label_Single.isChecked():
            if event.button == 1 and self.is_left_button_pressed:
                self.is_left_button_pressed = False
                if self.temp_point:
                    self.label2 = np.append(self.label2, self.temp_point[0])
                    self.points_y_2 = np.append(self.points_y_2, self.temp_point[1])
                    info(f"Added point: ({str(self.temp_point[0])},{str(self.temp_point[1])}).")
                    self.textBrowser_update(f"操作：新增点：({str(self.temp_point[0])},{str(self.temp_point[1])})")
                    self.temp_point = None
                    self.redraw_plot()
            elif event.button == 3 and self.is_right_button_pressed:
                self.is_right_button_pressed = False
                if self.temp_point:
                    distances = np.abs(np.array(self.label2) - self.temp_point[0])
                    if len(distances) > 0:
                        idx = distances.argmin()
                        info(f"Removed point {idx + 1}: ({str(self.label2[idx])},{str(self.points_y_2[idx])}).")
                        self.textBrowser_update(f"操作：删除第{idx + 1}点：({str(self.label2[idx])},{str(self.points_y_2[idx])})")
                        self.label2 = np.delete(self.label2, idx)
                        self.points_y_2 = np.delete(self.points_y_2, idx)
                        self.temp_point = None
                        self.redraw_plot()
            self.label2.sort()
            self.points_y_2 = [self.data2[x] for x in self.label2]
            self.update_tableWidget_and_info()
            np.savetxt(Path(self.lineEdit_savepath.text()), self.label2, fmt='%d', newline='\n')
        elif self.figToolbar.action_Label_Multiple.isChecked():
            if self.figToolbar.rect_start_x is not None:
                self.figToolbar.rect_end_x = event.xdata
                if self.figToolbar.rect_start_x is not None and self.figToolbar.rect_end_x is not None:
                    if self.figToolbar.rect_start_x < self.figToolbar.rect_end_x:
                        rect_left = self.figToolbar.rect_start_x
                        rect_right = self.figToolbar.rect_end_x
                    elif self.figToolbar.rect_start_x > self.figToolbar.rect_end_x:
                        rect_left = self.figToolbar.rect_end_x
                        rect_right = self.figToolbar.rect_start_x
                    else:
                        rect_left = self.figToolbar.rect_start_x
                        rect_right = self.figToolbar.rect_start_x
                else:
                    rect_left = self.figToolbar.rect_start_x
                    rect_right = self.figToolbar.rect_start_x
                if event.button == 1 and self.is_left_button_pressed:
                    self.is_left_button_pressed = False
                    if rect_left < 0:
                        rect_left = 0
                    if rect_right >= len(self.data2):
                        rect_right = len(self.data2) - 1
                    selected_area_for_add_points = self.data2[int(rect_left):int(rect_right)]
                    peaks_idx, _ = find_peaks(selected_area_for_add_points, height=float(self.lineEdit_findpeaks_min_height.text()), distance=float(self.lineEdit_findpeaks_min_interval.text()))
                    peaks_idx = peaks_idx + int(rect_left)
                    if len(peaks_idx) != 0:
                        info(f"Added {len(peaks_idx)} points.")
                        self.textBrowser_update(f"操作：新增{len(peaks_idx)}个点")
                    else:
                        info(f"No point added in selected area.")
                        self.textBrowser_update(f"操作：所选区间内无新增点")
                    self.label2 = np.append(self.label2, peaks_idx)
                    self.points_y_2 = np.append(self.points_y_2, self.data2[peaks_idx])
                    self.redraw_plot()
                elif event.button == 3 and self.is_right_button_pressed:
                    self.is_right_button_pressed = False
                    left_label2_to_delete = self.label2 - rect_left
                    right_label2_to_delete = self.label2 - rect_right
                    self.left_label2_to_delete_idx = len(left_label2_to_delete[left_label2_to_delete < 0])
                    self.right_label2_to_delete_idx = len(right_label2_to_delete[right_label2_to_delete < 0])
                    if self.left_label2_to_delete_idx != self.right_label2_to_delete_idx:
                        info(f"Removed points from {self.left_label2_to_delete_idx + 1} to {self.right_label2_to_delete_idx}.")
                        self.textBrowser_update(f"操作：删除第{self.left_label2_to_delete_idx + 1}到{self.right_label2_to_delete_idx}点")
                    else:
                        info(f"No point to delete in selected area.")
                        self.textBrowser_update(f"操作：所选区间内无删除点")
                    self.label2 = np.delete(self.label2, np.arange(self.left_label2_to_delete_idx, self.right_label2_to_delete_idx))
                    self.points_y_2 = np.delete(self.points_y_2, np.arange(self.left_label2_to_delete_idx, self.right_label2_to_delete_idx))
                    self.redraw_plot()
                self.figToolbar.rect_start_x = None
                self.figToolbar.rect_end_x = None
                self.label2.sort()
                self.points_y_2 = [self.data2[x] for x in self.label2]
                self.update_tableWidget_and_info()
                np.savetxt(Path(self.lineEdit_savepath.text()), self.label2, fmt='%d', newline='\n')
                # 移除矩形patch
                if self.figToolbar.rect_patch is not None:
                    self.figToolbar.rect_patch.remove()
                    self.figToolbar.rect_patch = None
                self.canvas.draw()

    def on_hold(self, event):
        if self.figToolbar.rect_start_x is not None and event.xdata is not None:
            self.figToolbar.rect_end_x = event.xdata

            # 如果矩形patch不存在，则创建一个新的
            if self.figToolbar.rect_patch is None:
                if self.is_left_button_pressed:
                    self.figToolbar.rect_patch = plt.Rectangle((0, 0), 1, 1, fill=True, alpha=0.2, color='#ff00ff')
                elif self.is_right_button_pressed:
                    self.figToolbar.rect_patch = plt.Rectangle((0, 0), 1, 1, fill=True, alpha=0.2, color='r')
                self.ax1.add_patch(self.figToolbar.rect_patch)

            # 更新矩形patch的位置和大小
            x_start = self.figToolbar.rect_start_x
            x_end = self.figToolbar.rect_end_x
            y_min, y_max = self.ax1.get_ylim()
            self.figToolbar.rect_patch.set_xy((min(x_start, x_end), y_min))
            self.figToolbar.rect_patch.set_width(abs(x_end - x_start))
            self.figToolbar.rect_patch.set_height(y_max - y_min)

            self.canvas.draw()

    def on_motion(self, event):
        if event.inaxes:
            # Clear previous reference lines and temporary points
            for line in self.ax0.lines[1:]:
                if line.get_label() == 'vline' or line.get_label() == 'hline' or line.get_label() == 'temp_point':
                    line.remove()
            for line in self.ax1.lines[1:]:
                if line.get_label() == 'vline' or line.get_label() == 'hline' or line.get_label() == 'temp_point':
                    line.remove()

            # Draw vertical and horizontal reference lines
            self.ax0.axvline(event.xdata, color='gray', linestyle='--', label='vline')
            self.ax0.axhline(event.ydata, color='gray', linestyle='--', label='hline')
            self.ax1.axvline(event.xdata, color='gray', linestyle='--', label='vline')
            self.ax1.axhline(event.ydata, color='gray', linestyle='--', label='hline')

            # Draw temporary point if left button is pressed
            if self.is_left_button_pressed:
                self.add_temp_point(event)
            elif self.is_right_button_pressed:
                self.remove_temp_point(event)

            self.canvas.draw()

    def add_temp_point(self, event):
        if self.figToolbar.action_Label_Single.isChecked():
            # Find the closest x value on the curve
            idx = np.abs(np.arange(len(self.data2)) - event.xdata).argmin()
            x_point = np.arange(len(self.data2))[idx]
            y_point = self.data2[idx]

            # Store the temporary point
            self.temp_point = (x_point, y_point)

            # Plot the temporary point
            self.ax1.plot(x_point, y_point, marker='o', color='#ff00ff', label='temp_point')
            self.ax1.plot(x_point, y_point, marker='x', color='#ff00ff', markersize=30, label='temp_point')

    def remove_temp_point(self, event):
        if self.figToolbar.action_Label_Single.isChecked():
            if self.label2.any():
                # Find the closest point to the current x coordinate
                distances = np.abs(np.array(self.label2) - event.xdata)
                idx = distances.argmin()
                x_point = self.label2[idx]
                y_point = self.points_y_2[idx]

                # Store the temporary point to be removed
                self.temp_point = (x_point, y_point)

                # Plot the temporary point to be removed
                self.ax1.plot(x_point, y_point, marker='x', color='r', markersize=30, label='temp_point')

    def redraw_plot(self):
        self.point_label2.remove()
        self.point_label2, = self.ax1.plot(self.label2, self.points_y_2, 'ro', label='label2')
        self.canvas.draw()

    def on_xlim_change(self, event_ax):
        try:
            if self.annotation_tableWidget:
                self.annotation_tableWidget.remove()
        except AttributeError:
            pass
        self.annotation_tableWidget = None
        self.canvas.draw()

class CustomNavigationToolbar(NavigationToolbar2QT):
    def __init__(self, canvas, parent):
        super().__init__(canvas, parent)
        # 初始化画框工具栏
        self.action_Label_Single = QAction('逐一更改标签(Z)', self)
        self.action_Label_Single.setFont(QFont("黑体", 14))
        self.action_Label_Single.setCheckable(True)
        self.action_Label_Single.setShortcut(QCoreApplication.translate("MainWindow", "Z"))
        self.action_Label_Multiple = QAction('批量更改标签(X)', self)
        self.action_Label_Multiple.setFont(QFont("黑体", 14))
        self.action_Label_Multiple.setCheckable(True)
        self.action_Label_Multiple.setShortcut(QCoreApplication.translate("MainWindow", "X"))
        self.insertAction(self._actions['pan'], self.action_Label_Single)
        self.insertAction(self._actions['pan'], self.action_Label_Multiple)

        self._actions['pan'].setShortcut(QCoreApplication.translate("MainWindow", "C"))

        # 用于存储事件连接ID
        self.cid_mouse_press = None
        self.cid_mouse_release = None
        self.cid_mouse_hold = None

        # 初始化矩形选择区域
        self.rect_start_x = None
        self.rect_end_x = None
        self.rect_patch = None  # 用于绘制矩形的patch

    def home(self, *args):
        pass

    def zoom(self, *args):
        super().zoom(*args)
        self.deactivate_figToorbar_changeLabel_mode()

    def pan(self, *args):
        super().pan(*args)
        self.deactivate_figToorbar_changeLabel_mode()

    def deactivate_figToorbar_changeLabel_mode(self):
        if self.action_Label_Single.isChecked():
            self.action_Label_Single.setChecked(False)
            if self.cid_mouse_press is not None:
                self.canvas.mpl_disconnect(self.cid_mouse_press)
                self.cid_mouse_press = None
            if self.cid_mouse_release is not None:
                self.canvas.mpl_disconnect(self.cid_mouse_release)
                self.cid_mouse_release = None
        elif self.action_Label_Multiple.isChecked():
            self.action_Label_Multiple.setChecked(False)
            if self.cid_mouse_press is not None:
                self.canvas.mpl_disconnect(self.cid_mouse_press)
                self.cid_mouse_press = None
            if self.cid_mouse_release is not None:
                self.canvas.mpl_disconnect(self.cid_mouse_release)
                self.cid_mouse_release = None
            if self.cid_mouse_hold is not None:
                self.canvas.mpl_disconnect(self.cid_mouse_hold)
                self.cid_mouse_hold = None

class CustomMessageBox(QDialog):
    def __init__(self, parent=None):
        super().__init__(parent)

        self.resize(300, 300)
        self.setWindowTitle("警告：确认操作")
        layout = QVBoxLayout()
        self.label = QLabel("是否执行<片段合并>？请在执行前确保此份数据的所有片段都已被打标！请输入相应参数后执行任务。")
        self.label1 = QLabel(" ")
        self.label_fs = QLabel("信号采样率(Hz)：")
        self.label_th1 = QLabel("寻峰阈值(个)：")
        self.lineEdit_fs = QLineEdit()
        self.lineEdit_th1 = QLineEdit()

        font = QtGui.QFont()
        font.setFamily("黑体")
        font.setPointSize(14)
        self.label.setFont(font)
        self.label1.setFont(font)
        self.label_fs.setFont(font)
        self.label_th1.setFont(font)
        self.lineEdit_fs.setFont(font)
        self.lineEdit_th1.setFont(font)

        self.label.setWordWrap(True)
        self.label.setFixedWidth(300)
        self.lineEdit_fs.setText("1000")
        self.lineEdit_th1.setText("130")

        layout.addWidget(self.label)
        layout.addWidget(self.label1)
        layout.addWidget(self.label_fs)
        layout.addWidget(self.lineEdit_fs)
        layout.addWidget(self.label_th1)
        layout.addWidget(self.lineEdit_th1)

        self.button_box = QDialogButtonBox(QDialogButtonBox.Yes | QDialogButtonBox.No)
        self.button_box.setFont(font)
        self.button_box.accepted.connect(self.accept)
        self.button_box.rejected.connect(self.reject)
        layout.addWidget(self.button_box)
        self.setLayout(layout)

# 主函数
if __name__ == '__main__':
    app = QApplication(sys.argv)
    mainWindow = MainWindow()
    mainWindow.show()
    sys.exit(app.exec_())