package net.minecraft.world.level.levelgen.structure.pools;

import com.mojang.datafixers.util.Either;
import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.function.Function;
import net.minecraft.util.ExtraCodecs;

public record DimensionPadding(int bottom, int top) {
	private static final Codec<DimensionPadding> RECORD_CODEC = RecordCodecBuilder.create(
		instance -> instance.group(
				ExtraCodecs.NON_NEGATIVE_INT.lenientOptionalFieldOf("bottom", 0).forGetter(dimensionPadding -> dimensionPadding.bottom),
				ExtraCodecs.NON_NEGATIVE_INT.lenientOptionalFieldOf("top", 0).forGetter(dimensionPadding -> dimensionPadding.top)
			)
			.apply(instance, DimensionPadding::new)
	);
	public static final Codec<DimensionPadding> CODEC = Codec.either(ExtraCodecs.NON_NEGATIVE_INT, RECORD_CODEC)
		.xmap(
			either -> either.map(DimensionPadding::new, Function.identity()),
			dimensionPadding -> dimensionPadding.hasEqualTopAndBottom() ? Either.left(dimensionPadding.bottom) : Either.right(dimensionPadding)
		);
	public static final DimensionPadding ZERO = new DimensionPadding(0);

	public DimensionPadding(int padding) {
		this(padding, padding);
	}

	public boolean hasEqualTopAndBottom() {
		return this.top == this.bottom;
	}
}