NaturesAura/src/main/java/de/ellpeck/naturesaura/gen/WorldGenAuraBloom.java

package de.ellpeck.naturesaura.gen;

import com.mojang.serialization.Codec;
import de.ellpeck.naturesaura.blocks.tiles.BlockEntityAuraBloom;
import net.minecraft.block.Block;
import net.minecraft.block.BlockState;
import net.minecraft.tileentity.BlockEntity;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Mth;
import net.minecraft.level.ISeedReader;
import net.minecraft.level.gen.ChunkGenerator;
import net.minecraft.level.gen.Heightmap;
import net.minecraft.level.gen.feature.Feature;
import net.minecraft.level.gen.feature.IFeatureConfig;
import net.minecraft.level.gen.feature.NoFeatureConfig;

import java.util.Random;

public class LevelGenAuraBloom extends Feature<NoFeatureConfig> {

    private final Block block;
    private final int chance;
    private final boolean nether;

    public LevelGenAuraBloom(Block block, int chance, boolean nether) {
        super(Codec.unit(IFeatureConfig.NO_FEATURE_CONFIG));
        this.block = block;
        this.chance = chance;
        this.nether = nether;
    }

    @Override
    public boolean func_241855_a(ISeedReader levelIn, ChunkGenerator gen, Random rand, BlockPos pos, NoFeatureConfig config) {
        if (rand.nextInt(this.chance) != 0)
            return false;
        int startX = pos.getX() + rand.nextInt(16);
        int startZ = pos.getZ() + rand.nextInt(16);
        boolean any = false;
        for (int i = Mth.nextInt(rand, 3, 8); i > 0; i--) {
            int offX = startX + Mth.nextInt(rand, -5, 5);
            int offZ = startZ + Mth.nextInt(rand, -5, 5);
            if (this.nether) {
                int y = Mth.nextInt(rand, 0, 128);
                for (int off = 0; off < 64; off++) {
                    // try to find a good location in both directions of the random pos
                    if (this.tryPlace(levelIn, new BlockPos(offX, y - off, offZ)) || this.tryPlace(levelIn, new BlockPos(offX, y + off, offZ))) {
                        any = true;
                        break;
                    }
                }
            } else {
                BlockPos placePos = new BlockPos(offX, levelIn.getHeight(Heightmap.Type.MOTION_BLOCKING_NO_LEAVES, offX, offZ), offZ);
                if (this.tryPlace(levelIn, placePos))
                    any = true;
            }
        }
        return any;
    }

    private boolean tryPlace(ISeedReader level, BlockPos pos) {
        BlockState state = this.block.getDefaultState();
        if (this.block.isValidPosition(state, level, pos)) {
            level.setBlockState(pos, state, 3);
            BlockEntity tile = level.getBlockEntity(pos);
            if (tile instanceof BlockEntityAuraBloom)
                ((BlockEntityAuraBloom) tile).justGenerated = true;
            return true;
        }
        return false;
    }
}