From 50d99a85e1943928e7594c95504128936feda516 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 9 Oct 2017 10:07:31 -0500
Subject: [PATCH] Add more knobs in mesh.vis.draw_curve
---
meshmode/mesh/visualization.py | 19 ++++++++++++++-----
1 file changed, 14 insertions(+), 5 deletions(-)
diff --git a/meshmode/mesh/visualization.py b/meshmode/mesh/visualization.py
index 52711a0a..a1b14707 100644
--- a/meshmode/mesh/visualization.py
+++ b/meshmode/mesh/visualization.py
@@ -142,14 +142,23 @@ def draw_2d_mesh(mesh, draw_vertex_numbers=True, draw_element_numbers=True,
# {{{ draw_curve
-def draw_curve(mesh):
- import matplotlib.pyplot as pt
- pt.plot(mesh.vertices[0], mesh.vertices[1], "o")
+def draw_curve(mesh,
+ el_bdry_style="o", el_bdry_kwargs=None,
+ node_style="x-", node_kwargs=None):
+ import matplotlib.pyplot as plt
+
+ if el_bdry_kwargs is None:
+ el_bdry_kwargs = {}
+ if node_kwargs is None:
+ node_kwargs = {}
+
+ plt.plot(mesh.vertices[0], mesh.vertices[1], el_bdry_style, **el_bdry_kwargs)
for i, group in enumerate(mesh.groups):
- pt.plot(
+ plt.plot(
group.nodes[0].ravel(),
- group.nodes[1].ravel(), "-x", label="Group %d" % i)
+ group.nodes[1].ravel(), node_style, label="Group %d" % i,
+ **node_kwargs)
# }}}
--
GitLab