demos/spheres_ao.py
author Radek Brich <radek.brich@devl.cz>
Sun, 20 Apr 2008 16:48:24 +0200
branchpyrit
changeset 72 7c3f38dff082
parent 60 a23b5089b9c3
child 75 20dee9819b17
permissions -rwxr-xr-x
kd-tree building - check all axes for best split, add additional shape-bbox check extent Container bounds by Eps to fix invisible triangles on borders new Camera constructor with more intuitive lookat/up vectors fix camera axes (mirrored images) better camera angle-of-view change capitalization of addShape and addLight

#!/usr/bin/python

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

rt = Raytracer()
rt.ambientocclusion(samples=100, distance=16.0, angle=0.5)

light1 = Light(position=(0.0, 5.0, -5.0), colour=(0.7, 0.3, 0.6))
light1.castshadows(False)
rt.addlight(light1)

light2 = Light(position=(-2.0, 10.0, -2.0), colour=(0.4, 0.6, 0.3))
light2.castshadows(False)
rt.addlight(light2)

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

box = Box(L=(-20.0, -1.2, -20.0), H=(20.0, -1.0, 20.0), material=mat0)
rt.addshape(box)

mat1 = Material(colour=(1.0, 0.0, 0.0))
bigsphere = Sphere(centre=(3.0, 2.0, -7.0), radius=3.0, material=mat1)
rt.addshape(bigsphere)

mat2 = Material(colour=(0.0, 1.0, 0.0))
smallsphere = Sphere(centre=(-5.5, 1.5, -8.0), radius=2.0, material=mat2)
rt.addshape(smallsphere)

mat3 = Material(colour=(0.0, 0.0, 1.0))
tinysphere = Sphere(centre=(-1.2, 0.0, -2.0), radius=0.5, material=mat3)
rt.addshape(tinysphere)

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