annotate test/ray.d @ 94:61615fa85940 trunk

[svn r98] Added support for std.c.stdlib.alloca via pragma(LLVM_internal, "alloca"). Added support for array .sort and .reverse properties. Fixed some bugs with pointer arithmetic. Disabled some DMD AST optimizations that was messing things up, destroying valuable information. Added a KDevelop project file, this is what I use for coding LLVMDC now :) Other minor stuff.
author lindquist
date Mon, 12 Nov 2007 06:32:46 +0100
parents 70d6113eeb8c
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
92
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
1 //import std.stdio, std.math, std.string;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
2 //import tools.base;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
3
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
4 int atoi(char[] s) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
5 int i, fac=1;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
6 bool neg = (s.length) && (s[0] == '-');
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
7 char[] a = neg ? s[1..$] : s;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
8 foreach_reverse(c; a) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
9 i += (c-'0') * fac;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
10 fac *= 10;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
11 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
12 return !neg ? i : -i;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
13 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
14
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
15 pragma(LLVM_internal, "intrinsic", "llvm.sqrt.f64")
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
16 double sqrt(double val);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
17
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
18 double delta;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
19 static this() { delta=sqrt(real.epsilon); }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
20
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
21 struct Vec {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
22 double x, y, z;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
23 Vec opAdd(ref Vec other) { return Vec(x+other.x, y+other.y, z+other.z); }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
24 Vec opSub(ref Vec other) { return Vec(x-other.x, y-other.y, z-other.z); }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
25 Vec opMul(double a) { return Vec(x*a, y*a, z*a); }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
26 double dot(ref Vec other) { return x*other.x+y*other.y+z*other.z; }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
27 Vec unitise() { return opMul(1.0/sqrt(dot(*this))); }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
28 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
29
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
30 struct Pair(T, U) { T first; U second; }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
31 typedef Pair!(double, Vec) Hit;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
32
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
33 struct Ray { Vec orig, dir; }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
34
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
35 class Scene {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
36 //abstract void intersect(ref Hit, ref Ray);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
37 void intersect(ref Hit, ref Ray) {}
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
38 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
39
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
40 class Sphere : Scene {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
41 Vec center;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
42 double radius;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
43 //mixin This!("center, radius");
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
44 this(ref Vec c, double r)
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
45 {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
46 center = c;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
47 radius = r;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
48 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
49 double ray_sphere(ref Ray ray) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
50 auto v = center - ray.orig, b = v.dot(ray.dir), disc=b*b - v.dot(v) + radius*radius;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
51 if (disc < 0) return double.infinity;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
52 auto d = sqrt(disc), t2 = b + d;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
53 if (t2 < 0) return double.infinity;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
54 auto t1 = b - d;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
55 return (t1 > 0 ? t1 : t2);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
56 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
57 void intersect(ref Hit hit, ref Ray ray) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
58 auto lambda = ray_sphere(ray);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
59 if (lambda < hit.first)
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
60 hit = Hit(lambda, (ray.orig + lambda*ray.dir - center).unitise);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
61 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
62 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
63
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
64 class Group : Scene {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
65 Sphere bound;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
66 Scene[] children;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
67 //mixin This!("bound, children");
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
68 this (Sphere s, Scene[] c)
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
69 {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
70 bound = s;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
71 children = c;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
72 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
73 void intersect(ref Hit hit, ref Ray ray) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
74 auto l = bound.ray_sphere(ray);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
75 if (l < hit.first) foreach (child; children) child.intersect(hit, ray);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
76 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
77 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
78
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
79 double ray_trace(ref Vec light, ref Ray ray, Scene s) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
80 auto hit=Hit(double.infinity, Vec(0, 0, 0));
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
81 s.intersect(hit, ray);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
82 if (hit.first == double.infinity) return 0.0;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
83 auto g = hit.second.dot(light);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
84 if (g >= 0) return 0.0;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
85 auto p = ray.orig + ray.dir*hit.first + hit.second*delta;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
86 auto hit2=Hit(double.infinity, Vec(0, 0, 0));
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
87 s.intersect(hit2, Ray(p, light*-1.0));
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
88 return (hit2.first < double.infinity ? 0 : -g);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
89 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
90
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
91 Scene create(int level, ref Vec c, double r) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
92 auto s = new Sphere(c, r);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
93 if (level == 1) return s;
94
61615fa85940 [svn r98] Added support for std.c.stdlib.alloca via pragma(LLVM_internal, "alloca").
lindquist
parents: 92
diff changeset
94 Scene[] children;
61615fa85940 [svn r98] Added support for std.c.stdlib.alloca via pragma(LLVM_internal, "alloca").
lindquist
parents: 92
diff changeset
95 children ~= s;
92
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
96 double rn = 3*r/sqrt(12.0);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
97 for (int dz=-1; dz<=1; dz+=2)
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
98 for (int dx=-1; dx<=1; dx+=2)
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
99 children~=create(level-1, c + Vec(dx, 1, dz)*rn, r/2);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
100 return new Group(new Sphere(c, 3*r), children);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
101 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
102
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
103 void main(string[] args) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
104 int level = (args.length==3 ? args[1].atoi() : 9),
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
105 n = (args.length==3 ? args[2].atoi() : 512), ss = 4;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
106 auto light = Vec(-1, -3, 2).unitise();
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
107 auto s=create(level, Vec(0, -1, 0), 1);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
108 printf("P5\n%d %d\n255", n, n);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
109 for (int y=n-1; y>=0; --y)
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
110 for (int x=0; x<n; ++x) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
111 double g=0;
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
112 for (int d=0; d<ss*ss; ++d) {
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
113 auto dir=Vec(x+(d%ss)*1.0/ss-n/2.0, y+(d/ss)*1.0/ss-n/2.0, n).unitise();
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
114 g += ray_trace(light, Ray(Vec(0, 0, -4), dir), s);
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
115 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
116 printf("%c", cast(ubyte)(0.5 + 255.0 * g / (ss*ss)));
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
117 }
70d6113eeb8c [svn r96] Updated to DMD 1.023.
lindquist
parents:
diff changeset
118 }