more features

Content/Python/ue_workshop/scene.py

@@ -52,11 +52,11 @@ class Scene:
         actor.set_folder_path("Workshop")
         return actor
 
-    def mesh(self, asset, name="Model", location=(0, 0, 0), scale=(1, 1, 1), material=None):
+    def mesh(self, asset, name="Model", location=(0, 0, 0), scale=(1, 1, 1), material=None, rotation=(0, 0, 0)):
         mesh = unreal.load_asset(asset) if isinstance(asset, str) else asset
         if not isinstance(mesh, unreal.StaticMesh):
             raise TypeError(f"Oczekiwano StaticMesh: {asset}")
-        actor = self.spawn(unreal.StaticMeshActor, name, location)
+        actor = self.spawn(unreal.StaticMeshActor, name, location, rotation)
         actor.static_mesh_component.set_mobility(unreal.ComponentMobility.MOVABLE)
         actor.static_mesh_component.set_static_mesh(mesh)
         actor.set_actor_scale3d(unreal.Vector(*scale))
@@ -69,7 +69,7 @@ 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), reimport=False):
+    def model(self, filename, name="Character", height=200, rotation=(0, 0, 0), reimport=False, location=(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
@@ -102,6 +102,7 @@ class Scene:
         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)
+        root.set_actor_location(unreal.Vector(*location), False, False)
         return root
 
     def skeletal_model(self, filename, name="Character", height=200, rotation=(0, 0, 0),
@@ -145,19 +146,20 @@ class Scene:
         self._skeletal_models[root.get_path_name()] = (child, imported)
         return root
 
-    def character(self, material=None, head_scale=1.0):
-        root = self.spawn(unreal.StaticMeshActor, "Character")
+    def character(self, material=None, head_scale=1.0, name="Character", location=(0, 0, 0)):
+        root = self.spawn(unreal.StaticMeshActor, name)
         root.static_mesh_component.set_mobility(unreal.ComponentMobility.MOVABLE)
-        for shape, name, position, scale in [
+        for shape, part, position, scale in [
             ("Cube", "Body", (0, 0, 90), (0.55, 0.8, 0.9)),
             ("Sphere", "Head", (0, 0, 175), (0.7 * head_scale,) * 3),
             ("Cube", "LeftLeg", (0, -23, 25), (0.35, 0.3, 0.5)),
             ("Cube", "RightLeg", (0, 23, 25), (0.35, 0.3, 0.5)),
             ("Sphere", "Nose", (34, 0, 177), (0.24, 0.2, 0.2)),
         ]:
-            actor = self.primitive(shape, name=name, location=position, scale=scale, material=material)
+            actor = self.primitive(shape, name=name + part, location=position, scale=scale, material=material)
             actor.attach_to_actor(root, "", unreal.AttachmentRule.KEEP_WORLD,
                                   unreal.AttachmentRule.KEEP_WORLD, unreal.AttachmentRule.KEEP_WORLD, False)
+        root.set_actor_location(unreal.Vector(*location), False, False)
         return root
 
     def save(self):