From c28744e83ac67c16ca2fbd3a111685ea78cf8783 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 2 Dec 2018 17:32:24 -0600
Subject: [PATCH 1/2] Add mesh_to_tikz

---
 examples/mesh-to-tikz.py       | 14 ++++++++++++
 meshmode/mesh/visualization.py | 40 ++++++++++++++++++++++++++++++++++
 test/test_meshmode.py          | 24 ++++++++++++++++++++
 3 files changed, 78 insertions(+)
 create mode 100644 examples/mesh-to-tikz.py

diff --git a/examples/mesh-to-tikz.py b/examples/mesh-to-tikz.py
new file mode 100644
index 00000000..2041e995
--- /dev/null
+++ b/examples/mesh-to-tikz.py
@@ -0,0 +1,14 @@
+from meshmode.mesh.io import generate_gmsh, FileSource
+
+h = 0.3
+order = 1
+
+mesh = generate_gmsh(
+        FileSource("../test/blob-2d.step"), 2, order=order,
+        force_ambient_dim=2,
+        other_options=[
+            "-string", "Mesh.CharacteristicLengthMax = %s;" % h]
+        )
+
+from meshmode.mesh.visualization import mesh_to_tikz
+print(mesh_to_tikz(mesh))
diff --git a/meshmode/mesh/visualization.py b/meshmode/mesh/visualization.py
index afee1b34..de60f13d 100644
--- a/meshmode/mesh/visualization.py
+++ b/meshmode/mesh/visualization.py
@@ -29,6 +29,7 @@ __doc__ = """
 .. autofunction:: draw_2d_mesh
 .. autofunction:: draw_curve
 .. autofunction:: write_vertex_vtk_file
+.. autofunction:: mesh_to_tikz
 """
 
 
@@ -237,4 +238,43 @@ def write_vertex_vtk_file(mesh, file_name,
 
 # }}}
 
+
+# {{{ mesh_to_tikz
+
+def mesh_to_tikz(mesh):
+    lines = []
+
+    lines.append(r"\def\nelements{%s}" % (sum(grp.nelements for grp in mesh.groups)))
+    lines.append(r"\def\nvertices{%s}" % mesh.nvertices)
+    lines.append("")
+
+    drawel_lines = []
+    drawel_lines.append(r"\def\drawelements#1{")
+
+    for grp in mesh.groups:
+        for iel, el in enumerate(grp.vertex_indices):
+            el_nr = grp.element_nr_base+iel+1
+            elverts = mesh.vertices[:, el]
+
+            centroid = np.average(elverts, axis=1)
+            lines.append(r"\coordinate (cent%d) at (%s);"
+                    % (el_nr,
+                        ", ".join("%.5f" % (vi) for vi in centroid)))
+
+            for ivert, vert in enumerate(elverts.T):
+                lines.append(r"\coordinate (v%d-%d) at (%s);"
+                        % (el_nr, ivert+1, ", ".join("%.5f" % vi for vi in vert)))
+            drawel_lines.append(
+                    r"\draw [#1] %s -- cycle;"
+                    % " -- ".join(
+                        "(v%d-%d)" % (el_nr, vi+1) for vi in range(elverts.shape[1])))
+    drawel_lines.append("}")
+    lines.append("")
+
+    lines.extend(drawel_lines)
+
+    return "\n".join(lines)
+
+# }}}
+
 # vim: foldmethod=marker
diff --git a/test/test_meshmode.py b/test/test_meshmode.py
index e7ccf52a..1ee97d94 100644
--- a/test/test_meshmode.py
+++ b/test/test_meshmode.py
@@ -1002,6 +1002,8 @@ def test_quad_multi_element():
 # }}}
 
 
+# {{{ test_vtk_overwrite
+
 def test_vtk_overwrite(ctx_getter):
     pytest.importorskip("pyvisfile")
 
@@ -1049,6 +1051,28 @@ def test_vtk_overwrite(ctx_getter):
     _try_write_vtk(lambda x, y, **kwargs:
             write_nodal_adjacency_vtk_file(x, discr.mesh, **kwargs), discr.mesh)
 
+# }}}
+
+
+# {{{ test_mesh_to_tikz
+def test_mesh_to_tikz():
+    from meshmode.mesh.io import generate_gmsh, FileSource
+
+    h = 0.3
+    order = 1
+
+    mesh = generate_gmsh(
+            FileSource("../test/blob-2d.step"), 2, order=order,
+            force_ambient_dim=2,
+            other_options=[
+                "-string", "Mesh.CharacteristicLengthMax = %s;" % h]
+            )
+
+    from meshmode.mesh.visualization import mesh_to_tikz
+    mesh_to_tikz(mesh)
+
+# }}}
+
 
 if __name__ == "__main__":
     import sys
-- 
GitLab


From 1df1c67f035b6c8360a17ab7cd5c7978d21c427a Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 2 Dec 2018 17:46:58 -0600
Subject: [PATCH 2/2] Placate flake8

---
 meshmode/mesh/visualization.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/meshmode/mesh/visualization.py b/meshmode/mesh/visualization.py
index de60f13d..abc586af 100644
--- a/meshmode/mesh/visualization.py
+++ b/meshmode/mesh/visualization.py
@@ -267,7 +267,8 @@ def mesh_to_tikz(mesh):
             drawel_lines.append(
                     r"\draw [#1] %s -- cycle;"
                     % " -- ".join(
-                        "(v%d-%d)" % (el_nr, vi+1) for vi in range(elverts.shape[1])))
+                        "(v%d-%d)" % (el_nr, vi+1)
+                        for vi in range(elverts.shape[1])))
     drawel_lines.append("}")
     lines.append("")
 
-- 
GitLab