Use specular or diffuse reflection for invalied cone angles

ViennaTools · Feb 6, 2025 · fca5465 · fca5465
1 parent 818fc90
commit fca5465
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/include/viennaray/rayReflection.hpp b/include/viennaray/rayReflection.hpp
@@ -60,8 +60,13 @@ template <typename NumericType, int D>
   // (https://math.stackexchange.com/a/182936)
   std::uniform_real_distribution<NumericType> uniDist;
 
-  assert(maxConeAngle >= 0. && maxConeAngle <= M_PI / 2. &&
-         "Cone angle not allowed");
+  if (maxConeAngle <= 0.) {
+    return ReflectionSpecular<NumericType>(rayDir, geomNormal);
+  }
+
+  if (maxConeAngle >= M_PI_2) {
+    return ReflectionDiffuse<NumericType, D>(geomNormal, rngState);
+  }
 
   Vec3D<NumericType> direction;
   // compute specular direction