203 lines
7.8 KiB
Python
203 lines
7.8 KiB
Python
from pathlib import Path
|
|
|
|
from PySide6.QtWidgets import QMainWindow, QMessageBox, QFileDialog
|
|
from matplotlib import use
|
|
from yaml import dump, load, FullLoader
|
|
|
|
from func.utils.PublicFunc import PublicFunc
|
|
from ui.MainWindow.MainWindow_menu import Ui_Signal_Label
|
|
|
|
from func.Module_approximately_align import MainWindow_approximately_align
|
|
from func.Module_preprocess import MainWindow_preprocess
|
|
from func.Module_detect_Jpeak import MainWindow_detect_Jpeak
|
|
from func.Module_detect_Rpeak import MainWindow_detect_Rpeak
|
|
from func.Module_label_check import MainWindow_label_check
|
|
from func.Module_precisely_align import MainWindow_precisely_align
|
|
from func.Module_cut_PSG import MainWindow_cut_PSG
|
|
|
|
from func.utils.Constants import Constants, ConfigParams
|
|
|
|
|
|
use("QtAgg")
|
|
|
|
|
|
Config = {
|
|
|
|
}
|
|
|
|
|
|
class MainWindow(QMainWindow, Ui_Signal_Label):
|
|
|
|
def __init__(self):
|
|
super(MainWindow, self).__init__()
|
|
self.ui = Ui_Signal_Label()
|
|
self.ui.setupUi(self)
|
|
|
|
self.setFixedSize(720, 1080) # 设置固定大小，禁止缩放
|
|
|
|
# 消息弹窗初始化
|
|
self.msgBox = QMessageBox()
|
|
self.msgBox.setWindowTitle(Constants.MAINWINDOW_MSGBOX_TITLE)
|
|
|
|
self.__read_config__()
|
|
self.ui.plainTextEdit_root_path.setPlainText(Config["Path"]["Root"])
|
|
self.seek_sampID(Path(Config["Path"]["Root"]) / Path(ConfigParams.PUBLIC_PATH_ORGBCG_TEXT))
|
|
|
|
self.approximately_align = None
|
|
self.preprocess = None
|
|
self.detect_Jpeak = None
|
|
self.detect_Rpeak = None
|
|
self.label_check = None
|
|
self.precisely_align = None
|
|
self.cut_PSG = None
|
|
|
|
# 绑定槽函数
|
|
self.ui.pushButton_open.clicked.connect(self.__slot_btn_open__)
|
|
self.ui.pushButton_approximately_align.clicked.connect(self.__slot_btn_approximately_align__)
|
|
self.ui.pushButton_preprocess_BCG.clicked.connect(self.__slot_btn_preprocess__)
|
|
self.ui.pushButton_preprocess_ECG.clicked.connect(self.__slot_btn_preprocess__)
|
|
self.ui.pushButton_detect_Jpeak.clicked.connect(self.__slot_btn_detect_Jpeak__)
|
|
self.ui.pushButton_detect_Rpeak.clicked.connect(self.__slot_btn_detect_Rpeak__)
|
|
self.ui.pushButton_label_check_BCG.clicked.connect(self.__slot_btn_label_check__)
|
|
self.ui.pushButton_label_check_ECG.clicked.connect(self.__slot_btn_label_check__)
|
|
self.ui.pushButton_precisely_align.clicked.connect(self.__slot_btn_precisely_align__)
|
|
self.ui.pushButton_cut_PSG.clicked.connect(self.__slot_btn_cut_PSG__)
|
|
|
|
@staticmethod
|
|
def __read_config__():
|
|
if not Path(ConfigParams.PUBLIC_CONFIG_FILE_PATH).exists():
|
|
Path(ConfigParams.PUBLIC_CONFIG_FILE_PATH).parent.mkdir(parents=True, exist_ok=True)
|
|
with open(ConfigParams.PUBLIC_CONFIG_FILE_PATH, "w") as f:
|
|
dump(ConfigParams.PUBLIC_CONFIG_NEW_CONTENT, f)
|
|
|
|
with open(ConfigParams.PUBLIC_CONFIG_FILE_PATH, "r") as f:
|
|
file_config = load(f.read(), Loader=FullLoader)
|
|
Config.update(file_config)
|
|
|
|
@staticmethod
|
|
def __write_config__():
|
|
with open(Path(ConfigParams.PUBLIC_CONFIG_FILE_PATH), "w") as f:
|
|
dump(Config, f)
|
|
|
|
def __slot_btn_open__(self):
|
|
file_dialog = QFileDialog()
|
|
file_dialog.setFileMode(QFileDialog.Directory)
|
|
file_dialog.setOption(QFileDialog.ShowDirsOnly, True)
|
|
if file_dialog.exec_() == QFileDialog.Accepted:
|
|
self.seek_sampID(Path(file_dialog.selectedFiles()[0]) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT)
|
|
self.ui.plainTextEdit_root_path.setPlainText(file_dialog.selectedFiles()[0])
|
|
|
|
# 修改配置
|
|
Config["Path"]["Root"] = str(file_dialog.selectedFiles()[0])
|
|
self.__write_config__()
|
|
else:
|
|
PublicFunc.msgbox_output(self, Constants.OPERATION_CANCELED, Constants.MSGBOX_TYPE_INFO)
|
|
|
|
def __slot_btn_approximately_align__(self):
|
|
self.approximately_align = MainWindow_approximately_align()
|
|
root_path = self.ui.plainTextEdit_root_path.toPlainText()
|
|
sampID = self.ui.comboBox_sampID.currentText()
|
|
if not self.check_root_path():
|
|
return
|
|
if not self.check_sampID():
|
|
return
|
|
self.approximately_align.show(root_path, int(sampID))
|
|
|
|
def __slot_btn_preprocess__(self):
|
|
self.preprocess = MainWindow_preprocess()
|
|
|
|
sender = self.sender()
|
|
root_path = self.ui.plainTextEdit_root_path.toPlainText()
|
|
sampID = self.ui.comboBox_sampID.currentText()
|
|
if sender == self.ui.pushButton_preprocess_BCG:
|
|
mode = "BCG"
|
|
elif sender == self.ui.pushButton_preprocess_ECG:
|
|
mode = "ECG"
|
|
else:
|
|
raise ValueError("模式不存在")
|
|
if not self.check_root_path():
|
|
return
|
|
if not self.check_sampID():
|
|
return
|
|
self.preprocess.show(mode, root_path, int(sampID))
|
|
|
|
def __slot_btn_detect_Jpeak__(self):
|
|
self.detect_Jpeak = MainWindow_detect_Jpeak()
|
|
root_path = self.ui.plainTextEdit_root_path.toPlainText()
|
|
sampID = self.ui.comboBox_sampID.currentText()
|
|
if not self.check_root_path():
|
|
return
|
|
if not self.check_sampID():
|
|
return
|
|
self.detect_Jpeak.show(root_path, int(sampID))
|
|
|
|
def __slot_btn_detect_Rpeak__(self):
|
|
self.detect_Rpeak = MainWindow_detect_Rpeak()
|
|
root_path = self.ui.plainTextEdit_root_path.toPlainText()
|
|
sampID = self.ui.comboBox_sampID.currentText()
|
|
if not self.check_root_path():
|
|
return
|
|
if not self.check_sampID():
|
|
return
|
|
self.detect_Rpeak.show(root_path, int(sampID))
|
|
|
|
def __slot_btn_label_check__(self):
|
|
self.label_check = MainWindow_label_check()
|
|
|
|
sender = self.sender()
|
|
root_path = self.ui.plainTextEdit_root_path.toPlainText()
|
|
sampID = self.ui.comboBox_sampID.currentText()
|
|
|
|
if not self.check_root_path():
|
|
return
|
|
if not self.check_sampID():
|
|
return
|
|
|
|
if sender == self.ui.pushButton_label_check_BCG:
|
|
mode = "BCG"
|
|
elif sender == self.ui.pushButton_label_check_ECG:
|
|
mode = "ECG"
|
|
else:
|
|
raise ValueError("模式不存在")
|
|
self.label_check.show(mode, root_path, int(sampID))
|
|
|
|
def __slot_btn_precisely_align__(self):
|
|
self.precisely_align = MainWindow_precisely_align()
|
|
root_path = self.ui.plainTextEdit_root_path.toPlainText()
|
|
sampID = self.ui.comboBox_sampID.currentText()
|
|
if not self.check_root_path():
|
|
return
|
|
if not self.check_sampID():
|
|
return
|
|
self.precisely_align.show(root_path, int(sampID))
|
|
|
|
def __slot_btn_cut_PSG__(self):
|
|
self.cut_PSG = MainWindow_cut_PSG()
|
|
root_path = self.ui.plainTextEdit_root_path.toPlainText()
|
|
sampID = self.ui.comboBox_sampID.currentText()
|
|
if not self.check_root_path():
|
|
return
|
|
if not self.check_sampID():
|
|
return
|
|
self.cut_PSG.show(root_path, int(sampID))
|
|
|
|
def seek_sampID(self, path):
|
|
if not Path(path).exists():
|
|
PublicFunc.msgbox_output(self, Constants.MAINWINDOW_ROOT_PATH_NOT_EXIST, Constants.MSGBOX_TYPE_ERROR)
|
|
return
|
|
|
|
sub_folders = [item.name for item in Path(path).iterdir() if item.is_dir()]
|
|
self.ui.comboBox_sampID.addItems(sub_folders)
|
|
|
|
def check_root_path(self):
|
|
if self.ui.plainTextEdit_root_path.toPlainText() == Constants.STRING_IS_EMPTY:
|
|
PublicFunc.msgbox_output(self, Constants.MAINWINDOW_ROOT_PATH_NOT_EXIST, Constants.MSGBOX_TYPE_ERROR)
|
|
return False
|
|
return True
|
|
|
|
def check_sampID(self):
|
|
if self.ui.comboBox_sampID.currentText() == Constants.STRING_IS_EMPTY:
|
|
PublicFunc.msgbox_output(self, Constants.MAINWINDOW_SAMPID_EMPTY, Constants.MSGBOX_TYPE_ERROR)
|
|
return False
|
|
return True
|