From 3555a8c239a046bb4f9d1cc9bdf58199958a0917 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Sat, 2 May 2026 04:58:07 +0200
Subject: [PATCH] Improve foliage rebuilding logic and add visibility check for
 foliage renderer

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

diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index ba34070b..7bb103f4 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -1707,6 +1707,11 @@ export class ViewportHost {
     this.applyViewModePose();
     this.applyAdvancedRenderingCameraFar(this.currentAdvancedRenderingSettings);
     this.syncFoliageVisibility();
+
+    if (this.currentDocument !== null) {
+      this.rebuildFoliage(this.currentDocument);
+    }
+
     this.syncTerrainBrushPreview();
 
     if (this.currentAdvancedRenderingSettings !== null) {
@@ -6643,6 +6648,12 @@ export class ViewportHost {
 
   private rebuildFoliage(document: SceneDocument) {
     this.syncFoliageVisibility();
+
+    if (!this.foliageRenderer.group.visible) {
+      this.foliageRenderer.dispose();
+      return;
+    }
+
     this.foliageRenderer.sync({
       terrains: document.terrains,
       foliageLayers: document.foliageLayers,