PushConstantsLayout.h

#pragma once
 
#include "spark/render/Export.h"
#include "spark/render/PushConstantsRange.h"
 
#include <algorithm>
#include <vector>
 
namespace spark::render
{
    class SPARK_RENDER_EXPORT IPushConstantsLayout
    {
    public:
        virtual ~IPushConstantsLayout() noexcept = default;
 
        [[nodiscard]] virtual unsigned int size() const noexcept = 0;
 
        [[nodiscard]] virtual const IPushConstantsRange& range(ShaderStage stage) const = 0;
 
        [[nodiscard]] std::vector<const IPushConstantsRange*> ranges() const noexcept { return genericRanges(); }
 
    private:
        [[nodiscard]] virtual std::vector<const IPushConstantsRange*> genericRanges() const noexcept = 0;
    };
 
    template <typename PushConstantsRangeType>
    class PushConstantsLayout : public IPushConstantsLayout
    {
    public:
        using PushConstantsRange = PushConstantsRangeType;
 
    public:
        [[nodiscard]] const PushConstantsRangeType& range(ShaderStage stage) const override = 0;
 
        [[nodiscard]] virtual std::vector<const PushConstantsRangeType*> ranges() const noexcept = 0;
 
    private:
        [[nodiscard]] virtual std::vector<const IPushConstantsRange*> genericRanges() const noexcept final
        {
            std::vector<const IPushConstantsRange*> ranges_vector;
            ranges_vector.reserve(ranges().size());
            std::ranges::transform(ranges(), std::back_inserter(ranges_vector), [](const auto& range) { return static_cast<const IPushConstantsRange*>(range); });
            return ranges_vector;
        }
    };
}