Added normalize function

Added function to normalize the scale of meshes
tjs20011 · Mar 9, 2025 · 4acc015 · 4acc015
1 parent e685ab2
commit 4acc015
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/ShipGen/Generate.py b/ShipGen/Generate.py
@@ -82,6 +82,10 @@ def mesh_to_triangles(mesh:Mesh) -> np.ndarray:
 def mesh_list_to_triangles(mesh_list:list[Mesh]) -> list[np.ndarray]:
     return list(map(mesh_to_triangles, mesh_list))
 
+def normalize_x(mesh:Mesh, scale:float = 10):
+    mesh.vectors -= np.array([mesh.min_[0], 0, 0])
+    mesh.vectors *= scale / (mesh.max_[0] - mesh.min_[0])
+
 def load_data():
     # Load in the Data:
 

diff --git a/ShipGen/__init__.py b/ShipGen/__init__.py
@@ -3,4 +3,5 @@
     ShipCharacteristics,
     mesh_to_triangles,
     mesh_list_to_triangles,
+    normalize_x,
 )