vk_clear_attachments_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 VKClearAttachmentsData {
  uint32_t attachment_count;
  VkClearAttachment attachments[8];
  VkClearRect vk_clear_rect;
};
 
class VKClearAttachmentsNode : public VKNodeInfo<VKNodeType::CLEAR_ATTACHMENTS,
                                                 VKClearAttachmentsData,
                                                 VKClearAttachmentsData,
                                                 VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT |
                                                     VK_PIPELINE_STAGE_EARLY_FRAGMENT_TESTS_BIT |
                                                     VK_PIPELINE_STAGE_LATE_FRAGMENT_TESTS_BIT,
                                                 VKResourceType::IMAGE> {
 public:
  template<typename Node, typename Storage>
  void set_node_data(Node &node, Storage &storage, const CreateInfo &create_info)
  {
    node.storage_index = storage.clear_attachments.append_and_get_index(create_info);
  }

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

  void build_commands(VKCommandBufferInterface &command_buffer,
                      Data &data,
                      VKBoundPipelines & /*r_bound_pipelines*/) override
  {
    command_buffer.clear_attachments(
        data.attachment_count, data.attachments, 1, &data.vk_clear_rect);
  }
};
}  // namespace blender::gpu::render_graph