vk_copy_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_command_buffer_wrapper.hh"
#include "../vk_render_graph_links.hh"
#include "../vk_resource_state_tracker.hh"
#include "vk_common.hh"
#include "vk_node_info.hh"
 
namespace blender::gpu::render_graph {
struct VKCopyImageData {
  VkImage src_image;
  VkImage dst_image;
  VkImageCopy region;
};
 
struct VKCopyImageCreateInfo {
  VKCopyImageData node_data;
  VkImageAspectFlags vk_image_aspect;
};
 
class VKCopyImageNode : public VKNodeInfo<VKNodeType::COPY_IMAGE,
                                          VKCopyImageCreateInfo,
                                          VKCopyImageData,
                                          VK_PIPELINE_STAGE_TRANSFER_BIT,
                                          VKResourceType::IMAGE> {
 public:
  template<typename Node, typename Storage>
  static void set_node_data(Node &node, Storage &storage, const CreateInfo &create_info)
  {
    node.storage_index = storage.copy_image.append_and_get_index(create_info.node_data);
  }

  void build_links(VKResourceStateTracker &resources,
                   VKRenderGraphNodeLinks &node_links,
                   const CreateInfo &create_info) override
  {
    ResourceWithStamp src_resource = resources.get_image(create_info.node_data.src_image);
    ResourceWithStamp dst_resource = resources.get_image_and_increase_stamp(
        create_info.node_data.dst_image);
    node_links.inputs.append({src_resource,
                              VK_ACCESS_TRANSFER_READ_BIT,
                              VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
                              create_info.vk_image_aspect});
    node_links.outputs.append({dst_resource,
                               VK_ACCESS_TRANSFER_WRITE_BIT,
                               VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
                               create_info.vk_image_aspect});
  }

  void build_commands(VKCommandBufferInterface &command_buffer,
                      Data &data,
                      VKBoundPipelines & /*r_bound_pipelines*/) override
  {
    command_buffer.copy_image(data.src_image,
                              VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL,
                              data.dst_image,
                              VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL,
                              1,
                              &data.region);
  }
};
}  // namespace blender::gpu::render_graph