demos/demo_PIL.py
branchpyrit
changeset 90 f6a72eb99631
child 96 9eb71e76c7fd
--- /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')