#!/usr/bin/python

from pyrit import *
from lworeader import LoadLightwaveLwoFile
from plyreader import LoadStanfordPlyFile
from math import *
from time import time

rt = Raytracer()
top = KdTree()
rt.setTop(top)
rt.setBgColour((0.2,0.3,0.8))
cam = Camera(eye=(-8,3,4),lookat=(0,-0.5,1))
rt.setCamera(cam)

carmodel = "../models/lwo/Nissan300ZX.lwo"
bunnymodel = "../models/ply/bunny/bun_zipper.ply"

print "Loading file", carmodel
facenum = LoadLightwaveLwoFile(rt, carmodel, scale=0.4)
print "Loaded", facenum, "faces"

matbunny = Material(colour=(0.4, 0.5, 0.9))
matbunny.setTransmissivity(0.65, 1.3)
matbunny.setReflectivity(0.3)
matbunny.setSmooth(True)

print "Loading file", bunnymodel
facenum = LoadStanfordPlyFile(rt, bunnymodel,
	matbunny, scale=(20.0, 20.0, -20.0), trans=(-1,-1.8,3.5))
print "Loaded", facenum, "faces"

mat = Material(colour=(0.5, 0.5, 0.5))
ground = Box(L=(-10,-2,-10), H=(10,-1.2,10), material=mat)
rt.addShape(ground)

lwall = Box(L=(-10,-2,-4), H=(10,10,-3), material=mat)
rt.addShape(lwall)

mat = Material(colour=(0.5, 0.5, 0.5))
mat.setReflectivity(0.7)
backwall = Box(L=(8,-2, -10), H=(10,10,10), material=mat)
rt.addShape(backwall)

light1 = Light(position=(-5.0, 20.0, 8.0), colour=(0.9, 0.9, 0.9))
rt.addLight(light1)
light2 = Light(position=(5.0, 10.0, 10.0), colour=(0.9, 0.7, 0.7))
rt.addLight(light2)

t1 = time()
top.optimize()
t2 = time()
print "kd-tree built in", (t2-t1), "s"

sampler = DefaultSampler(1024, 768)
sampler.setOversample(1)
rt.setSampler(sampler)

t1 = time()
rt.render()
t2 = time()
print "rendered in", (t2-t1), "s"

sampler.getPixmap().writePNG('bench.png')