added labels to graphs axis
This commit is contained in:
parent
39e6f494bd
commit
5897f874b5
@ -1,7 +1,7 @@
|
|||||||
import vispy.plot as vp
|
import vispy.plot as vp
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
def render2D(values:list,title:str,show:bool=True):
|
def render2D(values:list,title:str,xlabel="",ylabel="",show:bool=True):
|
||||||
"""
|
"""
|
||||||
Render a 2D plot using vispy
|
Render a 2D plot using vispy
|
||||||
:param values: A list with the values
|
:param values: A list with the values
|
||||||
@ -9,13 +9,18 @@ def render2D(values:list,title:str,show:bool=True):
|
|||||||
fig = vp.Fig(size=(600, 500), show=False)
|
fig = vp.Fig(size=(600, 500), show=False)
|
||||||
plotwidget = fig[0, 0]
|
plotwidget = fig[0, 0]
|
||||||
fig.title = title
|
fig.title = title
|
||||||
plotwidget.plot(values,marker_size=0, width=2,title=title)
|
plotwidget.plot(values,
|
||||||
|
marker_size=0,
|
||||||
|
width=2,
|
||||||
|
title=title,
|
||||||
|
xlabel=xlabel,
|
||||||
|
ylabel=ylabel)
|
||||||
if show:
|
if show:
|
||||||
fig.show(run=True)
|
fig.show(run=True)
|
||||||
else:
|
else:
|
||||||
return fig
|
return fig
|
||||||
|
|
||||||
def cross_section(x_values:list, y_values:list,title:str, show:bool=True ):
|
def cross_section(x_values:list, y_values:list,title:str,xlabel="",ylabel="",show:bool=True ):
|
||||||
"""
|
"""
|
||||||
Render a 2D cross section using vispy
|
Render a 2D cross section using vispy
|
||||||
:param x: A list with the x values
|
:param x: A list with the x values
|
||||||
@ -23,9 +28,15 @@ def cross_section(x_values:list, y_values:list,title:str, show:bool=True ):
|
|||||||
"""
|
"""
|
||||||
color = (0.3, 0.5, 0.8,.8)
|
color = (0.3, 0.5, 0.8,.8)
|
||||||
fig = vp.Fig(show=False)
|
fig = vp.Fig(show=False)
|
||||||
line = fig[0:4, 0:4].plot(np.column_stack((x_values,y_values)), symbol='disc', width=0,
|
line = fig[0:4, 0:4].plot(np.column_stack((x_values,y_values)),
|
||||||
face_color=color, edge_color=None,
|
symbol='disc',
|
||||||
marker_size=8,title=title)
|
width=0,
|
||||||
|
face_color=color,
|
||||||
|
edge_color=None,
|
||||||
|
marker_size=8,
|
||||||
|
title=title,
|
||||||
|
xlabel=xlabel,
|
||||||
|
ylabel=ylabel)
|
||||||
line.set_gl_state(depth_test=False)
|
line.set_gl_state(depth_test=False)
|
||||||
if show:
|
if show:
|
||||||
fig.show(run=True)
|
fig.show(run=True)
|
||||||
|
|||||||
@ -249,11 +249,25 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
|
|||||||
if graph_type == "Mesh3D":
|
if graph_type == "Mesh3D":
|
||||||
current_slot.addWidget(render3D(obj,False).native)
|
current_slot.addWidget(render3D(obj,False).native)
|
||||||
if graph_type == "Coupe XZ":
|
if graph_type == "Coupe XZ":
|
||||||
current_slot.addWidget(cross_section(obj.get_x(),obj.get_z(),"Coupe XZ",False).native)
|
current_slot.addWidget(cross_section(obj.get_x(),
|
||||||
|
obj.get_z(),
|
||||||
|
"Coupe XZ",
|
||||||
|
"X (en mm)",
|
||||||
|
"Z (en mm)",
|
||||||
|
False).native)
|
||||||
if graph_type == "Coupe YZ":
|
if graph_type == "Coupe YZ":
|
||||||
current_slot.addWidget(cross_section(obj.get_y(),obj.get_z(),"Coupe YZ",False).native)
|
current_slot.addWidget(cross_section(obj.get_y(),
|
||||||
|
obj.get_z(),
|
||||||
|
"Coupe YZ",
|
||||||
|
"Y (en mm)",
|
||||||
|
"Z (en mm)",
|
||||||
|
False).native)
|
||||||
if graph_type == "Evolution du rayon moyen":
|
if graph_type == "Evolution du rayon moyen":
|
||||||
current_slot.addWidget(render2D(list(zip(discrete_data['Z moy (en mm)'],discrete_data['Rayon moyen (en mm)'])),"Evolution du rayon moyen",False).native)
|
current_slot.addWidget(render2D(list(zip(discrete_data['Z moy (en mm)'],discrete_data['Rayon moyen (en mm)'])),
|
||||||
|
"Evolution du rayon moyen en fonction de Z",
|
||||||
|
"Z (en mm)",
|
||||||
|
"Rayon moyen (en mm)\n",
|
||||||
|
False).native)
|
||||||
|
|
||||||
def clear_graphs(self):
|
def clear_graphs(self):
|
||||||
"""
|
"""
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user