vk_clear_color_image_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 VKClearColorImageData {
  VkImage vk_image;
  VkClearColorValue vk_clear_color_value;
  VkImageSubresourceRange vk_image_subresource_range;
};
 
class VKClearColorImageNode : public VKNodeInfo<VKNodeType::CLEAR_COLOR_IMAGE,
                                                VKClearColorImageData,
                                                VKClearColorImageData,
                                                VK_PIPELINE_STAGE_TRANSFER_BIT,
                                                VKResourceType::IMAGE> {
 public:
  template<typename Node, typename Storage>
  void set_node_data(Node &node, Storage & /* storage */, const CreateInfo &create_info)
  {
    node.clear_color_image = create_info;
  }

  void build_links(VKResourceStateTracker &resources,
                   VKRenderGraphNodeLinks &node_links,
                   const CreateInfo &create_info) override
  {
    ResourceWithStamp resource = resources.get_image_and_increase_stamp(create_info.vk_image);
    node_links.outputs.append({resource,
                               VK_ACCESS_TRANSFER_WRITE_BIT,
                               VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
                               VK_IMAGE_ASPECT_COLOR_BIT});
  }

  void build_commands(VKCommandBufferInterface &command_buffer,
                      Data &data,
                      VKBoundPipelines & /*r_bound_pipelines*/) override
  {
    command_buffer.clear_color_image(data.vk_image,
                                     VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
                                     &data.vk_clear_color_value,
                                     1,
                                     &data.vk_image_subresource_range);
  }
};
}  // namespace blender::gpu::render_graph