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 be38083482b48ce714174a23a1ad771180fbaf07..dd6450fdbce88f7b364018ffe93634e86ac9cda1 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 0000000000000000000000000000000000000000..86627b172d37bacf4fd527c604d9603807be574d --- /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