diff --git a/meshmode/mesh/visualization.py b/meshmode/mesh/visualization.py index 7cacd1234fa68e4c794dfe4536bf804a74fffff4..2b19da7f38c77d0a9d44af862a3164978e5b3ab2 100644 --- a/meshmode/mesh/visualization.py +++ b/meshmode/mesh/visualization.py @@ -29,7 +29,8 @@ import numpy as np # {{{ draw_2d_mesh -def draw_2d_mesh(mesh, draw_numbers=True, **kwargs): +def draw_2d_mesh(mesh, draw_vertex_numbers=True, draw_element_numbers=True, + **kwargs): assert mesh.ambient_dim == 2 import matplotlib.pyplot as pt @@ -55,7 +56,7 @@ def draw_2d_mesh(mesh, draw_numbers=True, **kwargs): patch = mpatches.PathPatch(path, **kwargs) pt.gca().add_patch(patch) - if draw_numbers: + if draw_element_numbers: centroid = (np.sum(elverts, axis=1) / elverts.shape[1]) @@ -68,6 +69,12 @@ def draw_2d_mesh(mesh, draw_numbers=True, **kwargs): ha="center", va="center", bbox=dict(facecolor='white', alpha=0.5, lw=0)) + if draw_vertex_numbers: + for ivert, vert in enumerate(mesh.vertices.T): + pt.text(vert[0], vert[1], str(ivert), fontsize=15, + ha="center", va="center", color="blue", + bbox=dict(facecolor='white', alpha=0.5, lw=0)) + # }}} # vim: foldmethod=marker