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

import com.mojang.serialization.Codec;
import net.minecraft.core.BlockPos;
import net.minecraft.util.RandomSource;
import net.minecraft.world.level.WorldGenLevel;
import net.minecraft.world.level.levelgen.feature.configurations.BlockColumnConfiguration;

public class BlockColumnFeature extends Feature<BlockColumnConfiguration> {
	public BlockColumnFeature(Codec<BlockColumnConfiguration> codec) {
		super(codec);
	}

	@Override
	public boolean place(FeaturePlaceContext<BlockColumnConfiguration> context) {
		WorldGenLevel worldGenLevel = context.level();
		BlockColumnConfiguration blockColumnConfiguration = context.config();
		RandomSource randomSource = context.random();
		int i = blockColumnConfiguration.layers().size();
		int[] is = new int[i];
		int j = 0;

		for (int k = 0; k < i; k++) {
			is[k] = ((BlockColumnConfiguration.Layer)blockColumnConfiguration.layers().get(k)).height().sample(randomSource);
			j += is[k];
		}

		if (j == 0) {
			return false;
		} else {
			BlockPos.MutableBlockPos mutableBlockPos = context.origin().mutable();
			BlockPos.MutableBlockPos mutableBlockPos2 = mutableBlockPos.mutable().move(blockColumnConfiguration.direction());

			for (int l = 0; l < j; l++) {
				if (!blockColumnConfiguration.allowedPlacement().test(worldGenLevel, mutableBlockPos2)) {
					truncate(is, j, l, blockColumnConfiguration.prioritizeTip());
					break;
				}

				mutableBlockPos2.move(blockColumnConfiguration.direction());
			}

			for (int l = 0; l < i; l++) {
				int m = is[l];
				if (m != 0) {
					BlockColumnConfiguration.Layer layer = (BlockColumnConfiguration.Layer)blockColumnConfiguration.layers().get(l);

					for (int n = 0; n < m; n++) {
						worldGenLevel.setBlock(mutableBlockPos, layer.state().getState(randomSource, mutableBlockPos), 2);
						mutableBlockPos.move(blockColumnConfiguration.direction());
					}
				}
			}

			return true;
		}
	}

	private static void truncate(int[] layerHeights, int totalHeight, int currentHeight, boolean prioritizeTip) {
		int i = totalHeight - currentHeight;
		int j = prioritizeTip ? 1 : -1;
		int k = prioritizeTip ? 0 : layerHeights.length - 1;
		int l = prioritizeTip ? layerHeights.length : -1;

		for (int m = k; m != l && i > 0; m += j) {
			int n = layerHeights[m];
			int o = Math.min(n, i);
			i -= o;
			layerHeights[m] -= o;
		}
	}
}