Shader.h

Go to the documentation of this file.

#pragma once
 
#include <filesystem>
#include <string>
#include <vector>
#include <vulkan/vulkan_core.h>
 
 
class Shader {
public:
    virtual ~Shader() = default;
    virtual void destroyShaders(VkDevice device) { }
 
  protected:
    void destroyShader(VkShaderModule shaderModule, const VkDevice &device);
    VkShaderModule createShaderModule(VkDevice device, const std::vector<char> &code);
};
 
class ComputeShader : public Shader
{
public:
    ComputeShader(VkDevice device, const std::filesystem::path& shaderPath);
 
    void destroyShaders(VkDevice device) override;
    [[nodiscard]] VkShaderModule getComputeShaderModule() const { return computeShaderModule; }
 
private:
    VkShaderModule computeShaderModule = nullptr;
};
 
class MeshShader : public Shader
{
public:
    MeshShader(VkDevice device, const std::filesystem::path& mesh_shader_path, const std::filesystem::path& fragment_shader_path);
    [[nodiscard]] VkShaderModule getMeshShader() const;
    [[nodiscard]] VkShaderModule getFragmentShader() const;
 
    void destroyShaders(VkDevice device) override;
private:
    VkShaderModule meshShader = nullptr;
    VkShaderModule fragmentShader = nullptr;
};
 
class VertexShader : public Shader
{
    public:
    VertexShader(VkDevice device, const std::filesystem::path& vertex_shader_path, const std::filesystem::path& fragment_shader_path);
    [[nodiscard]] VkShaderModule getVertexShader() const;
    [[nodiscard]] VkShaderModule getFragmentShader() const;
 
    void destroyShaders(VkDevice device) override;
private:
    VkShaderModule vertexShader = nullptr;
    VkShaderModule fragmentShader = nullptr;
};