Add MeshioFromData

polpo/preprocessing/mesh/_meshio.py

@@ -29,3 +29,21 @@ def apply(self, data):
         mesh.write(path, **self.kwargs)
 
         return path
+
+
+class MeshioFromData(PreprocessingStep):
+    def __init__(self, cell_type="triangle"):
+        super().__init__()
+        self.cell_type = cell_type
+
+    def apply(self, mesh):
+        # TODO: also consider colors
+        if len(mesh) == 3:
+            vertices, faces, _ = mesh
+        else:
+            vertices, faces = mesh
+            _ = None
+        return meshio.Mesh(
+            points=vertices,
+            cells=[meshio.CellBlock(cell_type=self.cell_type, data=faces)],
+        )

polpo/preprocessing/mesh/conversion.py

@@ -21,6 +21,11 @@
 except ImportError:
     pass
 
+try:
+    from ._meshio import MeshioFromData  # noqa:F401
+except ImportError:
+    pass
+
 from polpo.macro import create_to_classes_from_from
 from polpo.preprocessing.base import PreprocessingStep