diff -r fcf1487b398b -r f6a72eb99631 demos/demo_PIL.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/demos/demo_PIL.py Tue Apr 29 23:31:08 2008 +0200 @@ -0,0 +1,42 @@ +#!/usr/bin/python + +from pyrit import * +import Image + +rt = Raytracer() +top = KdTree() +rt.setTop(top) +rt.setCamera(Camera()) + +light1 = Light(position=(0.0, 5.0, -5.0), colour=(0.7, 0.3, 0.6)) +rt.addLight(light1) + +light2 = Light(position=(-2.0, 10.0, -2.0), colour=(0.4, 0.6, 0.3)) +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) + +top.optimize() +imagesize = (800, 600) +sampler = DefaultSampler(imagesize) +rt.setSampler(sampler) +rt.render() +pixmap = sampler.getPixmap() +data = pixmap.getCharData() +img = Image.fromstring("RGB", imagesize, data) +img.save('demo_PIL.png')