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

import net.minecraft.world.level.levelgen.feature.configurations.DecoratorConfiguration;
import net.minecraft.world.level.levelgen.Heightmap;
import java.util.stream.IntStream;
import java.util.stream.Stream;
import net.minecraft.core.BlockPos;
import java.util.Random;
import net.minecraft.world.level.levelgen.ChunkGeneratorSettings;
import net.minecraft.world.level.chunk.ChunkGenerator;
import net.minecraft.world.level.LevelAccessor;
import com.mojang.datafixers.Dynamic;
import java.util.function.Function;

public class CountHeightmapDecorator extends FeatureDecorator<FrequencyDecoratorConfiguration> {
    public CountHeightmapDecorator(final Function<Dynamic<?>, ? extends FrequencyDecoratorConfiguration> function) {
        super(function);
    }
    
    @Override
    public Stream<BlockPos> getPositions(final LevelAccessor bju, final ChunkGenerator<? extends ChunkGeneratorSettings> bzx, final Random random, final FrequencyDecoratorConfiguration cka, final BlockPos fk) {
        final int integer5;
        final int integer6;
        final int integer7;
        return IntStream.range(0, cka.count).<BlockPos>mapToObj(integer -> {
            integer5 = random.nextInt(16) + fk.getX();
            integer6 = random.nextInt(16) + fk.getZ();
            integer7 = bju.getHeight(Heightmap.Types.MOTION_BLOCKING, integer5, integer6);
            return new BlockPos(integer5, integer7, integer6);
        });
    }
}