From e689903512caf22ceebbbc36e4b4e361fc221673 Mon Sep 17 00:00:00 2001
From: Justin Parsell <justin@parsell.me>
Date: Thu, 25 Feb 2021 22:00:43 -0500
Subject: [PATCH] Limited biomes to forest and wooded hills

---
 src/main/java/net/parsell/cherry/core/CherryFeatures.java | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main/java/net/parsell/cherry/core/CherryFeatures.java b/src/main/java/net/parsell/cherry/core/CherryFeatures.java
index 7957ca8..75d0f65 100644
--- a/src/main/java/net/parsell/cherry/core/CherryFeatures.java
+++ b/src/main/java/net/parsell/cherry/core/CherryFeatures.java
@@ -9,6 +9,7 @@ import net.minecraft.util.Identifier;
 import net.minecraft.util.registry.BuiltinRegistries;
 import net.minecraft.util.registry.Registry;
 import net.minecraft.util.registry.RegistryKey;
+import net.minecraft.world.biome.BiomeKeys;
 import net.minecraft.world.gen.GenerationStep;
 import net.minecraft.world.gen.UniformIntDistribution;
 import net.minecraft.world.gen.decorator.ChanceDecoratorConfig;
@@ -52,7 +53,8 @@ public class CherryFeatures {
     	Registry.register(BuiltinRegistries.CONFIGURED_FEATURE, cherryTree1.getValue(), CHERRY_TREE_1_C_R);
 
         // Modify the biomes
-		BiomeModifications.addFeature(BiomeSelectors.all(), GenerationStep.Feature.VEGETAL_DECORATION, cherryTree1);
+		BiomeModifications.addFeature(BiomeSelectors.includeByKey(BiomeKeys.FOREST), GenerationStep.Feature.VEGETAL_DECORATION, cherryTree1);
+        BiomeModifications.addFeature(BiomeSelectors.includeByKey(BiomeKeys.WOODED_HILLS), GenerationStep.Feature.VEGETAL_DECORATION, cherryTree1);
     }
 
     public static void init(){