#pragma once

#include <GLFW/glfw3.h>

#include <iostream>
#include <vector>
#include <cstring>
#include <optional>
#include <set>
#include <fstream>
#include <numeric>
#include <chrono>
#include <array>
#include <memory>
#include <vulkan/vulkan_core.h>
#include <map>
#include <glm/vec2.hpp>
#include "constraints.hpp"
#include <glm/vec3.hpp>

using std::unique_ptr, std::make_unique;
using std::vector;

class SoftBody;
class Instance;
class Swapchain;
class GraphicsPipeline;
class Buffer;
class CommandPool;
class Image;
class ComputePipeline;
class Fence;
class Semaphore;
class Camera;
class DescriptorPool;

class Application {
public:
	explicit Application();
	void mainLoop();
	~Application();
private:

	void createSyncObjects();
	unique_ptr<Semaphore> imageAvailable;
	unique_ptr<Semaphore> renderFinished;
	unique_ptr<Semaphore> computeSemaphore;
	unique_ptr<Semaphore> transferFinished;
	unique_ptr<Fence> renderFence;
	unique_ptr<Fence> computeFence;
	unique_ptr<Fence> transferFence;
	std::mutex submitMutex;

	unique_ptr<Swapchain> swapchain;
	unique_ptr<DescriptorPool> descriptorPool;
	unique_ptr<GraphicsPipeline> graphicsPipeline;
	unique_ptr<Buffer> uniformBuffer;

	unique_ptr<Camera> camera;

	void createMeshBuffers();
	size_t currentDrawVertexBuffer = 0;
	unique_ptr<Buffer> vertexBuffers[2];
	unique_ptr<Buffer> faceBuffer;
	unique_ptr<Buffer> edgeBuffer;
	unique_ptr<Buffer> triangleBuffer;
	unique_ptr<Buffer> tetrahedronBuffer;
	ConstraintData constraintData {};
	vector<unique_ptr<SoftBody>> softBodies;

	struct SizeInformation {
		uint32_t vertexCount;
		uint32_t faceCount;
		uint32_t edgeCount;
		uint32_t triangleCount;
		uint32_t tetrahedronCount;
	};
	unique_ptr<Buffer> sizeInformationBuffer;

	struct Properties {
		glm::vec3 gravity;
		// Delta time in seconds
		float dt;
		uint32_t k;
	};
	unique_ptr<Buffer> propertiesBuffer;
	Properties properties {};

	void createComputePipelines();
	unique_ptr<ComputePipeline> pbdPipeline;
	unique_ptr<ComputePipeline> normalPipeline;

	void updateUniformBuffer();
	void recordDrawCommands(VkCommandBuffer cmdBuffer);
	void drawFrame();

	void recordComputeCommands(VkCommandBuffer cmdBuffer);
	struct PBDPushData {
		uint32_t state;
		ConstraintData::Partition edgePartition;
		ConstraintData::Partition tetrahedronPartition;
	};

	void update();
};