Added conversion

Added helper functions to convert stl.Mesh's to triangle arrays
tjs20011 · Mar 5, 2025 · 968cbac · 968cbac
1 parent 13a589f
commit 968cbac
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/ShipGen/Generate.py b/ShipGen/Generate.py
@@ -77,6 +77,10 @@ def generate_hulls(find_best:int,
             print('Error at hull {}'.format(valid_idx[idxs[i]]))
     return meshs
 
+def mesh_to_triangles(mesh:Mesh) -> np.ndarray:
+    return np.array([mesh.v0, mesh.v1, mesh.v2]).swapaxes(0, 1)
+def mesh_list_to_triangles(mesh_list:list[Mesh]) -> list[np.ndarray]:
+    return list(map(mesh_to_triangles, mesh_list))
 
 def load_data():
     # Load in the Data:

diff --git a/ShipGen/__init__.py b/ShipGen/__init__.py
@@ -1 +1,6 @@
-from .Generate import generate_hulls, ShipCharacteristics
+from .Generate import (
+    generate_hulls,
+    ShipCharacteristics,
+    mesh_to_triangles,
+    mesh_list_to_triangles,
+)