"""Uruchom plik w UE: File > Execute Python Script. Zmieniaj i uruchamiaj ponownie."""
import importlib
import sys

# Ponowny import bibliotek (ue cachuje zaimportowaną biblioteke i jeśli nie bedzie 
# jawnego reloadu/restartu edytora to będzie wykonywał stary kod)
if "ue_workshop" in sys.modules:
    _render_module = sys.modules.get("ue_workshop.render")
    if _render_module is not None and _render_module.is_rendering():
        raise RuntimeError("Poczekaj na zakonczenie renderu przed przeladowaniem biblioteki.")
    # Najpierw zaleznosci, potem moduly, ktore z nich korzystaja.
    for _name in ("paths", "progress", "video", "materials", "camera", "lighting", "assets",
                  "_gltf", "blender", "scene", "sequence", "render"):
        _module = sys.modules.get("ue_workshop." + _name)
        if _module is not None:
            importlib.reload(_module)
    importlib.reload(sys.modules["ue_workshop"])

from ue_workshop import Scene, Sequence, Renderer, camera, lighting, materials

# 1. MODEL I SCENA. Jednostki: centymetry. Os Z wskazuje gore.
scene = Scene("MyShot")
blue = materials.color("Character", rgb=(0.04, 0.35, 0.8), metallic=0.2)
floor = materials.color("Floor", rgb=(0.07, 0.08, 0.11))
character = scene.character(material=blue, head_scale=1.0)
scene.primitive("Cube", name="Floor", location=(0, 0, -10), scale=(20, 20, 0.2), material=floor)
lighting.studio(scene, key_color=(1.0, 0.75, 0.5), fill_color=(0.3, 0.6, 1.0))
cam = camera.create(scene, location=(450, -550, 300), focal_length=45)

# 2. ANIMACJA. 48 klatek przy 24 FPS = 2 sekundy. Ostatnia klatka ma numer 47.
shot = Sequence(scene, fps=24, frames=48)
shot.camera(cam)
# Rotacja: (roll, pitch, yaw), w stopniach.
shot.transform_key(character, 0, rotation=(0, 0, 0))
shot.transform_key(character, 47, rotation=(0, 0, 360))
shot.save()

# 3. WYNIK. Najpierw obejrzyj animacje w Sequencerze, potem zmien RENDER na True.
RENDER = True
renderer = Renderer(width=1920, height=1080, samples=4, warmup=32, mp4=True)
# Mozesz tez zaczac od presetu i nadpisac wybrane wartosci:
# renderer = Renderer.from_preset("final", width=1920, height=1080)
if RENDER:
    renderer.render(shot)
    # Szybki podglad zamiast filmu: renderer.render(shot, single_frame=24)