class Window(QtWidgets.QDialog):
def __init__(self, parent=None):
super().__init__(parent)
self.figure = plt.figure()
self.axes = self.figure.add_subplot(111)
# We want the axes cleared every time plot() is called
self.axes.hold(False)
self.canvas = FigureCanvas(self.figure)
self.toolbar = NavigationToolbar(self.canvas, self)
self.toolbar.hide()
# Just some button
self.button1 = QtWidgets.QPushButton('Plot')
self.button1.clicked.connect(self.plot)
self.button2 = QtWidgets.QPushButton('Zoom')
self.button2.clicked.connect(self.zoom)
self.button3 = QtWidgets.QPushButton('Pan')
self.button3.clicked.connect(self.pan)
self.button4 = QtWidgets.QPushButton('Home')
self.button4.clicked.connect(self.home)
# set the layout
layout = QtWidgets.QVBoxLayout()
layout.addWidget(self.toolbar)
layout.addWidget(self.canvas)
btnlayout = QtWidgets.QHBoxLayout()
btnlayout.addWidget(self.button1)
btnlayout.addWidget(self.button2)
btnlayout.addWidget(self.button3)
btnlayout.addWidget(self.button4)
qw = QtWidgets.QWidget(self)
qw.setLayout(btnlayout)
layout.addWidget(qw)
self.setLayout(layout)
def home(self):
self.toolbar.home()
def zoom(self):
self.toolbar.zoom()
def pan(self):
self.toolbar.pan()
def plot(self):
''' plot some random stuff '''
data = [random.random() for i in range(25)]
self.axes.plot(data, '*-')
self.canvas.draw()
class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
# Init 0 for draw...
plt.rcParams['agg.path.chunksize'] = 0
matplotlib.rcParams['agg.path.chunksize'] = 0
matplotlib.rcParams.update(matplotlib.rc_params())
self.dpi = 100
self.signalframe = self.widget_Signal_TimeDomain
self.figure = Figure((11.3, 6.3), dpi=self.dpi)
self.canvas = FigureCanvas(self.figure)
self.canvas.setParent(self.signalframe)
self.axes = self.figure.add_subplot(111)
#self.axes.set_title('Signal')
self.axes.set_xlabel('Time(μs)')
self.axes.set_ylabel('Voltage')
#plt.subplots_adjust(left=0.2, bottom=0.2, right=0.8, top=0.8, hspace=0.2, wspace=0.3)
timespan = self.getRecordLength() * 1024 / self.getSampleRate(
) # in us
x = np.linspace(0, timespan, self.getRecordLength() * 1024)
normalLimY = self.getVoltageScale() * 10
self.axes.set_ylim(-normalLimY / 2 + self.getOffset(),
normalLimY / 2 + self.getOffset())
ymajorLocator = MultipleLocator(self.getVoltageScale())
yminorLocator = MultipleLocator(self.getVoltageScale() / 2)
self.axes.yaxis.set_major_locator(ymajorLocator)
self.axes.yaxis.set_minor_locator(yminorLocator)
self.axes.grid(True)
self.figure.tight_layout() # Adjust spaces
#self.NavToolbar = NavigationToolbar(self.canvas, self.signalframe)
#self.addToolBar(QtCore.Qt.RightToolBarArea, NavigationToolbar(self.canvas, self.signalframe))
self.toolbar = NavigationToolbar(self.canvas, self.signalframe)
self.toolbar.hide()
# Button slots
self.pushButton_Home.clicked.connect(self.home)
self.pushButton_Back.clicked.connect(self.back)
self.pushButton_Forward.clicked.connect(self.forward)
self.pushButton_Pan.clicked.connect(self.pan)
self.pushButton_Zoom.clicked.connect(self.zoom)
self.pushButton_SavePic.clicked.connect(self.savepic)
# Init Socket
self.udpSocketClient = UDPSocketClient()
# Init FrameNum
self.checkBox_FrameMode.setEnabled(True)
self.frameNum = self.getFrameNumber()
# 0x4
self.sendCmdFramNum(self.frameNum)
# Init Length 0x8
self.comboBox_RecordLength.setCurrentIndex(6) # 64k
# Set RecordLength
self.sendCmdRecordLength(self.getRecordLength() *
self.getFrameNumber())
#Init Trigger 0x2, bit2
self.comboBox_TriggerDomain.setCurrentIndex(1) # External
self.sendCmdTriggerType(self.getTriggerType())
# Reset PFGA to read to capture data
if (self.getTriggerType() == 0):
self.sendCmdWRREG(0x2, 0x20)
self.sendCmdWRREG(0x2, 0x28)
else:
self.sendCmdWRREG(0xa, 0x0)
self.sendCmdWRREG(0x2, 0x2c)
self.sendCmdWRREG(0x2, 0x2d)
# Read sampleRate
value = self.readCmdSampleRate()
if value > 5:
value = 0
self.comboBox_SampleRate.setCurrentIndex(value)
# The last data
self.lastChAData = []
self.lastChBData = []
self.realTimeThread = None
self.externalTriggerThread = None
def home(self):
self.toolbar.home()
def back(self):
self.toolbar.back()
def forward(self):
self.toolbar.forward()
def zoom(self):
self.toolbar.zoom()
def pan(self):
self.toolbar.pan()
def savepic(self):
self.toolbar.save_figure()
def sendcommand(self, cmdid, status, msgid, len, type, offset, apiversion,
pad, CRC16, cmdData):
cmdid = struct.pack('H', htons(cmdid))
status = struct.pack('H', htons(status))
msgid = struct.pack('H', htons(msgid))
len = struct.pack('H', htons(len))
type = struct.pack('H', htons(type))
offset = struct.pack('H', htons(offset))
apiversion = struct.pack('B', apiversion) # 1 Byte unsigned char
pad = struct.pack('B', pad) # 1 Byte unsigned char
CRC16 = struct.pack('H', htons(CRC16)) # 2 Byte unsigned short
cmdHeader = cmdid + status + msgid + len + type + offset + apiversion + pad + CRC16
if (cmdData != None):
self.udpSocketClient.mData = cmdHeader + cmdData
else:
self.udpSocketClient.mData = cmdHeader
self.udpSocketClient.sendData()
def sendCmdTriggerType(self, value):
regAddr = 0x2 # 0x2, Bit[2], 0: Auto, 1: External
currentValue = self.readCmdTriggerType()
if value == 0: # Auto Trigger
mask = 0b1111111111111011
currentValue = currentValue & mask
elif value == 1: # External
mask = 0b100
currentValue = currentValue | mask
self.sendCmdWRREG(regAddr, currentValue)
def readCmdTriggerType(self):
print("readCmdTriggerType ")
self.sendCmdRDREG(0x02, 0x00)
data = self.udpSocketClient.receiveData()
value = ntohl(int(struct.unpack('L', data[20:24])[0]))
return value
def readDataCount(self):
self.sendCmdRDREG(0x10, 0x00)
data = self.udpSocketClient.receiveData()
data = data[20:24]
lowValue = ntohl(int(struct.unpack('L', data)[0]))
print("0x10 Value:", hex(lowValue))
self.sendCmdRDREG(0x12, 0x00)
data = self.udpSocketClient.receiveData()
data = data[20:24]
highValue = ntohl(int(struct.unpack('L', data)[0]))
print("0x12 Value:", hex(highValue))
value = highValue << 16 | lowValue
return value
#return 1024
def sendCmdSampleRate(self, value):
global gSocketBodySize
gSocketBodySize = 4
self.udpSocketClient.setBufSize(gSocketBodySize + gSocketHeaderSize)
cmdData = struct.pack('L', htonl(value))
self.sendcommand(0x5a09, 0x0000, 0x5a09, 0x0004, 0x0000, 0x0000, 0x00,
0x00, 0x0000, cmdData)
self.udpSocketClient.receiveData() # Do nothing
def readCmdSampleRate(self):
global gSocketBodySize
gSocketBodySize = 4
self.udpSocketClient.setBufSize(gSocketBodySize + gSocketHeaderSize)
# Len is not cared
self.sendcommand(0x5a0a, 0x0000, 0x5a0a, 0x0004, 0x0000, 0x0000, 0x00,
0x00, 0x0000, None)
data = self.udpSocketClient.receiveData()
value = ntohl(int(struct.unpack('L', data[16:20])[0]))
return value
def sendCmdRecordLength(self, length):
#recordLength = self.getRecordLength()
regAddr = 0x8
regValue = length
self.sendCmdWRREG(regAddr, regValue)
def receiveCmdRecordLength(self):
global gSocketBodySize
gSocketBodySize = 8
self.udpSocketClient.setBufSize(gSocketBodySize + gSocketHeaderSize)
self.udpSocketClient.receiveData() # Do nothing
def sendCmdRAW_AD_SAMPLE(self, length):
#print (sys._getframe().f_code.co_name)
global gSocketBodySize
gSocketBodySize = length * 1024
self.udpSocketClient.setBufSize(gSocketBodySize + gSocketHeaderSize)
len = 0 #self.getRecordLength()
self.sendcommand(0x5a04, 0x0000, 0x5a04, len, 0x0000, 0x0000, 0x00,
0x00, 0x0000, None)
def receiveCmdRAW_AD_SAMPLE(self, length):
global gSocketBodySize
gSocketBodySize = length * 1024
self.udpSocketClient.setBufSize(gSocketBodySize + gSocketHeaderSize)
self.udpSocketClient.receiveData()
def sendCmdFramNum(self, frameNum):
if (frameNum <= 2**16 - 1):
regAddr = 0x4
regValue = frameNum
self.sendCmdWRREG(regAddr, regValue)
else:
# Low
regAddr = 0x4
regValue = frameNum & (2**16 - 1)
self.sendCmdWRREG(regAddr, regValue)
# High
regAddr = 0x6
regValue = frameNum >> 16
self.sendCmdWRREG(regAddr, regValue)
def sendCmdWRREG(self, regAddress, regValue):
#print (sys._getframe().f_code.co_name)
global gSocketBodySize
gSocketBodySize = 8
self.udpSocketClient.setBufSize(gSocketBodySize + gSocketHeaderSize)
cmdData = struct.pack('L', htonl(regAddress)) + struct.pack(
'L', htonl(regValue))
self.sendcommand(0x5a02, 0x0000, 0x5a02, 0x0008, 0x0000, 0x0000, 0x00,
0x00, 0x0000, cmdData)
self.udpSocketClient.receiveData() # Do nothing
def sendCmdRDREG(self, regAddress, regValue):
#print (sys._getframe().f_code.co_name)
global gSocketBodySize
gSocketBodySize = 8
self.udpSocketClient.setBufSize(gSocketBodySize + gSocketHeaderSize)
cmdData = struct.pack('L', htonl(regAddress)) + struct.pack(
'L', htonl(regValue))
self.sendcommand(0x5a01, 0x0000, 0x5a01, 0x0008, 0x0000, 0x0000, 0x00,
0x00, 0x0000, cmdData)
def getTriggerType(self):
index = self.comboBox_TriggerDomain.currentIndex()
return int(index)
def getSampleRate(self):
index = self.comboBox_SampleRate.currentText()
return int(index)
def getRecordLength(self):
index = self.comboBox_RecordLength.currentIndex()
if (index == 18):
return 40
else:
return 2**index
def getVoltageScale(self):
volScale = 200
volScaleStr = self.lineEdit_VolScale.text()
volScale = 200
if (('-') == volScaleStr or "" == volScaleStr):
volScale = 200
else:
volScale = int(volScaleStr)
return volScale
def getOffset(self):
offset = 0
offsetStr = self.lineEdit_Offset.text()
if (('-') == offsetStr or "" == offsetStr):
offset = 0
else:
offset = int(offsetStr)
return offset
def getFrameNumber(self):
if (self.checkBox_FrameMode.isChecked()
and self.checkBox_FrameMode.isEnabled()):
return int(self.lineEdit_FrameNum.text())
else:
return 1
@pyqtSlot()
def on_pushButton_Stop_TimeDomain_clicked(self):
"""
Slot documentation goes here.
"""
if self.realTimeThread != None:
self.realTimeThread.stop()
if self.externalTriggerThread != None:
self.externalTriggerThread.stop()
self.pushButton_Start_TimeDomain.setEnabled(True)
self.pushButton_Stop_TimeDomain.setEnabled(False)
self.pushButton_Save_TimeDomain.setEnabled(True)
self.comboBox_RecordLength.setEnabled(True)
# Disable them for zhengzhou
self.comboBox_SampleRate.setEnabled(True)
self.comboBox_TriggerDomain.setEnabled(True)
self.lineEdit_FrameNum.setEnabled(True)
@pyqtSlot()
def on_pushButton_Start_TimeDomain_clicked(self):
"""
Slot documentation goes here.
"""
# Restart LCD display
self.lcdNumber_FrameNum.display(0)
self.pushButton_Start_TimeDomain.setEnabled(False)
self.pushButton_Stop_TimeDomain.setEnabled(True)
self.pushButton_Save_TimeDomain.setEnabled(False)
self.comboBox_RecordLength.setEnabled(False)
self.comboBox_SampleRate.setEnabled(False)
self.comboBox_TriggerDomain.setEnabled(False)
self.lineEdit_FrameNum.setEnabled(False)
if self.getTriggerType() == 0: # Auto Trigger Type
self.realTimeThread = RealTimeThread(
self.axes, self.canvas, self.radioButton_CHA.isChecked(), 1.0,
False)
self.realTimeThread.setDaemon(True)
self.realTimeThread.start()
elif self.getTriggerType() == 1: # External Trigger Type
self.externalTriggerThread = RealTimeThread(
self.axes, self.canvas, self.radioButton_CHA.isChecked(), 1.0,
True)
self.externalTriggerThread.setDaemon(True)
self.externalTriggerThread.start()
@pyqtSlot()
def on_pushButton_Save_TimeDomain_clicked(self):
"""
Slot documentation goes here.
"""
# Write into file
now = datetime.datetime.now()
currentTime = now.strftime('%Y-%m-%d-%H-%M-%S')
if (len(self.lastChAData) == 1):
FileName_CHA = "ChA-" + currentTime + ".txt"
File_CHA = open(FileName_CHA, 'w')
FileName_CHB = "ChB-" + currentTime + ".txt"
File_CHB = open(FileName_CHB, 'w')
for pos in range(0, len(self.lastChAData[0])):
File_CHA.write(str(self.lastChAData[0][pos]))
File_CHA.write('\n')
File_CHB.write(str(self.lastChBData[0][pos]))
File_CHB.write('\n')
File_CHA.close()
File_CHB.close()
else:
for fileIndex in range(0, len(self.lastChAData)):
FileName_CHA = "ChA-" + currentTime + "-" + str(fileIndex +
1) + ".txt"
File_CHA = open(FileName_CHA, 'w')
FileName_CHB = "ChB-" + currentTime + "-" + str(fileIndex +
1) + ".txt"
File_CHB = open(FileName_CHB, 'w')
for pos in range(0, len(self.lastChAData[fileIndex])):
File_CHA.write(str(self.lastChAData[fileIndex][pos]))
File_CHA.write('\n')
File_CHB.write(str(self.lastChBData[fileIndex][pos]))
File_CHB.write('\n')
File_CHA.close()
File_CHB.close()
# Do not clear it
#self.lastChAData = []
#self.lastChBData = []
@pyqtSlot(int)
def on_comboBox_TriggerDomain_currentIndexChanged(self, index):
self.sendCmdTriggerType(index)
# Enable/Disable Frame Mode
if index == 1:
self.checkBox_FrameMode.setEnabled(True)
self.label_FrameNum.setEnabled(True)
self.lineEdit_FrameNum.setEnabled(True)
self.label_FrameNum_Current.setEnabled(True)
self.lcdNumber_FrameNum.setEnabled(True)
else:
self.checkBox_FrameMode.setEnabled(False)
self.label_FrameNum.setEnabled(False)
self.lineEdit_FrameNum.setEnabled(False)
self.label_FrameNum_Current.setEnabled(False)
self.lcdNumber_FrameNum.setEnabled(False)
# Restor RecordLength
self.sendCmdRecordLength(self.getRecordLength() *
self.getFrameNumber())
@pyqtSlot(int)
def on_comboBox_SampleRate_currentIndexChanged(self, index):
if index > -1:
self.sendCmdSampleRate(index)
@pyqtSlot(int)
def on_comboBox_RecordLength_currentIndexChanged(self, index):
if (index == 18): # this is 40k
self.sendCmdRecordLength(40 * self.getFrameNumber())
else:
self.sendCmdRecordLength((2**index) * self.getFrameNumber())
@pyqtSlot()
def on_lineEdit_FrameNum_editingFinished(self):
self.frameNum = int(self.lineEdit_FrameNum.text())
self.sendCmdFramNum(self.frameNum)
self.sendCmdRecordLength(self.getRecordLength() * self.frameNum)
@pyqtSlot(bool)
def on_checkBox_FrameMode_clicked(self, checked):
"""
Slot documentation goes here.
@param checked DESCRIPTION
@type bool
"""
# Set recordLength value to register
self.frameNum = self.getFrameNumber()
self.sendCmdFramNum(self.frameNum)
self.sendCmdRecordLength(self.getRecordLength() * self.frameNum)
class Window(QtWidgets.QWidget):
flag = 1
count = 1
def __init__(self, parent=None):
super().__init__(parent)
self.canvas = MplCanvas()
self.toolbar = NavigationToolbar(self.canvas, self)
# Just some button
self.button1 = QtWidgets.QPushButton('StartPlot')
self.button1.clicked.connect(self.startplot)
self.button2 = QtWidgets.QPushButton('Zoom')
self.button2.clicked.connect(self.zoom)
self.button3 = QtWidgets.QPushButton('Pan')
self.button3.clicked.connect(self.pan)
self.button4 = QtWidgets.QPushButton('Home')
self.button4.clicked.connect(self.home)
self.button5 = QtWidgets.QPushButton('Save')
self.button5.clicked.connect(self.save)
self.button6 = QtWidgets.QPushButton('Clear')
self.button6.clicked.connect(self.clear)
self.button7 = QtWidgets.QPushButton('StopPlot')
self.button7.clicked.connect(self.stopplot)
# set the layout
layout = QtWidgets.QVBoxLayout()
layout.addWidget(self.toolbar)
layout.addWidget(self.canvas)
btnlayout = QtWidgets.QHBoxLayout()
btnlayout.addWidget(self.button1)
btnlayout.addWidget(self.button2)
btnlayout.addWidget(self.button3)
btnlayout.addWidget(self.button4)
btnlayout.addWidget(self.button5)
btnlayout.addWidget(self.button6)
btnlayout.addWidget(self.button7)
self.dataX = []
self.dataY = []
self.dataX2 = []
self.dataY2 = []
qw = QtWidgets.QWidget(self)
qw.setLayout(btnlayout)
layout.addWidget(qw)
self.setLayout(layout)
def getdata(self):
return numpy.array([i for i in range(100)])
def home(self):
self.toolbar.home()
#self.toolbar2.home()
def zoom(self):
self.toolbar.zoom()
#self.toolbar2.zoom()
def pan(self):
self.toolbar.pan()
#self.toolbar2.pan()
def save(self):
self.toolbar.save_figure()
def startplot(self):
''' plot some random stuff '''
self.tData = threading.Thread(name="dataGenerator",
target=self.generateData)
self.__flag = threading.Event()
self.__flag.set()
self.__running = threading.Event()
self.__running.set()
self.tData.start()
def stopplot(self):
if Window.flag == 1:
Window.flag = -2
self.__running.clear()
else:
Window.flag = 1
self.__running.set()
def generateData(self):
# 在当前文件夹下创建保存数据的csv文件
self.Now_Time = str(datetime.now())[:10]
if not os.path.exists('./data'):
os.mkdir('./data')
if not os.path.exists('./data/DataFile'):
os.mkdir('./data/DataFile')
with open('./data/DataFile/{time}.csv'.format(time=self.Now_Time),
'a+',
newline='') as f1:
writer1 = csv.DictWriter(f1, ['Time', 'Data'])
writer1.writeheader()
while 1:
if self.__running.isSet():
# 串口读取数据，数据的读取和解析需要自己改写,然后将输出值传递给s
# s = ser.readline()
s = 100
newTime = date2num(datetime.now())
self.dataX.append(newTime)
self.dataY.append(s)
self.canvas.plot(self.dataX, self.dataY)
# 将数据保存到文件中去
with open('./data/DataFile/{time}.csv'.format(
time=self.Now_Time),
'a+',
newline='') as f2:
writer2 = csv.writer(f2)
writer2.writerow([str(datetime.now())[11:], str(s)])
time.sleep(0.5)
def clear(self):
self.axes.cla()
self.axes.set_xlabel('Frequency')
self.axes.set_ylabel('Amplification')
self.axes.set_title('test')
self.tData.join()
self.canvas.draw()
class Window(QtWidgets.QMainWindow, Ui_MainWindow):
"""
重写修改该类即可实现自定义后端界面，相加什么按钮可以随便加，目前还只是个demo
self.canvas.draw() 每执行该函数，图形重绘
"""
def __init__(self, figure):
super(Window, self).__init__()
self.setupUi(self) # 先执行父类方法，以产生成员变量
self.figure = figure
self.canvas = FigureCanvas(self.figure) # 这里的canvas就是曲线图
self.toolbar = NavigationToolbar(self.canvas, self)
self.toolbar.hide() # 隐藏QT原来的工具栏
# self.graphicsView.addWidget(self.canvas) # 将canvas渲染到布局中
self.scene = QGraphicsScene()
self.scene.addWidget(self.canvas)
self.graphicsView.setScene(self.scene)
self.graphicsView.show()
self.actionX_X.triggered.connect(self.axes_control_slot)
# 初始化当前界面
self.init_gui()
# 槽函数连接
# 当前子图对象切换
self.current_path = os.path.dirname(__file__)
self.saveAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/save.png')), 'save',
self)
self.saveAction.setShortcut('Ctrl+S')
self.saveAction.triggered.connect(self.save_slot)
self.toolBar.addAction(self.saveAction)
self.settingAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/setting.png')),
'setting', self)
self.settingAction.triggered.connect(self.axes_control_slot)
self.toolBar.addAction(self.settingAction)
self.homeAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/home.png')), 'home',
self)
self.homeAction.setShortcut('Ctrl+H')
self.homeAction.triggered.connect(self.home_slot)
self.toolBar.addAction(self.homeAction)
self.backAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/back.png')), 'back',
self)
self.backAction.triggered.connect(self.back_slot)
self.toolBar.addAction(self.backAction)
self.frontAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/front.png')), 'front',
self)
self.frontAction.triggered.connect(self.front_slot)
self.toolBar.addAction(self.frontAction)
self.zoomAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/zoom.png')), 'zoom',
self)
self.zoomAction.triggered.connect(self.zoom_slot)
self.toolBar.addAction(self.zoomAction)
self.panAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/pan.png')), '平移',
self)
self.panAction.triggered.connect(self.pan_slot)
self.toolBar.addAction(self.panAction)
self.rotateAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/rotate.png')),
'rotate', self)
self.rotateAction.triggered.connect(self.rotate_slot)
self.toolBar.addAction(self.rotateAction)
self.textAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/text.png')), 'text',
self)
self.textAction.triggered.connect(self.add_text_slot)
self.toolBar.addAction(self.textAction)
self.rectAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/rect.png')), 'rect',
self)
self.rectAction.triggered.connect(self.add_rect_slot)
self.toolBar.addAction(self.rectAction)
self.ovalAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/oval.png')), 'oval',
self)
self.ovalAction.triggered.connect(self.add_oval_slot)
self.toolBar.addAction(self.ovalAction)
self.arrowAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/arrow.png')), 'arrow',
self)
self.arrowAction.triggered.connect(self.add_arrow_slot)
self.toolBar.addAction(self.arrowAction)
self.pointAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/point.png')), 'point',
self)
self.pointAction.triggered.connect(self.add_point_slot)
self.toolBar.addAction(self.pointAction)
self.styleAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/style.png')), 'style',
self)
self.styleAction.triggered.connect(self.add_style_slot)
self.toolBar.addAction(self.styleAction)
# 将下拉菜单放在最右边
self.toolBar.addSeparator()
spacer = QWidget()
spacer.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.toolBar.addWidget(spacer)
self.toolBar.addWidget(self.comboBox)
# 以上为工具栏1
self.gridAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/grid.png')),
'显示/隐藏网格', self)
self.gridAction.triggered.connect(self.show_grid_slot)
self.toolBar_2.addAction(self.gridAction)
self.legendAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/legend.png')),
'显示/隐藏图例', self)
self.legendAction.triggered.connect(self.show_legend_slot)
self.toolBar_2.addAction(self.legendAction)
self.colorbarAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/colorbar.png')),
'显示/隐藏colorbar', self)
self.colorbarAction.triggered.connect(self.show_colorbar_slot)
self.toolBar_2.addAction(self.colorbarAction)
self.layoutAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/layout.png')), '改变布局',
self)
# self.layoutAction.triggered.connect(self.show_layout_slot)
self.toolBar_2.addAction(self.layoutAction)
self.mainViewAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/mainView.png')),
'mainView', self)
self.mainViewAction.triggered.connect(self.mainView_slot)
self.toolBar_2.addAction(self.mainViewAction)
self.leftViewAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/leftView.png')),
'leftView', self)
self.leftViewAction.triggered.connect(self.leftView_slot)
self.toolBar_2.addAction(self.leftViewAction)
self.rightViewAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/rightView.png')),
'rightView', self)
self.rightViewAction.triggered.connect(self.rightView_slot)
self.toolBar_2.addAction(self.rightViewAction)
self.topViewAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/topView.png')),
'topView', self)
self.topViewAction.triggered.connect(self.topView_slot)
self.toolBar_2.addAction(self.topViewAction)
self.bottomViewAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/bottomView.png')),
'bottomView', self)
self.bottomViewAction.triggered.connect(self.bottomView_slot)
self.toolBar_2.addAction(self.bottomViewAction)
self.backViewAction = QAction(
QIcon(os.path.join(self.current_path, 'icons/backView.png')),
'backView', self)
self.backViewAction.triggered.connect(self.backView_slot)
self.toolBar_2.addAction(self.backViewAction)
# 样式右键菜单功能集
self.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.customContextMenuRequested.connect(self.rightMenuShow)
self.rightMenuShow() # 创建上下文菜单
self.comboBox.currentIndexChanged.connect(self.combobox_slot)
# 获取子图对象
self.axes = self.canvas.figure.get_axes()
if not self.axes:
QtWidgets.QMessageBox.warning(self.canvas.parent(), "Error",
"There are no axes to edit.")
return
elif len(self.axes) == 1:
self.current_subplot, = self.axes
titles = ['图1']
else:
titles = ['图' + str(i + 1) for i in range(len(self.axes))]
# 将三维图的初始视角保存下来，便于旋转之后可以复原
self.init_views = [(index, item.azim, item.elev)
for index, item in enumerate(self.axes)
if hasattr(item, 'azim')]
self.comboBox.addItems(titles)
# 鼠标拖拽，实现三维图形旋转功能
self.canvas.mpl_connect('motion_notify_event', self.on_rotate)
# 鼠标拖拽，实现画矩形功能
self.canvas.mpl_connect('motion_notify_event', self.add_rect)
self.mouse_pressed = False
# 鼠标拖拽，实现画椭圆和按住shift画圆的功能实现
self.canvas.mpl_connect('motion_notify_event', self.add_oval)
# 画箭头
self.canvas.mpl_connect('motion_notify_event', self.add_arrow)
# 画点，并且按住点能够移动
self.canvas.mpl_connect('button_press_event', self.add_point)
self.canvas.mpl_connect('motion_notify_event', self.add_point)
self.canvas.mpl_connect('button_release_event', self.add_point)
self.canvas.mpl_connect('pick_event', self.add_point)
self.press_time = 0
# 记录鼠标运动的位置
self.rotate_mouse_point = None
self.canvas.mpl_connect('button_press_event', self.add_text)
# 为曲线添加样式的功能实现
self.canvas.mpl_connect('button_press_event', self.add_style)
self.canvas.mpl_connect('pick_event', self.add_style)
# 为图例绑定监听事件
self.canvas.mpl_connect('button_press_event', self.change_legend)
self.canvas.mpl_connect('pick_event', self.change_legend)
# 所有的按钮标志
self.make_flag_invalid()
self.artist = None
for ax in self.axes:
for line in ax.lines:
line.set_picker(True)
line.set_pickradius(5)
def make_flag_invalid(self):
self.add_rect_flag = False
self.add_oval_flag = False
self.add_text_flag = False
self.rotate_flag = False
self.home_flag = False
self.pan_flag = False
self.zoom_flag = False
self.add_arrow_flag = False
self.add_point_flag = False
self.add_style_flag = False
self.show_grid_flag = False
self.show_legend_flag = False
# 禁用移动和缩放
self.toolbar.mode = None
def home_slot(self):
self.make_flag_invalid()
self.home_flag = not self.home_flag
self.home()
def home(self):
"""
matplotlib lines里面放曲线，patches可以放图形，artists也可以放东西，设为空则可以删除对应的对象
"""
if not self.home_flag:
return
self.toolbar.home()
# 将三维图视角还原
for item in self.init_views:
self.axes[item[0]].view_init(azim=item[1], elev=item[2])
# 将所有添加的形状去除
for item in self.axes:
item.patches = []
item.artists = [] # 去除画在图中的点，是否需要去掉有待考究
self.canvas.draw()
def zoom_slot(self):
self.make_flag_invalid()
self.zoom_flag = not self.zoom_flag
self.zoom()
def zoom(self):
if not self.zoom_flag:
return
self.toolbar.zoom()
def pan_slot(self):
self.make_flag_invalid()
self.pan_flag = not self.pan_flag
self.pan()
def pan(self):
if not self.pan_flag:
return
self.toolbar.pan()
def save_slot(self):
self.toolbar.save_figure()
def front_slot(self):
self.toolbar._nav_stack.forward()
self.toolbar._update_view()
def back_slot(self):
self.toolbar._nav_stack.back()
self.toolbar._update_view()
def add_text_slot(self):
self.make_flag_invalid()
self.add_text_flag = not self.add_text_flag
def add_text(self, event):
if not self.add_text_flag:
return
if self.add_text_flag and event.xdata and event.ydata and not hasattr(
event.inaxes, 'azim'):
text, ok = QtWidgets.QInputDialog.getText(self.canvas.parent(),
'输入文字', '添加注释')
if ok and text:
event.inaxes.text(event.xdata, event.ydata, text)
# plt.text(event.xdata, event.ydata, text)
self.canvas.draw()
def rotate_slot(self):
self.make_flag_invalid()
self.rotate_flag = not self.rotate_flag
def on_rotate(self, event):
"""
通过观察发现，旋转时产生的xdata,ydata是以图像中心为原点，正负0.1范围内的数值
"""
if not self.rotate_flag:
return
# 如果鼠标移动过程有按下，视为拖拽，判断当前子图是否有azim属性来判断当前是否3D
if event.button and hasattr(event.inaxes, 'azim'):
for item in self.init_views:
if self.axes[item[0]] == event.inaxes:
delta_azim = 180 * event.xdata * -1 / 0.1
delta_elev = 180 * event.ydata / 0.1
azim = delta_azim + item[1]
elev = delta_elev + item[2]
event.inaxes.view_init(azim=azim, elev=elev)
self.canvas.draw()
def add_rect_slot(self):
# 除了本标记，其余全置False
self.make_flag_invalid()
self.add_rect_flag = not self.add_rect_flag
def add_rect(self, event):
if not self.add_rect_flag:
return
if not event.button and event.inaxes:
self.ax_init = event
if self.mouse_pressed and event.inaxes.patches:
event.inaxes.add_patch(event.inaxes.patches[0])
self.mouse_pressed = False
# 仅能在二维图形中画矩形，鼠标按下，且当前子图和初始点的子图相同
try:
if event.button and event.inaxes and event.inaxes == self.ax_init.inaxes and not hasattr(
event.inaxes, 'azim'):
self.mouse_pressed = True
if event.inaxes.patches:
event.inaxes.patches.pop()
rect = plt.Rectangle((self.ax_init.xdata, self.ax_init.ydata),
event.xdata - self.ax_init.xdata,
event.ydata - self.ax_init.ydata,
fill=False,
edgecolor='red',
linewidth=1)
event.inaxes.add_patch(rect)
self.canvas.draw()
self.canvas.flush_events()
except Exception:
pass
def add_oval_slot(self):
self.make_flag_invalid()
self.add_oval_flag = not self.add_oval_flag
def add_oval(self, event):
if not self.add_oval_flag:
return
if not event.button and event.inaxes:
self.ax_init = event
if self.mouse_pressed and event.inaxes.patches:
event.inaxes.add_patch(event.inaxes.patches[0])
self.mouse_pressed = False
try:
if event.button and event.inaxes and event.inaxes == self.ax_init.inaxes and not hasattr(
event.inaxes, 'azim'):
self.mouse_pressed = True
if event.inaxes.patches:
event.inaxes.patches.pop()
oval = Ellipse(xy=(self.ax_init.xdata, self.ax_init.ydata),
width=abs(event.xdata - self.ax_init.xdata) * 2,
height=abs(event.ydata - self.ax_init.ydata) *
2,
angle=0,
fill=False,
edgecolor='red',
linewidth=1)
event.inaxes.add_patch(oval)
self.canvas.draw()
self.canvas.flush_events()
except Exception:
pass
def add_arrow_slot(self):
self.make_flag_invalid()
self.add_arrow_flag = not self.add_arrow_flag
def add_arrow(self, event):
if not self.add_arrow_flag:
return
if not event.button and event.inaxes:
self.ax_init = event
if self.mouse_pressed and event.inaxes.patches:
event.inaxes.add_patch(event.inaxes.patches[0])
self.mouse_pressed = False
try:
if event.button and event.inaxes and event.inaxes == self.ax_init.inaxes and not hasattr(
event.inaxes, 'azim'):
self.mouse_pressed = True
if event.inaxes.patches:
event.inaxes.patches.pop()
arrow = event.inaxes.arrow(self.ax_init.xdata,
self.ax_init.ydata,
event.xdata - self.ax_init.xdata,
event.ydata - self.ax_init.ydata,
width=0.01,
length_includes_head=True,
head_width=0.05,
head_length=0.1,
fc='r',
ec='r')
event.inaxes.add_patch(arrow)
# 请恕我无知，我也不懂这里为什么还要pop一次，我不想思考，但的确这样是正确的。
if event.inaxes.patches:
event.inaxes.patches.pop()
self.canvas.draw()
self.canvas.flush_events()
except Exception:
pass
def add_point_slot(self):
self.make_flag_invalid()
self.add_point_flag = not self.add_point_flag
# def add_point(self, event):
# if not self.add_point_flag:
# return
# if event.inaxes and event.button and not hasattr(event.inaxes, 'azim'):
# x_range = np.array(event.inaxes.get_xlim())
# y_range = np.array(event.inaxes.get_ylim())
# self.offset = np.sqrt(np.sum((x_range - y_range) ** 2)) / 20 # 将坐标轴范围的1/50视为误差
# self.nearest_point = None
# d_min = 10 * self.offset
# for point in event.inaxes.artists:
# xt, yt = point.get_data()
# d = ((xt - event.xdata) ** 2 + (yt - event.ydata) ** 2) ** 0.5
# if d <= self.offset and d < d_min: # 如果在误差范围内，移动该点
# d_min = d
# self.nearest_point = point
# if self.nearest_point:
# new_point = Line2D([event.xdata], [event.ydata], ls="",
# marker='o', markerfacecolor='r',
# animated=False)
# event.inaxes.add_artist(new_point)
# event.inaxes.artists.remove(self.nearest_point)
# self.canvas.restore_region(self.bg)
# event.inaxes.draw_artist(new_point)
# self.canvas.blit(event.inaxes.bbox)
# self.bg = self.canvas.copy_from_bbox(event.inaxes.bbox)
def add_point(self, event):
if not self.add_point_flag:
return
if event.name == 'pick_event':
self.artist = event.artist
return
if event.name == 'button_press_event' and not self.artist and hasattr(
event, 'inaxes') and not hasattr(event.inaxes, 'azim'):
point = Line2D([event.xdata], [event.ydata],
ls="",
marker='o',
markerfacecolor='r',
animated=False,
pickradius=5,
picker=True)
event.inaxes.add_artist(point)
self.canvas.draw()
return
if event.name == 'motion_notify_event' and self.artist and hasattr(
event, 'inaxes') and event.button and not hasattr(
event.inaxes, 'azim'):
xy = self.artist.get_data()
if len(xy[0]) == 1: # 判断该对象是否是一个点。
self.artist.set_data(([event.xdata], [event.ydata]))
self.canvas.draw()
if event.name == 'button_release_event':
self.artist = None
def add_style_slot(self):
self.make_flag_invalid()
self.add_style_flag = not self.add_style_flag
def add_style(self, event):
if not self.add_style_flag:
return
if event.name == 'pick_event':
self.artist = event.artist
if self.artist and event.name == 'button_press_event':
for line in event.inaxes.lines:
if self.artist != line:
line.set_alpha(0.5)
else:
line.set_alpha(1)
self.canvas.draw_idle()
if event.button == 3:
self.contextMenu.popup(QCursor.pos()) # 2菜单显示的位置
self.contextMenu.show()
return
elif event.button == 1:
self.artist = None
return
if not self.artist and event.name == 'button_press_event' and event.button == 1:
for line in event.inaxes.lines:
line.set_alpha(1)
self.canvas.draw_idle()
def show_legend_slot(self):
legend_titles = []
for index in range(len(self.current_subplot.lines)):
legend_titles.append('curve ' + str(index + 1)) # 从1开始算
if self.current_subplot.lines: # 如果存在曲线才允许画图例
leg = self.current_subplot.legend(self.current_subplot.lines,
legend_titles)
leg.set_draggable(True) # 设置legend可拖拽
for legline in leg.get_lines():
legline.set_pickradius(10)
legline.set_picker(True) # 给每个legend设置可点击
self.canvas.draw()
def change_legend(self, event):
if event.name == 'pick_event':
self.artist = event.artist
if self.artist and event.name == 'button_press_event' and event.button == 3:
self.contextMenu.popup(QCursor.pos()) # 2菜单显示的位置
self.contextMenu.show()
def show_colorbar_slot(self):
# print(self.current_subplot.curves)
pass
# self.canvas.figure.colorbar(self.canvas,self.current_subplot)
def rightMenuShow(self):
self.contextMenu = QMenu()
self.actionStyle = self.contextMenu.addAction('修改曲线样式')
self.actionLegend = self.contextMenu.addAction('修改图例样式')
self.actionCurve = self.contextMenu.addAction('修改曲线类型')
self.actionStyle.triggered.connect(self.styleHandler)
self.actionLegend.triggered.connect(self.legendHandler)
self.actionLegend.triggered.connect(self.curveHandler)
def styleHandler(self):
print(self.artist)
def legendHandler(self):
print(self.artist)
def curveHandler(self):
print(self.artist)
def mainView_slot(self):
if hasattr(self.current_subplot, 'azim'):
self.current_subplot.view_init(azim=0, elev=0)
self.canvas.draw()
def leftView_slot(self):
if hasattr(self.current_subplot, 'azim'):
self.current_subplot.view_init(azim=90, elev=0)
self.canvas.draw()
def rightView_slot(self):
if hasattr(self.current_subplot, 'azim'):
self.current_subplot.view_init(azim=-90, elev=0)
self.canvas.draw()
def topView_slot(self):
if hasattr(self.current_subplot, 'azim'):
self.current_subplot.view_init(azim=0, elev=90)
self.canvas.draw()
def bottomView_slot(self):
if hasattr(self.current_subplot, 'azim'):
self.current_subplot.view_init(azim=0, elev=-90)
self.canvas.draw()
def backView_slot(self):
if hasattr(self.current_subplot, 'azim'):
self.current_subplot.view_init(azim=180, elev=0)
self.canvas.draw()
def show_grid_slot(self):
self.show_grid_flag = not self.show_grid_flag
self.current_subplot.grid(self.show_grid_flag)
self.canvas.draw_idle()
def init_gui(self):
self.toolbar._update_view()
def combobox_slot(self):
self.current_subplot = self.axes[
self.comboBox.currentIndex()] # 将当前选择付给子图对象
def axes_control_slot(self):
if not self.current_subplot:
QtWidgets.QMessageBox.warning(self.canvas.parent(), "错误",
"没有可选的子图！")
return
Ui_Form_Manager(self.current_subplot, self.canvas)
def show(self):
super().show()
class Window(QtWidgets.QWidget):
def __init__(self, parent=None):
super(Window, self).__init__(parent)
# self.figure = plt.figure()
# self.axes = self.figure.add_subplot(111)
# # We want the axes cleared every time plot() is called
# self.axes.hold(False)
# self.canvas = FigureCanvas(self.figure)
# self.canvas = Figure_Canvas()
# self.canvas.plotleadone()
self.viewer = MyView()
self.toolbar = NavigationToolbar(self.viewer.canvas, self)
# self.toolbar.hide()
# Just some button
self.button1 = QtWidgets.QPushButton('Plot')
self.button1.clicked.connect(self.plot)
self.button2 = QtWidgets.QPushButton('Zoom')
self.button2.clicked.connect(self.zoom)
self.button3 = QtWidgets.QPushButton('Pan')
self.button3.clicked.connect(self.pan)
self.button4 = QtWidgets.QPushButton('Home')
self.button4.clicked.connect(self.home)
# set the layout
layout = QtWidgets.QVBoxLayout()
layout.addWidget(self.toolbar)
#
# self.graphicview = QGraphicsView() # 第一步，创建一个QGraphicsView
# self.graphicview.setObjectName("figure_graphicview")
# graphicscene = QGraphicsScene() # 第三步，创建一个QGraphicsScene，因为加载的图形（FigureCanvas）不能直接放到graphicview控件中，必须先放到graphicScene，然后再把graphicscene放到graphicview中
# graphicscene.addWidget(self.canvas) # 第四步，把图形放到QGraphicsScene中，注意：图形是作为一个QWidget放到QGraphicsScene中的
# self.graphicview.setScene(graphicscene) # 第五步，把QGraphicsScene放入QGraphicsView
# self.graphicview.show() # 最后，调用show方法呈现图形！Voila!!
layout.addWidget(self.viewer)
# layout.addWidget(self.graphicview)
# layout.addWidget(self.canvas)
btnlayout = QtWidgets.QHBoxLayout()
btnlayout.addWidget(self.button1)
btnlayout.addWidget(self.button2)
btnlayout.addWidget(self.button3)
btnlayout.addWidget(self.button4)
qw = QtWidgets.QWidget(self)
qw.setLayout(btnlayout)
layout.addWidget(qw)
self.setLayout(layout)
def home(self):
self.toolbar.home()
def zoom(self):
self.toolbar.zoom()
def pan(self):
self.toolbar.pan()
def plot(self):
''' plot some random stuff '''
# data = [random.ra
pass
class Window(QtWidgets.QMainWindow, Data_to_plot):
def __init__(self, parent=None):
super().__init__(parent)
wid = QWidget(self)
self.setCentralWidget(wid)
self.figure = plt.figure()
self.axes = self.figure.add_subplot(111)
# We want the axes cleared every time plot() is called
self.axes.hold(False)
self.canvas = FigureCanvas(self.figure)
self.toolbar = NavigationToolbar(self.canvas, self)
self.button_to_plot = QtWidgets.QPushButton('Plot')
self.button_to_plot.clicked.connect(self.plot)
self.button_to_plot_twinx = QtWidgets.QPushButton('Plot twinx')
self.button_to_plot_twinx.clicked.connect(self.plot_twinx)
self.button_to_zoom = QtWidgets.QPushButton('Zoom')
self.button_to_zoom.clicked.connect(self.zoom)
self.button_to_home = QtWidgets.QPushButton('Home')
self.button_to_home.clicked.connect(self.home)
self.button_to_save_fig = QtWidgets.QPushButton('Save')
self.button_to_save_fig.clicked.connect(self.save_fig)
self.button_to_set_xlabel = QtWidgets.QPushButton('Set xlabel')
self.button_to_set_xlabel.clicked.connect(self.set_xlabel)
self.text_xlable = QtWidgets.QLineEdit()
self.button_to_set_ylabel = QtWidgets.QPushButton('Set ylabel')
self.button_to_set_ylabel.clicked.connect(self.set_ylabel)
self.text_ylable = QtWidgets.QLineEdit()
self.button_to_set_title = QtWidgets.QPushButton('Set title')
self.button_to_set_title.clicked.connect(self.set_title)
self.text_title = QtWidgets.QLineEdit()
self.button_to_import_data = QtWidgets.QPushButton('import data x,y')
self.button_to_import_data.clicked.connect(self.import_data)
self.textEdit = QtWidgets.QTextEdit()
self.button_to_import_data_xyz = QtWidgets.QPushButton(
'import data x,y,z')
# set the layout
layout = QtWidgets.QVBoxLayout()
layout.addWidget(self.toolbar)
layout.addWidget(self.canvas)
btnlayout1 = QtWidgets.QHBoxLayout()
btnlayout1.addWidget(self.button_to_plot)
btnlayout1.addWidget(self.button_to_plot_twinx)
btnlayout1.addWidget(self.button_to_zoom)
btnlayout1.addWidget(self.button_to_home)
btnlayout1.addWidget(self.button_to_save_fig)
btnlayout2 = QtWidgets.QHBoxLayout()
btnlayout2.addWidget(self.button_to_set_xlabel)
btnlayout2.addWidget(self.button_to_set_ylabel)
btnlayout2.addWidget(self.button_to_set_title)
btnlayout3 = QtWidgets.QHBoxLayout()
btnlayout3.addWidget(self.text_xlable)
btnlayout3.addWidget(self.text_ylable)
btnlayout3.addWidget(self.text_title)
btnlayout4 = QtWidgets.QHBoxLayout()
btnlayout4.addWidget(self.button_to_import_data)
btnlayout4.addWidget(self.textEdit)
btnlayout4.addWidget(self.button_to_import_data_xyz)
qw = QtWidgets.QWidget(self)
qw.setLayout(btnlayout1)
qw2 = QtWidgets.QWidget(self)
qw2.setLayout(btnlayout2)
qw3 = QtWidgets.QWidget(self)
qw3.setLayout(btnlayout3)
qw4 = QtWidgets.QWidget(self)
qw4.setLayout(btnlayout4)
layout.addWidget(qw)
layout.addWidget(qw2)
layout.addWidget(qw3)
layout.addWidget(qw4)
wid.setLayout(layout)
def home(self):
self.toolbar.home()
def zoom(self):
self.toolbar.zoom()
def import_data(self):
fname = QtWidgets.QFileDialog.getOpenFileName(self, 'Open file',
os.getcwd())
if fname[0]:
try:
self.x, self.y = np.loadtxt(fname[0], unpack=True, skiprows=1)
with open(fname[0], 'r') as f:
data = f.read()
self.textEdit.setText(data)
except Exception as err:
print(err)
def import_data_xyz(self):
fname = QtWidgets.QFileDialog.getOpenFileName(self, 'Open file',
os.getcwd())
if fname[0]:
try:
self.x, self.y, self.z = np.loadtxt(fname[0],
unpack=True,
skiprows=1)
with open(fname[0], 'r') as f:
data = f.read()
self.textEdit.setText(data)
except Exception as err:
print(err)
def plot(self):
self.axes.plot(self.x, self.y, 'bo')
self.canvas.draw()
def plot_twinx(self):
self.axes.plot(self.x, self.y, 'bo')
self.axes2 = self.axes.twinx()
self.axes2.plot(self.x, self.z, 'r<')
self.canvas.draw()
def set_title(self):
self.title = self.text_title.text()
self.axes.set_title(self.title)
self.canvas.draw()
def set_xlabel(self):
self.xlabel = self.text_xlable.text()
self.axes.set_xlabel(self.xlabel)
self.canvas.draw()
def set_ylabel(self):
self.ylabel = self.text_ylable.text()
self.axes.set_ylabel(self.ylabel)
self.canvas.draw()
def save_fig(self):
path_to_save_fig = os.path.join(os.getcwd(), "wykres.jpg")
plt.savefig(path_to_save_fig)
