computecomponent.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 "renderservice.h"
#include "materialinstance.h"
#include "shadervariabledeclarations.h"
 
// External Includes
#include <nap/resourceptr.h>
#include <component.h>
 
namespace nap
{
    // Forward Declares
    class ComputeComponentInstance;
 
    class NAPAPI ComputeComponent : public Component
    {
        RTTI_ENABLE(Component)
        DECLARE_COMPONENT(ComputeComponent, ComputeComponentInstance)
    public:
        ComputeMaterialInstanceResource     mComputeMaterialInstanceResource;   
        uint                                mInvocations = 1;                   
    };
 
 
    class NAPAPI ComputeComponentInstance : public ComponentInstance
    {
        RTTI_ENABLE(ComponentInstance)
    public:
        ComputeComponentInstance(EntityInstance& entity, Component& resource);
 
        virtual bool init(utility::ErrorState& errorState) override;
 
        void compute(VkCommandBuffer commandBuffer);
 
        void compute(VkCommandBuffer commandBuffer, uint numInvocations);
 
        ComputeMaterialInstance& getMaterialInstance()          { return mComputeMaterialInstance; }
 
        glm::u32vec3 getWorkGroupSize() const                           { return mComputeMaterialInstance.getMaterial().getShader().getWorkGroupSize(); }
 
        void setInvocations(uint numInvocations)                        { mInvocations = numInvocations; }
 
        uint getInvocations() const                                     { return mInvocations; }
 
    protected:
 
        virtual void onCompute(VkCommandBuffer commandBuffer, uint numInvocations);
 
        void insertBarriers(VkCommandBuffer commandBuffer);
 
        RenderService*                      mRenderService = nullptr;
        ComputeMaterialInstance             mComputeMaterialInstance;
        uint                                mInvocations = 1;
    };
}