lsfg-vk/lsfg-vk-backend/src/shaderchains/delta1.hpp

#pragma once

#include "../helpers/managed_shader.hpp"
#include "../helpers/utils.hpp"
#include "lsfg-vk-common/helpers/pointers.hpp"
#include "lsfg-vk-common/vulkan/command_buffer.hpp"
#include "lsfg-vk-common/vulkan/image.hpp"

#include <vector>

#include <vulkan/vulkan_core.h>

namespace ctx { struct Ctx; }

namespace chains {
    /// gamma shaderchain
    class Delta1 {
    public:
        /// create a gamma shaderchain
        /// @param ctx context
        /// @param idx generated frame index
        /// @param sourceImages0 source images
        /// @param sourceImages1 source images
        /// @param additionalInput0 additional input image
        /// @param additionalInput1 additional input image
        /// @param additionalInput2 additional input image
        Delta1(const ls::Ctx& ctx, size_t idx,
            const std::vector<vk::Image>& sourceImages0,
            const std::vector<vk::Image>& sourceImages1,
            const vk::Image& additionalInput0,
            const vk::Image& additionalInput1,
            const vk::Image& additionalInput2);

        /// prepare the shaderchain initially
        /// @param cmd command buffer
        void prepare(const vk::CommandBuffer& cmd) const;

        /// render the gamma shaderchain
        /// @param cmd command buffer
        void render(const vk::CommandBuffer& cmd) const;

        /// get the first generated image
        /// @return image
        [[nodiscard]] const auto& getImage0() const { return *this->image0; }

        /// get the second generated image
        /// @return image
        [[nodiscard]] const auto& getImage1() const { return *this->image1; }
    private:
        std::vector<vk::Image> tempImages0;
        std::vector<vk::Image> tempImages1;
        ls::lazy<vk::Image> image0;
        ls::lazy<vk::Image> image1;

        std::vector<ls::ManagedShader> sets;
        VkExtent2D dispatchExtent{};
    };
}