2 |
2 |
3 from raytracer import Raytracer, Material, Box, Sphere, Light |
3 from raytracer import Raytracer, Material, Box, Sphere, Light |
4 import Image |
4 import Image |
5 |
5 |
6 rt = Raytracer() |
6 rt = Raytracer() |
7 rt.ambientocclusion(samples=100, distance=16.0, angle=0.5) |
7 rt.ambientOcclusion(samples=100, distance=16.0, angle=0.5) |
8 |
8 |
9 light1 = Light(position=(0.0, 5.0, -5.0), colour=(0.7, 0.3, 0.6)) |
9 light1 = Light(position=(0.0, 5.0, -5.0), colour=(0.7, 0.3, 0.6)) |
10 light1.castshadows(False) |
10 light1.castShadows(False) |
11 rt.addlight(light1) |
11 rt.addLight(light1) |
12 |
12 |
13 light2 = Light(position=(-2.0, 10.0, -2.0), colour=(0.4, 0.6, 0.3)) |
13 light2 = Light(position=(-2.0, 10.0, -2.0), colour=(0.4, 0.6, 0.3)) |
14 light2.castshadows(False) |
14 light2.castShadows(False) |
15 rt.addlight(light2) |
15 rt.addLight(light2) |
16 |
16 |
17 mat0 = Material(colour=(0.7, 0.7, 0.7)) |
17 mat0 = Material(colour=(0.7, 0.7, 0.7)) |
18 |
18 |
19 box = Box(L=(-20.0, -1.2, -20.0), H=(20.0, -1.0, 20.0), material=mat0) |
19 box = Box(L=(-20.0, -1.2, -20.0), H=(20.0, -1.0, 20.0), material=mat0) |
20 rt.addshape(box) |
20 rt.addShape(box) |
21 |
21 |
22 mat1 = Material(colour=(1.0, 0.0, 0.0)) |
22 mat1 = Material(colour=(1.0, 0.0, 0.0)) |
23 bigsphere = Sphere(centre=(3.0, 2.0, -7.0), radius=3.0, material=mat1) |
23 bigsphere = Sphere(centre=(3.0, 2.0, -7.0), radius=3.0, material=mat1) |
24 rt.addshape(bigsphere) |
24 rt.addShape(bigsphere) |
25 |
25 |
26 mat2 = Material(colour=(0.0, 1.0, 0.0)) |
26 mat2 = Material(colour=(0.0, 1.0, 0.0)) |
27 smallsphere = Sphere(centre=(-5.5, 1.5, -8.0), radius=2.0, material=mat2) |
27 smallsphere = Sphere(centre=(-5.5, 1.5, -8.0), radius=2.0, material=mat2) |
28 rt.addshape(smallsphere) |
28 rt.addShape(smallsphere) |
29 |
29 |
30 mat3 = Material(colour=(0.0, 0.0, 1.0)) |
30 mat3 = Material(colour=(0.0, 0.0, 1.0)) |
31 tinysphere = Sphere(centre=(-1.2, 0.0, -2.0), radius=0.5, material=mat3) |
31 tinysphere = Sphere(centre=(-1.2, 0.0, -2.0), radius=0.5, material=mat3) |
32 rt.addshape(tinysphere) |
32 rt.addShape(tinysphere) |
33 |
33 |
34 imagesize = (800, 600) |
34 imagesize = (800, 600) |
35 data = rt.render(imagesize) |
35 data = rt.render(imagesize) |
36 img = Image.fromstring("RGB", imagesize, data) |
36 img = Image.fromstring("RGB", imagesize, data) |
37 img.save('spheres_ao.png') |
37 img.save('spheres_ao.png') |