From 5cd2d2fa9fc3db39a6c444d33b5c852bcf460a5e Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Sat, 2 May 2026 04:14:10 +0200
Subject: [PATCH] Track foliage mask value keys in ViewportHost

---
 src/viewport-three/viewport-host.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 612ee5f3..db5d66ee 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -9764,6 +9764,11 @@ export class ViewportHost {
       dirtyBounds: initialStamp.dirtyBounds,
       heightSampleIndices: new Set(initialStamp.heightSampleIndices),
       paintWeightIndices: new Set(initialStamp.paintWeightIndices),
+      foliageMaskValueKeys: new Set(
+        initialStamp.foliageMaskValueIndices.map((entry) =>
+          createTerrainFoliageMaskValueKey(entry.layerId, entry.index)
+        )
+      ),
       referenceHeight,
       lastAppliedPoint: {
         x: hit.point.x,