package net.minecraft.world.attribute.modifier;

import com.mojang.serialization.Codec;
import net.minecraft.util.Mth;
import net.minecraft.world.attribute.EnvironmentAttribute;
import net.minecraft.world.attribute.LerpFunction;

public interface FloatModifier<Argument> extends AttributeModifier<Float, Argument> {
	FloatModifier<FloatWithAlpha> ALPHA_BLEND = new FloatModifier<FloatWithAlpha>() {
		public Float apply(final Float subject, final FloatWithAlpha argument) {
			return Mth.lerp(argument.alpha(), subject, argument.value());
		}

		@Override
		public Codec<FloatWithAlpha> argumentCodec(final EnvironmentAttribute<Float> type) {
			return FloatWithAlpha.CODEC;
		}

		@Override
		public LerpFunction<FloatWithAlpha> argumentKeyframeLerp(final EnvironmentAttribute<Float> type) {
			return (alpha, from, to) -> new FloatWithAlpha(Mth.lerp(alpha, from.value(), to.value()), Mth.lerp(alpha, from.alpha(), to.alpha()));
		}
	};
	FloatModifier<Float> ADD = Float::sum;
	FloatModifier<Float> SUBTRACT = (FloatModifier.Simple)(a, b) -> a - b;
	FloatModifier<Float> MULTIPLY = (FloatModifier.Simple)(a, b) -> a * b;
	FloatModifier<Float> MINIMUM = Math::min;
	FloatModifier<Float> MAXIMUM = Math::max;

	@FunctionalInterface
	public interface Simple extends FloatModifier<Float> {
		@Override
		default Codec<Float> argumentCodec(final EnvironmentAttribute<Float> type) {
			return Codec.FLOAT;
		}

		@Override
		default LerpFunction<Float> argumentKeyframeLerp(final EnvironmentAttribute<Float> type) {
			return LerpFunction.ofFloat();
		}
	}
}