Fun with ray tracing Distribute rays over pixels with some randomness Allowing the rays to make some random decisions along the way a k a Monte Carlo path tracing means lots more samples but produces a nicer more realistic picture with less of the over perfect problems Simple Ray Tracer with Monte Carlo function validf val if typeof val undefined throw Error invalid value if isNaN val throw Error NaN value if val 9999999 throw Error number warning range if val 9999999 throw Error number warning range function validv val if typeof val undefined throw Error invalid value validf val x validf val y validf val z function vec3 x y z if typeof x undefined x 0 if typeof y undefined y 0 if typeof z undefined z 0 if typeof x number throw Error invalid value not a number if typeof y number throw Error invalid value not a number if typeof z number throw Error invalid value not a number this x x this y y this z z validv this return this vec3 add function v0 v1 return new vec3 v0 x v1 x v0 y v1 y v0 z v1 z vec3 add3 function v0 v1 v2 validv v0 validv v1 validv v2 return vec3 add v0 vec3 add v1 v2 vec3 sub function v0 v1 return new vec3 v0 x v1 x v0 y v1 y v0 z v1 z vec3 scale function v0 s return new vec3 v0 x s v0 y s v0 z s vec3 mult function v0 v1 return new vec3 v0 x v1 x v0 y v1 y v0 z v1 z vec3 cross function v0 v1 return v0 x v1 y v0 y v1 x vec3 cross function v0 v1 return new vec3 v0 y v1 z v0 z v1 y v0 z v1 x v0 x v1 z v0 x v1 y v0 y v1 x vec3 dot function v0 v1 return v0 x v1 x v0 y v1 y v0 z v1 z note you can t use length word vec3 dist function v0 return Math sqrt v0 x v0 x v0 y v0 y v0 z v0 z vec3 norm function v0 let ln Math sqrt v0 x v0 x v0 y v0 y v0 z v0 z validf ln return vec3 scale v0 1 0 ln function Ray o_ d_ this o o_ this d d_ validv this o validv this d const REFL DIFF 0 SPEC 1 REFR 2 material types function Sphere r_ p_ e_ c_ refl_ this r r_ radius this p p_ position this e e_ emission this c c_ color this refl refl_ reflection type DIFFuse SPECular REFRactive this intersect function ray returns distance 0 if nohit let op vec3 sub this p ray o Solve t 2 d d 2 t o p d o p o p R 2 0 let eps 1e 4 let b vec3 dot op ray d let det b b vec3 dot op op this r this r if det 0 return 0 else det Math sqrt det let t b det return t eps t t b det eps t 0 const spheres Scene radius position emission color material new Sphere 1e5 new vec3 1e5 1 40 8 81 6 new vec3 new vec3 75 25 25 REFL DIFF Left new Sphere 1e5 new vec3 1e5 99 40 8 81 6 new vec3 new vec3 25 25 75 REFL DIFF Rght new Sphere 1e5 new vec3 50 40 8 1e5 new vec3 new vec3 75 75 75 REFL DIFF Back new Sphere 1e5 new vec3 50 40 8 1e5 170 new vec3 new vec3 REFL DIFF Frnt new Sphere 1e5 new vec3 50 1e5 81 6 new vec3 new vec3 75 75 75 REFL DIFF Botm new Sphere 1e5 new vec3 50 1e5 81 6 81 6 new vec3 new vec3 75 75 75 REFL DIFF Top new Sphere 16 5 new vec3 27 16 5 47 new vec3 new vec3 1 1 1 999 REFL SPEC Mirr new Sphere 16 5 new vec3 73 16 5 78 new vec3 new vec3 1 1 1 999 REFL REFR Glas new Sphere 600 new vec3 50 681 6 27 81 6 new vec3 12 12 12 new vec3 REFL DIFF Lite const spheres Scene radius position emission color material new Sphere 1e5 new vec3 1e5 1 40 8 81 6 new vec3 new vec3 1 0 25 25 REFL DIFF Left new Sphere 1e5 new vec3 1e5 99 40 8 81 6 new vec3 new vec3 25 1 0 75 REFL DIFF Rght new Sphere 1e5 new vec3 50 40 8 1e5 new vec3 new vec3 25 25 1 0 REFL DIFF Back new Sphere 1e5 new vec3 50 40 8 1e5 170 new vec3 new vec3 REFL DIFF Frnt new Sphere 1e5 new vec3 50 1e5 81 6 new vec3 new vec3 1 0 1 0 25 REFL DIFF Botm new Sphere 1e5 new vec3 50 1e5 81 6 81 6 new vec3 new vec3 0 25 1 0 1 0 REFL DIFF Top new Sphere 16 5 new vec3 27 16 5 47 new vec3 new vec3 1 0 1 REFL SPEC Mirr new Sphere 16 5 new vec3 73 16 5 78 new vec3 new vec3 1 1 1 REFL REFR Glas new Sphere 600 new vec3 50 681 6 27 81 6 new vec3 12 12 12 new vec3 REFL DIFF Lite function clamp x return x 0 0 x 1 1 x function toInt x return Math round Math pow clamp x 1 2 2 255 5 function intersect o o ray t id const n spheres length let inf 1e20 o t inf for let i n i let d spheres i intersect o ray if d d o t o t d o id i return o t inf function erand48 return Math random const M_PI Math PI let cos Math cos let sin Math sin let sqrt Math sqrt let fabs Math abs function trace ray depth Xi const Ray r int depth unsigned short Xi let t distance to intersection let id 0 id of intersected object let o ray ray t t id id if intersect o return new vec3 if miss return black id o id t o t const obj spheres id the hit object let x vec3 add ray o vec3 scale ray d t validv x console log x x let n vec3 norm vec3 sub x obj p validv n let nl vec3 dot n ray d 0 vec3 scale n 1 0 vec3 scale n 1 validv nl return obj c if depth 5 return obj e let f new vec3 obj c x obj c y obj c z validv f let p f x f y f x f z f x f y f z f y f z max refl if depth 5 if erand48 Xi p f f 1 p else let res1 new vec3 obj e x obj e y obj e z validv res1 return res1 R R if obj refl REFL DIFF Ideal DIFFUSE reflection let r1 2 M_PI erand48 Xi let r2 erand48 Xi let r2s sqrt r2 let w nl let nnn vec3 cross fabs w x 1 new vec3 0 1 new vec3 1 w let u vec3 norm nnn let v vec3 cross w u validv v let aa11 vec3 scale u cos r1 r2s let aa22 vec3 scale v sin r1 r2s let aa33 vec3 scale w sqrt 1 r2 validv aa11 validv aa22 validv aa33 let aabb vec3 add3 aa11 aa22 aa33 validv aabb let d vec3 norm aabb validv d let tt trace new Ray x d depth Xi validv tt let res0 vec3 add obj e vec3 mult f tt console log res0 res0 return res0 else if obj refl REFL SPEC Ideal SPECULAR reflection let gg vec3 sub ray d vec3 scale n 2 vec3 dot n ray d let tt trace new Ray x gg depth Xi let res2 vec3 add obj e vec3 mult f tt console log res2 res2 validv res2 return res2 let reflRay new Ray x vec3 sub ray d vec3 scale n 2 vec3 dot n ray d Ideal dielectric REFRACTION console log reflRay reflRay let into vec3 dot n nl 0 Ray from outside going in validf into let nc 1 let nt 1 5 let nnt into nc nt nt nc let ddn vec3 dot ray d nl let cos2t 1 nnt nnt 1 ddn ddn console log ddn ddn console log cos2t cos2t if cos2t 0 Total internal reflection console log total inter let res3 vec3 add obj e vec3 mult f trace reflRay depth Xi console log res3 res3 return res3 let aaa vec3 norm vec3 scale ray d nnt console log cos2t cos2t let bbb vec3 scale n into 1 1 ddn nnt sqrt cos2t console log bbb bbb let tdir vec3 sub aaa bbb console log tdir tdir let a nt nc b nt nc R0 a a b b c 1 into ddn vec3 dot tdir n let Re R0 1 R0 c c c c c let Tr 1 Re let P 25 5 Re let RP Re P let TP Tr 1 P let dd if depth 2 Russian roulette console log reflRay Xi P depth RP TP let gggg trace reflRay depth Xi console log gggg gggg dd erand48 Xi P vec3 scale trace reflRay depth Xi RP vec3 scale trace new Ray x tdir depth Xi TP else console log Re Tr dd vec3 add vec3 scale trace reflRay depth Xi Re vec3 scale trace new Ray x tdir depth Xi Tr console log dd dd let res4 vec3 add obj e vec3 mult f dd console log res4 res4 return res4 function main let w 256 256 1024 let h 256 256 768 let samps 4 0 samples let cam new Ray new vec3 50 52 295 6 vec3 norm new vec3 0 0 042612 1 cam pos dir let cx new vec3 w 5135 h let cy vec3 scale vec3 norm vec3 cross cx cam d 0 5135 let c Array from Array w h map return vec3 for let y 0 y h y Loop over image rows fprintf stderr rRendering d spp 5 2f samps 4 100 y h 1 for let x 0 x w x Loop cols let Xi new Array 3 0 0 y y y for let sy 0 sy 2 sy 2x2 subpixel rows let i h y 1 w x for let sx 0 sx 2 sx 2x2 subpixel cols let r new vec3 for let s 0 s samps s let r1 2 erand48 Xi dx r1 1 sqrt r1 1 1 sqrt 2 r1 let r2 2 erand48 Xi dy r2 1 sqrt r2 1 1 sqrt 2 r2 dx 0 5 dy 0 5 let d vec3 add3 vec3 scale cx sx 5 dx 2 x w 5 vec3 scale cy sy 5 dy 2 y h 5 cam d let cd2 vec3 add cam o vec3 scale d 140 let rr2 trace new Ray cd2 vec3 norm d 0 Xi c i vec3 add rr2 c i let t2 vec3 scale rr2 1 0 samps r vec3 add r t2 s Camera rays are pushed forward to start in interior c i vec3 add c i vec3 scale vec3 clamp r x clamp r y clamp r z 0 25 return return sx sy x y let canvas document createElement canvas document body appendChild canvas canvas width w canvas height h let context2d canvas getContext 2d let image_data context2d getImageData 0 0 w h raw_data image_data data for let x 0 x w x for let y 0 y h y let i x y w image_data data i 4 0 toInt c i x image_data data i 4 1 toInt c i y image_data data i 4 2 toInt c i z image_data data i 4 3 255 context2d putImageData image_data 0 0 console log finished render main console log ready
the hit object let x vec3 add ray o vec3 scale ray d t validv x console log x x let n vec3 norm vec3 sub x obj p validv n let nl vec3 dot n ray d 0 vec3 scale n 1 0 vec3 scale n 1 validv nl return obj c if depth 5 return obj e let f new vec3 obj c x obj c y obj c z validv f let p f x f y f x f z f x f y f z f y f z max refl if depth 5 if erand48 Xi p f f 1 p else let res1 new vec3 obj e x obj e y obj e z validv res1 return res1 R R if obj refl REFL DIFF Ideal DIFFUSE reflection let r1 2 M_PI erand48 Xi let r2 erand48 Xi let r2s sqrt r2 let w nl let nnn vec3 cross fabs w x 1 new vec3 0 1 new vec3 1 w let u vec3 norm nnn let v vec3 cross w u validv v let aa11 vec3 scale u cos r1 r2s let aa22 vec3 scale v sin r1 r2s let aa33 vec3 scale w sqrt 1 r2 validv aa11 validv aa22 validv aa33 let aabb vec3 add3 aa11 aa22 aa33 validv aabb let d vec3 norm aabb validv d let tt trace new Ray x d depth Xi validv tt let res0 vec3 add obj e vec3 mult f tt console log res0 res0 return res0 else if obj refl REFL SPEC Ideal SPECULAR reflection let gg vec3 sub ray d vec3 scale n 2 vec3 dot n ray d let tt trace new Ray x gg depth Xi let res2 vec3 add obj e vec3 mult f tt console log res2 res2 validv res2 return res2 let reflRay new Ray x vec3 sub ray d vec3 scale n 2 vec3 dot n ray d Ideal dielectric REFRACTION console log reflRay reflRay let into vec3 dot n nl 0 Ray from outside going in validf into let nc 1 let nt 1 5 let nnt into nc nt nt nc let ddn vec3 dot ray d nl let cos2t 1 nnt nnt 1 ddn ddn console log ddn ddn console log cos2t cos2t if cos2t 0 Total internal reflection console log total inter let res3 vec3 add obj e vec3 mult f trace reflRay depth Xi console log res3 res3 return res3 let aaa vec3 norm vec3 scale ray d nnt console log cos2t cos2t let bbb vec3 scale n into 1 1 ddn nnt sqrt cos2t console log bbb bbb let tdir vec3 sub aaa bbb console log tdir tdir let a nt nc b nt nc R0 a a b b c 1 into ddn vec3 dot tdir n let Re R0 1 R0 c c c c c let Tr 1 Re let P 25 5 Re let RP Re P let TP Tr 1 P let dd if depth 2 Russian roulette console log reflRay Xi P depth RP TP let gggg trace reflRay depth Xi console log gggg gggg dd erand48 Xi P vec3 scale trace reflRay depth Xi RP vec3 scale trace new Ray x tdir depth Xi TP else console log Re Tr dd vec3 add vec3 scale trace reflRay depth Xi Re vec3 scale trace new Ray x tdir depth Xi Tr console log dd dd let res4 vec3 add obj e vec3 mult f dd console log res4 res4 return res4 function main let w 256 256 1024 let h 256 256 768 let samps 4 0 samples let cam new Ray new vec3 50 52 295 6 vec3 norm new vec3 0 0 042612 1 cam pos dir let cx new vec3 w 5135 h let cy vec3 scale vec3 norm vec3 cross cx cam d 0 5135 let c Array from Array w h map return vec3 for let y 0 y h y Loop over image rows fprintf stderr rRendering d spp 5 2f samps 4 100 y h 1 for let x 0 x w x Loop cols let Xi new Array 3 0 0 y y y for let sy 0 sy 2 sy 2x2 subpixel rows let i h y 1 w x for let sx 0 sx 2 sx 2x2 subpixel cols let r new vec3 for let s 0 s samps s let r1 2 erand48 Xi dx r1 1 sqrt r1 1 1 sqrt 2 r1 let r2 2 erand48 Xi dy r2 1 sqrt r2 1 1 sqrt 2 r2 dx 0 5 dy 0 5 let d vec3 add3 vec3 scale cx sx 5 dx 2 x w 5 vec3 scale cy sy 5 dy 2 y h 5 cam d let cd2 vec3 add cam o vec3 scale d 140 let rr2 trace new Ray cd2 vec3 norm d 0 Xi c i vec3 add rr2 c i let t2 vec3 scale rr2 1 0 samps r vec3 add r t2 s Camera rays are pushed forward to start in interior c i vec3 add c i vec3 scale vec3 clamp r x clamp r y clamp r z 0 25 return return sx sy x y let canvas document createElement canvas document body appendChild canvas canvas width w canvas height h let context2d canvas getContext 2d let image_data context2d getImageData 0 0 w h raw_data image_data data for let x 0 x w x for let y 0 y h y let i x y w image_data data i 4 0 toInt c i x image_data data i 4 1 toInt c i y image_data data i 4 2 toInt c i z image_data data i 4 3 255 context2d putImageData image_data 0 0 console log finished render main console log ready