vk_end_rendering_node.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2024 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

 
#pragma once
 
#include "vk_node_info.hh"
 
namespace blender::gpu::render_graph {

struct VKEndRenderingData {
  /* Render pass used (when dynamic rendering is not supported). */
  VkRenderPass vk_render_pass;
};

class VKEndRenderingNode : public VKNodeInfo<VKNodeType::END_RENDERING,
                                             VKEndRenderingData,
                                             VKEndRenderingData,
                                             VK_PIPELINE_STAGE_ALL_GRAPHICS_BIT,
                                             VKResourceType::NONE> {
 public:
  template<typename Node, typename Storage>
  void set_node_data(Node &node, Storage & /*storage*/, const CreateInfo &create_info)
  {
    node.end_rendering = create_info;
  }

  void build_links(VKResourceStateTracker & /*resources*/,
                   VKRenderGraphNodeLinks & /*node_links*/,
                   const CreateInfo & /*create_info*/) override
  {
  }

  void build_commands(VKCommandBufferInterface &command_buffer,
                      Data &data,
                      VKBoundPipelines & /*r_bound_pipelines*/) override
  {
    const bool is_dynamic_rendering = data.vk_render_pass == VK_NULL_HANDLE;
    if (is_dynamic_rendering) {
      command_buffer.end_rendering();
    }
    else {
      command_buffer.end_render_pass();
    }
  }
};
}  // namespace blender::gpu::render_graph