使用pyqt 显示控件实现曲线的绘制
代码中brain_node_disk_session 为我项目中需要显示的数据源,自行替换。
class XchartView(QWidget): def __init__(self,brain_node_disk_session): super(XchartView, self).__init__() self.brain_node_disk_session = brain_node_disk_session main_layout =QVBoxLayout() comb_box =QHBoxLayout() self.combox1 = QComboBox() self.combox2 = QComboBox() self.combox1.addItems(utils_ui_list_sig) self.combox2.addItems(utils_ui_list_sig) self.combox1.setStyleSheet("QWidget{color:red}")#color self.combox2.setStyleSheet("QWidget{color:green}") comb_box.addWidget(self.combox1) comb_box.addWidget(self.combox2) self.win_charts_sensor = pyqtgraph.GraphicsLayoutWidget(show=True) self.label = pyqtgraph.LabelItem(justify='right') self.win_charts_sensor.addItem(self.label) self.win_charts_sensor.nextRow() # self.win_charts_sensor = pyqtgraph.PlotWidget() self.p1 = self.win_charts_sensor.addPlot()#(title="value") # self.p1 = self.win_charts_sensor.plotItem #self.p1.setLabels(left='axis 1') self.p2 = pyqtgraph.ViewBox() self.p1.showAxis('right') self.p1.scene().addItem(self.p2) self.p1.getAxis('right').linkToView(self.p2) self.p2.setXLink(self.p1) #self.p1.getAxis('right').setLabel('axis2', color='#0000ff') # self.label.setGeometry(QRectF(0, 0, 100, 100)) self.vLine = pyqtgraph.InfiniteLine(angle=90, movable=False) self.hLine = pyqtgraph.InfiniteLine(angle=0, movable=False) #self.curpose_Line = pyqtgraph.InfiniteLine(angle=90, movable=False)#current slide position self.p1.addItem(self.vLine, ignoreBounds=True) self.p1.addItem(self.hLine, ignoreBounds=True) #self.p1.addItem(self.curpose_Line, ignoreBounds=True) self.vb = self.p1.vb main_layout.addLayout(comb_box) main_layout.addWidget(self.win_charts_sensor) main_layout.setContentsMargins(0, 0, 0, 0) self.setLayout(main_layout) self.combox1_data = [] self.combox2_data = [] self.curve1 = self.p1.plot(self.combox1_data, pen="r") self.p2.addItem(pyqtgraph.PlotCurveItem(self.combox2_data, pen='g')) self.combox1.currentIndexChanged[int].connect(self.update_comb1) self.combox2.currentIndexChanged[int].connect(self.update_comb2) self.ts = 0 self.timer_init() self.index1_choose = None self.index2_choose = None def timer_init(self): #使用QTimer,2秒触发一次,更新数据 self.timer = QTimer(self) self.timer.timeout.connect(self.drawLine) self.timer.start(200) def drawLine(self): # print("self.index1_choose,self.index2_choose:",self.index1_choose,self.index2_choose) if self.index1_choose != None and self.index2_choose != None: self.combox1_data = np.array(self.get_val_from_str(utils_ui_list_sig[self.index1_choose])) self.combox2_data = np.array(self.get_val_from_str(utils_ui_list_sig[self.index2_choose])) elif self.index1_choose == None and self.index2_choose != None: self.combox1_data = [] self.combox2_data = np.array(self.get_val_from_str(utils_ui_list_sig[self.index2_choose])) elif self.index1_choose != None and self.index2_choose == None: self.combox1_data = np.array(self.get_val_from_str(utils_ui_list_sig[self.index1_choose])) self.combox2_data = [] else: self.combox1_data = [] self.combox2_data = [] # print("self.combox1_data's length:",len(self.combox1_data)) self.curve1.setData(self.combox1_data) self.p2.addItem(pyqtgraph.PlotCurveItem(self.combox2_data, pen='g')) def update_comb1(self, index): self.index1_choose = index self.combox1_data = [] self.curve1.setData(self.combox1_data) self.combox1_data = np.array(self.get_val_from_str(utils_ui_list_sig[index])) if np.any(self.combox1_data) : self.curve1.setData(self.combox1_data) else: print("self.combox2_data is null") self.proxy = pyqtgraph.SignalProxy(self.p1.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved) def update_comb2(self, index): self.index2_choose = index self.p2.setGeometry(self.p1.vb.sceneBoundingRect()) self.p2.linkedViewChanged(self.p1.vb, self.p2.XAxis) self.combox2_data = [] self.p2.clear() self.combox2_data = np.array(self.get_val_from_str(utils_ui_list_sig[index])) # self.curve2 = self.plot_view_v1.plot(self.combox2_data, pen="g") if np.any(self.combox2_data): self.p2.addItem(pyqtgraph.PlotCurveItem(self.combox2_data, pen='g')) else: print("self.combox2_data is null") self.proxy = pyqtgraph.SignalProxy(self.p1.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved) def mouseMoved(self,evt): pos = evt[0] ## using signal proxy turns original arguments into a tuple if self.p1.sceneBoundingRect().contains(pos): mousePoint = self.vb.mapSceneToView(pos) mouse_index = int(mousePoint.x()) combox1_data_str, combox2_data_str = self.get_cur_combox_value(mouse_index) try: self.ts = self.brain_node_disk_session.data_lidar.timestamps[mouse_index]/1000 except: self.ts = self.brain_node_disk_session.data_lidar.timestamps[-1] / 1000 # print("ts:",self.ts) self.label.setText( "<span style='font-size: 10pt'>ts=%s, <span style='font-size: 10pt'>x=%0.3f, <span style='color: red'>y1=%s</span>, <span style='color: green'>y2=%s</span>" % ( time_convert(self.ts), mouse_index, combox1_data_str, combox2_data_str)) self.vLine.setPos(mousePoint.x()) self.hLine.setPos(mousePoint.y()) def set_p(self,pos_index): mouse_index = pos_index self.vLine.setPos(pos_index) combox1_data_str, combox2_data_str=self.get_cur_combox_value(mouse_index) self.ts = self.brain_node_disk_session.data_imu.timestamps[pos_index]/1000 self.label.setText( "<span style='font-size: 10pt'>ts=%s, <span style='font-size: 10pt'>x=%0.3f, <span style='color: red'>y1=%s</span>, <span style='color: green'>y2=%s</span>" % ( time_convert(self.ts),mouse_index, combox1_data_str, combox2_data_str)) def get_cur_combox_value(self,mouse_index): len_comb1 = len(self.combox1_data) len_comb2 = len(self.combox2_data) if len_comb1 > 0 and mouse_index > 0 and mouse_index < len_comb1: combox1_data_str = '{:.3f}'.format(self.combox1_data[mouse_index]) else: combox1_data_str = '_' if len_comb2 > 0 and mouse_index > 0 and mouse_index < len_comb2: combox2_data_str = '{:.3f}'.format(self.combox2_data[mouse_index]) else: combox2_data_str = '_' return combox1_data_str,combox2_data_str def set_y_range(self, up, down): self.p1.setRange(yRange=(down,up))
TRANSLATE with