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 #include "render_node_create_default_camera_gpu_images.h"
17
18 #include <3d/render/intf_render_data_store_default_camera.h>
19 #include <core/log.h>
20 #include <core/namespace.h>
21 #include <render/datastore/intf_render_data_store.h>
22 #include <render/datastore/intf_render_data_store_manager.h>
23 #include <render/device/intf_gpu_resource_manager.h>
24 #include <render/nodecontext/intf_node_context_pso_manager.h>
25 #include <render/nodecontext/intf_render_command_list.h>
26 #include <render/nodecontext/intf_render_node_context_manager.h>
27 #include <render/nodecontext/intf_render_node_graph_share_manager.h>
28 #include <render/nodecontext/intf_render_node_parser_util.h>
29 #include <render/render_data_structures.h>
30
31 CORE3D_BEGIN_NAMESPACE()
32 using namespace BASE_NS;
33 using namespace RENDER_NS;
34
InitNode(IRenderNodeContextManager & renderNodeContextMgr)35 void RenderNodeCreateDefaultCameraGpuImages::InitNode(IRenderNodeContextManager& renderNodeContextMgr)
36 {
37 renderNodeContextMgr_ = &renderNodeContextMgr;
38 ParseRenderNodeInputs();
39
40 const auto& renderNodeGraphData = renderNodeContextMgr_->GetRenderNodeGraphData();
41 stores_ = RenderNodeSceneUtil::GetSceneRenderDataStores(
42 renderNodeContextMgr, renderNodeGraphData.renderNodeGraphDataStoreName);
43
44 if (jsonInputs_.gpuImageDescs.empty()) {
45 CORE_LOG_W("RenderNodeCreateDefaultCameraGpuImages: No gpu image descs given to");
46 }
47
48 descs_.reserve(jsonInputs_.gpuImageDescs.size());
49 resourceHandles_.reserve(jsonInputs_.gpuImageDescs.size());
50
51 auto& gpuResourceMgr = renderNodeContextMgr.GetGpuResourceManager();
52 for (const auto& ref : jsonInputs_.gpuImageDescs) {
53 descs_.push_back(ref);
54
55 GpuImageDesc desc = ref.desc;
56 desc.width = Math::max(1u, static_cast<uint32_t>(static_cast<float>(desc.width) * ref.dependencySizeScale));
57 desc.height = Math::max(1u, static_cast<uint32_t>(static_cast<float>(desc.height) * ref.dependencySizeScale));
58 resourceHandles_.push_back(gpuResourceMgr.Create(ref.name, desc));
59 }
60 // broadcast the resources
61 for (size_t idx = 0; idx < resourceHandles_.size(); ++idx) {
62 IRenderNodeGraphShareManager& rngShareMgr = renderNodeContextMgr_->GetRenderNodeGraphShareManager();
63 rngShareMgr.RegisterRenderNodeOutput(descs_[idx].shareName, resourceHandles_[idx].GetHandle());
64 }
65 }
66
PreExecuteFrame()67 void RenderNodeCreateDefaultCameraGpuImages::PreExecuteFrame()
68 {
69 CORE_ASSERT(resourceHandles_.size() == descs_.size());
70 UpdateRenderTargets();
71
72 // broadcast the resources
73 for (size_t idx = 0; idx < resourceHandles_.size(); ++idx) {
74 IRenderNodeGraphShareManager& rngShareMgr = renderNodeContextMgr_->GetRenderNodeGraphShareManager();
75 rngShareMgr.RegisterRenderNodeOutput(descs_[idx].shareName, resourceHandles_[idx].GetHandle());
76 }
77 }
78
ParseRenderNodeInputs()79 void RenderNodeCreateDefaultCameraGpuImages::ParseRenderNodeInputs()
80 {
81 const IRenderNodeParserUtil& parserUtil = renderNodeContextMgr_->GetRenderNodeParserUtil();
82 const auto jsonVal = renderNodeContextMgr_->GetNodeJson();
83 jsonInputs_.gpuImageDescs = parserUtil.GetGpuImageDescs(jsonVal, "gpuImageDescs");
84 jsonInputs_.customCameraName = parserUtil.GetStringValue(jsonVal, "customCameraName");
85 jsonInputs_.customCameraId = parserUtil.GetUintValue(jsonVal, "customCameraId");
86 }
87
UpdateRenderTargets()88 void RenderNodeCreateDefaultCameraGpuImages::UpdateRenderTargets()
89 {
90 auto& gpuResourceMgr = renderNodeContextMgr_->GetGpuResourceManager();
91 const auto& renderDataStoreMgr = renderNodeContextMgr_->GetRenderDataStoreManager();
92 const auto* dataStoreCamera =
93 static_cast<IRenderDataStoreDefaultCamera*>(renderDataStoreMgr.GetRenderDataStore(stores_.dataStoreNameCamera));
94 if (!dataStoreCamera) {
95 return;
96 }
97 const auto& cameras = dataStoreCamera->GetCameras();
98 if (cameras.empty() || descs_.empty()) {
99 return;
100 }
101 for (size_t idx = 0; idx < descs_.size(); ++idx) {
102 // If no camera given, we default to zero camera (main camera)
103 uint32_t cameraIndex = 0U;
104 if (jsonInputs_.customCameraId != INVALID_CAM_ID) {
105 cameraIndex = dataStoreCamera->GetCameraIndex(jsonInputs_.customCameraId);
106 } else if (!jsonInputs_.customCameraName.empty()) {
107 cameraIndex = dataStoreCamera->GetCameraIndex(jsonInputs_.customCameraName);
108 }
109 if (cameraIndex >= (uint32_t)cameras.size()) {
110 continue;
111 }
112 const auto& currCamera = cameras[cameraIndex];
113 auto& currDesc = descs_[idx];
114 if ((currCamera.renderResolution.x == currDesc.desc.width) &&
115 (currCamera.renderResolution.y == currDesc.desc.height)) {
116 continue;
117 }
118
119 currDesc.desc.width = currCamera.renderResolution.x;
120 currDesc.desc.height = currCamera.renderResolution.y;
121 // additional copy
122 GpuImageDesc desc = currDesc.desc;
123 desc.width =
124 Math::max(1u, static_cast<uint32_t>(static_cast<float>(desc.width) * currDesc.dependencySizeScale));
125 desc.height =
126 Math::max(1u, static_cast<uint32_t>(static_cast<float>(desc.height) * currDesc.dependencySizeScale));
127 // replace the handle
128 resourceHandles_[idx] = gpuResourceMgr.Create(resourceHandles_[idx], desc);
129 }
130 }
131
132 // for plugin / factory interface
Create()133 RENDER_NS::IRenderNode* RenderNodeCreateDefaultCameraGpuImages::Create()
134 {
135 return new RenderNodeCreateDefaultCameraGpuImages();
136 }
137
Destroy(IRenderNode * instance)138 void RenderNodeCreateDefaultCameraGpuImages::Destroy(IRenderNode* instance)
139 {
140 delete static_cast<RenderNodeCreateDefaultCameraGpuImages*>(instance);
141 }
142 CORE3D_END_NAMESPACE()
143