From c8b21682bbbecfa0c24decf5bb186fe603e08720 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Sat, 2 May 2026 04:15:32 +0200
Subject: [PATCH] Add foliage mask paint controls and status display

---
 src/app/App.tsx | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/src/app/App.tsx b/src/app/App.tsx
index c64d070e..4ff6c6b5 100644
--- a/src/app/App.tsx
+++ b/src/app/App.tsx
@@ -20654,6 +20654,41 @@ export function App({ store, draftStorage = null, initialStatusMessage }: AppPro
                     </div>
                   </div>
 
+                  <div className="form-section">
+                    <div className="label">Foliage Mask Paint</div>
+                    <div
+                      className="viewport-panel__control-group"
+                      role="group"
+                      aria-label="Foliage mask paint controls"
+                    >
+                      <button
+                        className={`viewport-panel__button ${armedTerrainBrushTool === "foliagePaint" ? "viewport-panel__button--active" : ""}`}
+                        type="button"
+                        data-testid="terrain-brush-tool-foliage-paint"
+                        aria-pressed={armedTerrainBrushTool === "foliagePaint"}
+                        disabled={activeFoliageLayer === null}
+                        onClick={() => handleArmTerrainBrushTool("foliagePaint")}
+                      >
+                        Paint Mask
+                      </button>
+                      <button
+                        className={`viewport-panel__button ${armedTerrainBrushTool === "foliageErase" ? "viewport-panel__button--active" : ""}`}
+                        type="button"
+                        data-testid="terrain-brush-tool-foliage-erase"
+                        aria-pressed={armedTerrainBrushTool === "foliageErase"}
+                        disabled={activeFoliageLayer === null}
+                        onClick={() => handleArmTerrainBrushTool("foliageErase")}
+                      >
+                        Erase Mask
+                      </button>
+                    </div>
+                    <div className="material-summary">
+                      {activeFoliageLayer === null
+                        ? "Create or select a foliage layer in the Foliage panel before painting terrain masks."
+                        : `${activeFoliageLayer.name} mask has ${selectedTerrainActiveFoliageMaskPaintedSampleCount} painted sample${selectedTerrainActiveFoliageMaskPaintedSampleCount === 1 ? "" : "s"} on this terrain.`}
+                    </div>
+                  </div>
+
                   <div className="form-section">
                     <div className="label">Brush Settings</div>
                     <label className="form-field">