Normalize throughput and kill the bounced ray stochastically.

Allows "infinite" bounces without freezing the GPU.
w23 · Nov 13, 2021 · 5ae66bc · 5ae66bc
1 parent 24ebc1f
commit 5ae66bc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ref_vk/shaders/ray.rgen b/ref_vk/shaders/ray.rgen
@@ -489,6 +489,12 @@ void main() {
 
 		if (bounce == 0)
 			first_bounce_brdf_type = brdfType;
+
+		float maxThroughput = max(max(throughput.x, throughput.y), throughput.z);
+		if (rand01() > maxThroughput) {
+			break;
+		}
+		throughput /= maxThroughput;
 	} // for all bounces
 
 	if (first_bounce_brdf_type == DIFFUSE_TYPE) {