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

import com.mojang.serialization.Codec;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.util.RandomSource;
import net.minecraft.world.level.WorldGenLevel;
import net.minecraft.world.level.block.BambooStalkBlock;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.block.state.properties.BambooLeaves;
import net.minecraft.world.level.levelgen.Heightmap;
import net.minecraft.world.level.levelgen.feature.configurations.ProbabilityFeatureConfiguration;

public class BambooFeature extends Feature<ProbabilityFeatureConfiguration> {
	private static final BlockState BAMBOO_TRUNK = Blocks.BAMBOO
		.defaultBlockState()
		.setValue(BambooStalkBlock.AGE, 1)
		.setValue(BambooStalkBlock.LEAVES, BambooLeaves.NONE)
		.setValue(BambooStalkBlock.STAGE, 0);
	private static final BlockState BAMBOO_FINAL_LARGE = BAMBOO_TRUNK.setValue(BambooStalkBlock.LEAVES, BambooLeaves.LARGE).setValue(BambooStalkBlock.STAGE, 1);
	private static final BlockState BAMBOO_TOP_LARGE = BAMBOO_TRUNK.setValue(BambooStalkBlock.LEAVES, BambooLeaves.LARGE);
	private static final BlockState BAMBOO_TOP_SMALL = BAMBOO_TRUNK.setValue(BambooStalkBlock.LEAVES, BambooLeaves.SMALL);

	public BambooFeature(Codec<ProbabilityFeatureConfiguration> codec) {
		super(codec);
	}

	@Override
	public boolean place(FeaturePlaceContext<ProbabilityFeatureConfiguration> context) {
		int i = 0;
		BlockPos blockPos = context.origin();
		WorldGenLevel worldGenLevel = context.level();
		RandomSource randomSource = context.random();
		ProbabilityFeatureConfiguration probabilityFeatureConfiguration = context.config();
		BlockPos.MutableBlockPos mutableBlockPos = blockPos.mutable();
		BlockPos.MutableBlockPos mutableBlockPos2 = blockPos.mutable();
		if (worldGenLevel.isEmptyBlock(mutableBlockPos)) {
			if (Blocks.BAMBOO.defaultBlockState().canSurvive(worldGenLevel, mutableBlockPos)) {
				int j = randomSource.nextInt(12) + 5;
				if (randomSource.nextFloat() < probabilityFeatureConfiguration.probability) {
					int k = randomSource.nextInt(4) + 1;

					for (int l = blockPos.getX() - k; l <= blockPos.getX() + k; l++) {
						for (int m = blockPos.getZ() - k; m <= blockPos.getZ() + k; m++) {
							int n = l - blockPos.getX();
							int o = m - blockPos.getZ();
							if (n * n + o * o <= k * k) {
								mutableBlockPos2.set(l, worldGenLevel.getHeight(Heightmap.Types.WORLD_SURFACE, l, m) - 1, m);
								if (isDirt(worldGenLevel.getBlockState(mutableBlockPos2))) {
									worldGenLevel.setBlock(mutableBlockPos2, Blocks.PODZOL.defaultBlockState(), 2);
								}
							}
						}
					}
				}

				for (int k = 0; k < j && worldGenLevel.isEmptyBlock(mutableBlockPos); k++) {
					worldGenLevel.setBlock(mutableBlockPos, BAMBOO_TRUNK, 2);
					mutableBlockPos.move(Direction.UP, 1);
				}

				if (mutableBlockPos.getY() - blockPos.getY() >= 3) {
					worldGenLevel.setBlock(mutableBlockPos, BAMBOO_FINAL_LARGE, 2);
					worldGenLevel.setBlock(mutableBlockPos.move(Direction.DOWN, 1), BAMBOO_TOP_LARGE, 2);
					worldGenLevel.setBlock(mutableBlockPos.move(Direction.DOWN, 1), BAMBOO_TOP_SMALL, 2);
				}
			}

			i++;
		}

		return i > 0;
	}
}