From 6c6367eafacea66d5620c84760ea8c1d070eb3a2 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Tue, 27 Oct 2015 16:52:47 -0500 Subject: [PATCH] 2D discretization visualization: don't connect element lines --- meshmode/discretization/visualization.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/meshmode/discretization/visualization.py b/meshmode/discretization/visualization.py index 8d18e780..5e3772a0 100644 --- a/meshmode/discretization/visualization.py +++ b/meshmode/discretization/visualization.py @@ -255,12 +255,15 @@ def draw_curve(discr): import matplotlib.pyplot as pt pt.plot(mesh.vertices[0], mesh.vertices[1], "o") + color = pt.cm.rainbow(np.linspace(0, 1, len(discr.groups))) 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) + group_nodes[0].T, + group_nodes[1].T, "-x", + label="Group %d" % i, + color=color[i]) # }}} -- GitLab