import numpy as np from matplotlib import pyplot as plt plt.rcParams['font.sans-serif'] = ['SimSun'] # 使用宋体 plt.rcParams['axes.unicode_minus'] = False # 正常显示负号 from global_var import global_var_init cycle, fs, record_name, data_path = global_var_init() from processing import processing (signal_dwt , waves_dwt , rpeaks , ecg_signal, signal, signal.v, signal.fs, rows_to_extract, t_ecg, t_ppg, on_column_list, on_column_list_toArea, on_values, on_values_toArea, sp_column_list, sp_values, dn_column_list, dn_values, dp_column_list, dp_values, u_column_list, u_column_list_toArea, u_values, u_values_toArea, v_column_list, v_values, w_column_list, w_values, a_column_list, a_column_list_toArea, a_values, a_values_toArea, b_column_list, b_values, c_column_list, c_values, e_column_list, e_values, f_column_list, f_values, signal.ppg, signal.vpg, signal.apg, signal.jpg) = processing() from ecg_peaks_val import ecg_peaks_val (P_peaks, Q_peaks, R_peaks, S_peaks, T_peaks, P_onsets, P_offsets, T_onsets, T_offsets, P_peaks_values, Q_peaks_values, R_peaks_values, S_peaks_values, T_peaks_values, P_onsets_values, P_offsets_values, T_onsets_values, T_offsets_values, PQ_baseline) = ecg_peaks_val() from ppg_peaks_val import ppg_peaks_val (on, on_toArea, sp, dn, dp, on_values, on_toArea_values, sp_values, dn_values, dp_values, u, u_toArea, v, w, u_values, u_toArea_values, v_values, w_values, a, a_toArea, b, c, e, f, a_values, a_toArea_values, b_values, c_values, e_values, f_values) = ppg_peaks_val() #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #----------------------------------- 创建拖动类 ---------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- class DraggablePoints: def __init__(self, ax, points, signal_type, signal_name): self.ax = ax self.points = points # 这里是 Line2D 对象的列表 self.signal_type = signal_type self.signal_name = signal_name self.selected_point = None self.press = None self.new_x = None self.real_y = None # 最终获得的点，其纵坐标数值应在信号线上 self.point_x = None # 连接鼠标事件 self.cidpress = ax.figure.canvas.mpl_connect('button_press_event', self.on_press) self.cidmotion = ax.figure.canvas.mpl_connect('motion_notify_event', self.on_motion) self.cidrelease = ax.figure.canvas.mpl_connect('button_release_event', self.on_release) def on_press(self, event): if self.ax == event.inaxes: for point in self.points: point_x = point.get_xdata() point_y = point.get_ydata() distance = np.hypot(point_x - event.xdata, point_y - event.ydata) if distance < 0.05: # 可以根据需要调整这个阈值 self.selected_point = point self.point_x = point.get_xdata() self.point_y = point.get_ydata() self.press = (point_x, point_y, event.xdata, event.ydata) break def on_motion(self, event): if self.selected_point is None or event.inaxes != self.ax: return if self.signal_name == "ECG": # 计算点的新的位置 self.new_x = self.press[0] + (event.xdata - self.press[2]) self.real_y = self.signal_type[int(self.new_x*fs)] self.real_y = np.array([self.real_y]) #格式转换，从float64 转换成了 ndarray，为了set_data（）能读取 # 更新选中点的位置 self.selected_point.set_data(self.new_x, self.real_y) self.ax.figure.canvas.draw() # 使用 draw 更新 else : # 计算点的新的位置 self.new_x = self.press[0] + (event.xdata - self.press[2]) self.real_y = self.signal_type[int(self.new_x*fs)] self.real_y = np.array([self.real_y]) #格式转换，从float64 转换成了 ndarray，为了set_data（）能读取 # 更新选中点的位置 self.selected_point.set_data(self.new_x, self.real_y) self.ax.figure.canvas.draw() # 使用 draw 更新 def on_release(self, event): if self.selected_point: if self.signal_name == "ECG": print(" 点所属的信号 ：",self.signal_name) print(" 点的坐标(t) ：",self.point_x, self.point_y) print(" 更新后点坐标(t) ：",np.round(self.new_x, 3) , "[", self.real_y, "]" ) print(" 点的位置 : ",int(self.point_x*fs)) print(" 更新后点的位置 ：",int(self.new_x*fs) ) print("------------------------------------------") else: print(" 点所属的信号 ：",self.signal_name) print(" 点的坐标(t) ：",self.point_x, self.point_y) print(" 更新后点坐标(t) ：",np.round(self.new_x, 3) , "[", self.real_y, "]" ) print(" 点的位置 : ",int(self.point_x*fs)) print(" 更新后点的位置 ：",int(self.new_x*fs) ) print("------------------------------------------") self.selected_point = None self.press = None self.new_x = None #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #----------------------------------- 创建绘图函数 ---------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- def create_plot_with_draggable_points(save_path = None): fig, (ax1, ax2, ax3, ax4) = plt.subplots(4, 1, sharex=True) # 绘制归一化后的ECG信号 ax1.plot(t_ecg, ecg_signal) ax1.set(xlabel='', ylabel='ECG') # 绘制归一化后的PPG信号 ax2.plot(t_ppg, signal.ppg) ax2.set(xlabel='', ylabel='PPG') # 绘制归一化后的一阶导数 ax3.plot(t_ppg, signal.vpg) ax3.set(xlabel='', ylabel='vpg') # 绘制归一化后的二阶导数 ax4.plot(t_ppg, signal.apg) ax4.set(xlabel='Time (s)', ylabel='apg') # 存储并绘制这些点对象 points_1 = [ax1.plot(x, y, 'o')[0] for x, y in zip(t_ecg[P_peaks], P_peaks_values)] points_1 += [ax1.plot(x, y, 'o', color='cyan')[0] for x, y in zip(t_ecg[Q_peaks], Q_peaks_values)] points_1 += [ax1.plot(x, y, 'o', color='purple')[0] for x, y in zip(t_ecg[R_peaks], R_peaks_values)] points_1 += [ax1.plot(x, y, 'o', color='blue')[0] for x, y in zip(t_ecg[S_peaks], S_peaks_values)] points_1 += [ax1.plot(x, y, 'o', color='black')[0] for x, y in zip(t_ecg[T_peaks], T_peaks_values)] points_1 += [ax1.plot(x, y, 'x', color='red')[0] for x, y in zip(t_ecg[P_onsets], P_onsets_values)] points_1 += [ax1.plot(x, y, 'x', color='magenta')[0] for x, y in zip(t_ecg[P_offsets], P_offsets_values)] points_1 += [ax1.plot(x, y, 'x', color='black')[0] for x, y in zip(t_ecg[T_onsets], T_onsets_values)] points_1 += [ax1.plot(x, y, 'x', color='gold')[0] for x, y in zip(t_ecg[T_offsets], T_offsets_values)] points_2 = [ax2.plot(x, y, 'o', color='red')[0] for x, y in zip(t_ppg[on], on_values)] points_2 += [ax2.plot(x, y, 'o', color='blue')[0] for x, y in zip(t_ppg[sp], sp_values)] points_2 += [ax2.plot(x, y, 'x', color='orange')[0] for x, y in zip(t_ppg[dn], dn_values)] points_2 += [ax2.plot(x, y, 'o', color='purple')[0] for x, y in zip(t_ppg[dp], dp_values)] points_3 = [ax3.plot(x, y, 'o', color='cyan')[0] for x, y in zip(t_ppg[u], u_values)] points_3 += [ax3.plot(x, y, 'o', color='magenta')[0] for x, y in zip(t_ppg[v], v_values)] points_3 += [ax3.plot(x, y, 'o', color='black')[0] for x, y in zip(t_ppg[w], w_values)] points_4 = [ax4.plot(x, y, 'o', color='black')[0] for x, y in zip(t_ppg[a], a_values)] points_4 += [ax4.plot(x, y, 'o', color='blue')[0] for x, y in zip(t_ppg[b], b_values)] points_4 += [ax4.plot(x, y, 'o', color='green')[0] for x, y in zip(t_ppg[c], c_values)] points_4 += [ax4.plot(x, y, 'o', color='orange')[0] for x, y in zip(t_ppg[e], e_values)] points_4 += [ax4.plot(x, y, 'o', color='purple')[0] for x, y in zip(t_ppg[f], f_values)] draggable_points_1 = DraggablePoints(ax1, points_1, ecg_signal, signal_name = "ECG") draggable_points_2 = DraggablePoints(ax2, points_2, signal.ppg, signal_name = "PPG") draggable_points_3 = DraggablePoints(ax3, points_3, signal.vpg, signal_name = "VPG") draggable_points_4 = DraggablePoints(ax4, points_4, signal.apg, signal_name = "APG") # 如果指定了保存路径，则保存图像 if save_path: plt.savefig(save_path) plt.tight_layout() plt.show() # 将图形嵌入到 Qt 的 canvas 中 # canvas.figure = fig # canvas.draw() #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #----------------------------------- 调用绘图函数 ---------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- #-------------------------------------------------------------------------------------------------------- if __name__ == '__main__': print(len(P_peaks)) print(len(Q_peaks)) print(len(R_peaks)) print(len(S_peaks)) print(len(T_peaks)) print(len(P_onsets)) print(len(P_offsets)) print(len(T_onsets)) print(len(T_offsets)) print(len(u)) print(len(v)) print(len(w)) print(len(a)) print(len(b)) print(len(c)) print(len(e)) print(len(f)) create_plot_with_draggable_points()