/*
* Copyright (c) 2024 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef RENDER_RENDER__NODE__RENDER_NODE_COMPUTE_GENERIC_H
#define RENDER_RENDER__NODE__RENDER_NODE_COMPUTE_GENERIC_H
#include
#include
#include
#include
#include
#include
#include
RENDER_BEGIN_NAMESPACE()
class IRenderCommandList;
class IRenderNodeContextManager;
struct RenderNodeGraphInputs;
class RenderNodeComputeGeneric final : public IRenderNode {
public:
RenderNodeComputeGeneric() = default;
~RenderNodeComputeGeneric() override = default;
void InitNode(IRenderNodeContextManager& renderNodeContextMgr) override;
void PreExecuteFrame() override;
void ExecuteFrame(IRenderCommandList& cmdList) override;
ExecuteFlags GetExecuteFlags() const override
{
return 0U;
}
// for plugin / factory interface
static constexpr BASE_NS::Uid UID { "dc5da6df-0234-4275-b7a3-d68421e76313" };
static constexpr char const* TYPE_NAME = "RenderNodeComputeGeneric";
static constexpr IRenderNode::BackendFlags BACKEND_FLAGS = IRenderNode::BackendFlagBits::BACKEND_FLAG_BITS_DEFAULT;
static constexpr IRenderNode::ClassType CLASS_TYPE = IRenderNode::ClassType::CLASS_TYPE_NODE;
static IRenderNode* Create();
static void Destroy(IRenderNode* instance);
private:
IRenderNodeContextManager* renderNodeContextMgr_ { nullptr };
void ParseRenderNodeInputs();
RenderHandle GetPsoHandle(IRenderNodeContextManager& renderNodeContextMgr);
// Json resources which might need re-fetching
struct JsonInputs {
RenderNodeGraphInputs::InputResources resources;
RenderNodeGraphInputs::InputResources dispatchResources;
RenderNodeGraphInputs::RenderDataStore renderDataStore;
RenderNodeGraphInputs::RenderDataStore renderDataStoreSpecialization;
bool hasChangeableResourceHandles { false };
bool hasChangeableDispatchHandles { false };
};
JsonInputs jsonInputs_;
RenderNodeHandles::InputResources inputResources_;
RenderNodeHandles::InputResources dispatchResources_;
RenderHandle shader_;
IPipelineDescriptorSetBinder::Ptr pipelineDescriptorSetBinder_;
PipelineLayout pipelineLayout_;
RenderHandle psoHandle_;
// data store push constant
bool useDataStorePushConstant_ { false };
// data store shader specialization
bool useDataStoreShaderSpecialization_ { false };
struct ShaderSpecilizationData {
BASE_NS::vector constants;
BASE_NS::vector data;
};
ShaderSpecilizationData shaderSpecializationData_;
ShaderThreadGroup threadGroupSize_ { 1u, 1u, 1u };
};
RENDER_END_NAMESPACE()
#endif // CORE__RENDER__NODE__RENDER_NODE_COMPUTE_GENERIC_H