完成了<预处理>的信号重采样部分,当输入输出采样率不同时,将执行重采样
This commit is contained in:
@ -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"]
|
||||||
|
|
||||||
|
|||||||
@ -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,
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
@ -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">
|
||||||
|
|||||||
Reference in New Issue
Block a user