renderbloomcomponent.h

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
 
#pragma once
 
// Local Includes
#include "rendertotexturecomponent.h"
#include "blurshader.h"
 
namespace nap
{
    // Forward Declares
    class RenderBloomComponentInstance;
 
    class NAPAPI RenderBloomComponent : public RenderableComponent
    {
        RTTI_ENABLE(RenderableComponent)
        DECLARE_COMPONENT(RenderBloomComponent, RenderBloomComponentInstance)
    public:
        ResourcePtr<RenderTexture2D>    mInputTexture = nullptr;                            
        ResourcePtr<RenderTexture2D>    mOutputTexture = nullptr;                           
        EBlurSamples                    mKernel = EBlurSamples::X5;                         
        uint                            mPassCount = 1;                                     
    };
 
 
    class NAPAPI RenderBloomComponentInstance : public  RenderableComponentInstance
    {
        RTTI_ENABLE(RenderableComponentInstance)
    public:
        RenderBloomComponentInstance(EntityInstance& entity, Component& resource);
 
        virtual bool init(utility::ErrorState& errorState) override;
 
        void draw();
 
        Texture2D& getOutputTexture() { return *mOutputTexture; }
 
    protected:
        virtual void onDraw(IRenderTarget& renderTarget, VkCommandBuffer commandBuffer, const glm::mat4& viewMatrix, const glm::mat4& projectionMatrix) override;
 
    private:
        using DoubleBufferedRenderTexture2D = std::array<std::unique_ptr<RenderTexture2D>, 2>;
        using DoubleBufferedRenderTarget = std::array<std::unique_ptr<RenderTarget>, 2>;
 
        RenderService*              mRenderService = nullptr;           
        RenderTexture2D*            mInputTexture = nullptr;            
        RenderTexture2D*            mOutputTexture = nullptr;           
        
        std::vector<DoubleBufferedRenderTexture2D> mBloomTextures;      
        std::vector<DoubleBufferedRenderTarget> mBloomRTs;              
 
        MaterialInstanceResource    mMaterialInstanceResource;          
        MaterialInstance            mMaterialInstance;                  
 
        RenderableMesh              mRenderableMesh;                    
        nap::PlaneMesh              mPlane;                             
 
        glm::mat4x4                 mModelMatrix;                       
 
        UniformMat4Instance*        mModelMatrixUniform = nullptr;      
        UniformMat4Instance*        mProjectMatrixUniform = nullptr;    
        UniformMat4Instance*        mViewMatrixUniform = nullptr;       
        UniformStructInstance*      mMVPStruct = nullptr;               
 
        Sampler2DInstance*          mColorTextureSampler = nullptr;     
        UniformVec2Instance*        mDirectionUniform = nullptr;        
        UniformVec2Instance*        mTextureSizeUniform = nullptr;      
 
        UniformMat4Instance* ensureUniform(const std::string& uniformName, nap::UniformStructInstance& mvpStruct, utility::ErrorState& error);
    };
}