blender customization

Content/Python/ue_workshop/scene.py

@@ -1,7 +1,10 @@
 """An isolated generated map, simple meshes and a ready-to-use character."""
+import math
+from pathlib import Path
+
 import unreal
 
-from .paths import GENERATED, slug
+from .paths import GENERATED, PROJECT, slug
 
 
 class Scene:
@@ -65,6 +68,41 @@ class Scene:
             raise ValueError("Ksztalt: Cube, Sphere, Cylinder, Cone lub Plane.")
         return self.mesh(f"/Engine/BasicShapes/{shape}.{shape}", **kwargs)
 
+    def model(self, filename, name="Character", height=200, rotation=(0, 0, 0)):
+        """Import a model, fit its height (cm), and give it a floor-centered pivot.
+
+        Returns a parent actor for transform animation; imported materials stay
+        on its mesh child. Rotation is (roll, pitch, yaw), applied before fitting.
+        """
+        if height is not None and (not isinstance(height, (int, float)) or
+                                   not math.isfinite(height) or height <= 0):
+            raise ValueError("Wysokosc modelu musi byc dodatnia; None zachowuje rozmiar pliku.")
+        source = Path(filename).expanduser()
+        source = source if source.is_absolute() else PROJECT / source
+        if source.suffix.lower() == ".blend":
+            from .blender import import_character
+            mesh = import_character(source, name=name)
+        else:
+            from .assets import import_model
+            mesh = import_model(source, name=name)
+        actor = self.mesh(mesh, name=name + "Mesh")
+        roll, pitch, yaw = rotation
+        actor.set_actor_rotation(unreal.Rotator(roll=roll, pitch=pitch, yaw=yaw), False)
+        origin, extent = actor.get_actor_bounds(False)
+        if extent.z <= 0:
+            self.subsystem.destroy_actor(actor)
+            raise ValueError("Model ma zerowa wysokosc. Sprawdz geometrie i MODEL_ROTATION.")
+        if height is not None:
+            factor = float(height) / (2 * extent.z)
+            actor.set_actor_scale3d(unreal.Vector(factor, factor, factor))
+        origin, extent = actor.get_actor_bounds(False)
+        actor.set_actor_location(unreal.Vector(-origin.x, -origin.y, extent.z - origin.z), False, False)
+        root = self.spawn(unreal.StaticMeshActor, name)
+        root.static_mesh_component.set_mobility(unreal.ComponentMobility.MOVABLE)
+        actor.attach_to_actor(root, "", unreal.AttachmentRule.KEEP_WORLD,
+                              unreal.AttachmentRule.KEEP_WORLD, unreal.AttachmentRule.KEEP_WORLD, False)
+        return root
+
     def character(self, material=None, head_scale=1.0):
         root = self.spawn(unreal.StaticMeshActor, "Character")
         root.static_mesh_component.set_mobility(unreal.ComponentMobility.MOVABLE)