Shader.h

#pragma once
 
#include "spark/render/Export.h"
#include "spark/render/Pipeline.h"
#include "spark/render/ShaderStages.h"
 
#include <algorithm>
#include <memory>
#include <string>
 
namespace spark::render
{
    class SPARK_RENDER_EXPORT IShaderModule
    {
    public:
        virtual ~IShaderModule() noexcept = default;
 
        [[nodiscard]] virtual ShaderStage stage() const noexcept = 0;
 
        [[nodiscard]] virtual const std::string& fileName() const noexcept = 0;
 
        [[nodiscard]] virtual const std::string& entryPoint() const noexcept = 0;
    };
 
    class SPARK_RENDER_EXPORT IShaderProgram
    {
    public:
        virtual ~IShaderProgram() noexcept = default;
 
        [[nodiscard]] std::vector<const IShaderModule*> shaders() const noexcept { return genericShaders(); }
 
        [[nodiscard]] virtual std::shared_ptr<IPipelineLayout> reflectPipelineLayout() const = 0;
 
    private:
        [[nodiscard]] virtual std::vector<const IShaderModule*> genericShaders() const noexcept = 0;
    };
 
    template <typename ShaderModuleType>
    class ShaderProgram : public IShaderProgram
    {
    public:
        using shader_module_type = ShaderModuleType;
 
    public:
        [[nodiscard]] virtual std::vector<const shader_module_type*> shaders() const noexcept = 0;
 
    private:
        [[nodiscard]] std::vector<const IShaderModule*> genericShaders() const noexcept final
        {
            auto tmp = shaders();
            std::vector<const IShaderModule*> result;
            result.reserve(tmp.size());
            std::ranges::transform(tmp, std::back_inserter(result), [](const auto& shader) { return static_cast<const IShaderModule*>(shader); });
            return result;
        }
    };
}