minecraft-kajetans-mod/src/main/java/me/km/world/ModBiomeProvider.java

package me.km.world;

import java.util.Set;
import net.minecraft.util.registry.Registry;
import net.minecraft.world.biome.Biome;
import net.minecraft.world.biome.provider.BiomeProvider;
import net.minecraft.world.biome.provider.OverworldBiomeProviderSettings;
import net.minecraft.world.gen.layer.Layer;
import net.minecraft.world.gen.layer.LayerUtil;

public class ModBiomeProvider extends BiomeProvider {
    private final Layer genBiomes;
    private final Biome[] biomeArray;

    public ModBiomeProvider(Set<Biome> biomes, OverworldBiomeProviderSettings settingsProvider) {
        super(biomes);
        biomeArray = biomes.toArray(new Biome[biomes.size()]);
        this.genBiomes = LayerUtil.func_227474_a_(settingsProvider.getSeed(), settingsProvider.getWorldType(), settingsProvider.getGeneratorSettings());
    }

    @Override
    public Biome getNoiseBiome(int x, int y, int z) {
        Biome vanillaBiome = genBiomes.func_215738_a(x, z);

        float min = Float.MAX_VALUE;
        float min2 = Float.MAX_VALUE;
        Biome minBiome = null;
        Biome minBiome2 = null;
        for(Biome b : biomeArray) {
            float distance = Math.abs(b.getDepth() - vanillaBiome.getDepth());
            if(distance < min) {
                min2 = min;
                minBiome2 = minBiome;

                min = distance;
                minBiome = b;
            } else if(distance < min2) {
                min2 = distance;
                minBiome2 = b;
            }
        }

        if((Registry.BIOME.getId(vanillaBiome) % 2) == 1) {
            return minBiome;
        }
        return minBiome2;
    }
}