package net.minecraft.data.worldgen.placement;

import java.util.List;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Holder;
import net.minecraft.core.registries.Registries;
import net.minecraft.data.worldgen.BootstrapContext;
import net.minecraft.resources.ResourceKey;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.util.random.WeightedList;
import net.minecraft.util.valueproviders.ConstantInt;
import net.minecraft.util.valueproviders.IntProvider;
import net.minecraft.util.valueproviders.WeightedListInt;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.levelgen.Heightmap;
import net.minecraft.world.level.levelgen.VerticalAnchor;
import net.minecraft.world.level.levelgen.blockpredicates.BlockPredicate;
import net.minecraft.world.level.levelgen.feature.ConfiguredFeature;
import net.minecraft.world.level.levelgen.feature.Feature;
import net.minecraft.world.level.levelgen.feature.configurations.FeatureConfiguration;
import net.minecraft.world.level.levelgen.placement.BlockPredicateFilter;
import net.minecraft.world.level.levelgen.placement.CountPlacement;
import net.minecraft.world.level.levelgen.placement.HeightRangePlacement;
import net.minecraft.world.level.levelgen.placement.HeightmapPlacement;
import net.minecraft.world.level.levelgen.placement.PlacedFeature;
import net.minecraft.world.level.levelgen.placement.PlacementFilter;
import net.minecraft.world.level.levelgen.placement.PlacementModifier;

public class PlacementUtils {
	public static final PlacementModifier HEIGHTMAP = HeightmapPlacement.onHeightmap(Heightmap.Types.MOTION_BLOCKING);
	public static final PlacementModifier HEIGHTMAP_NO_LEAVES = HeightmapPlacement.onHeightmap(Heightmap.Types.MOTION_BLOCKING_NO_LEAVES);
	public static final PlacementModifier HEIGHTMAP_TOP_SOLID = HeightmapPlacement.onHeightmap(Heightmap.Types.OCEAN_FLOOR_WG);
	public static final PlacementModifier HEIGHTMAP_WORLD_SURFACE = HeightmapPlacement.onHeightmap(Heightmap.Types.WORLD_SURFACE_WG);
	public static final PlacementModifier HEIGHTMAP_OCEAN_FLOOR = HeightmapPlacement.onHeightmap(Heightmap.Types.OCEAN_FLOOR);
	public static final PlacementModifier FULL_RANGE = HeightRangePlacement.uniform(VerticalAnchor.bottom(), VerticalAnchor.top());
	public static final PlacementModifier RANGE_10_10 = HeightRangePlacement.uniform(VerticalAnchor.aboveBottom(10), VerticalAnchor.belowTop(10));
	public static final PlacementModifier RANGE_8_8 = HeightRangePlacement.uniform(VerticalAnchor.aboveBottom(8), VerticalAnchor.belowTop(8));
	public static final PlacementModifier RANGE_4_4 = HeightRangePlacement.uniform(VerticalAnchor.aboveBottom(4), VerticalAnchor.belowTop(4));
	public static final PlacementModifier RANGE_BOTTOM_TO_MAX_TERRAIN_HEIGHT = HeightRangePlacement.uniform(VerticalAnchor.bottom(), VerticalAnchor.absolute(256));

	public static void bootstrap(BootstrapContext<PlacedFeature> context) {
		AquaticPlacements.bootstrap(context);
		CavePlacements.bootstrap(context);
		EndPlacements.bootstrap(context);
		MiscOverworldPlacements.bootstrap(context);
		NetherPlacements.bootstrap(context);
		OrePlacements.bootstrap(context);
		TreePlacements.bootstrap(context);
		VegetationPlacements.bootstrap(context);
		VillagePlacements.bootstrap(context);
	}

	public static ResourceKey<PlacedFeature> createKey(String key) {
		return ResourceKey.create(Registries.PLACED_FEATURE, ResourceLocation.withDefaultNamespace(key));
	}

	public static void register(
		BootstrapContext<PlacedFeature> context,
		ResourceKey<PlacedFeature> key,
		Holder<ConfiguredFeature<?, ?>> configuredFeature,
		List<PlacementModifier> placements
	) {
		context.register(key, new PlacedFeature(configuredFeature, List.copyOf(placements)));
	}

	public static void register(
		BootstrapContext<PlacedFeature> context, ResourceKey<PlacedFeature> key, Holder<ConfiguredFeature<?, ?>> configuredFeature, PlacementModifier... placements
	) {
		register(context, key, configuredFeature, List.of(placements));
	}

	public static PlacementModifier countExtra(int baseValue, float chance, int addedAmount) {
		float f = 1.0F / chance;
		if (Math.abs(f - (int)f) > 1.0E-5F) {
			throw new IllegalStateException("Chance data cannot be represented as list weight");
		} else {
			WeightedList<IntProvider> weightedList = WeightedList.<IntProvider>builder()
				.add(ConstantInt.of(baseValue), (int)f - 1)
				.add(ConstantInt.of(baseValue + addedAmount), 1)
				.build();
			return CountPlacement.of(new WeightedListInt(weightedList));
		}
	}

	public static PlacementFilter isEmpty() {
		return BlockPredicateFilter.forPredicate(BlockPredicate.ONLY_IN_AIR_PREDICATE);
	}

	public static BlockPredicateFilter filteredByBlockSurvival(Block block) {
		return BlockPredicateFilter.forPredicate(BlockPredicate.wouldSurvive(block.defaultBlockState(), BlockPos.ZERO));
	}

	public static Holder<PlacedFeature> inlinePlaced(Holder<ConfiguredFeature<?, ?>> feature, PlacementModifier... placements) {
		return Holder.direct(new PlacedFeature(feature, List.of(placements)));
	}

	public static <FC extends FeatureConfiguration, F extends Feature<FC>> Holder<PlacedFeature> inlinePlaced(
		F feature, FC config, PlacementModifier... placements
	) {
		return inlinePlaced(Holder.direct(new ConfiguredFeature(feature, config)), placements);
	}

	public static <FC extends FeatureConfiguration, F extends Feature<FC>> Holder<PlacedFeature> onlyWhenEmpty(F feature, FC config) {
		return filtered(feature, config, BlockPredicate.ONLY_IN_AIR_PREDICATE);
	}

	public static <FC extends FeatureConfiguration, F extends Feature<FC>> Holder<PlacedFeature> filtered(F feature, FC config, BlockPredicate predicate) {
		return inlinePlaced(feature, config, BlockPredicateFilter.forPredicate(predicate));
	}
}