Pass.h

#pragma once
 
#include "Common.h"
 
#include "Device.h"
#include "Swapchain.h"
 
namespace Mandrill
{
    class Pass
    {
    public:
        MANDRILL_NON_COPYABLE(Pass)
 
        
        MANDRILL_API Pass(ptr<Device> pDevice, std::vector<ptr<Image>> colorAttachments, ptr<Image> pDepthAttachment);
 
        MANDRILL_API Pass(ptr<Device> pDevice, VkExtent2D extent, VkFormat format, uint32_t colorAttachmentCount = 1,
                          bool depthAttachment = true, VkSampleCountFlagBits sampleCount = VK_SAMPLE_COUNT_1_BIT);
 
        MANDRILL_API Pass(ptr<Device> pDevice, VkExtent2D extent, std::vector<VkFormat> formats,
                          bool depthAttachment = true, VkSampleCountFlagBits sampleCount = VK_SAMPLE_COUNT_1_BIT);
 
        MANDRILL_API ~Pass();
 
        MANDRILL_API void transitionForRendering(VkCommandBuffer cmd, ptr<Image> pImage) const;
 
        MANDRILL_API void transitionForBlitting(VkCommandBuffer cmd, ptr<Image> pImage) const;
 
        MANDRILL_API void begin(VkCommandBuffer cmd);
 
        MANDRILL_API void begin(VkCommandBuffer cmd, glm::vec4 clearColor,
                                VkAttachmentLoadOp loadOp = VK_ATTACHMENT_LOAD_OP_CLEAR);
 
        MANDRILL_API void begin(VkCommandBuffer cmd, ptr<Image> pImage);
 
        MANDRILL_API void end(VkCommandBuffer cmd) const;
 
        MANDRILL_API void end(VkCommandBuffer cmd, ptr<Image> pImage) const;
 
        MANDRILL_API void update(std::vector<ptr<Image>> colorAttachments, ptr<Image> pDepthAttachment);
 
        MANDRILL_API void update(VkExtent2D extent);
 
        MANDRILL_API VkPipelineRenderingCreateInfo getPipelineRenderingCreateInfo() const
        {
            return mPipelineRenderingCreateInfo;
        }
 
        MANDRILL_API const std::vector<ptr<Image>>& getColorAttachments() const
        {
            return mColorAttachments;
        }
 
        MANDRILL_API ptr<Image> getOutput() const
        {
            return mpResolveAttachment ? mpResolveAttachment : mColorAttachments[0];
        }
 
        MANDRILL_API VkExtent2D getExtent() const
        {
            return mExtent;
        }
 
        MANDRILL_API VkSampleCountFlagBits getSampleCount() const
        {
            return mpResolveAttachment ? mpDevice->getSampleCount() : VK_SAMPLE_COUNT_1_BIT;
        }
 
    private:
        void createExplicitPass(std::vector<ptr<Image>> colorAttachments, ptr<Image> depthAttachment);
        void createImplicitPass(bool depthAttachment, VkSampleCountFlagBits sampleCount);
 
        ptr<Device> mpDevice;
 
        VkPipelineRenderingCreateInfo mPipelineRenderingCreateInfo;
        VkExtent2D mExtent;
 
        std::vector<VkFormat> mFormats;
 
        bool mImplicitAttachments;
        std::vector<ptr<Image>> mColorAttachments;
        ptr<Image> mpDepthAttachment;
        ptr<Image> mpResolveAttachment;
    };
} // namespace Mandrill