From 6bd2a08726588cdd2724ea1e99a7831ab0a1ba8d Mon Sep 17 00:00:00 2001
From: paulevsGitch <paulevs@yandex.ru>
Date: Thu, 23 Dec 2021 07:10:58 +0300
Subject: [PATCH] Cave floor generation

---
 .../ru/betterend/world/features/terrain/EndLakeFeature.java   | 4 ----
 .../world/features/terrain/caves/EndCaveFeature.java          | 3 ++-
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/src/main/java/ru/betterend/world/features/terrain/EndLakeFeature.java b/src/main/java/ru/betterend/world/features/terrain/EndLakeFeature.java
index 591c412d..7a9b7dba 100644
--- a/src/main/java/ru/betterend/world/features/terrain/EndLakeFeature.java
+++ b/src/main/java/ru/betterend/world/features/terrain/EndLakeFeature.java
@@ -129,10 +129,6 @@ public class EndLakeFeature extends DefaultFeature {
 								pos = POS.below();
 								if (world.getBlockState(pos).is(TagAPI.BLOCK_GEN_TERRAIN)) {
 									state = EndBiome.findTopMaterial(world, pos);
-									//state = world.getBiome(pos)
-//												 .getGenerationSettings()
-//												 .getSurfaceBuilderConfig()
-//												 .getTopMaterial();
 									if (y > waterLevel + 1) BlocksHelper.setWithoutUpdate(world, pos, state);
 									else if (y > waterLevel)
 										BlocksHelper.setWithoutUpdate(
diff --git a/src/main/java/ru/betterend/world/features/terrain/caves/EndCaveFeature.java b/src/main/java/ru/betterend/world/features/terrain/caves/EndCaveFeature.java
index dd0403ea..cb3cbee9 100644
--- a/src/main/java/ru/betterend/world/features/terrain/caves/EndCaveFeature.java
+++ b/src/main/java/ru/betterend/world/features/terrain/caves/EndCaveFeature.java
@@ -22,6 +22,7 @@ import ru.bclib.world.biomes.BCLBiome;
 import ru.bclib.world.features.DefaultFeature;
 import ru.betterend.registry.EndBiomes;
 import ru.betterend.util.BlockFixer;
+import ru.betterend.world.biome.EndBiome;
 import ru.betterend.world.biome.cave.EndCaveBiome;
 
 import java.util.List;
@@ -75,7 +76,7 @@ public abstract class EndCaveFeature extends DefaultFeature {
 					}
 				});
 
-				BlockState surfaceBlock = Blocks.END_STONE.defaultBlockState(); //biome.getBiome().getGenerationSettings().getSurfaceBuilderConfig().getTopMaterial();
+				BlockState surfaceBlock = EndBiome.findTopMaterial(biome);
 				placeFloor(world, biome, floorPositions, random, surfaceBlock);
 				placeCeil(world, biome, ceilPositions, random);
 				placeWalls(world, biome, caveBlocks, random);