From 52815f64ef63eaf14ae645449936dce317b21bbc Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 17 May 2015 21:46:04 -0500
Subject: [PATCH] Add discretization.visualization.draw_curve
---
meshmode/discretization/visualization.py | 18 ++++++++++++++++++
1 file changed, 18 insertions(+)
diff --git a/meshmode/discretization/visualization.py b/meshmode/discretization/visualization.py
index 3c2565fd..099527c0 100644
--- a/meshmode/discretization/visualization.py
+++ b/meshmode/discretization/visualization.py
@@ -247,6 +247,24 @@ def make_visualizer(queue, discr, vis_order):
# }}}
+# {{{ draw_curve
+
+def draw_curve(discr):
+ mesh = discr.mesh
+
+ import matplotlib.pyplot as pt
+ pt.plot(mesh.vertices[0], mesh.vertices[1], "o")
+
+ with cl.CommandQueue(discr.cl_context) as queue:
+ for i, group in enumerate(discr.groups):
+ group_nodes = group.view(discr.nodes()).get(queue=queue)
+ pt.plot(
+ group_nodes[0].reshape(-1),
+ group_nodes[1].reshape(-1), "-x", label="Group %d" % i)
+
+# }}}
+
+
# {{{ connectivity
def write_mesh_connectivity_vtk_file(file_name, mesh, compressor=None,):
--
GitLab