vk_draw_indirect_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 "render_graph/vk_resource_access_info.hh"
#include "vk_node_info.hh"
 
namespace blender::gpu::render_graph {

struct VKDrawIndirectData {
  VKPipelineDataGraphics graphics;
  VKVertexBufferBindings vertex_buffers;
  VkBuffer indirect_buffer;
  VkDeviceSize offset;
  uint32_t draw_count;
  uint32_t stride;
};
 
struct VKDrawIndirectCreateInfo {
  VKDrawIndirectData node_data = {};
  const VKResourceAccessInfo &resources;
  VKDrawIndirectCreateInfo(const VKResourceAccessInfo &resources) : resources(resources) {}
};
 
class VKDrawIndirectNode : public VKNodeInfo<VKNodeType::DRAW_INDIRECT,
                                             VKDrawIndirectCreateInfo,
                                             VKDrawIndirectData,
                                             VK_PIPELINE_STAGE_ALL_GRAPHICS_BIT,
                                             VKResourceType::IMAGE | VKResourceType::BUFFER> {
 public:
  template<typename Node, typename Storage>
  static void set_node_data(Node &node, Storage &storage, const CreateInfo &create_info)
  {
    node.storage_index = storage.draw_indirect.append_and_get_index(create_info.node_data);
    vk_pipeline_data_copy(storage.draw_indirect[node.storage_index].graphics,
                          create_info.node_data.graphics);
  }

  void build_links(VKResourceStateTracker &resources,
                   VKRenderGraphNodeLinks &node_links,
                   const CreateInfo &create_info) override
  {
    create_info.resources.build_links(resources, node_links);
    vk_vertex_buffer_bindings_build_links(
        resources, node_links, create_info.node_data.vertex_buffers);
    ResourceWithStamp buffer_resource = resources.get_buffer(
        create_info.node_data.indirect_buffer);
    node_links.inputs.append({buffer_resource, VK_ACCESS_INDIRECT_COMMAND_READ_BIT});
  }

  void build_commands(VKCommandBufferInterface &command_buffer,
                      Data &data,
                      VKBoundPipelines &r_bound_pipelines) override
  {
    vk_pipeline_dynamic_graphics_build_commands(
        command_buffer, data.graphics.viewport, data.graphics.line_width, r_bound_pipelines);
    vk_pipeline_data_build_commands(command_buffer,
                                    data.graphics.pipeline_data,
                                    r_bound_pipelines.graphics.pipeline,
                                    VK_PIPELINE_BIND_POINT_GRAPHICS,
                                    VK_SHADER_STAGE_ALL_GRAPHICS);
    vk_vertex_buffer_bindings_build_commands(
        command_buffer, data.vertex_buffers, r_bound_pipelines.graphics.vertex_buffers);
 
    command_buffer.draw_indirect(data.indirect_buffer, data.offset, data.draw_count, data.stride);
  }
 
  void free_data(Data &data)
  {
    vk_pipeline_data_free(data.graphics);
  }
};
}  // namespace blender::gpu::render_graph