319ff28b7b
Motivation: When discussing with @Jeroen-Bakker and @aras_p about how to approach sorting when rendering Gaussian splats in Blender, we realised that compute shaders could help there (and have many other use cases), and that also due to Blender 4.0 being on OpenGL >= 4.3, we can now rely on compute shaders existing. This PR is an initial pass for that functionality. It comes with a Python example, which runs a compute shader and saves the output to a texture, which is then rendered in the viewport. There is no exposed support for storage buffers yet, but I expect I'll be able to work on them soon if this is accepted. The newly added parts are: 1. `gpu.compute.dispatch()` 2. a way set the compute source to `GPUShaderCreateInfo`: `shader_info.compute_source()` 3. a way to set the image store for `GPUShaderCreateInfo`: `shader_info.image()` 4. a way to set the `local_group_size` for `GPUShaderCreateInfo`: `shader_info.local_group_size(x,y,z)` 5. a way to get `max_work_group_size` from capabilities: `gpu.capabilities.max_work_group_size_get(index)` 6. a way to get `max_work_group_count` from capabilities: `gpu.capabilities.max_work_group_count_get(index)` Pull Request: https://projects.blender.org/blender/blender/pulls/114238
88 lines
2.7 KiB
C++
88 lines
2.7 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
/** \file
|
|
* \ingroup bpygpu
|
|
*
|
|
* Experimental Python API, not considered public yet (called '_gpu'),
|
|
* we may re-expose as public later.
|
|
*
|
|
* - Use `bpygpu_` for local API.
|
|
* - Use `BPyGPU` for public API.
|
|
*/
|
|
|
|
#include <Python.h>
|
|
|
|
#include "BLI_utildefines.h"
|
|
|
|
#include "gpu_py_capabilities.h"
|
|
#include "gpu_py_matrix.h"
|
|
#include "gpu_py_platform.h"
|
|
#include "gpu_py_select.h"
|
|
#include "gpu_py_state.h"
|
|
#include "gpu_py_types.h"
|
|
#include "gpu_py_compute.h"
|
|
|
|
#include "gpu_py.h"
|
|
#include "gpu_py_api.h" /* Own include. */
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name GPU Module
|
|
* \{ */
|
|
|
|
PyDoc_STRVAR(pygpu_doc,
|
|
"This module provides Python wrappers for the GPU implementation in Blender.\n"
|
|
"Some higher level functions can be found in the `gpu_extras` module.");
|
|
static PyModuleDef pygpu_module_def = {
|
|
/*m_base*/ PyModuleDef_HEAD_INIT,
|
|
/*m_name*/ "gpu",
|
|
/*m_doc*/ pygpu_doc,
|
|
/*m_size*/ 0,
|
|
/*m_methods*/ nullptr,
|
|
/*m_slots*/ nullptr,
|
|
/*m_traverse*/ nullptr,
|
|
/*m_clear*/ nullptr,
|
|
/*m_free*/ nullptr,
|
|
};
|
|
|
|
PyObject *BPyInit_gpu()
|
|
{
|
|
PyObject *sys_modules = PyImport_GetModuleDict();
|
|
PyObject *submodule;
|
|
PyObject *mod;
|
|
|
|
mod = bpygpu_create_module(&pygpu_module_def);
|
|
|
|
PyModule_AddObject(mod, "types", (submodule = bpygpu_types_init()));
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
|
|
|
PyModule_AddObject(mod, "capabilities", (submodule = bpygpu_capabilities_init()));
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
|
|
|
PyModule_AddObject(mod, "matrix", (submodule = bpygpu_matrix_init()));
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
|
|
|
PyModule_AddObject(mod, "platform", (submodule = bpygpu_platform_init()));
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
|
|
|
PyModule_AddObject(mod, "select", (submodule = bpygpu_select_init()));
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
|
|
|
PyModule_AddObject(mod, "shader", (submodule = bpygpu_shader_init()));
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
|
|
|
PyModule_AddObject(mod, "state", (submodule = bpygpu_state_init()));
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
|
|
|
PyModule_AddObject(mod, "texture", (submodule = bpygpu_texture_init()));
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
|
|
|
PyModule_AddObject(mod, "compute", (submodule = bpygpu_compute_init()));
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
|
|
|
return mod;
|
|
}
|
|
|
|
/** \} */
|