Clément Foucault
d7df0dcccb
This uses a StorageBuf as the source of indirect dispatch argument. The user needs to make sure the parameters are in the right order. There is no support for argument offset for the moment as there is no need for it. But this might be added in the future. Note that the indirect buffer is synchronized at the backend level. This is done for practical reasons and because this feature is almost always used for GPU driven pipeline.
150 lines
3.6 KiB
C++
150 lines
3.6 KiB
C++
/* SPDX-License-Identifier: GPL-2.0-or-later
|
|
* Copyright 2022 Blender Foundation. All rights reserved. */
|
|
|
|
/** \file
|
|
* \ingroup gpu
|
|
*/
|
|
|
|
#include "BKE_global.h"
|
|
|
|
#include "BLI_string.h"
|
|
|
|
#include "gpu_backend.hh"
|
|
#include "gpu_context_private.hh"
|
|
|
|
#include "gl_backend.hh"
|
|
#include "gl_debug.hh"
|
|
#include "gl_storage_buffer.hh"
|
|
#include "gl_vertex_buffer.hh"
|
|
|
|
namespace blender::gpu {
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Creation & Deletion
|
|
* \{ */
|
|
|
|
GLStorageBuf::GLStorageBuf(size_t size, GPUUsageType usage, const char *name)
|
|
: StorageBuf(size, name)
|
|
{
|
|
usage_ = usage;
|
|
/* Do not create ubo GL buffer here to allow allocation from any thread. */
|
|
BLI_assert(size <= GLContext::max_ssbo_size);
|
|
}
|
|
|
|
GLStorageBuf::~GLStorageBuf()
|
|
{
|
|
GLContext::buf_free(ssbo_id_);
|
|
}
|
|
|
|
/** \} */
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Data upload / update
|
|
* \{ */
|
|
|
|
void GLStorageBuf::init()
|
|
{
|
|
BLI_assert(GLContext::get());
|
|
|
|
glGenBuffers(1, &ssbo_id_);
|
|
glBindBuffer(GL_SHADER_STORAGE_BUFFER, ssbo_id_);
|
|
glBufferData(GL_SHADER_STORAGE_BUFFER, size_in_bytes_, nullptr, to_gl(this->usage_));
|
|
|
|
debug::object_label(GL_SHADER_STORAGE_BUFFER, ssbo_id_, name_);
|
|
}
|
|
|
|
void GLStorageBuf::update(const void *data)
|
|
{
|
|
if (ssbo_id_ == 0) {
|
|
this->init();
|
|
}
|
|
glBindBuffer(GL_SHADER_STORAGE_BUFFER, ssbo_id_);
|
|
glBufferSubData(GL_SHADER_STORAGE_BUFFER, 0, size_in_bytes_, data);
|
|
glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0);
|
|
}
|
|
|
|
/** \} */
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Usage
|
|
* \{ */
|
|
|
|
void GLStorageBuf::bind(int slot)
|
|
{
|
|
if (slot >= GLContext::max_ssbo_binds) {
|
|
fprintf(
|
|
stderr,
|
|
"Error: Trying to bind \"%s\" ssbo to slot %d which is above the reported limit of %d.",
|
|
name_,
|
|
slot,
|
|
GLContext::max_ssbo_binds);
|
|
return;
|
|
}
|
|
|
|
if (ssbo_id_ == 0) {
|
|
this->init();
|
|
}
|
|
|
|
if (data_ != nullptr) {
|
|
this->update(data_);
|
|
MEM_SAFE_FREE(data_);
|
|
}
|
|
|
|
slot_ = slot;
|
|
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, slot_, ssbo_id_);
|
|
|
|
#ifdef DEBUG
|
|
BLI_assert(slot < 16);
|
|
/* TODO */
|
|
// GLContext::get()->bound_ssbo_slots |= 1 << slot;
|
|
#endif
|
|
}
|
|
|
|
void GLStorageBuf::bind_as(GLenum target)
|
|
{
|
|
BLI_assert_msg(ssbo_id_ != 0,
|
|
"Trying to use storage buf as indirect buffer but buffer was never filled.");
|
|
glBindBuffer(target, ssbo_id_);
|
|
}
|
|
|
|
void GLStorageBuf::unbind()
|
|
{
|
|
#ifdef DEBUG
|
|
/* NOTE: This only unbinds the last bound slot. */
|
|
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, slot_, 0);
|
|
/* Hope that the context did not change. */
|
|
/* TODO */
|
|
// GLContext::get()->bound_ssbo_slots &= ~(1 << slot_);
|
|
#endif
|
|
slot_ = 0;
|
|
}
|
|
|
|
void GLStorageBuf::clear(eGPUTextureFormat internal_format, eGPUDataFormat data_format, void *data)
|
|
{
|
|
if (ssbo_id_ == 0) {
|
|
this->init();
|
|
}
|
|
|
|
if (GLContext::direct_state_access_support) {
|
|
glClearNamedBufferData(ssbo_id_,
|
|
to_gl_internal_format(internal_format),
|
|
to_gl_data_format(internal_format),
|
|
to_gl(data_format),
|
|
data);
|
|
}
|
|
else {
|
|
/* WATCH(@fclem): This should be ok since we only use clear outside of drawing functions. */
|
|
glBindBuffer(GL_SHADER_STORAGE_BUFFER, ssbo_id_);
|
|
glClearBufferData(GL_SHADER_STORAGE_BUFFER,
|
|
to_gl_internal_format(internal_format),
|
|
to_gl_data_format(internal_format),
|
|
to_gl(data_format),
|
|
data);
|
|
glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0);
|
|
}
|
|
}
|
|
|
|
/** \} */
|
|
|
|
} // namespace blender::gpu
|