1 #!/usr/bin/python2.4

     2 

     3 import sys

     4 sys.path.append("../")

     5 

     6 from raytracer import Raytracer, Material, Plane, Sphere, Light

     7 import Image

     8 

     9 rt = Raytracer()

    10 rt.ambientocclusion(samples=60, distance=6.0, angle=0.5)

    11 

    12 light1 = Light(position=(0.0, 5.0, 5.0), colour=(0.7, 0.3, 0.6))

    13 light1.castshadows(False)

    14 rt.addlight(light1)

    15 

    16 light2 = Light(position=(-2.0, 10.0, 2.0), colour=(0.4, 0.6, 0.3))

    17 light2.castshadows(False)

    18 rt.addlight(light2)

    19 

    20 mat0 = Material(colour=(0.7, 0.7, 0.7))

    21 

    22 plane = Plane(normal=(0.0, 1.0, 0.0), d=1.0, material=mat0)

    23 rt.addshape(plane)

    24 

    25 mat1 = Material(colour=(1.0, 0.0, 0.0))

    26 bigsphere = Sphere(centre=(3.0, 2.0, 7.0), radius=3.0, material=mat1)

    27 rt.addshape(bigsphere)

    28 

    29 mat2 = Material(colour=(0.0, 1.0, 0.0))

    30 smallsphere = Sphere(centre=(-5.5, 1.5, 8.0), radius=2.0, material=mat2)

    31 rt.addshape(smallsphere)

    32 

    33 mat3 = Material(colour=(0.0, 0.0, 1.0))

    34 tinysphere = Sphere(centre=(-1.2, 0.0, 2.0), radius=0.5, material=mat3)

    35 rt.addshape(tinysphere)

    36 

    37 imagesize = (800, 600)

    38 data = rt.render(imagesize)

    39 img = Image.fromstring("RGB", imagesize, data)

    40 img.save('spheres_ao.png')