vk_staging_buffer.hh

Go to the documentation of this file.

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

 
#pragma once
 
#include "vk_buffer.hh"
#include "vk_common.hh"
 
namespace blender::gpu {

class VKStagingBuffer {
 public:
  enum class Direction {
    HostToDevice,
    DeviceToHost,
  };
 
 private:
  const VKBuffer &device_buffer_;

  VKBuffer host_buffer_;
 
  VkDeviceSize device_buffer_offset_;
  VkDeviceSize region_size_;
 
 public:
  VKStagingBuffer(const VKBuffer &device_buffer,
                  Direction direction,
                  VkDeviceSize device_buffer_offset = 0,
                  VkDeviceSize region_size = UINT64_MAX);

  void copy_to_device(VKContext &context);

  void copy_from_device(VKContext &context);

  VKBuffer &host_buffer_get()
  {
    return host_buffer_;
  }

  void free();
 
  VkDeviceSize size_in_bytes_get() const
  {
    return region_size_;
  }
};
}  // namespace blender::gpu