mirror of
https://github.com/PancakeTAS/lsfg-vk.git
synced 2025-10-30 07:01:10 +00:00
61 lines
1.8 KiB
C++
61 lines
1.8 KiB
C++
#pragma once
|
|
|
|
#include "core/buffer.hpp"
|
|
#include "core/commandbuffer.hpp"
|
|
#include "core/descriptorset.hpp"
|
|
#include "core/image.hpp"
|
|
#include "core/pipeline.hpp"
|
|
#include "core/sampler.hpp"
|
|
#include "core/shadermodule.hpp"
|
|
#include "common/utils.hpp"
|
|
|
|
#include <array>
|
|
#include <cstdint>
|
|
|
|
namespace LSFG::Shaders {
|
|
|
|
///
|
|
/// Beta shader.
|
|
///
|
|
class Beta {
|
|
public:
|
|
Beta() = default;
|
|
|
|
///
|
|
/// Initialize the shaderchain.
|
|
///
|
|
/// @param inImgs Three sets of four RGBA images, corresponding to a frame count % 3.
|
|
///
|
|
/// @throws LSFG::vulkan_error if resource creation fails.
|
|
///
|
|
Beta(Vulkan& vk, std::array<std::array<Core::Image, 4>, 3> inImgs);
|
|
|
|
///
|
|
/// Dispatch the shaderchain.
|
|
///
|
|
void Dispatch(const Core::CommandBuffer& buf, uint64_t frameCount);
|
|
|
|
/// Get the output images
|
|
[[nodiscard]] const auto& getOutImages() const { return this->outImgs; }
|
|
|
|
/// Trivially copyable, moveable and destructible
|
|
Beta(const Beta&) noexcept = default;
|
|
Beta& operator=(const Beta&) noexcept = default;
|
|
Beta(Beta&&) noexcept = default;
|
|
Beta& operator=(Beta&&) noexcept = default;
|
|
~Beta() = default;
|
|
private:
|
|
std::array<Core::ShaderModule, 5> shaderModules;
|
|
std::array<Core::Pipeline, 5> pipelines;
|
|
std::array<Core::Sampler, 2> samplers;
|
|
Core::Buffer buffer;
|
|
std::array<Core::DescriptorSet, 3> firstDescriptorSet;
|
|
std::array<Core::DescriptorSet, 4> descriptorSets;
|
|
|
|
std::array<std::array<Core::Image, 4>, 3> inImgs;
|
|
std::array<Core::Image, 2> tempImgs1;
|
|
std::array<Core::Image, 2> tempImgs2;
|
|
std::array<Core::Image, 6> outImgs;
|
|
};
|
|
|
|
}
|