#pragma once

#include <vector>
#include <string>

#include "Common/Log.h"
#include "VulkanLoader.h"

// Simple scoped based profiler, initially meant for instant one-time tasks like texture uploads
// etc. Supports recursive scopes. Scopes are not yet tracked separately for each command buffer.
// For the pass profiler in VulkanQueueRunner, a purpose-built separate profiler that can take only
// one measurement between each pass makes more sense.
//
// Put the whole thing in a FrameData to allow for overlap.

struct ProfilerScope {
	std::string name;
	int startQueryId;
	int endQueryId;
	int level;
};

class VulkanContext; 


class VulkanProfiler {
public:
	void Init(VulkanContext *vulkan);
	void Shutdown();

	void BeginFrame(VulkanContext *vulkan, VkCommandBuffer firstCommandBuffer);

	void Begin(VkCommandBuffer cmdBuf, VkPipelineStageFlagBits stage, const char *fmt, ...)
#ifdef __GNUC__
		__attribute__((format(printf, 4, 5)))
#endif
		;
	void End(VkCommandBuffer cmdBuf, VkPipelineStageFlagBits stage);

	void SetEnabledPtr(bool *enabledPtr) {
		enabledPtr_ = enabledPtr;
	}
private:
	VulkanContext *vulkan_;

	VkQueryPool queryPool_ = VK_NULL_HANDLE;
	std::vector<ProfilerScope> scopes_;
	int numQueries_ = 0;
	bool firstFrame_ = true;
	bool *enabledPtr_ = nullptr;

	std::vector<size_t> scopeStack_;

	const int MAX_QUERY_COUNT = 1024;
};