import unreal

from .camera import look_at


def area(scene, name="Light", location=(200, -200, 350), target=(0, 0, 100),
         intensity=12000, color=(1, 1, 1), size=180):
    actor = scene.spawn(unreal.RectLight, name, location)
    light = actor.get_component_by_class(unreal.RectLightComponent)
    light.set_mobility(unreal.ComponentMobility.MOVABLE)
    light.set_editor_property("intensity", float(intensity))
    light.set_editor_property("source_width", float(size))
    light.set_editor_property("source_height", float(size))
    light.set_editor_property("attenuation_radius", 2000.0)
    light.set_light_color(unreal.LinearColor(*color, 1.0))
    look_at(actor, target)
    return actor


def studio(scene, key_color=(1.0, 0.82, 0.65), fill_color=(0.5, 0.7, 1.0)):
    return [area(scene, "Key", (200, -250, 350), color=key_color),
            area(scene, "Fill", (100, 300, 250), intensity=7000, color=fill_color),
            area(scene, "Rim", (-200, 0, 300), intensity=15000)]