1 #!/usr/bin/python |
1 #!/usr/bin/python |
2 |
2 |
3 import sys |
3 import sys |
4 sys.path.append("..") |
4 sys.path.append("..") |
5 |
5 |
6 from raytracer import Raytracer, Material, Plane, Sphere, Light |
6 from raytracer import Raytracer, Material, Box, Sphere, Light |
7 import Image |
7 import Image |
8 |
8 |
9 rt = Raytracer() |
9 rt = Raytracer() |
10 |
10 |
11 light1 = Light(position=(0.0, 5.0, 5.0), colour=(0.7, 0.3, 0.6)) |
11 light1 = Light(position=(0.0, 5.0, 5.0), colour=(0.7, 0.3, 0.6)) |
14 light2 = Light(position=(-2.0, 10.0, 2.0), colour=(0.4, 0.6, 0.3)) |
14 light2 = Light(position=(-2.0, 10.0, 2.0), colour=(0.4, 0.6, 0.3)) |
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 plane = Plane(normal=(0.0, 1.0, 0.0), d=1.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(plane) |
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 |