basic advanced example

tools/smoke_keep_end_position.py

@@ -0,0 +1,51 @@
+"""Verify moving slash root travel becomes persistent parent Transform keys."""
+import json
+import math
+
+import unreal
+
+from ue_workshop import Scene, Sequence, Renderer, camera, lighting
+from ue_workshop.animations import FighterAnimations as F
+from ue_workshop.characters import FIGHTER_MODEL
+from ue_workshop.paths import PROJECT
+
+scene = Scene("KeepEndSmoke")
+actor = scene.skeletal_model(FIGHTER_MODEL, name="Fighter", height=200)
+child, imported = scene._skeletal_models[actor.get_path_name()]
+original = imported.animations[F.GREAT_SWORD_SLASH_2.key]
+bone = child.skeletal_mesh_component.get_bone_name(0)
+before = unreal.AnimationLibrary.get_bone_pose_for_time(original, bone, original.get_play_length(), False)
+scene.primitive("Cube", name="Floor", location=(0, 0, -10), scale=(20, 20, 0.2))
+lighting.studio(scene)
+cam = camera.create(scene, location=(650, -650, 450), target=(0, 150, 100), fov=55)
+shot = Sequence(scene, fps=30, frames=300)
+shot.camera(cam)
+section = shot.animation(actor, F.GREAT_SWORD_SLASH_2, start=57, keep_end_position=True)
+shot.animation(actor, F.GREAT_SWORD_SLASH_3, start=161)
+copy = section.get_editor_property("params").animation
+assert copy != original
+end = unreal.AnimationLibrary.get_bone_pose_for_time(copy, bone, copy.get_play_length(), False)
+start = unreal.AnimationLibrary.get_bone_pose_for_time(copy, bone, 0, False)
+assert abs(end.translation.y - start.translation.y) < 0.001
+after = unreal.AnimationLibrary.get_bone_pose_for_time(original, bone, original.get_play_length(), False)
+assert abs(after.translation.y - before.translation.y) < 0.001
+channels = shot._transforms[actor.get_path_name()]
+last = [channel.get_keys()[-1].get_value() for channel in channels[:3]]
+assert math.hypot(last[0], last[1]) > 100, last
+assert channels[0].get_keys()[-1].get_time().frame_number.value == section.get_end_frame()
+shot.save(open_editor=False)
+unreal.LevelSequenceEditorBlueprintLibrary.open_level_sequence(shot.asset)
+positions = {}
+hips_positions = {}
+for frame in (57, 160, 161, 162, 163, 164, 190):
+    unreal.LevelSequenceEditorBlueprintLibrary.set_current_time(frame)
+    unreal.LevelSequenceEditorBlueprintLibrary.refresh_current_level_sequence()
+    positions[frame] = actor.get_actor_location().to_tuple()
+    hips_positions[frame] = child.skeletal_mesh_component.get_socket_location(bone).to_tuple()
+assert math.dist(positions[163], positions[190]) < 0.01, positions
+assert math.dist(positions[162], positions[163]) < 20, positions
+assert math.dist(hips_positions[162], hips_positions[163]) < 30, hips_positions
+assert abs((hips_positions[160][1] - hips_positions[57][1]) -
+           (positions[160][1] - positions[57][1])) < 1, hips_positions
+(PROJECT / "Saved/Workshop/keep_end_position.json").write_text(json.dumps(positions, indent=2), encoding="utf-8")
+renderer = Renderer(width=640, height=360, samples=1, warmup=0, mp4=False)