From 87ad6c2afbe2964e012ec4642895199a74b2b2d6 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Sat, 2 May 2026 04:13:20 +0200
Subject: [PATCH] Implement foliage paint and erase brush modes

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

diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index f9d112a7..28281b7b 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -9352,6 +9352,10 @@ export class ViewportHost {
         return TERRAIN_BRUSH_PREVIEW_SMOOTH_COLOR;
       case "flatten":
         return TERRAIN_BRUSH_PREVIEW_FLATTEN_COLOR;
+      case "foliagePaint":
+        return TERRAIN_BRUSH_PREVIEW_FOLIAGE_PAINT_COLOR;
+      case "foliageErase":
+        return TERRAIN_BRUSH_PREVIEW_FOLIAGE_ERASE_COLOR;
       case "paint": {
         const terrain =
           this.getDisplayedTerrainState(brushState.terrainId) ??