完成了<预处理>的信号重采样部分,当输入输出采样率不同时,将执行重采样

This commit is contained in:
Yorusora
2025-04-28 16:57:36 +08:00
parent 2a13ceac39
commit 2d2a2f6495
4 changed files with 74 additions and 9 deletions

View File

@ -8,6 +8,7 @@ from matplotlib.backends.backend_qt import NavigationToolbar2QT
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg
from overrides import overrides from overrides import overrides
from pandas import read_csv, DataFrame from pandas import read_csv, DataFrame
from scipy.signal import resample
from yaml import dump, load, FullLoader from yaml import dump, load, FullLoader
from func.utils.PublicFunc import PublicFunc from func.utils.PublicFunc import PublicFunc
@ -54,6 +55,7 @@ class SettingWindow(QMainWindow):
self.__read_config__() self.__read_config__()
self.ui.spinBox_input_freq.valueChanged.connect(self.__update_ui__) self.ui.spinBox_input_freq.valueChanged.connect(self.__update_ui__)
self.ui.spinBox_output_freq.valueChanged.connect(self.__update_ui__)
self.ui.pushButton_confirm.clicked.connect(self.__write_config__) self.ui.pushButton_confirm.clicked.connect(self.__write_config__)
self.ui.pushButton_cancel.clicked.connect(self.__rollback_config__) self.ui.pushButton_cancel.clicked.connect(self.__rollback_config__)
self.ui.pushButton_cancel.clicked.connect(self.close) self.ui.pushButton_cancel.clicked.connect(self.close)
@ -78,7 +80,7 @@ class SettingWindow(QMainWindow):
ConfigParams.ENDSWITH_TXT))), ConfigParams.ENDSWITH_TXT))),
"Save": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT / "Save": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT /
Path(str(self.sampID)) / Path(ConfigParams.PREPROCESS_SAVE_BCG_FILENAME + Path(str(self.sampID)) / Path(ConfigParams.PREPROCESS_SAVE_BCG_FILENAME +
str(Config["InputConfig"]["Freq"]) + str(Config["OutputConfig"]["Freq"]) +
ConfigParams.ENDSWITH_TXT))) ConfigParams.ENDSWITH_TXT)))
}, },
"Mode": self.mode "Mode": self.mode
@ -92,7 +94,7 @@ class SettingWindow(QMainWindow):
ConfigParams.ENDSWITH_TXT))), ConfigParams.ENDSWITH_TXT))),
"Save": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_TEXT / "Save": str((Path(self.root_path) / ConfigParams.PUBLIC_PATH_PSG_TEXT /
Path(str(self.sampID)) / Path(ConfigParams.PREPROCESS_SAVE_ECG_FILENAME + Path(str(self.sampID)) / Path(ConfigParams.PREPROCESS_SAVE_ECG_FILENAME +
str(Config["InputConfig"]["Freq"]) + str(Config["OutputConfig"]["Freq"]) +
ConfigParams.ENDSWITH_TXT))) ConfigParams.ENDSWITH_TXT)))
}, },
"Mode": self.mode "Mode": self.mode
@ -100,6 +102,7 @@ class SettingWindow(QMainWindow):
# 数据回显 # 数据回显
self.ui.spinBox_input_freq.setValue(Config["InputConfig"]["Freq"]) self.ui.spinBox_input_freq.setValue(Config["InputConfig"]["Freq"])
self.ui.spinBox_output_freq.setValue(Config["OutputConfig"]["Freq"])
self.ui.plainTextEdit_file_path_input.setPlainText(Config["Path"]["Input"]) self.ui.plainTextEdit_file_path_input.setPlainText(Config["Path"]["Input"])
self.ui.plainTextEdit_file_path_save.setPlainText(Config["Path"]["Save"]) self.ui.plainTextEdit_file_path_save.setPlainText(Config["Path"]["Save"])
@ -107,11 +110,13 @@ class SettingWindow(QMainWindow):
# 从界面写入配置 # 从界面写入配置
Config["InputConfig"]["Freq"] = self.ui.spinBox_input_freq.value() Config["InputConfig"]["Freq"] = self.ui.spinBox_input_freq.value()
Config["OutputConfig"]["Freq"] = self.ui.spinBox_output_freq.value()
Config["Path"]["Input"] = self.ui.plainTextEdit_file_path_input.toPlainText() Config["Path"]["Input"] = self.ui.plainTextEdit_file_path_input.toPlainText()
Config["Path"]["Save"] = self.ui.plainTextEdit_file_path_save.toPlainText() Config["Path"]["Save"] = self.ui.plainTextEdit_file_path_save.toPlainText()
# 保存配置到文件 # 保存配置到文件
self.config["InputConfig"]["Freq"] = self.ui.spinBox_input_freq.value() self.config["InputConfig"]["Freq"] = self.ui.spinBox_input_freq.value()
self.config["OutputConfig"]["Freq"] = self.ui.spinBox_output_freq.value()
with open(ConfigParams.PREPROCESS_CONFIG_FILE_PATH, "w") as f: with open(ConfigParams.PREPROCESS_CONFIG_FILE_PATH, "w") as f:
dump(self.config, f) dump(self.config, f)
@ -137,7 +142,7 @@ class SettingWindow(QMainWindow):
ConfigParams.PUBLIC_PATH_ORGBCG_TEXT / ConfigParams.PUBLIC_PATH_ORGBCG_TEXT /
Path(str(self.sampID)) / Path(str(self.sampID)) /
Path(ConfigParams.PREPROCESS_SAVE_BCG_FILENAME + Path(ConfigParams.PREPROCESS_SAVE_BCG_FILENAME +
str(self.ui.spinBox_input_freq.value()) + str(self.ui.spinBox_output_freq.value()) +
ConfigParams.ENDSWITH_TXT)))) ConfigParams.ENDSWITH_TXT))))
else: else:
self.ui.plainTextEdit_file_path_input.setPlainText( self.ui.plainTextEdit_file_path_input.setPlainText(
@ -152,7 +157,7 @@ class SettingWindow(QMainWindow):
ConfigParams.PUBLIC_PATH_PSG_TEXT / ConfigParams.PUBLIC_PATH_PSG_TEXT /
Path(str(self.sampID)) / Path(str(self.sampID)) /
Path(ConfigParams.PREPROCESS_SAVE_ECG_FILENAME + Path(ConfigParams.PREPROCESS_SAVE_ECG_FILENAME +
str(self.ui.spinBox_input_freq.value()) + str(self.ui.spinBox_output_freq.value()) +
ConfigParams.ENDSWITH_TXT)))) ConfigParams.ENDSWITH_TXT))))
@ -491,18 +496,22 @@ class Data:
return False, Constants.PREPROCESS_PROCESS_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Raw_Data_Not_Exist"] return False, Constants.PREPROCESS_PROCESS_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Raw_Data_Not_Exist"]
try: try:
if Config["InputConfig"]["Freq"] != Config["OutputConfig"]["Freq"]:
self.raw_data = resample(self.raw_data,
int(len(self.raw_data) *
(Config["OutputConfig"]["Freq"] / Config["InputConfig"]["Freq"])))
if Config["Mode"] == "BCG": if Config["Mode"] == "BCG":
self.processed_data = Butterworth_for_BCG_PreProcess(self.raw_data, type='bandpass', self.processed_data = Butterworth_for_BCG_PreProcess(self.raw_data, type='bandpass',
low_cut=Config["Filter"]["BCGBandPassLow"], low_cut=Config["Filter"]["BCGBandPassLow"],
high_cut=Config["Filter"]["BCGBandPassHigh"], high_cut=Config["Filter"]["BCGBandPassHigh"],
order=Config["Filter"]["BCGBandPassOrder"], order=Config["Filter"]["BCGBandPassOrder"],
sample_rate=Config["InputConfig"]["Freq"]) sample_rate=Config["OutputConfig"]["Freq"])
else: else:
self.processed_data = Butterworth_for_ECG_PreProcess(self.raw_data, type='bandpass', self.processed_data = Butterworth_for_ECG_PreProcess(self.raw_data, type='bandpass',
low_cut=Config["Filter"]["ECGBandPassLow"], low_cut=Config["Filter"]["ECGBandPassLow"],
high_cut=Config["Filter"]["ECGBandPassHigh"], high_cut=Config["Filter"]["ECGBandPassHigh"],
order=Config["Filter"]["ECGBandPassOrder"], order=Config["Filter"]["ECGBandPassOrder"],
sample_rate=Config["InputConfig"]["Freq"]) sample_rate=Config["OutputConfig"]["Freq"])
except Exception: except Exception:
return False, Constants.PREPROCESS_PROCESS_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Filter_Exception"] return False, Constants.PREPROCESS_PROCESS_FAILURE + Constants.PREPROCESS_FAILURE_REASON["Filter_Exception"]

View File

@ -36,6 +36,9 @@ class ConfigParams:
"InputConfig": { "InputConfig": {
"Freq": 1000 "Freq": 1000
}, },
"OutputConfig": {
"Freq": 1000
},
"Filter": { "Filter": {
"BCGBandPassOrder": 4, "BCGBandPassOrder": 4,
"BCGBandPassLow": 2, "BCGBandPassLow": 2,

View File

@ -23,7 +23,7 @@ class Ui_MainWindow_preprocess_input_setting(object):
def setupUi(self, MainWindow_preprocess_input_setting): def setupUi(self, MainWindow_preprocess_input_setting):
if not MainWindow_preprocess_input_setting.objectName(): if not MainWindow_preprocess_input_setting.objectName():
MainWindow_preprocess_input_setting.setObjectName(u"MainWindow_preprocess_input_setting") MainWindow_preprocess_input_setting.setObjectName(u"MainWindow_preprocess_input_setting")
MainWindow_preprocess_input_setting.resize(480, 320) MainWindow_preprocess_input_setting.resize(480, 440)
self.centralwidget = QWidget(MainWindow_preprocess_input_setting) self.centralwidget = QWidget(MainWindow_preprocess_input_setting)
self.centralwidget.setObjectName(u"centralwidget") self.centralwidget.setObjectName(u"centralwidget")
self.gridLayout = QGridLayout(self.centralwidget) self.gridLayout = QGridLayout(self.centralwidget)
@ -80,6 +80,25 @@ class Ui_MainWindow_preprocess_input_setting(object):
self.groupBox_file_path_save.setObjectName(u"groupBox_file_path_save") self.groupBox_file_path_save.setObjectName(u"groupBox_file_path_save")
self.verticalLayout_4 = QVBoxLayout(self.groupBox_file_path_save) self.verticalLayout_4 = QVBoxLayout(self.groupBox_file_path_save)
self.verticalLayout_4.setObjectName(u"verticalLayout_4") self.verticalLayout_4.setObjectName(u"verticalLayout_4")
self.horizontalLayout_2 = QHBoxLayout()
self.horizontalLayout_2.setObjectName(u"horizontalLayout_2")
self.label_2 = QLabel(self.groupBox_file_path_save)
self.label_2.setObjectName(u"label_2")
self.label_2.setFont(font)
self.horizontalLayout_2.addWidget(self.label_2)
self.spinBox_output_freq = QSpinBox(self.groupBox_file_path_save)
self.spinBox_output_freq.setObjectName(u"spinBox_output_freq")
self.spinBox_output_freq.setFont(font)
self.spinBox_output_freq.setMinimum(1)
self.spinBox_output_freq.setMaximum(1000000)
self.horizontalLayout_2.addWidget(self.spinBox_output_freq)
self.verticalLayout_4.addLayout(self.horizontalLayout_2)
self.plainTextEdit_file_path_save = QPlainTextEdit(self.groupBox_file_path_save) self.plainTextEdit_file_path_save = QPlainTextEdit(self.groupBox_file_path_save)
self.plainTextEdit_file_path_save.setObjectName(u"plainTextEdit_file_path_save") self.plainTextEdit_file_path_save.setObjectName(u"plainTextEdit_file_path_save")
@ -88,6 +107,8 @@ class Ui_MainWindow_preprocess_input_setting(object):
self.verticalLayout_2.addWidget(self.groupBox_file_path_save) self.verticalLayout_2.addWidget(self.groupBox_file_path_save)
self.verticalLayout_2.setStretch(0, 1)
self.verticalLayout_2.setStretch(1, 1)
self.gridLayout.addWidget(self.groupBox, 0, 0, 1, 4) self.gridLayout.addWidget(self.groupBox, 0, 0, 1, 4)
@ -113,6 +134,7 @@ class Ui_MainWindow_preprocess_input_setting(object):
self.plainTextEdit_file_path_input.setPlainText("") self.plainTextEdit_file_path_input.setPlainText("")
self.plainTextEdit_file_path_input.setPlaceholderText(QCoreApplication.translate("MainWindow_preprocess_input_setting", u"\u6587\u4ef6\u8def\u5f84", None)) self.plainTextEdit_file_path_input.setPlaceholderText(QCoreApplication.translate("MainWindow_preprocess_input_setting", u"\u6587\u4ef6\u8def\u5f84", None))
self.groupBox_file_path_save.setTitle(QCoreApplication.translate("MainWindow_preprocess_input_setting", u"\u6ee4\u6ce2\u540e\u4fdd\u5b58\u8def\u5f84", None)) self.groupBox_file_path_save.setTitle(QCoreApplication.translate("MainWindow_preprocess_input_setting", u"\u6ee4\u6ce2\u540e\u4fdd\u5b58\u8def\u5f84", None))
self.label_2.setText(QCoreApplication.translate("MainWindow_preprocess_input_setting", u"\u91c7\u6837\u7387(Hz)\uff1a", None))
self.plainTextEdit_file_path_save.setPlaceholderText(QCoreApplication.translate("MainWindow_preprocess_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None)) self.plainTextEdit_file_path_save.setPlaceholderText(QCoreApplication.translate("MainWindow_preprocess_input_setting", u"\u4fdd\u5b58\u8def\u5f84", None))
self.pushButton_confirm.setText(QCoreApplication.translate("MainWindow_preprocess_input_setting", u"\u786e\u5b9a", None)) self.pushButton_confirm.setText(QCoreApplication.translate("MainWindow_preprocess_input_setting", u"\u786e\u5b9a", None))
# retranslateUi # retranslateUi

View File

@ -7,7 +7,7 @@
<x>0</x> <x>0</x>
<y>0</y> <y>0</y>
<width>480</width> <width>480</width>
<height>320</height> <height>440</height>
</rect> </rect>
</property> </property>
<property name="windowTitle"> <property name="windowTitle">
@ -37,7 +37,7 @@
<property name="title"> <property name="title">
<string>文件路径</string> <string>文件路径</string>
</property> </property>
<layout class="QVBoxLayout" name="verticalLayout_2"> <layout class="QVBoxLayout" name="verticalLayout_2" stretch="1,1">
<item> <item>
<widget class="QGroupBox" name="groupBox_file_path_input"> <widget class="QGroupBox" name="groupBox_file_path_input">
<property name="title"> <property name="title">
@ -94,6 +94,37 @@
<string>滤波后保存路径</string> <string>滤波后保存路径</string>
</property> </property>
<layout class="QVBoxLayout" name="verticalLayout_4"> <layout class="QVBoxLayout" name="verticalLayout_4">
<item>
<layout class="QHBoxLayout" name="horizontalLayout_2">
<item>
<widget class="QLabel" name="label_2">
<property name="font">
<font>
<pointsize>12</pointsize>
</font>
</property>
<property name="text">
<string>采样率(Hz)</string>
</property>
</widget>
</item>
<item>
<widget class="QSpinBox" name="spinBox_output_freq">
<property name="font">
<font>
<pointsize>12</pointsize>
</font>
</property>
<property name="minimum">
<number>1</number>
</property>
<property name="maximum">
<number>1000000</number>
</property>
</widget>
</item>
</layout>
</item>
<item> <item>
<widget class="QPlainTextEdit" name="plainTextEdit_file_path_save"> <widget class="QPlainTextEdit" name="plainTextEdit_file_path_save">
<property name="placeholderText"> <property name="placeholderText">