From 1cc7eaa0c069bc6d84072b641bdaf4af35694aa3 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Sat, 2 May 2026 04:21:34 +0200
Subject: [PATCH] Add tests for foliage mask painting and erasing functionality

---
 tests/geometry/terrain-brush.test.ts | 61 ++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/tests/geometry/terrain-brush.test.ts b/tests/geometry/terrain-brush.test.ts
index 788bc9d6..a5e5d348 100644
--- a/tests/geometry/terrain-brush.test.ts
+++ b/tests/geometry/terrain-brush.test.ts
@@ -3,6 +3,7 @@ import { describe, expect, it } from "vitest";
 import { createDefaultTerrainBrushSettings } from "../../src/core/terrain-brush";
 import {
   createTerrain,
+  getTerrainFoliageMask,
   getTerrainSampleLayerWeights
 } from "../../src/document/terrains";
 import {
@@ -146,6 +147,7 @@ describe("terrain brush geometry", () => {
     });
     expect(result.heightSampleIndices).toEqual([12]);
     expect(result.paintWeightIndices).toEqual([]);
+    expect(result.foliageMaskValueIndices).toEqual([]);
     expect(terrain.heights).toBe(originalHeights);
     expect(terrain.paintWeights).toBe(originalPaintWeights);
     expect(terrain.heights[2 + 2 * 5]).toBeCloseTo(0.5);
@@ -192,6 +194,65 @@ describe("terrain brush geometry", () => {
         after: 0.4
       }
     ]);
+    expect(patch.foliageMaskValues).toEqual([]);
+  });
+
+  it("paints and erases foliage mask density for a foliage layer", () => {
+    const foliageLayerId = "foliage-layer-brush";
+    const terrain = createTerrain({
+      id: "terrain-foliage-brush",
+      position: { x: 0, y: 0, z: 0 },
+      sampleCountX: 3,
+      sampleCountZ: 3,
+      cellSize: 1
+    });
+
+    const paintedTerrain = applyTerrainBrushStamp({
+      terrain,
+      center: { x: 1, z: 1 },
+      settings: {
+        radius: 0.6,
+        strength: 0.5,
+        falloff: 0
+      },
+      tool: "foliagePaint",
+      foliageLayerId
+    });
+    const paintedMask = getTerrainFoliageMask(paintedTerrain, foliageLayerId);
+
+    expect(paintedMask?.values[4]).toBeCloseTo(0.5);
+
+    const erasedTerrain = applyTerrainBrushStamp({
+      terrain: paintedTerrain,
+      center: { x: 1, z: 1 },
+      settings: {
+        radius: 0.6,
+        strength: 0.5,
+        falloff: 0
+      },
+      tool: "foliageErase",
+      foliageLayerId
+    });
+    const erasedMask = getTerrainFoliageMask(erasedTerrain, foliageLayerId);
+
+    expect(erasedMask?.values[4]).toBeCloseTo(0.25);
+
+    const patch = createTerrainBrushPatchFromTerrains({
+      before: terrain,
+      after: paintedTerrain,
+      heightSampleIndices: [],
+      paintWeightIndices: [],
+      foliageMaskValueIndices: [{ layerId: foliageLayerId, index: 4 }]
+    });
+
+    expect(patch.foliageMaskValues).toEqual([
+      {
+        layerId: foliageLayerId,
+        index: 4,
+        before: 0,
+        after: 0.5
+      }
+    ]);
   });
 
   it("paints terrain layer weights toward the active layer while preserving a normalized blend", () => {