#include <iostream>
#include <exception>
#include <thread>
#include <vulkan/vulkan.h>
import Crafter.Graphics;
import Crafter.Asset;
import Crafter.Event;
import Crafter.Math;
using namespace Crafter;

typedef VulkanShader<"MeshShaderXYZUV.spirv", "main", VK_SHADER_STAGE_MESH_BIT_EXT, 3, {{{VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, 0}, {VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 1}, {VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, 2}}}> MeshVulkanShader;
typedef VulkanShader<"FragmentShaderTexture.spirv", "main", VK_SHADER_STAGE_FRAGMENT_BIT, 1, {{{VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, 0}}}> FragmentShader;
typedef VulkanPipeline<MeshVulkanShader, FragmentShader> Pipeline;

int main() {
    VulkanDevice::CreateDevice();
    MeshVulkanShader::CreateShader();
    FragmentShader::CreateShader();
    Pipeline::CreatePipeline();

    WindowWaylandVulkan window("Crafter.Graphics", 1280, 720);

    Asset asset;
    asset.LoadFull("cannon.cras");

    Camera camera(1.57079633, 16 / 9, 0.01, 512);
    camera.Update();
    Mesh<VertexUV>* mesh = Mesh<VertexUV>::FromAssetUV(asset.entries[0].data.data());
    DescriptorSet descriptors;
    Pipeline::GetDescriptorSet(descriptors);

    MeshShader<VertexUV> meshShader(mesh, &camera);
    meshShader.WriteDescriptors(descriptors);
    meshShader.Update();

    Asset asset2;
    asset2.LoadFull("texture.cras");

    VkCommandBuffer cmd = window.StartInit();
    VulkanTexture<Pixel_RU8_GU8_BU8_AU8>* txt = VulkanTexture<Pixel_RU8_GU8_BU8_AU8>::FromAsset(asset2.entries[0].data.data(), cmd);
    
    TextureShader texShader(txt);
    texShader.WriteDescriptors(descriptors);

    window.FinishInit();

    EventListener<VkCommandBuffer> listener(&window.onDraw, [&descriptors, &meshShader](VkCommandBuffer cmd){
        vkCmdBindDescriptorSets(cmd, VK_PIPELINE_BIND_POINT_GRAPHICS, Pipeline::pipelineLayout, 0, 2, &descriptors.set[0], 0, NULL);
        vkCmdBindPipeline(cmd, VK_PIPELINE_BIND_POINT_GRAPHICS, Pipeline::pipeline);
        VulkanDevice::vkCmdDrawMeshTasksEXTProc(cmd, meshShader.threadCount, 1, 1);
    });

    window.Start();
    while(true) {
        
    }
}