#!/usr/bin/python

import sys
sys.path.append("..")

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

def LoadWavefrontObjFile(rt, mat, filename):
	vertices = []
	fp = file(filename)
	while True:
		ln = fp.readline()
		if ln == "":
			break;
		ln = ln.split()
		if ln[0] == "v":
			v = [1.5*float(x) for x in ln[1:4]]
			vertices.append(tuple(v))
		if ln[0] == "f":
			f = [vertices[int(x)-1] for x in ln[1:4]]
			face = Triangle(f[0], f[1], f[2], mat)
			rt.addshape(face)

rt = Raytracer()
mat = Material(colour=(0.9, 0.9, 0.9))
LoadWavefrontObjFile(rt, mat, "monkey.obj")

light = Light(position=(-5.0, 2.0, -8.0), colour=(0.9, 0.3, 0.6))
rt.addlight(light)

imagesize = (800, 600)
data = rt.render(imagesize)
img = Image.fromstring("RGB", imagesize, data)
img.save('triangles_monkey.png')