-
Notifications
You must be signed in to change notification settings - Fork 3
/
raytracer.sql
173 lines (169 loc) · 12 KB
/
raytracer.sql
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
DROP TABLE IF EXISTS sphere_sample CASCADE;
CREATE TABLE IF NOT EXISTS sphere_sample (x DOUBLE PRECISION NOT NULL, y DOUBLE PRECISION NOT NULL, z DOUBLE PRECISION NOT NULL,
a DOUBLE PRECISION NOT NULL, b DOUBLE PRECISION NOT NULL, c DOUBLE PRECISION NOT NULL, sampleno INTEGER NOT NULL, n_samples INTEGER NOT NULL);
INSERT INTO sphere_sample
WITH square_sample AS (SELECT 2.0*(RANDOM() - 0.5) AS a1, 2.0*(RANDOM() - 0.5) AS b1, 2.0*(RANDOM() - 0.5) AS c1
FROM generate_series(1, 5000)),
ball_sample AS (SELECT a1 AS a, b1 AS b, c1 AS c, SQRT(a1*a1+b1*b1+c1*c1) AS radius FROM square_sample WHERE 1>=(a1*a1+b1*b1+c1*c1)),
sphere_sample AS (SELECT a/radius AS x, b/radius AS y, c/radius AS z, a, b, c, ROW_NUMBER() OVER () AS sampleno, COUNT(*) OVER () AS n_samples FROM ball_sample)
SELECT x,y,z,a,b,c,sampleno,n_samples FROM sphere_sample;
DROP INDEX IF EXISTS idx_ss;
CREATE INDEX IF NOT EXISTS idx_ss ON sphere_sample(sampleno);
DROP VIEW IF EXISTS rays CASCADE;
CREATE VIEW rays AS
WITH RECURSIVE
xs AS (SELECT 0 AS u, 0.0 AS img_frac_x UNION ALL SELECT u+1, (u+1.0)/img.res_x FROM xs, img WHERE xs.u<img.res_x-1),
ys AS (SELECT 0 AS v, 0.0 AS img_frac_y UNION ALL SELECT v+1, (v+1.0)/img.res_y FROM ys, img WHERE ys.v<img.res_y-1),
px_sample_n(px_sample_n, px_jitter_u, px_jitter_v) AS (SELECT 1, (RANDOM()-0.5) * (fov_rad_x/res_x), (RANDOM()-0.5) * (fov_rad_y/res_y)
FROM camera, img
UNION ALL SELECT px_sample_n+1, (RANDOM()-0.5) * (fov_rad_x/res_x), (RANDOM()-0.5) * (fov_rad_y/res_y)
FROM px_sample_n, camera, img WHERE px_sample_n<camera.samples_per_px),
rs(img_x, img_y, sceneid, depth, max_ray_depth, samples_per_px, px_sample_n, color_mult,
ray_col_r, ray_col_g, ray_col_b,
x1, y1, z1,
dir_x, dir_y, dir_z,
dir_lensquared,
n_x, n_y, n_z, n_len,
stop_tracing, ray_len_idx, hit_sphereid, n_sphere_samples, inside_dielectric) AS
-- Send out initial set of rays from camera
(SELECT xs.u, ys.v, c.sceneid, -1, max_ray_depth, samples_per_px, px_sample_n, CAST(2.0 AS DOUBLE PRECISION),
CAST(NULL AS DOUBLE PRECISION), CAST(NULL AS DOUBLE PRECISION), CAST(NULL AS DOUBLE PRECISION),
c.x, c.y, c.z,
(SIN(c.rot_y-(fov_rad_x/2.0)+img_frac_x*fov_rad_x) + px_jitter_u) /
SQRT(((SIN(c.rot_y-(fov_rad_x/2.0)+img_frac_x*fov_rad_x) + px_jitter_u)*(SIN(c.rot_y-(fov_rad_x/2.0)+img_frac_x*fov_rad_x) + px_jitter_u) +
(SIN(c.rot_x-(fov_rad_y/2.0)+img_frac_y*fov_rad_y) + px_jitter_v)*(SIN(c.rot_x-(fov_rad_y/2.0)+img_frac_y*fov_rad_y) + px_jitter_v) + 1.0)),
(SIN(c.rot_x-(fov_rad_y/2.0)+img_frac_y*fov_rad_y) + px_jitter_v) /
SQRT(((SIN(c.rot_y-(fov_rad_x/2.0)+img_frac_x*fov_rad_x) + px_jitter_u)*(SIN(c.rot_y-(fov_rad_x/2.0)+img_frac_x*fov_rad_x) + px_jitter_u) +
(SIN(c.rot_x-(fov_rad_y/2.0)+img_frac_y*fov_rad_y) + px_jitter_v)*(SIN(c.rot_x-(fov_rad_y/2.0)+img_frac_y*fov_rad_y) + px_jitter_v) + 1.0)),
CAST(1.0 AS DOUBLE PRECISION) /
SQRT(((SIN(c.rot_y-(fov_rad_x/2.0)+img_frac_x*fov_rad_x) + px_jitter_u)*(SIN(c.rot_y-(fov_rad_x/2.0)+img_frac_x*fov_rad_x) + px_jitter_u) +
(SIN(c.rot_x-(fov_rad_y/2.0)+img_frac_y*fov_rad_y) + px_jitter_v)*(SIN(c.rot_x-(fov_rad_y/2.0)+img_frac_y*fov_rad_y) + px_jitter_v) + 1.0)),
CAST(1.0 AS DOUBLE PRECISION),
CAST(NULL AS DOUBLE PRECISION), CAST(NULL AS DOUBLE PRECISION), CAST(NULL AS DOUBLE PRECISION), CAST(NULL AS DOUBLE PRECISION),
CAST(0 AS BOOLEAN), CAST(1 AS BIGINT), CAST(NULL AS INTEGER),
(SELECT COUNT(*) FROM sphere_sample), FALSE
FROM camera c, img, xs, ys, px_sample_n
UNION ALL
-- Collide all rays with spheres
SELECT img_x, img_y, rs.sceneid, depth+1, max_ray_depth, samples_per_px, px_sample_n,
(CASE WHEN norm_x IS NULL THEN 0.5 ELSE mirror_frac END)*color_mult,
CASE WHEN discrim>0 THEN (CASE
WHEN shade_normal THEN mat_col_r*(1+norm_x)/2
ELSE mat_col_r
END)
ELSE 1.0-(0.5*((dir_y/SQRT(dir_lensquared)+1.0)))+0.2*(0.5*((dir_y/SQRT(dir_lensquared)+1.0))) END,
CASE WHEN discrim>0 THEN (CASE
WHEN shade_normal THEN mat_col_g*(1+norm_y)/2
ELSE mat_col_g
END)
ELSE 1.0-(0.5*((dir_y/SQRT(dir_lensquared)+1.0)))+0.3*(0.5*((dir_y/SQRT(dir_lensquared)+1.0))) END,
CASE WHEN discrim>0 THEN (CASE
WHEN shade_normal THEN mat_col_b*(1+norm_z)/2
ELSE mat_col_b
END)
ELSE 1.0-(0.5*((dir_y/SQRT(dir_lensquared)+1.0)))+1.0*(0.5*((dir_y/SQRT(dir_lensquared)+1.0))) END,
-- x1, y1, z1
hit_x, hit_y, hit_z,
-- dir_x, dir_y, dir_z
CASE WHEN is_metal OR (is_dielectric AND must_reflect)
THEN (dir_x - 2 * norm_x * dot_ray_norm) / reflection_len
WHEN is_dielectric THEN (reflec_dir_x + refrac_dir_x) / refrac_len
ELSE diffuse_dir_x/diffuse_dir_len
END,
CASE WHEN is_metal OR (is_dielectric AND must_reflect)
THEN (dir_y - 2 * norm_y * dot_ray_norm) / reflection_len
WHEN is_dielectric THEN (reflec_dir_y + refrac_dir_y) / refrac_len
ELSE diffuse_dir_y/diffuse_dir_len
END,
CASE WHEN is_metal OR (is_dielectric AND must_reflect)
THEN (dir_z - 2 * norm_z * dot_ray_norm) / reflection_len
WHEN is_dielectric THEN (reflec_dir_z + refrac_dir_z) / refrac_len
ELSE diffuse_dir_z/diffuse_dir_len
END,
1.0,
norm_x, norm_y, norm_z, 1.0,
discrim IS NULL, ROW_NUMBER() OVER (PARTITION BY img_x, img_y, depth+1, px_sample_n
ORDER BY t),
sphereid, n_sphere_samples, (inside_dielectric AND NOT must_reflect) OR (NOT inside_dielectric AND NOT is_dielectric)
FROM rs
LEFT JOIN LATERAL
(SELECT s.*, (((x1-cx)*dir_x+(y1-cy)*dir_y+(z1-cz)*dir_z)*((x1-cx)*dir_x+(y1-cy)*dir_y+(z1-cz)*dir_z)-
(((x1-cx)*(x1-cx)+(y1-cy)*(y1-cy)+(z1-cz)*(z1-cz)-radius2)*dir_lensquared)) discrim,
(-((x1-cx)*dir_x+(y1-cy)*dir_y+(z1-cz)*dir_z)
-SQRT(((x1-cx)*dir_x+(y1-cy)*dir_y+(z1-cz)*dir_z)*((x1-cx)*dir_x+(y1-cy)*dir_y+(z1-cz)*dir_z)-
(((x1-cx)*(x1-cx)+(y1-cy)*(y1-cy)+(z1-cz)*(z1-cz)-radius2)*dir_lensquared))/dir_lensquared) t
FROM sphere s
WHERE s.sceneid=rs.sceneid
) hit_sphere ON discrim>0 AND t>0
LEFT JOIN LATERAL
(SELECT x1+dir_x*t AS hit_x, y1+dir_y*t AS hit_y, z1+dir_z*t AS hit_z,
x1+dir_x*t-cx AS norm_x_nonunit, y1+dir_y*t-cy AS norm_y_nonunit, z1+dir_z*t-cz AS norm_z_nonunit,
SQRT((x1+dir_x*t-cx)*(x1+dir_x*t-cx)+(y1+dir_y*t-cy)*(y1+dir_y*t-cy)+(z1+dir_z*t-cz)*(z1+dir_z*t-cz)) AS norm_len,
ROW_NUMBER() OVER (PARTITION BY img_x, img_y, depth, px_sample_n ORDER BY t ASC) AS t_idx
WHERE t>0
) sphere_normal ON t_idx=1
LEFT JOIN LATERAL
(SELECT norm_x_nonunit/norm_len AS norm_x, norm_y_nonunit/norm_len AS norm_y, norm_z_nonunit/norm_len AS norm_z
) sphere_unit_normal ON norm_x IS NOT NULL
LEFT JOIN LATERAL
(SELECT dir_x*norm_x + dir_y*norm_y + dir_z*norm_z AS dot_ray_norm,
SQRT((dir_x - 2 * norm_x * (dir_x*norm_x + dir_y*norm_y + dir_z*norm_z)) * (dir_x - 2 * norm_x * (dir_x*norm_x + dir_y*norm_y + dir_z*norm_z)) +
(dir_y - 2 * norm_y * (dir_x*norm_x + dir_y*norm_y + dir_z*norm_z)) * (dir_y - 2 * norm_y * (dir_x*norm_x + dir_y*norm_y + dir_z*norm_z)) +
(dir_z - 2 * norm_z * (dir_x*norm_x + dir_y*norm_y + dir_z*norm_z)) * (dir_z - 2 * norm_z * (dir_x*norm_x + dir_y*norm_y + dir_z*norm_z))) AS reflection_len
) dot_ray_norm ON norm_x IS NOT NULL
LEFT JOIN material ON material.materialid=hit_sphere.materialid
LEFT JOIN LATERAL
(SELECT x, y, z,
x+norm_x AS diffuse_dir_x, y+norm_y AS diffuse_dir_y, z+norm_z AS diffuse_dir_z,
SQRT((x+norm_x)*(x+norm_x)+(y+norm_y)*(y+norm_y)+(z+norm_z)*(z+norm_z)) AS diffuse_dir_len
FROM sphere_sample ss WHERE ss.sampleno=1+CAST(FLOOR(ABS((100000*dir_x)-FLOOR(100000*dir_x))*n_sphere_samples) AS INTEGER)
) diffuse_scatter ON norm_x IS NOT NULL
LEFT JOIN LATERAL
(SELECT (CASE WHEN is_dielectric THEN 1.0/eta ELSE eta END) AS ir) index_of_refraction ON norm_x IS NOT NULL
LEFT JOIN LATERAL
(SELECT LEAST(1.0, (-(dir_x*norm_x) -(dir_y*norm_y) -(dir_z*norm_z))) AS cos_theta,
((1.0-ir)/(1.0+ir))*((1.0-ir)/(1.0+ir)) AS r0
) refract_cos_theta ON norm_x IS NOT NULL
LEFT JOIN LATERAL
(SELECT ir * (dir_x + cos_theta * norm_x) AS refrac_dir_x,
ir * (dir_y + cos_theta * norm_y) AS refrac_dir_y,
ir * (dir_z + cos_theta * norm_z) AS refrac_dir_z,
SQRT(1.0-cos_theta*cos_theta) AS sin_theta,
r0 + (1.0 - r0)*pow(1.0-cos_theta, 5) AS reflectance
) refrac_vec ON norm_x IS NOT NULL
LEFT JOIN LATERAL
(SELECT -norm_x*SQRT(ABS(1.0 - (refrac_dir_x*refrac_dir_x + refrac_dir_y*refrac_dir_y + refrac_dir_z*refrac_dir_z))) AS reflec_dir_x,
-norm_y*SQRT(ABS(1.0 - (refrac_dir_x*refrac_dir_x + refrac_dir_y*refrac_dir_y + refrac_dir_z*refrac_dir_z))) AS reflec_dir_y,
-norm_z*SQRT(ABS(1.0 - (refrac_dir_x*refrac_dir_x + refrac_dir_y*refrac_dir_y + refrac_dir_z*refrac_dir_z))) AS reflec_dir_z,
(sin_theta*ir>1.0) OR (reflectance > RANDOM()) AS must_reflect
) reflec_vec ON norm_x IS NOT NULL
LEFT JOIN LATERAL
(SELECT SQRT((reflec_dir_x+refrac_dir_x)*(reflec_dir_x+refrac_dir_x)+
(reflec_dir_y+refrac_dir_y)*(reflec_dir_y+refrac_dir_y)+
(reflec_dir_z+refrac_dir_z)*(reflec_dir_z+refrac_dir_z)) refrac_len
) refrac_len ON norm_x IS NOT NULL
WHERE depth<max_ray_depth AND NOT stop_tracing AND ray_len_idx=1
)
SELECT * FROM rs WHERE ray_len_idx=1;
-- select * from rays;
DROP VIEW IF EXISTS do_render;
CREATE VIEW do_render AS
SELECT A.img_x, -A.img_y,
GREATEST(0.0, LEAST(1.0, SUM(POW(A.color_mult * COALESCE(A.ray_col_r, 0.0)/A.samples_per_px, gamma)))) col_r,
GREATEST(0.0, LEAST(1.0, SUM(POW(A.color_mult * COALESCE(A.ray_col_g, 0.0)/A.samples_per_px, gamma)))) col_g,
GREATEST(0.0, LEAST(1.0, SUM(POW(A.color_mult * COALESCE(A.ray_col_b, 0.0)/A.samples_per_px, gamma)))) col_b
FROM rays A, img
WHERE A.depth>=0
GROUP BY -A.img_y, A.img_x
ORDER BY -A.img_y, A.img_x;
DROP VIEW IF EXISTS ppm;
CREATE VIEW ppm AS
WITH maxcol(mc) AS (SELECT 255)
SELECT 'P3'
UNION ALL
SELECT res_x || ' ' || res_y || ' ' || mc FROM img, maxcol
UNION ALL
SELECT CAST(col_r*mc AS INTEGER) || ' ' || CAST(col_g*mc AS INTEGER) || ' ' || CAST(col_b*mc AS INTEGER)
FROM do_render, maxcol;
;
-- SELECT * FROM rays WHERE img_x=2 AND img_y=2;