Drive basic trees from biome feature stages

internal/world/trees_test.go

@@ -0,0 +1,61 @@
+package world
+
+import (
+	"testing"
+
+	"regionio/internal/worldgen"
+)
+
+func TestStraightBlobTreeFromDatapack(t *testing.T) {
+	set, err := worldgen.LoadFeatureSet()
+	if err != nil {
+		t.Fatal(err)
+	}
+	config, err := set.Tree("minecraft:oak")
+	if err != nil {
+		t.Fatal(err)
+	}
+	chunk := NewChunk(0, 0, BiomePlains)
+	chunk.setBlockRaw(8, 63, 8, StateGrass)
+	if !placeStraightBlobTree(chunk, worldgen.NewLegacy(42), 8, 64, 8, config) {
+		t.Fatal("datapack oak placement failed")
+	}
+	logs, leaves := 0, 0
+	for y := 64; y < 80; y++ {
+		for x := 0; x < 16; x++ {
+			for z := 0; z < 16; z++ {
+				switch chunk.GetBlock(x, y, z) {
+				case StateOakLog:
+					logs++
+				case StateOakLeaf:
+					leaves++
+				}
+			}
+		}
+	}
+	if logs < 4 || leaves < 10 {
+		t.Fatalf("tree has %d logs and %d leaves", logs, leaves)
+	}
+}
+
+func TestBiomeTreeStageProducesTrees(t *testing.T) {
+	gen := NewVanillaGenerator(12345)
+	trees := 0
+	for cx := int32(-12); cx <= 12; cx += 3 {
+		for cz := int32(-12); cz <= 12; cz += 3 {
+			chunk := gen(cx, cz)
+			for y := SeaLevel; y < 160; y++ {
+				for x := 0; x < 16; x++ {
+					for z := 0; z < 16; z++ {
+						if chunk.GetBlock(x, y, z) == StateOakLog {
+							trees++
+						}
+					}
+				}
+			}
+		}
+	}
+	if trees == 0 {
+		t.Fatal("biome vegetation stages produced no oak logs")
+	}
+}