From 990cbadbb148184c23319c4df7067e03aa3a37d2 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 17 Feb 2015 14:17:52 -0600
Subject: [PATCH] Allow drawing vertex numbers
---
meshmode/mesh/visualization.py | 11 +++++++++--
1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/meshmode/mesh/visualization.py b/meshmode/mesh/visualization.py
index 7cacd12..2b19da7 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
--
GitLab