#pragma once

#include "cve_device.hpp"

#include <string>
#include <vector>

namespace cve {
  struct PipelineConfigInfo {
    PipelineConfigInfo(const PipelineConfigInfo&) = delete;
    PipelineConfigInfo& operator=(const PipelineConfigInfo&) = delete;

    VkPipelineViewportStateCreateInfo viewportInfo;
    VkPipelineInputAssemblyStateCreateInfo inputAssemblyInfo;
    VkPipelineRasterizationStateCreateInfo rasterizationInfo;
    VkPipelineMultisampleStateCreateInfo multisampleInfo;
    VkPipelineColorBlendAttachmentState colorBlendAttachment;
    VkPipelineColorBlendStateCreateInfo colorBlendInfo;
    VkPipelineDepthStencilStateCreateInfo depthStencilInfo;
    std::vector<VkDynamicState> dynamicStateEnables;
    VkPipelineDynamicStateCreateInfo dynamicStateInfo;
    VkPipelineLayout pipelineLayout = nullptr;
    VkRenderPass renderPass = nullptr;
    uint32_t subpass = 0;
  };
  class CvePipeline {
    public:
      CvePipeline(CveDevice &device, const std::string& vertFilepath, 
          const std::string& fragFilepath, const PipelineConfigInfo &configInfo);
      ~CvePipeline();

      CvePipeline(const CvePipeline&) = delete;
      CvePipeline& operator=(const CvePipeline&) = delete;

      void bind(VkCommandBuffer commandBuffer);
      static void defaultPipelineConfigInfo(PipelineConfigInfo& configInfo);

    private:
      static std::vector<char> readFile(const std::string& filepath);

      void createGraphicsPipeline(const std::string& vertFilepath, 
          const std::string& fragFilepath, const PipelineConfigInfo &configInfo);
    
    void createShaderModule(const std::vector<char>& code, VkShaderModule *shaderModule);
    
    CveDevice& cveDevice;
    VkPipeline graphicsPipeline;
    VkShaderModule vertShaderModule;
    VkShaderModule fragShaderModule;
  };
}