[Bf-blender-cvs] [63a40ed422b] master: Cycles: Fix uniform cone sampling

[Lukas Stockner]

Commit: 63a40ed422bc61e9cd952282a636c53a9d8f3fde
Author: Lukas Stockner
Date:   Sat Jun 6 20:08:48 2020 +0200
Branches: master
https://developer.blender.org/rB63a40ed422bc61e9cd952282a636c53a9d8f3fde

Cycles: Fix uniform cone sampling

This code is currently only used for the Glossy Toon BSDF, but it's a generic
building block that might be used for other things in the future.

To see why the current code does not give a uniform distribution, consider that
it chooses both angles uniformly, but the smaller the angle from the center of
the cone is, the smaller the differential solid angle is (similar to how
sampling disks by choosing radius and phi uniformly does not work).

Differential Revision: https://developer.blender.org/D7948

===================================================================

M	intern/cycles/kernel/kernel_montecarlo.h

===================================================================

diff --git a/intern/cycles/kernel/kernel_montecarlo.h b/intern/cycles/kernel/kernel_montecarlo.h
index acd5086be3a..ac40840133f 100644
--- a/intern/cycles/kernel/kernel_montecarlo.h
+++ b/intern/cycles/kernel/kernel_montecarlo.h
@@ -85,8 +85,9 @@ ccl_device_inline void sample_uniform_hemisphere(
 ccl_device_inline void sample_uniform_cone(
     const float3 N, float angle, float randu, float randv, float3 *omega_in, float *pdf)
 {
-  float z = cosf(angle * randu);
-  float r = sqrtf(max(0.0f, 1.0f - z * z));
+  float zMin = cosf(angle);
+  float z = lerp(zMin, 1.0f, randu);
+  float r = safe_sqrtf(1.0f - sqr(z));
   float phi = M_2PI_F * randv;
   float x = r * cosf(phi);
   float y = r * sinf(phi);
@@ -94,7 +95,7 @@ ccl_device_inline void sample_uniform_cone(
   float3 T, B;
   make_orthonormals(N, &T, &B);
   *omega_in = x * T + y * B + z * N;
-  *pdf = 0.5f * M_1_PI_F / (1.0f - cosf(angle));
+  *pdf = M_1_2PI_F / (1.0f - zMin);
 }
 
 /* sample uniform point on the surface of a sphere */