From 0472a538048999bbe909783ce79b855e53de421e Mon Sep 17 00:00:00 2001
From: Cloudburst <serek.lego@gmail.com>
Date: Wed, 11 Aug 2021 12:40:42 +0200
Subject: [PATCH] added deset pyramids to .locate

---
 .../java/cloudburst/rejects/utils/WorldGenUtils.java   | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/main/java/cloudburst/rejects/utils/WorldGenUtils.java b/src/main/java/cloudburst/rejects/utils/WorldGenUtils.java
index 93b9ddc..308b284 100644
--- a/src/main/java/cloudburst/rejects/utils/WorldGenUtils.java
+++ b/src/main/java/cloudburst/rejects/utils/WorldGenUtils.java
@@ -61,6 +61,11 @@ public class WorldGenUtils {
         put(Feature.mineshaft, Collections.singletonList(
             Blocks.RAIL
         ));
+        put(Feature.desert_pyramid, Arrays.asList(
+            Blocks.TNT,
+            Blocks.CHISELED_SANDSTONE,
+            Blocks.STONE_PRESSURE_PLATE
+        ));
     }};
 
     private static final HashMap<Feature, List<Class<? extends Entity>>> FEATURE_ENTITIES = new HashMap<>(){{
@@ -103,7 +108,8 @@ public class WorldGenUtils {
         end_city,
         village,
         mineshaft,
-        slime_chunk
+        slime_chunk,
+        desert_pyramid
     }
 
     public static BlockPos locateFeature(Feature feature, BlockPos center) {
@@ -257,6 +263,7 @@ public class WorldGenUtils {
             case village -> { return Dimension.OVERWORLD; }
             case mineshaft -> { return Dimension.OVERWORLD; }
             case end_city -> { return Dimension.END; }
+            case desert_pyramid -> { return Dimension.OVERWORLD; }
             default -> { return Dimension.OVERWORLD; }
         }
     }
@@ -272,6 +279,7 @@ public class WorldGenUtils {
             case end_city -> { return new EndCity(version); }
             case village -> { return new Village(version); }
             case mineshaft -> { return new Mineshaft(version); }
+            case desert_pyramid -> { return new DesertPyramid(version); }
             default -> { return null;}
         }
     }