package net.minecraft.util.valueproviders;

import com.mojang.serialization.Codec;
import com.mojang.serialization.DataResult;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import net.minecraft.util.Mth;
import net.minecraft.util.RandomSource;

public class UniformFloat extends FloatProvider {
	public static final MapCodec<UniformFloat> CODEC = RecordCodecBuilder.<UniformFloat>mapCodec(
			instance -> instance.group(
					Codec.FLOAT.fieldOf("min_inclusive").forGetter(uniformFloat -> uniformFloat.minInclusive),
					Codec.FLOAT.fieldOf("max_exclusive").forGetter(uniformFloat -> uniformFloat.maxExclusive)
				)
				.apply(instance, UniformFloat::new)
		)
		.validate(
			uniformFloat -> uniformFloat.maxExclusive <= uniformFloat.minInclusive
				? DataResult.error(() -> "Max must be larger than min, min_inclusive: " + uniformFloat.minInclusive + ", max_exclusive: " + uniformFloat.maxExclusive)
				: DataResult.success(uniformFloat)
		);
	private final float minInclusive;
	private final float maxExclusive;

	private UniformFloat(float minInclusive, float maxExclusive) {
		this.minInclusive = minInclusive;
		this.maxExclusive = maxExclusive;
	}

	public static UniformFloat of(float minInclusive, float maxExclusive) {
		if (maxExclusive <= minInclusive) {
			throw new IllegalArgumentException("Max must exceed min");
		} else {
			return new UniformFloat(minInclusive, maxExclusive);
		}
	}

	@Override
	public float sample(RandomSource random) {
		return Mth.randomBetween(random, this.minInclusive, this.maxExclusive);
	}

	@Override
	public float getMinValue() {
		return this.minInclusive;
	}

	@Override
	public float getMaxValue() {
		return this.maxExclusive;
	}

	@Override
	public FloatProviderType<?> getType() {
		return FloatProviderType.UNIFORM;
	}

	public String toString() {
		return "[" + this.minInclusive + "-" + this.maxExclusive + "]";
	}
}