#pragma once

#include <cstdint>

#include <mutex>
#include <condition_variable>

#include "Common/GPU/Vulkan/VulkanContext.h"

struct VKRStep;

enum {
	MAX_TIMESTAMP_QUERIES = 128,
};

enum class VKRRunType {
	END,
	SYNC,
};

struct QueueProfileContext {
	VkQueryPool queryPool;
	std::vector<std::string> timestampDescriptions;
	std::string profileSummary;
	double cpuStartTime;
	double cpuEndTime;
};

struct FrameDataShared {
	// Permanent objects
	VkSemaphore acquireSemaphore = VK_NULL_HANDLE;
	VkSemaphore renderingCompleteSemaphore = VK_NULL_HANDLE;

	void Init(VulkanContext *vulkan);
	void Destroy(VulkanContext *vulkan);
};

enum class FrameSubmitType {
	Pending,
	Sync,
	Present,
};

// Per-frame data, round-robin so we can overlap submission with execution of the previous frame.
struct FrameData {
	std::mutex push_mutex;
	std::condition_variable push_condVar;

	std::mutex pull_mutex;
	std::condition_variable pull_condVar;

	bool readyForFence = true;
	bool readyForRun = false;  // protected by pull_mutex
	bool skipSwap = false;

	VkFence fence;
	VkFence readbackFence;  // Strictly speaking we might only need one global of these.

	// These are on different threads so need separate pools.
	VkCommandPool cmdPoolInit;  // Written to from main thread
	VkCommandPool cmdPoolMain;  // Written to from render thread, which also submits

	VkCommandBuffer initCmd;
	VkCommandBuffer mainCmd;
	VkCommandBuffer presentCmd;

	bool hasInitCommands = false;
	bool hasMainCommands = false;
	bool hasPresentCommands = false;

	bool hasAcquired = false;

	std::vector<VKRStep *> steps;

	// Swapchain.
	uint32_t curSwapchainImage = -1;

	// Profiling.
	QueueProfileContext profile;
	bool profilingEnabled_;

	void Init(VulkanContext *vulkan, int index);
	void Destroy(VulkanContext *vulkan);

	void AcquireNextImage(VulkanContext *vulkan, FrameDataShared &shared);
	VkResult QueuePresent(VulkanContext *vulkan, FrameDataShared &shared);
	VkCommandBuffer GetInitCmd(VulkanContext *vulkan);

	// This will only submit if we are actually recording init commands.
	void SubmitPending(VulkanContext *vulkan, FrameSubmitType type, FrameDataShared &shared);

	VKRRunType RunType() const {
		return runType_;
	}

	VKRRunType runType_ = VKRRunType::END;

private:
	// Metadata for logging etc
	int index;
};