Thraix
64 lines
1.7 KiB
C++
64 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#include "Common.h"
|
|
#include "VertexDescriptor.h"
|
|
|
|
#include <vulkan/vulkan.hpp>
|
|
#include <map>
|
|
|
|
namespace Copium
|
|
{
|
|
class PipelineCreator
|
|
{
|
|
struct DescriptorSetBinding
|
|
{
|
|
uint32_t binding;
|
|
VkDescriptorType type;
|
|
uint32_t count;
|
|
VkShaderStageFlags flags;
|
|
};
|
|
friend class Pipeline;
|
|
private:
|
|
std::map<uint32_t, std::vector<DescriptorSetBinding>> descriptorSetLayouts{};
|
|
|
|
std::string vertexShader;
|
|
std::string fragmentShader;
|
|
VertexDescriptor vertexDescriptor{};
|
|
VkPrimitiveTopology topology = VK_PRIMITIVE_TOPOLOGY_TRIANGLE_LIST;
|
|
VkCullModeFlags cullMode = VK_CULL_MODE_BACK_BIT;
|
|
VkFrontFace frontFace = VK_FRONT_FACE_CLOCKWISE;
|
|
VkRenderPass renderPass = VK_NULL_HANDLE;
|
|
|
|
public:
|
|
PipelineCreator(VkRenderPass renderPass, const std::string& vertexShader, const std::string& fragmentShader)
|
|
: vertexShader{vertexShader}, fragmentShader{fragmentShader}, renderPass{renderPass}
|
|
{}
|
|
|
|
void SetVertexDescriptor(const VertexDescriptor& descriptor)
|
|
{
|
|
vertexDescriptor = descriptor;
|
|
}
|
|
|
|
void AddDescriptorSetLayoutBinding(uint32_t set, uint32_t binding, VkDescriptorType type, uint32_t count, VkShaderStageFlags stageFlags)
|
|
{
|
|
CP_ASSERT(set <= descriptorSetLayouts.size(), "AddDescriptorSetLayoutBinding : Cannot add descriptor set with set number greater than the current set count");
|
|
descriptorSetLayouts[set].emplace_back(DescriptorSetBinding{binding, type, count, stageFlags});
|
|
}
|
|
|
|
void SetPrimitiveTopology(VkPrimitiveTopology primitiveTopology)
|
|
{
|
|
topology = primitiveTopology;
|
|
}
|
|
|
|
void SetCullMode(VkCullModeFlags flags)
|
|
{
|
|
cullMode = flags;
|
|
}
|
|
|
|
void SetCullFrontFace(VkFrontFace cullFrontFace)
|
|
{
|
|
frontFace = cullFrontFace;
|
|
}
|
|
};
|
|
}
|