Fix bug in ptracer which would generate NaN rays

mitsuba-renderer · Nov 30, 2023 · 3d902a4 · 3d902a4
1 parent eba6c25
commit 3d902a4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/integrators/ptracer.cpp b/src/integrators/ptracer.cpp
@@ -82,7 +82,12 @@ class ParticleTracerIntegrator final : public AdjointIntegrator<Float, Spectrum>
 
         // Primary & further bounces illumination
         auto [ray, throughput] = prepare_ray(scene, sensor, sampler);
-        trace_light_ray(ray, scene, sensor, sampler, throughput, block, sample_scale);
+
+        Float throughput_max = dr::max(unpolarized_spectrum(throughput));
+        Mask active = dr::neq(throughput_max, 0.f);
+
+        trace_light_ray(ray, scene, sensor, sampler, throughput, block,
+                        sample_scale, active);
     }
 
     /**