vk_uniform_buffer.hh

Go to the documentation of this file.

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

 
#pragma once
 
#include "BLI_utility_mixins.hh"
 
#include "gpu_uniform_buffer_private.hh"
 
#include "vk_buffer.hh"
 
namespace blender::gpu {
 
class VKUniformBuffer : public UniformBuf, NonCopyable {
  VKBuffer buffer_;

  bool data_uploaded_ = false;
 
 public:
  VKUniformBuffer(size_t size, const char *name) : UniformBuf(size, name) {}
 
  void update(const void *data) override;
  void clear_to_zero() override;
  void bind(int slot) override;
  void bind_as_ssbo(int slot) override;

  void unbind() override;
 
  VkBuffer vk_handle() const
  {
    return buffer_.vk_handle();
  }
  inline VkDeviceAddress device_address_get() const
  {
    return buffer_.device_address_get();
  }
 
  size_t size_in_bytes() const
  {
    return size_in_bytes_;
  }
 
  void ensure_updated();

  void reset_data_uploaded()
  {
    data_uploaded_ = false;
  }
 
 private:
  void allocate();
};
 
BLI_INLINE UniformBuf *wrap(VKUniformBuffer *uniform_buffer)
{
  return static_cast<UniformBuf *>(uniform_buffer);
}
 
}  // namespace blender::gpu