From f220783aca9340ab1d8dedeec2721837de782463 Mon Sep 17 00:00:00 2001 From: Djalim Simaila <DjalimS.pro@outlook.fr> Date: Fri, 21 Apr 2023 17:54:22 +0200 Subject: [PATCH] added visplot and matplotlib cross section --- utils/graph2D/{plots.py => mpl_render.py} | 10 ++++++++++ utils/graph2D/visplot_render.py | 22 ++++++++++++++++++++++ 2 files changed, 32 insertions(+) rename utils/graph2D/{plots.py => mpl_render.py} (52%) create mode 100644 utils/graph2D/visplot_render.py diff --git a/utils/graph2D/plots.py b/utils/graph2D/mpl_render.py similarity index 52% rename from utils/graph2D/plots.py rename to utils/graph2D/mpl_render.py index be38083..dd6450f 100644 --- a/utils/graph2D/plots.py +++ b/utils/graph2D/mpl_render.py @@ -8,4 +8,14 @@ def render2D(values:list): fig = plt.figure() ax = fig.add_subplot() ax.plot(values) + plt.show() + +def cross_section(x:list,y:list): + """ + Render a 2D model using matplotlib + :param values: A list with the values + """ + fig = plt.figure() + ax = fig.add_subplot() + ax.scatter(x,y) plt.show() \ No newline at end of file diff --git a/utils/graph2D/visplot_render.py b/utils/graph2D/visplot_render.py new file mode 100644 index 0000000..86627b1 --- /dev/null +++ b/utils/graph2D/visplot_render.py @@ -0,0 +1,22 @@ +import vispy.plot as vp +import numpy as np + + + +def render2D(values:list): + fig = vp.Fig(size=(600, 500), show=False) + plotwidget = fig[0, 0] + fig.title = "bollu" + plotwidget.plot(values) + plotwidget.colorbar(position="top", cmap="autumn") + fig.show(run=True) + + +def cross_section(x:list,y:list): + color = (0.3, 0.5, 0.8) + fig = vp.Fig(show=False) + line = fig[0:4, 0:4].plot(np.column_stack((x,y)), symbol='o', width=0, + face_color=color + (0.02,), edge_color=None, + marker_size=8) + line.set_gl_state(depth_test=False) + fig.show(run=True) \ No newline at end of file -- GitLab