From 92d4ccab6cf9af9eae69f3423f69b3ad922c4a4d Mon Sep 17 00:00:00 2001
From: paulevsGitch <paulevs@yandex.ru>
Date: Sat, 14 Nov 2020 13:00:19 +0300
Subject: [PATCH] Generator optimization

---
 gradle.properties                                 |  2 +-
 .../java/ru/betterend/registry/EndBiomes.java     | 15 +++++++++++++++
 .../java/ru/betterend/world/biome/EndBiome.java   |  9 +++++++++
 .../world/generator/BetterEndBiomeSource.java     |  2 +-
 4 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/gradle.properties b/gradle.properties
index af314448..375124e0 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -8,7 +8,7 @@
 	loader_version=0.10.6+build.214
 
 # Mod Properties
-	mod_version = 0.6.2-beta
+	mod_version = 0.6.3-beta
 	maven_group = ru.betterend
 	archives_base_name = better-end
 
diff --git a/src/main/java/ru/betterend/registry/EndBiomes.java b/src/main/java/ru/betterend/registry/EndBiomes.java
index fe66b993..8e38d7f3 100644
--- a/src/main/java/ru/betterend/registry/EndBiomes.java
+++ b/src/main/java/ru/betterend/registry/EndBiomes.java
@@ -108,6 +108,21 @@ public class EndBiomes {
 			}
 		});
 		
+		LAND_BIOMES.getBiomes().forEach((endBiome) -> {
+			Biome biome = biomeRegistry.get(endBiome.getID());
+			endBiome.setActualBiome(biome);
+		});
+		
+		VOID_BIOMES.getBiomes().forEach((endBiome) -> {
+			Biome biome = biomeRegistry.get(endBiome.getID());
+			endBiome.setActualBiome(biome);
+		});
+		
+		SUBBIOMES.forEach((endBiome) -> {
+			Biome biome = biomeRegistry.get(endBiome.getID());
+			endBiome.setActualBiome(biome);
+		});
+		
 		CLIENT.clear();
 	}
 	
diff --git a/src/main/java/ru/betterend/world/biome/EndBiome.java b/src/main/java/ru/betterend/world/biome/EndBiome.java
index 72bf3ff8..f0761852 100644
--- a/src/main/java/ru/betterend/world/biome/EndBiome.java
+++ b/src/main/java/ru/betterend/world/biome/EndBiome.java
@@ -35,6 +35,7 @@ public class EndBiome {
 
 	private final float fogDensity;
 	private EndFeature structuresFeature;
+	private Biome actualBiome;
 
 	public EndBiome(BiomeDefinition definition) {
 		biome = definition.build();
@@ -157,4 +158,12 @@ public class EndBiome {
 	public EndFeature getStructuresFeature() {
 		return structuresFeature;
 	}
+	
+	public void setActualBiome(Biome biome) {
+		this.actualBiome = biome;
+	}
+	
+	public Biome getActualBiome() {
+		return this.actualBiome;
+	}
 }
diff --git a/src/main/java/ru/betterend/world/generator/BetterEndBiomeSource.java b/src/main/java/ru/betterend/world/generator/BetterEndBiomeSource.java
index 784d4a9a..6d626395 100644
--- a/src/main/java/ru/betterend/world/generator/BetterEndBiomeSource.java
+++ b/src/main/java/ru/betterend/world/generator/BetterEndBiomeSource.java
@@ -85,7 +85,7 @@ public class BetterEndBiomeSource extends BiomeSource {
 			mapLand.clearCache();
 			mapVoid.clearCache();
 		}
-		return biomeRegistry.get(endBiome.getID());
+		return endBiome.getActualBiome();
 	}
 
 	@Override