BiomePatternGenerator.java

package com.hypixel.hytale.server.worldgen.biome;

import com.hypixel.hytale.common.map.IWeightedMap;
import com.hypixel.hytale.math.util.HashUtil;
import com.hypixel.hytale.procedurallib.logic.ResultBuffer;
import com.hypixel.hytale.procedurallib.logic.point.IPointGenerator;
import com.hypixel.hytale.server.worldgen.zone.ZoneGeneratorResult;
import java.util.Arrays;
import java.util.Objects;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class BiomePatternGenerator {
   protected final IPointGenerator pointGenerator;
   @Nonnull
   protected final IWeightedMap<TileBiome> tileBiomes;
   @Nonnull
   protected final CustomBiome[] customBiomes;
   @Nonnull
   protected final Biome[] biomes;
   protected final int extents;

   public BiomePatternGenerator(IPointGenerator pointGenerator, @Nonnull IWeightedMap<TileBiome> tileBiomes, @Nonnull CustomBiome[] customBiomes) {
      this.pointGenerator = pointGenerator;
      this.tileBiomes = tileBiomes;
      this.customBiomes = customBiomes;
      this.biomes = new Biome[tileBiomes.size() + customBiomes.length];
      int n = 0;

      for(TileBiome biome : (TileBiome[])tileBiomes.internalKeys()) {
         this.biomes[n++] = biome;
      }

      for(CustomBiome biome : customBiomes) {
         this.biomes[n++] = biome;
      }

      this.extents = getExtents(this.biomes);
   }

   public int getExtents() {
      return this.extents;
   }

   @Nonnull
   public Biome[] getBiomes() {
      return this.biomes;
   }

   @Nonnull
   public CustomBiome[] getCustomBiomes() {
      return this.customBiomes;
   }

   @Nullable
   public TileBiome getBiome(int seed, int x, int z) {
      return this.tileBiomes.get(seed, x, z, (iSeed, ix, iz, generator) -> generator.getBiomeIndex(iSeed, ix, iz), this);
   }

   protected double getBiomeIndex(int seed, int x, int z) {
      ResultBuffer.ResultBuffer2d buf = this.pointGenerator.nearest2D(seed, (double)x, (double)z);
      return HashUtil.random((long)seed, (long)buf.ix, (long)buf.iy);
   }

   @Nullable
   public TileBiome getBiomeDirect(int seed, int x, int z) {
      return this.tileBiomes.get(HashUtil.random((long)seed, Double.doubleToLongBits((double)x), Double.doubleToLongBits((double)z)));
   }

   @Nonnull
   public Biome generateBiomeAt(@Nonnull ZoneGeneratorResult zoneResult, int seed, int x, int z) {
      TileBiome parentResult = this.getBiome(seed, x, z);
      CustomBiome customBiome = this.getCustomBiomeAt(seed, (double)x, (double)z, zoneResult, parentResult);
      return (Biome)Objects.requireNonNullElse(customBiome, parentResult);
   }

   @Nullable
   public CustomBiome getCustomBiomeAt(int seed, double x, double z, @Nonnull ZoneGeneratorResult zoneResult, @Nonnull Biome parentResult) {
      if (this.customBiomes.length > 0) {
         int parentBiomeIndex = parentResult.getId();

         for(CustomBiome customBiome : this.customBiomes) {
            CustomBiomeGenerator customBiomeGenerator = customBiome.getCustomBiomeGenerator();
            if (customBiomeGenerator.isValidParentBiome(parentBiomeIndex) && customBiomeGenerator.shouldGenerateAt(seed, x, z, zoneResult, customBiome)) {
               return customBiome;
            }
         }
      }

      return null;
   }

   @Nonnull
   public String toString() {
      String var10000 = String.valueOf(this.pointGenerator);
      return "BiomePatternGenerator{pointGenerator=" + var10000 + ", tileBiomes=" + String.valueOf(this.tileBiomes) + ", customBiomes=" + Arrays.toString(this.customBiomes) + ", biomes=" + Arrays.toString(this.biomes) + "}";
   }

   private static int getExtents(@Nonnull Biome[] biomes) {
      int maxExtent = 0;

      for(Biome biome : biomes) {
         if (biome.getPrefabContainer() != null) {
            maxExtent = Math.max(maxExtent, biome.getPrefabContainer().getMaxSize());
         }
      }

      return maxExtent;
   }
}