From 02ac1f80f95e0d234d7ddb41428497b7265c01b3 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 2 Feb 2017 23:54:52 -0600
Subject: [PATCH] Make draw_2d_mesh work for TensorProductElementGroup
---
meshmode/mesh/visualization.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/meshmode/mesh/visualization.py b/meshmode/mesh/visualization.py
index 601086c1..52711a0a 100644
--- a/meshmode/mesh/visualization.py
+++ b/meshmode/mesh/visualization.py
@@ -41,6 +41,11 @@ def draw_2d_mesh(mesh, draw_vertex_numbers=True, draw_element_numbers=True,
for iel, el in enumerate(grp.vertex_indices):
elverts = mesh.vertices[:, el]
+ from meshmode.mesh import TensorProductElementGroup
+ if isinstance(grp, TensorProductElementGroup) and grp.dim == 2:
+ elverts = elverts[:,
+ np.array([0, 1, 3, 2])]
+
pathdata = [
(Path.MOVETO, (elverts[0, 0], elverts[1, 0])),
]
--
GitLab