package net.minecraft.world.ticks;

import it.unimi.dsi.fastutil.Hash.Strategy;
import java.util.Comparator;
import net.minecraft.core.BlockPos;
import org.jspecify.annotations.Nullable;

public record ScheduledTick<T>(T type, BlockPos pos, long triggerTick, TickPriority priority, long subTickOrder) {
	public static final Comparator<ScheduledTick<?>> DRAIN_ORDER = (o1, o2) -> {
		int compare = Long.compare(o1.triggerTick, o2.triggerTick);
		if (compare != 0) {
			return compare;
		} else {
			compare = o1.priority.compareTo(o2.priority);
			return compare != 0 ? compare : Long.compare(o1.subTickOrder, o2.subTickOrder);
		}
	};
	public static final Comparator<ScheduledTick<?>> INTRA_TICK_DRAIN_ORDER = (o1, o2) -> {
		int compare = o1.priority.compareTo(o2.priority);
		return compare != 0 ? compare : Long.compare(o1.subTickOrder, o2.subTickOrder);
	};
	public static final Strategy<ScheduledTick<?>> UNIQUE_TICK_HASH = new Strategy<ScheduledTick<?>>() {
		public int hashCode(final ScheduledTick<?> o) {
			return 31 * o.pos().hashCode() + o.type().hashCode();
		}

		public boolean equals(@Nullable final ScheduledTick<?> a, @Nullable final ScheduledTick<?> b) {
			if (a == b) {
				return true;
			} else {
				return a != null && b != null ? a.type() == b.type() && a.pos().equals(b.pos()) : false;
			}
		}
	};

	public ScheduledTick(final T type, final BlockPos pos, final long triggerTick, final long subTickOrder) {
		this(type, pos, triggerTick, TickPriority.NORMAL, subTickOrder);
	}

	public ScheduledTick(T type, BlockPos pos, long triggerTick, TickPriority priority, long subTickOrder) {
		pos = pos.immutable();
		this.type = type;
		this.pos = pos;
		this.triggerTick = triggerTick;
		this.priority = priority;
		this.subTickOrder = subTickOrder;
	}

	public static <T> ScheduledTick<T> probe(final T type, final BlockPos pos) {
		return new ScheduledTick<>(type, pos, 0L, TickPriority.NORMAL, 0L);
	}

	public SavedTick<T> toSavedTick(final long currentTick) {
		return new SavedTick<>(this.type, this.pos, (int)(this.triggerTick - currentTick), this.priority);
	}
}