Fix bug in CFG dependencies in p5.strands

Author: davepagurek

src/strands/strands_for.js

@@ -297,8 +297,8 @@ export class StrandsFor {
     const scopeEndBlock = CFG.createBasicBlock(cfg, BlockType.SCOPE_END);
     CFG.addEdge(cfg, updateBlock, scopeEndBlock);
 
-    // Loop back to break check
-    CFG.addEdge(cfg, scopeEndBlock, breakCheckBlock);
+    // Connect end of for loop to the merge agter the loop
+    CFG.addEdge(cfg, scopeEndBlock, mergeBlock);
 
     // Break condition exits to merge
     CFG.addEdge(cfg, breakCheckBlock, mergeBlock);

test/unit/webgl/p5.Shader.js

@@ -1128,6 +1128,40 @@ suite('p5.Shader', function() {
         const pixelColor = myp5.get(25, 25);
         assert.approximately(pixelColor[0], 127, 5); // 0.5 * 255 ≈ 127
       });
+
+      test('handle statements after for loop before return', () => {
+        myp5.createCanvas(50, 50, myp5.WEBGL);
+
+        const testShader = myp5.baseMaterialShader().modify(() => {
+          myp5.getPixelInputs(inputs => {
+            let avg = myp5.vec3(0.0);
+            let total = 0.0;
+
+            // Simulate blur-like accumulation in for loop
+            for (let i = 0; i < 3; i++) {
+              const sample = myp5.vec3(0.2, 0.1, 0.3);
+              const weight = 1.0;
+              avg += weight * sample;
+              total += weight;
+            }
+
+            const blended = avg / total;
+
+            inputs.color = [blended.x, blended.y, blended.z, 1.0];
+            return inputs;
+          });
+        }, { myp5 });
+
+        myp5.noStroke();
+        myp5.shader(testShader);
+        myp5.plane(myp5.width, myp5.height);
+
+        // Expected result: (3 * [0.2, 0.1, 0.3]) / 3 = [0.2, 0.1, 0.3]
+        const pixelColor = myp5.get(25, 25);
+        assert.approximately(pixelColor[0], 51, 5);  // 0.2 * 255 ≈ 51
+        assert.approximately(pixelColor[1], 25, 5);  // 0.1 * 255 ≈ 25
+        assert.approximately(pixelColor[2], 77, 5);  // 0.3 * 255 ≈ 77
+      });
     });
 
     suite('passing data between shaders', () => {