43 lines
2.0 KiB
Python
43 lines
2.0 KiB
Python
"""UE integration: teaching orbit, two-clip starter and complete animation catalogs."""
|
|
import json
|
|
import math
|
|
import runpy
|
|
|
|
from ue_workshop import Renderer
|
|
from ue_workshop.paths import PROJECT
|
|
from ue_workshop.characters import FIGHTER_MODEL, CREATURE_MODEL, _clips
|
|
|
|
fighter_clips, creature_clips = _clips(FIGHTER_MODEL), _clips(CREATURE_MODEL)
|
|
|
|
basics = runpy.run_path(str(PROJECT / "exercises/basics.py"))
|
|
shot = basics["shot"]
|
|
assert shot.frames == 192
|
|
channels = shot._transforms[basics["cam"].get_path_name()]
|
|
x = [key.get_value() for key in channels[0].get_keys()]
|
|
y = [key.get_value() for key in channels[1].get_keys()]
|
|
yaw = [key.get_value() for key in channels[5].get_keys()]
|
|
assert len(x) == shot.frames == len(yaw)
|
|
assert abs(yaw[-1] - yaw[0] - 360) < 0.001
|
|
assert all(b > a for a, b in zip(yaw, yaw[1:])), "Camera reverses direction"
|
|
assert abs(x[0] - x[-1]) < 0.001 and abs(y[0] - y[-1]) < 0.001
|
|
assert all(abs(math.hypot(px, py) - basics["radius"]) < 0.01 for px, py in zip(x, y))
|
|
|
|
starter = runpy.run_path(str(PROJECT / "exercises/animation_starter.py"))
|
|
assert starter["shot"].frames == 360
|
|
models = starter["scene"]._skeletal_models
|
|
assert len(models) == 1
|
|
assert set(next(iter(models.values()))[1].animations) == {"mutant_idle", "mutant_swiping"}
|
|
|
|
example = runpy.run_path(str(PROJECT / "examples/mixamo_characters.py"))
|
|
for actor_name, clips in (("fighter", fighter_clips), ("creature", creature_clips)):
|
|
actor = example[actor_name]
|
|
child, imported = example["scene"]._skeletal_models[actor.get_path_name()]
|
|
assert set(imported.animations) == set(clips)
|
|
assert all(clip.get_editor_property("sequence_length") > 0 for clip in imported.animations.values())
|
|
|
|
summary = {"fighter_clips": len(fighter_clips), "creature_clips": len(creature_clips),
|
|
"orbit_frames": basics["shot"].frames, "starter_frames": starter["shot"].frames}
|
|
(PROJECT / "Saved/Workshop/workshop_files_smoke.json").write_text(json.dumps(summary), encoding="utf-8")
|
|
shot = example["shot"]
|
|
renderer = Renderer(width=640, height=360, samples=1, warmup=0, mp4=False)
|