demos/triangles_monkey.py
author Radek Brich <radek.brich@devl.cz>
Sun, 25 Nov 2007 15:50:01 +0100
branchpyrit
changeset 14 fc18ac4833f2
parent 6 d8d596d26f25
child 19 4e0955fca797
permissions -rwxr-xr-x
replace Plane with axis-aligned Box (because infinite Plane is not usable with kd-tree) fix memory leak in KdTree::nearest_intersection rename BBox::R to BBox::H new file: common.h (Eps and Inf constants)

#!/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')