more features

tools/smoke_basics.py

@@ -0,0 +1,58 @@
+"""Check environment, camera cuts, visibility, rotate, materials, tiling, then and audio."""
+import math
+import wave
+
+import unreal
+
+from ue_workshop import Scene, Sequence, Renderer, camera, lighting, materials
+from ue_workshop.animations import FighterAnimations as F
+from ue_workshop.characters import FIGHTER_MODEL
+from ue_workshop.paths import PROJECT
+
+scene = Scene("BasicsSmoke")
+lighting.environment(scene, fog=0.05)
+tiled = materials.texture_set("BasicsFloor", tiling=8,
+                              base_color="source_assets/character/creature/Maw J Laygo.fbm/MAW_diffuse.png")
+red = materials.color("BasicsRed", (0.8, 0.02, 0.01))
+scene.primitive("Cube", name="Floor", location=(0, 0, -10), scale=(20, 20, 0.2), material=tiled)
+ramp = scene.primitive("Cube", name="Ramp", location=(0, 250, 50), scale=(2, 2, 0.2), rotation=(0, 30, 0))
+assert abs(ramp.get_actor_rotation().pitch - 30) < 0.01
+fighter = scene.skeletal_model(FIGHTER_MODEL, name="Fighter", height=200, location=(0, -150, 0))
+materials.assign(fighter, red)
+skeletal = fighter.get_attached_actors()[0].skeletal_mesh_component
+assert all(skeletal.get_material(index) == red for index in range(skeletal.get_num_materials()))
+ghost = scene.character(material=red, name="Ghost", location=(-200, 100, 0))
+assert math.dist(ghost.get_attached_actors()[0].get_actor_location().to_tuple()[:2], (-200, 100)) < 60
+ball = scene.primitive("Sphere", name="Ball", location=(150, 0, 100), material=red)
+wide = camera.create(scene, location=(600, -700, 300), target=(0, 0, 100))
+close = camera.create(scene, location=(300, -400, 150), target=(0, -150, 120), fov=30)
+
+shot = Sequence(scene, fps=24, frames=48)
+assert shot.seconds(1.5) == 36
+shot.camera(wide)
+shot.camera(close, start=24, end=36)
+assert [cut[:2] for cut in shot._camera_cuts] == [(0, 24), (24, 36), (36, 48)]
+tracks = shot.asset.find_tracks_by_type(unreal.MovieSceneCameraCutTrack)
+assert len(tracks[0].get_sections()) == 3
+shot.visible(ball, 12)
+shot.visible(ball, 24, False)
+shot.visible(ghost, 12, False)
+shot.animation(fighter, F.GREAT_SWORD_IDLE, start=0, end=24)
+shot.then(fighter, F.GREAT_SWORD_IDLE, overlap=6, end=48)
+assert shot.rotation(fighter, 0) == (0, 0, 0)
+assert shot.rotate(fighter, 0, 24, (0, 0, 90)) == (0, 0, 90)
+assert abs(shot.rotation(fighter, 24)[2] - 90) < 0.001
+assert 0 < shot.rotation(fighter, 12)[2] < 90
+assert abs(shot.rotation(fighter, 47)[2] - 90) < 0.001
+assert shot.rotate(fighter, 24, 47, (0, 0, 360))[2] == 450
+shot.save(open_editor=False)
+
+music = PROJECT / "Saved/Workshop/basics_smoke.wav"
+music.parent.mkdir(parents=True, exist_ok=True)
+with wave.open(str(music), "wb") as target:
+    target.setnchannels(1)
+    target.setsampwidth(2)
+    target.setframerate(8000)
+    target.writeframes(b"\x00\x10" * 8000 * 5)
+renderer = Renderer(width=640, height=360, samples=1, warmup=8, audio=str(music))
+print("Basics smoke ready")