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 

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

    12 rt.addlight(light1)

    13 

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

    15 rt.addlight(light2)

    16 

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

    18 

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

    20 rt.addshape(plane)

    21 

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

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

    24 rt.addshape(bigsphere)

    25 

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

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

    28 rt.addshape(smallsphere)

    29 

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

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

    32 rt.addshape(tinysphere)

    33 

    34 imagesize = (800, 600)

    35 data = rt.render(imagesize)

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

    37 img.save('spheres_shadow.png')