1 #!/usr/bin/python2.4

     2 

     3 import sys

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

     5 

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

     7 import Image

     8 

     9 def LoadWavefrontObjFile(rt, mat, filename):

    10 	vertices = []

    11 	fp = file(filename)

    12 	while True:

    13 		ln = fp.readline()

    14 		if ln == "":

    15 			break;

    16 		ln = ln.split()

    17 		if ln[0] == "v":

    18 			v = [1.5*float(x) for x in ln[1:4]]

    19 			vertices.append(tuple(v))

    20 		if ln[0] == "f":

    21 			f = [vertices[int(x)-1] for x in ln[1:4]]

    22 			face = Triangle(f[0], f[1], f[2], mat)

    23 			rt.addshape(face)

    24 

    25 rt = Raytracer()

    26 mat = Material(colour=(0.9, 0.9, 0.9))

    27 LoadWavefrontObjFile(rt, mat, "monkey.obj")

    28 

    29 light = Light(position=(-5.0, 2.0, -8.0), colour=(0.9, 0.3, 0.6))

    30 rt.addlight(light)

    31 

    32 imagesize = (800, 600)

    33 data = rt.render(imagesize)

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

    35 img.save('triangles_monkey.png')