Fix curve shading when multiple curves are defined in a single object

mitsuba-renderer · May 11, 2023 · 3875f9a · 3875f9a
1 parent 0c984ad
commit 3875f9a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/shapes/bsplinecurve.cpp b/src/shapes/bsplinecurve.cpp
@@ -640,7 +640,8 @@ class BSplineCurve final : public Shape<Float, Spectrum> {
     }
 
     std::tuple<Point3f, Vector3f, Vector3f, Vector3f, Float, Float, Float>
-    cubic_interpolation(const Float v, const UInt32 idx, Mask active) const {
+    cubic_interpolation(const Float v, const UInt32 prim_idx, Mask active) const {
+        UInt32 idx = dr::gather<UInt32>(m_indices, prim_idx, active);
         Point4f c0 = dr::gather<Point4f>(m_control_points, idx + 0, active),
                 c1 = dr::gather<Point4f>(m_control_points, idx + 1, active),
                 c2 = dr::gather<Point4f>(m_control_points, idx + 2, active),

diff --git a/src/shapes/linearcurve.cpp b/src/shapes/linearcurve.cpp
@@ -330,8 +330,9 @@ class LinearCurve final : public Shape<Float, Spectrum> {
         // taken into account the changing radius: `v_local` is shifted such
         // that the normal can be easily computed as `si.p - c`, 
         // where `c = (1 - c_local) * cp1 + v_local * cp2`
-        Point4f c0 = dr::gather<Point4f>(m_control_points, prim_idx + 0, active),
-                c1 = dr::gather<Point4f>(m_control_points, prim_idx + 1, active);
+        UInt32 idx = dr::gather<UInt32>(m_indices, prim_idx, active);
+        Point4f c0 = dr::gather<Point4f>(m_control_points, idx, active),
+                c1 = dr::gather<Point4f>(m_control_points, idx + 1, active);
         Point3f p0 = Point3f(c0.x(), c0.y(), c0.z()),
                 p1 = Point3f(c1.x(), c1.y(), c1.z());