demos/demo.py
branchpyrit
changeset 0 3547b885df7e
child 1 e74bf781067e
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/demos/demo.py	Thu Oct 25 16:40:22 2007 +0200
@@ -0,0 +1,45 @@
+#!/usr/bin/python2.4
+
+import sys
+sys.path.append("../")
+
+from raytracer import Raytracer, Material, Plane, Sphere, Light
+#, SphericalLight
+import Image
+
+rt = Raytracer()
+
+light1 = Light(position=(0.0, 5.0, 3.0), colour=(0.9, 0.3, 0.6))
+rt.addlight(light1)
+
+#light2 = SphericalLight(position=(-2.0, 5.0, 1.0), radius=3.0, colour=(0.7, 1.0, 0.3))
+light2 = Light(position=(-2.0, -5.0, 1.0), colour=(0.7, 1.0, 0.3))
+rt.addlight(light2)
+
+mat0 = Material(colour=(0.1, 0.2, 0.9))
+plane = Plane(normal=(0.0, 1.0, 0.0), d=1.0, material=mat0)
+rt.addshape(plane)
+
+mat1 = Material(colour=(1.0, 0.2, 0.1))
+bigsphere = Sphere(centre=(2.0, 2.0, 5.0), radius=2.5, material=mat1)
+#      reflection=0.6)
+rt.addshape(bigsphere)
+
+mat2 = Material(colour=(0.1, 0.7, 1.0))
+smallsphere = Sphere(centre=(-5.5, 1.5, 8.0), radius=2.0, material=mat2)
+#      reflection=1.0, diffuse=0.1)
+rt.addshape(smallsphere)
+
+mat3 = Material(colour=(0.9, 0.9, 0.1))
+tinysphere = Sphere(centre=(-0.5, 0.0, 2.0), radius=0.5, material=mat3)
+#      reflection=1.0, diffuse=0.1)
+rt.addshape(tinysphere)
+
+#for i in range(100):
+#      sph=Sphere((-5.5+i/10.0, -0.5, 7.0), 2.0)
+#      rt.addshape(sph)
+
+rendersize = (800, 600)
+data = rt.render(rendersize)
+img = Image.fromstring("RGB", rendersize, data)
+img.save('out.png')