ADV/faculty_project
3
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Загрузить файлы в «GraphWidget»

Browse Source
This commit is contained in:
artem 2023-10-18 21:44:14 +03:00
parent f1ab7c2750
commit 6887247d8d
2 changed files with 46 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
GraphWidget/Graph.py
View File

@ -1,32 +1,16 @@
import math
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
import numexpr as ne
import openpyxl as op
from tkinter import ttk
from matplotlib.figure import Figure from matplotlib.figure import Figure
class Graph: class Graph:
def __init__(self,check_butt_cond,x, y,lab="f(x)",xl="x",yl="y",linetype='k',lims=False): def __init__(self,x, y,labels,mult_subplots=0,xl="x",yl="y",linetype='k',lims=False):
#словарь из настроек? #словарь из настроек?
self.labels=labels
self.__x=x self.__x=x
self.__y=y self.__y=y
self.__fig =Figure(figsize=(5,4),dpi=100) self.__fig =Figure(figsize=(5,4),dpi=100)
self.__chb=check_butt_cond self.__chb=mult_subplots
self.__ax=self.__fig.add_subplot() self.__ax=self.__fig.add_subplot()
#сделать отслеживание за состояние кнопок потом #сделать отслеживание за состояние кнопок потом
""" """
if self.__chb==0: if self.__chb==0:
@ -36,9 +20,10 @@ class Graph:
axes=self.__fig.add_subplot(n) axes=self.__fig.add_subplot(n)
axes.set_title(f"График №{i+1}") axes.set_title(f"График №{i+1}")
c=axes.plot(self.__x[i],self.__y[i],label=lab,linestyle=linetype,) c=axes.plot(self.__x[i],self.__y[i],label=lab)
else: else:
line=self.__ax.plot(self.__x,self.__y,label=lab,) line=self.__ax.plot(self.__x[0],self.__y[0],label=lab,)
print(self.__x)
else: else:
if len(self.__x)>1: #много графиков на одном холсте if len(self.__x)>1: #много графиков на одном холсте
@ -48,48 +33,44 @@ class Graph:
else: else:
line=self.__ax.plot(self.__x,self.__y,label=lab,) line=self.__ax.plot(self.__x[0],self.__y[0],label=lab,)
print(self.__x)
""" """
self.draw_n_func_plot() self.draw_n_func_plot()
self.draw_one_func() self.draw_one_func()
self.draw_subplots() self.draw_subplots()
self.__ax.set_xlabel(xl) self.__ax.set_xlabel(xl)
self.__ax.set_ylabel(yl) self.__ax.set_ylabel(yl)
self.__ax.legend()
self.figure.tight_layout()
def add_vline( ax,x=0, ymin=0, ymax=1): def add_vline( ax,x=0, ymin=0, ymax=1):
ax.axvline(x,ymin,ymax) ax.axvline(x,ymin,ymax)
#принимаем плот и на нем же рисуем #принимаем плот и на нем же рисуем
def add_hline(ax,y=0, xmin=0, xmax=1,): def add_hline(ax,y=0, xmin=0, xmax=1,):
ax.axhline(y, xmin, xmax) ax.axhline(y, xmin, xmax)
def draw_n_func_plot(self,): #много графиков на одном холсте def draw_n_func_plot(self,): #много графиков на одном холсте
if self.__chb==1 and len(self.__x)>1:
for i in range(len(self.__x)):
self.__ax.plot(self.__x[i],self.__y[i])
def draw_one_func(self,lab="f(x)"):
if len(self.__x)==1:
self.__ax.plot(self.__x,self.__y,label=lab,)
def draw_subplots(self,lab="f(x)"):
if self.__chb==0 and len(self.__x)>1: if self.__chb==0 and len(self.__x)>1:
for i in range(len(self.__x)):
self.__ax.plot(self.__x[i],self.__y[i],label=self.labels[i])
def draw_one_func(self,):
if len(self.__x)==1:
self.__ax.plot(self.__x[0],self.__y[0],label=self.labels[0])#
def draw_subplots(self,):
if self.__chb==1 and len(self.__x)>1:
for i in range(len(self.__x)): for i in range(len(self.__x)):
n=int(f"{len(self.__x)}1{i+1}") n=int(f"{len(self.__x)}1{i+1}")
axes=self.__fig.add_subplot(n) axes=self.__fig.add_subplot(n)
axes.set_title(f"График №{i+1}") axes.set_title(f"График №{i+1}")
axes.plot(self.__x[i],self.__y[i],label=lab,) axes.plot(self.__x[i],self.__y[i],label=self.labels[i])
@property @property
def figure(self,): def figure(self,):
return self.__fig return self.__fig

26
GraphWidget/GraphWidget.py
View File

@ -1,5 +1,23 @@
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg import matplotlib
matplotlib.use('Qt5Agg')
import Graph
from PyQt5 import QtWidgets
class GraphWidget(FigureCanvasQTAgg): from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg, NavigationToolbar2QT as NavigationToolbar
def __init__(self, *args):
...
class GraphWidget(QtWidgets.QWidget):
def __init__(self, x, y, labels, mult_plots=False):
super().__init__()
graph = Graph.Graph(x,y,labels, mult_subplots=mult_plots)
sc = FigureCanvasQTAgg(graph.figure)
# Create toolbar, passing canvas as first parament, parent (self, the MainWindow) as second.
toolbar = NavigationToolbar(sc, self)
layout = QtWidgets.QVBoxLayout()
layout.addWidget(toolbar)
layout.addWidget(sc)
self.setLayout(layout)