package net.minecraft.world.level.levelgen.feature;

import com.mojang.serialization.Codec;
import java.util.Random;
import net.minecraft.core.BaseBlockPosition;
import net.minecraft.core.BlockPosition;
import net.minecraft.world.level.GeneratorAccess;
import net.minecraft.world.level.IWorldWriter;
import net.minecraft.world.level.block.BlockHugeMushroom;
import net.minecraft.world.level.block.state.IBlockData;
import net.minecraft.world.level.levelgen.feature.configurations.WorldGenFeatureMushroomConfiguration;

public class WorldGenHugeMushroomRed extends WorldGenMushrooms {

    public WorldGenHugeMushroomRed(Codec<WorldGenFeatureMushroomConfiguration> codec) {
        super(codec);
    }

    @Override
    protected void a(GeneratorAccess generatoraccess, Random random, BlockPosition blockposition, int i, BlockPosition.MutableBlockPosition blockposition_mutableblockposition, WorldGenFeatureMushroomConfiguration worldgenfeaturemushroomconfiguration) {
        for (int j = i - 3; j <= i; ++j) {
            int k = j < i ? worldgenfeaturemushroomconfiguration.foliageRadius : worldgenfeaturemushroomconfiguration.foliageRadius - 1;
            int l = worldgenfeaturemushroomconfiguration.foliageRadius - 2;

            for (int i1 = -k; i1 <= k; ++i1) {
                for (int j1 = -k; j1 <= k; ++j1) {
                    boolean flag = i1 == -k;
                    boolean flag1 = i1 == k;
                    boolean flag2 = j1 == -k;
                    boolean flag3 = j1 == k;
                    boolean flag4 = flag || flag1;
                    boolean flag5 = flag2 || flag3;

                    if (j >= i || flag4 != flag5) {
                        blockposition_mutableblockposition.a((BaseBlockPosition) blockposition, i1, j, j1);
                        if (!generatoraccess.getType(blockposition_mutableblockposition).i(generatoraccess, blockposition_mutableblockposition)) {
                            IBlockData iblockdata = worldgenfeaturemushroomconfiguration.capProvider.a(random, blockposition);

                            if (iblockdata.b(BlockHugeMushroom.WEST) && iblockdata.b(BlockHugeMushroom.EAST) && iblockdata.b(BlockHugeMushroom.NORTH) && iblockdata.b(BlockHugeMushroom.SOUTH) && iblockdata.b(BlockHugeMushroom.UP)) {
                                iblockdata = (IBlockData) ((IBlockData) ((IBlockData) ((IBlockData) ((IBlockData) iblockdata.set(BlockHugeMushroom.UP, j >= i - 1)).set(BlockHugeMushroom.WEST, i1 < -l)).set(BlockHugeMushroom.EAST, i1 > l)).set(BlockHugeMushroom.NORTH, j1 < -l)).set(BlockHugeMushroom.SOUTH, j1 > l);
                            }

                            this.a((IWorldWriter) generatoraccess, blockposition_mutableblockposition, iblockdata);
                        }
                    }
                }
            }
        }

    }

    @Override
    protected int a(int i, int j, int k, int l) {
        int i1 = 0;

        if (l < j && l >= j - 3) {
            i1 = k;
        } else if (l == j) {
            i1 = k;
        }

        return i1;
    }
}