From dfce9f247d6b9c3f788dd259b6dc5fd6e9407d60 Mon Sep 17 00:00:00 2001
From: shedaniel <daniel@shedaniel.me>
Date: Wed, 20 Jul 2022 23:21:42 +0800
Subject: [PATCH] [ci skip] Add WorldGen testmod (#298)

(cherry picked from commit c259d62ba3ae9edde5e750eae044f28a4b6664c6)
---
 .../java/dev/architectury/test/TestMod.java   |  1 +
 .../test/worldgen/TestWorldGeneration.java    | 56 +++++++++++++++++++
 2 files changed, 57 insertions(+)
 create mode 100644 testmod-common/src/main/java/dev/architectury/test/worldgen/TestWorldGeneration.java

diff --git a/testmod-common/src/main/java/dev/architectury/test/TestMod.java b/testmod-common/src/main/java/dev/architectury/test/TestMod.java
index c1768bd6..009bcf22 100644
--- a/testmod-common/src/main/java/dev/architectury/test/TestMod.java
+++ b/testmod-common/src/main/java/dev/architectury/test/TestMod.java
@@ -35,6 +35,7 @@ import dev.architectury.test.registry.TestRegistries;
 import dev.architectury.test.registry.client.TestKeybinds;
 import dev.architectury.test.tags.TestTags;
 import dev.architectury.test.trade.TestTrades;
+import dev.architectury.test.worldgen.TestWorldGeneration;
 import dev.architectury.utils.Env;
 import dev.architectury.utils.EnvExecutor;
 import net.fabricmc.api.EnvType;
diff --git a/testmod-common/src/main/java/dev/architectury/test/worldgen/TestWorldGeneration.java b/testmod-common/src/main/java/dev/architectury/test/worldgen/TestWorldGeneration.java
new file mode 100644
index 00000000..e9aca40f
--- /dev/null
+++ b/testmod-common/src/main/java/dev/architectury/test/worldgen/TestWorldGeneration.java
@@ -0,0 +1,56 @@
+/*
+ * This file is part of architectury.
+ * Copyright (C) 2020, 2021, 2022 architectury
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+ */
+
+package dev.architectury.test.worldgen;
+
+import dev.architectury.event.events.common.LifecycleEvent;
+import dev.architectury.registry.level.biome.BiomeModifications;
+import dev.architectury.test.TestMod;
+import net.minecraft.core.Holder;
+import net.minecraft.data.worldgen.features.FeatureUtils;
+import net.minecraft.data.worldgen.features.OreFeatures;
+import net.minecraft.data.worldgen.placement.PlacementUtils;
+import net.minecraft.tags.BiomeTags;
+import net.minecraft.world.level.block.Blocks;
+import net.minecraft.world.level.levelgen.GenerationStep;
+import net.minecraft.world.level.levelgen.VerticalAnchor;
+import net.minecraft.world.level.levelgen.feature.ConfiguredFeature;
+import net.minecraft.world.level.levelgen.feature.Feature;
+import net.minecraft.world.level.levelgen.feature.configurations.OreConfiguration;
+import net.minecraft.world.level.levelgen.placement.*;
+
+import java.util.List;
+
+public class TestWorldGeneration {
+    public static void initialize() {
+        LifecycleEvent.SETUP.register(() -> {
+            Holder<ConfiguredFeature<OreConfiguration, ?>> configuredFeature = FeatureUtils.register(TestMod.MOD_ID + ":diamond_blocks", Feature.ORE,
+                    new OreConfiguration(OreFeatures.NATURAL_STONE, Blocks.DIAMOND_BLOCK.defaultBlockState(), 33));
+            Holder<PlacedFeature> placedFeature = PlacementUtils.register(TestMod.MOD_ID + ":diamond_blocks", configuredFeature,
+                    List.of(CountPlacement.of(4), InSquarePlacement.spread(),
+                            HeightRangePlacement.uniform(VerticalAnchor.bottom(), VerticalAnchor.absolute(15)),
+                            BiomeFilter.biome()));
+            BiomeModifications.addProperties((ctx, mutable) -> {
+                if (ctx.hasTag(BiomeTags.IS_FOREST)) {
+                    mutable.getGenerationProperties().addFeature(GenerationStep.Decoration.TOP_LAYER_MODIFICATION, placedFeature);
+                }
+            });
+        });
+    }
+}