package com.mojang.blaze3d.vulkan.checkpoints;

import com.mojang.blaze3d.vulkan.VulkanDevice;
import com.mojang.blaze3d.vulkan.VulkanQueue;
import java.util.List;
import java.util.function.Supplier;
import org.lwjgl.vulkan.VkCommandBuffer;

public interface CheckpointExtension extends AutoCloseable {
	CheckpointExtension.CheckpointStorage createStorage(VulkanDevice device, VulkanQueue queue, int maxFramesInFlight);

	List<CheckpointExtension.QueueCheckpoints> retrieveCheckpoints(boolean isDeviceLost);

	void close();

	public interface CheckpointStorage {
		void rotate();

		void recordCheckpoint(VkCommandBuffer commandBuffer, CheckpointExtension.CheckpointType type, Supplier<String> label);
	}

	public static enum CheckpointType {
		BEGIN_RENDER_PASS,
		END_RENDER_PASS;
	}

	public record QueueCheckpoints(long queue, List<CheckpointExtension.StageCheckpoint> checkpoints) {
	}

	public record StageCheckpoint(long stage, CheckpointExtension.CheckpointType type, String label) {
	}
}