1 /* 2 * Copyright (C) 2023 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef RENDER_RENDER__NODE__RENDER_NODE_COMPUTE_GENERIC_H 17 #define RENDER_RENDER__NODE__RENDER_NODE_COMPUTE_GENERIC_H 18 19 #include <base/math/vector.h> 20 #include <base/util/uid.h> 21 #include <render/namespace.h> 22 #include <render/nodecontext/intf_pipeline_descriptor_set_binder.h> 23 #include <render/nodecontext/intf_render_node.h> 24 #include <render/render_data_structures.h> 25 #include <render/resource_handle.h> 26 27 RENDER_BEGIN_NAMESPACE() 28 class IRenderCommandList; 29 class IRenderNodeContextManager; 30 struct RenderNodeGraphInputs; 31 32 class RenderNodeComputeGeneric final : public IRenderNode { 33 public: 34 RenderNodeComputeGeneric() = default; 35 ~RenderNodeComputeGeneric() override = default; 36 37 void InitNode(IRenderNodeContextManager& renderNodeContextMgr) override; 38 void PreExecuteFrame() override; 39 void ExecuteFrame(IRenderCommandList& cmdList) override; 40 41 // for plugin / factory interface 42 static constexpr BASE_NS::Uid UID { "dc5da6df-0234-4275-b7a3-d68421e76313" }; 43 static constexpr char const* TYPE_NAME = "RenderNodeComputeGeneric"; 44 static constexpr IRenderNode::BackendFlags BACKEND_FLAGS = IRenderNode::BackendFlagBits::BACKEND_FLAG_BITS_DEFAULT; 45 static constexpr IRenderNode::ClassType CLASS_TYPE = IRenderNode::ClassType::CLASS_TYPE_NODE; 46 static IRenderNode* Create(); 47 static void Destroy(IRenderNode* instance); 48 49 private: 50 IRenderNodeContextManager* renderNodeContextMgr_ { nullptr }; 51 52 void ParseRenderNodeInputs(); 53 RenderHandle GetPsoHandle(IRenderNodeContextManager& renderNodeContextMgr); 54 55 // Json resources which might need re-fetching 56 struct JsonInputs { 57 RenderNodeGraphInputs::InputResources resources; 58 59 RenderNodeGraphInputs::RenderDataStore renderDataStore; 60 RenderNodeGraphInputs::RenderDataStore renderDataStoreSpecialization; 61 62 bool hasChangeableResourceHandles { false }; 63 }; 64 JsonInputs jsonInputs_; 65 66 RenderNodeHandles::InputResources inputResources_; 67 RenderHandle shader_; 68 69 IPipelineDescriptorSetBinder::Ptr pipelineDescriptorSetBinder_; 70 PipelineLayout pipelineLayout_; 71 RenderHandle psoHandle_; 72 73 // data store push constant 74 bool useDataStorePushConstant_ { false }; 75 76 // data store shader specialization 77 bool useDataStoreShaderSpecialization_ { false }; 78 struct ShaderSpecilizationData { 79 BASE_NS::vector<ShaderSpecialization::Constant> constants; 80 BASE_NS::vector<uint32_t> data; 81 }; 82 ShaderSpecilizationData shaderSpecializationData_; 83 84 BASE_NS::Math::UVec3 targetSize_ { 1u, 1u, 1u }; 85 ShaderThreadGroup threadGroupSize_ { 1u, 1u, 1u }; 86 }; 87 RENDER_END_NAMESPACE() 88 89 #endif // CORE__RENDER__NODE__RENDER_NODE_COMPUTE_GENERIC_H 90