1 /*
2  * Copyright (c) 2024 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 CORE3D_RENDER__NODE__RENDER_NODE_CAMERA_SINGLE_POST_PROCESS_H
17 #define CORE3D_RENDER__NODE__RENDER_NODE_CAMERA_SINGLE_POST_PROCESS_H
18 
19 #include <3d/namespace.h>
20 #include <3d/render/intf_render_data_store_default_light.h>
21 #include <base/containers/string.h>
22 #include <base/util/uid.h>
23 #include <render/datastore/intf_render_data_store_post_process.h>
24 #include <render/datastore/render_data_store_render_pods.h>
25 #include <render/device/pipeline_state_desc.h>
26 #include <render/nodecontext/intf_pipeline_descriptor_set_binder.h>
27 #include <render/nodecontext/intf_render_node.h>
28 #include <render/render_data_structures.h>
29 #include <render/resource_handle.h>
30 
31 #include "render/node/render_copy_helper.h"
32 #include "render/render_node_scene_util.h"
33 
34 RENDER_BEGIN_NAMESPACE()
35 struct RenderNodeGraphInputs;
36 RENDER_END_NAMESPACE()
37 
38 CORE3D_BEGIN_NAMESPACE()
39 class IRenderDataStoreDefaultCamera;
40 class IRenderDataStoreDefaultScene;
41 
42 class RenderNodeCameraSinglePostProcess final : public RENDER_NS::IRenderNode {
43 public:
44     RenderNodeCameraSinglePostProcess() = default;
45     ~RenderNodeCameraSinglePostProcess() override = default;
46 
47     void InitNode(RENDER_NS::IRenderNodeContextManager& renderNodeContextMgr) override;
48     void PreExecuteFrame() override;
49     void ExecuteFrame(RENDER_NS::IRenderCommandList& cmdList) override;
GetExecuteFlags()50     ExecuteFlags GetExecuteFlags() const override
51     {
52         return 0U;
53     }
54 
55     // for plugin / factory interface
56     static constexpr BASE_NS::Uid UID { "6257f029-f747-44c7-8de9-87b41b8afd6a" };
57     static constexpr char const* TYPE_NAME = "RenderNodeCameraSinglePostProcess";
58     static constexpr IRenderNode::BackendFlags BACKEND_FLAGS = IRenderNode::BackendFlagBits::BACKEND_FLAG_BITS_DEFAULT;
59     static constexpr IRenderNode::ClassType CLASS_TYPE = IRenderNode::ClassType::CLASS_TYPE_NODE;
60     static IRenderNode* Create();
61     static void Destroy(IRenderNode* instance);
62 
63     struct ShadowBuffers {
64         RENDER_NS::RenderHandle pcfDepthHandle;
65         RENDER_NS::RenderHandle vsmColorHandle;
66 
67         RENDER_NS::RenderHandle pcfSamplerHandle;
68         RENDER_NS::RenderHandle vsmSamplerHandle;
69     };
70     struct DefaultImagesAndSamplers {
71         RENDER_NS::RenderHandle cubemapHandle;
72 
73         RENDER_NS::RenderHandle linearHandle;
74         RENDER_NS::RenderHandle nearestHandle;
75         RENDER_NS::RenderHandle linearMipHandle;
76 
77         RENDER_NS::RenderHandle colorPrePassHandle;
78     };
79 
80 private:
81     RENDER_NS::IRenderNodeContextManager* renderNodeContextMgr_ { nullptr };
82 
83     void RegisterOutputs();
84     void InitCreateBinders();
85     void ParseRenderNodeInputs();
86     void UpdateGlobalPostProcessUbo();
87     void UpdateImageData();
88     void ProcessPostProcessConfiguration();
89     void GetSceneUniformBuffers(const BASE_NS::string_view us);
90     void UpdateCurrentScene(const IRenderDataStoreDefaultScene& dataStoreScene,
91         const IRenderDataStoreDefaultCamera& dataStoreCamera, const IRenderDataStoreDefaultLight& dataStoreLight);
92     void UpdateSet0(RENDER_NS::IRenderCommandList& cmdList);
93     void BindDefaultResources(const uint32_t set, const RENDER_NS::DescriptorSetLayoutBindingResources& bindings);
94     void ExecuteSinglePostProcess(RENDER_NS::IRenderCommandList& cmdList);
95 
96     enum class DefaultOutputImage : uint32_t {
97         OUTPUT = 0,
98         INPUT_OUTPUT_COPY = 1,
99         INPUT = 2,
100         WHITE = 3,
101         BLACK = 4,
102     };
103 
104     static constexpr uint64_t INVALID_CAM_ID { 0xFFFFFFFFffffffff };
105     // Json resources which might need re-fetching
106     struct JsonInputs {
107         RENDER_NS::RenderNodeGraphInputs::InputRenderPass renderPass;
108         RENDER_NS::RenderNodeGraphInputs::InputResources resources;
109         RENDER_NS::RenderNodeGraphInputs::InputResources dispatchResources;
110 
111         RENDER_NS::RenderNodeGraphInputs::RenderDataStore renderDataStore;
112 
113         BASE_NS::string ppName;
114         DefaultOutputImage defaultOutputImage { DefaultOutputImage::OUTPUT };
115 
116         BASE_NS::string customCameraName;
117         uint64_t customCameraId { INVALID_CAM_ID };
118 
119         bool hasChangeableRenderPassHandles { false };
120         bool hasChangeableResourceHandles { false };
121         bool hasChangeableDispatchHandles { false };
122 
123         uint32_t inputIdx { ~0u };
124         uint32_t outputIdx { ~0u };
125     };
126     JsonInputs jsonInputs_;
127 
128     struct BuiltInVariables {
129         RENDER_NS::RenderHandle input;
130 
131         RENDER_NS::RenderHandle output;
132 
133         RENDER_NS::RenderHandle defBuffer;
134         RENDER_NS::RenderHandle defBlackImage;
135         RENDER_NS::RenderHandle defWhiteImage;
136         RENDER_NS::RenderHandle defSampler;
137     };
138     BuiltInVariables builtInVariables_;
139 
140     SceneBufferHandles sceneBuffers_;
141     SceneCameraBufferHandles cameraBuffers_;
142 
143     DefaultImagesAndSamplers defaultImagesAndSamplers_;
144 
145     struct CurrentScene {
146         RenderCamera camera;
147         RENDER_NS::RenderHandle cameraEnvRadianceHandle;
148         RENDER_NS::RenderHandle prePassColorTarget;
149 
150         bool hasShadow { false };
151         IRenderDataStoreDefaultLight::ShadowTypes shadowTypes {};
152         IRenderDataStoreDefaultLight::LightingFlags lightingFlags { 0u };
153     };
154     CurrentScene currentScene_;
155 
156     ShadowBuffers shadowBuffers_;
157 
158     RENDER_NS::RenderNodeHandles::InputRenderPass inputRenderPass_;
159     RENDER_NS::RenderNodeHandles::InputResources inputResources_;
160     RENDER_NS::RenderNodeHandles::InputResources dispatchResources_;
161     RENDER_NS::RenderHandle shader_;
162     RENDER_NS::PipelineLayout pipelineLayout_;
163     RENDER_NS::RenderHandle psoHandle_;
164     RENDER_NS::PushConstant pushConstant_;
165 
166     RENDER_NS::IPipelineDescriptorSetBinder::Ptr pipelineDescriptorSetBinder_;
167 
168     RENDER_NS::PostProcessConfiguration ppGlobalConfig_;
169     RENDER_NS::IRenderDataStorePostProcess::PostProcess ppLocalConfig_;
170 
171     bool graphics_ { true };
172     RENDER_NS::ShaderThreadGroup threadGroupSize_ { 1u, 1u, 1u };
173 
174     SceneRenderDataStores stores_;
175 
176     struct UboHandles {
177         // first 512 aligned is global post process
178         // after (256) we have effect local data
179         RENDER_NS::RenderHandleReference postProcess;
180     };
181     UboHandles ubos_;
182 
183     RenderCopyHelper renderCopy_;
184 
185     bool useAutoBindSet0_ { false };
186     bool valid_ { false };
187 };
188 CORE3D_END_NAMESPACE()
189 
190 #endif // CORE3D_RENDER__NODE__RENDER_NODE_CAMERA_SINGLE_POST_PROCESS_H
191