Germano Cavalcante
4430e8a008
This commit extends the gpu python API with:
```
gpu.types.Buffer #"__init__", "to_list"
gpu.types.GPUTexture #"__init__", "clear", "read", "format"
gpu.types.GPUFrameBuffer #"__init__", "bind", "clear", "is_bound", "viewport", ("__enter__", "__exit__" with "GPUFrameBufferStackContext")
gpu.types.GPUUniformBuf #"__init__", "update"
gpu.state #"blend_set", "blend_get", "depth_test_set", "depth_test_get", "depth_mask_set", "depth_mask_get", "viewport_set", "viewport_get", "line_width_set", "line_width_get", "point_size_set", "color_mask_set", "face_culling_set", "front_facing_set", "program_point_size_set"
```
Add these methods to existing objects:
```
gpu.types.GPUShader #"uniform_sample", "uniform_buffer"
```
Maniphest Tasks: T80481
Differential Revision: https://developer.blender.org/D8826
145 lines
3.9 KiB
C
145 lines
3.9 KiB
C
/*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
/** \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 "../generic/python_utildefines.h"
|
|
|
|
#include "GPU_init_exit.h"
|
|
#include "GPU_primitive.h"
|
|
|
|
#include "gpu_py_matrix.h"
|
|
#include "gpu_py_select.h"
|
|
#include "gpu_py_state.h"
|
|
#include "gpu_py_types.h"
|
|
|
|
#include "gpu_py_api.h" /* own include */
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Utils to invalidate functions
|
|
* \{ */
|
|
|
|
bool bpygpu_is_init_or_error(void)
|
|
{
|
|
if (!GPU_is_init()) {
|
|
PyErr_SetString(PyExc_SystemError,
|
|
"GPU functions for drawing are not available in background mode");
|
|
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
/** \} */
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Primitive Type Utils
|
|
* \{ */
|
|
|
|
int bpygpu_ParsePrimType(PyObject *o, void *p)
|
|
{
|
|
Py_ssize_t mode_id_len;
|
|
const char *mode_id = PyUnicode_AsUTF8AndSize(o, &mode_id_len);
|
|
if (mode_id == NULL) {
|
|
PyErr_Format(PyExc_ValueError, "expected a string, got %s", Py_TYPE(o)->tp_name);
|
|
return 0;
|
|
}
|
|
#define MATCH_ID(id) \
|
|
if (mode_id_len == strlen(STRINGIFY(id))) { \
|
|
if (STREQ(mode_id, STRINGIFY(id))) { \
|
|
mode = GPU_PRIM_##id; \
|
|
goto success; \
|
|
} \
|
|
} \
|
|
((void)0)
|
|
|
|
GPUPrimType mode;
|
|
MATCH_ID(POINTS);
|
|
MATCH_ID(LINES);
|
|
MATCH_ID(TRIS);
|
|
MATCH_ID(LINE_STRIP);
|
|
MATCH_ID(LINE_LOOP);
|
|
MATCH_ID(TRI_STRIP);
|
|
MATCH_ID(TRI_FAN);
|
|
MATCH_ID(LINES_ADJ);
|
|
MATCH_ID(TRIS_ADJ);
|
|
MATCH_ID(LINE_STRIP_ADJ);
|
|
|
|
#undef MATCH_ID
|
|
PyErr_Format(PyExc_ValueError, "unknown type literal: '%s'", mode_id);
|
|
return 0;
|
|
|
|
success:
|
|
(*(GPUPrimType *)p) = mode;
|
|
return 1;
|
|
}
|
|
|
|
/** \} */
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \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 struct PyModuleDef pygpu_module_def = {
|
|
PyModuleDef_HEAD_INIT,
|
|
.m_name = "gpu",
|
|
.m_doc = pygpu_doc,
|
|
};
|
|
|
|
PyObject *BPyInit_gpu(void)
|
|
{
|
|
PyObject *sys_modules = PyImport_GetModuleDict();
|
|
PyObject *submodule;
|
|
PyObject *mod;
|
|
|
|
mod = PyModule_Create(&pygpu_module_def);
|
|
|
|
PyModule_AddObject(mod, "types", (submodule = bpygpu_types_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, "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);
|
|
|
|
return mod;
|
|
}
|
|
|
|
/** \} */
|