Skip to content
Snippets Groups Projects
Commit 5897f874 authored by Djalim Simaila's avatar Djalim Simaila
Browse files

added labels to graphs axis

parent 39e6f494
No related branches found
No related tags found
1 merge request!15added labels to graphs axis
This commit is part of merge request !15. Comments created here will be created in the context of that merge request.
import vispy.plot as vp
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
: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)
plotwidget = fig[0, 0]
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:
fig.show(run=True)
else:
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
: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)
fig = vp.Fig(show=False)
line = fig[0:4, 0:4].plot(np.column_stack((x_values,y_values)), symbol='disc', width=0,
face_color=color, edge_color=None,
marker_size=8,title=title)
line = fig[0:4, 0:4].plot(np.column_stack((x_values,y_values)),
symbol='disc',
width=0,
face_color=color,
edge_color=None,
marker_size=8,
title=title,
xlabel=xlabel,
ylabel=ylabel)
line.set_gl_state(depth_test=False)
if show:
fig.show(run=True)
......
......@@ -249,11 +249,25 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
if graph_type == "Mesh3D":
current_slot.addWidget(render3D(obj,False).native)
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":
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":
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):
"""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment