from sightpy import *


# define materials to use
def setup_scene(width=400, height=300, environment="lake.png"):
    # Set Scene

    Sc = Scene(ambient_color=rgb(0.01, 0.01, 0.01))

    angle = -np.pi * 0.5
    Sc.add_Camera(
        screen_height=height,
        screen_width=width,
        look_from=vec3(4.0 * np.sin(angle), 0.00, 4.0 * np.cos(angle)),
        look_at=vec3(0.0, 0.05, 0.0),
    )

    soap_bubble = ThinFilmInterference(thickness=330, noise=60.0)
    Sc.add(
        Sphere(
            material=soap_bubble,
            center=vec3(1.0, 0.0, 1.5),
            radius=1.7,
            shadow=False,
            max_ray_depth=5,
        )
    )

    Sc.add_Background(environment, blur=10.0)

    return Sc