Merge branch 'blender2.7'
[blender.git] / source / blender / python / gpu / gpu_py_vertex_format.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * ***** END GPL LICENSE BLOCK *****
19  */
20
21 /** \file blender/python/gpu/gpu_py_vertex_format.c
22  *  \ingroup bpygpu
23  *
24  * - Use ``bpygpu_`` for local API.
25  * - Use ``BPyGPU`` for public API.
26  */
27
28 #include <Python.h>
29
30 #include "BLI_math.h"
31
32 #include "MEM_guardedalloc.h"
33
34 #include "../generic/py_capi_utils.h"
35 #include "../generic/python_utildefines.h"
36
37 #include "gpu_py_vertex_format.h" /* own include */
38
39 #ifdef __BIG_ENDIAN__
40    /* big endian */
41 #  define MAKE_ID2(c, d)  ((c) << 8 | (d))
42 #  define MAKE_ID3(a, b, c) ( (int)(a) << 24 | (int)(b) << 16 | (c) << 8 )
43 #  define MAKE_ID4(a, b, c, d) ( (int)(a) << 24 | (int)(b) << 16 | (c) << 8 | (d) )
44 #else
45    /* little endian  */
46 #  define MAKE_ID2(c, d)  ((d) << 8 | (c))
47 #  define MAKE_ID3(a, b, c) ( (int)(c) << 16 | (b) << 8 | (a) )
48 #  define MAKE_ID4(a, b, c, d) ( (int)(d) << 24 | (int)(c) << 16 | (b) << 8 | (a) )
49 #endif
50
51 /* -------------------------------------------------------------------- */
52
53 /** \name Enum Conversion
54  *
55  * Use with PyArg_ParseTuple's "O&" formatting.
56  * \{ */
57
58 static int bpygpu_parse_component_type(const char *str, int length)
59 {
60         if (length == 2) {
61                 switch (*((ushort *)str)) {
62                         case MAKE_ID2('I', '8'): return GPU_COMP_I8;
63                         case MAKE_ID2('U', '8'): return GPU_COMP_U8;
64                         default: break;
65                 }
66         }
67         else if (length == 3) {
68                 switch (*((uint *)str)) {
69                         case MAKE_ID3('I', '1', '6'): return GPU_COMP_I16;
70                         case MAKE_ID3('U', '1', '6'): return GPU_COMP_U16;
71                         case MAKE_ID3('I', '3', '2'): return GPU_COMP_I32;
72                         case MAKE_ID3('U', '3', '2'): return GPU_COMP_U32;
73                         case MAKE_ID3('F', '3', '2'): return GPU_COMP_F32;
74                         case MAKE_ID3('I', '1', '0'): return GPU_COMP_I10;
75                         default: break;
76                 }
77         }
78         return -1;
79 }
80
81 static int bpygpu_parse_fetch_mode(const char *str, int length)
82 {
83 #define MATCH_ID(id) \
84         if (length == strlen(STRINGIFY(id))) { \
85                 if (STREQ(str, STRINGIFY(id))) { \
86                         return GPU_FETCH_##id; \
87                 } \
88         } ((void)0)
89
90         MATCH_ID(FLOAT);
91         MATCH_ID(INT);
92         MATCH_ID(INT_TO_FLOAT_UNIT);
93         MATCH_ID(INT_TO_FLOAT);
94 #undef MATCH_ID
95
96         return -1;
97 }
98
99 static int bpygpu_ParseVertCompType(PyObject *o, void *p)
100 {
101         Py_ssize_t length;
102         const char *str = _PyUnicode_AsStringAndSize(o, &length);
103
104         if (str == NULL) {
105                 PyErr_Format(PyExc_ValueError,
106                              "expected a string, got %s",
107                              Py_TYPE(o)->tp_name);
108                 return 0;
109         }
110
111         int comp_type = bpygpu_parse_component_type(str, length);
112         if (comp_type == -1) {
113                 PyErr_Format(PyExc_ValueError,
114                              "unkown component type: '%s",
115                              str);
116                 return 0;
117         }
118
119         *((GPUVertCompType *)p) = comp_type;
120         return 1;
121 }
122
123 static int bpygpu_ParseVertFetchMode(PyObject *o, void *p)
124 {
125         Py_ssize_t length;
126         const char *str = _PyUnicode_AsStringAndSize(o, &length);
127
128         if (str == NULL) {
129                 PyErr_Format(PyExc_ValueError,
130                              "expected a string, got %s",
131                              Py_TYPE(o)->tp_name);
132                 return 0;
133         }
134
135         int fetch_mode = bpygpu_parse_fetch_mode(str, length);
136         if (fetch_mode == -1) {
137                 PyErr_Format(PyExc_ValueError,
138                              "unknown type literal: '%s'",
139                              str);
140                 return 0;
141         }
142
143         (*(GPUVertFetchMode *)p) = fetch_mode;
144         return 1;
145 }
146
147 /** \} */
148
149
150 /* -------------------------------------------------------------------- */
151
152 /** \name VertFormat Type
153  * \{ */
154
155 static PyObject *bpygpu_VertFormat_new(PyTypeObject *UNUSED(type), PyObject *args, PyObject *kwds)
156 {
157         if (PyTuple_GET_SIZE(args) || (kwds && PyDict_Size(kwds))) {
158                 PyErr_SetString(PyExc_ValueError, "This function takes no arguments");
159                 return NULL;
160         }
161         return BPyGPUVertFormat_CreatePyObject(NULL);
162 }
163
164 PyDoc_STRVAR(bpygpu_VertFormat_attr_add_doc,
165 ".. method:: attr_add(id, comp_type, len, fetch_mode)\n"
166 "\n"
167 "   Add a new attribute to the format.\n"
168 "\n"
169 "   :param id: Name the attribute. Often `position`, `normal`, ...\n"
170 "   :type id: str\n"
171 "   :param comp_type: The data type that will be used store the value in memory.\n"
172 "      Possible values are `I8`, `U8`, `I16`, `U16`, `I32`, `U32`, `F32` and `I10`.\n"
173 "   :type comp_type: `str`\n"
174 "   :param len: How many individual values the attribute consists of (e.g. 2 for uv coordinates).\n"
175 "   :type len: int\n"
176 "   :param fetch_mode: How values from memory will be converted when used in the shader.\n"
177 "      This is mainly useful for memory optimizations when you want to store values with reduced precision.\n"
178 "      E.g. you can store a float in only 1 byte but it will be converted to a normal 4 byte float when used.\n"
179 "      Possible values are `FLOAT`, `INT`, `INT_TO_FLOAT_UNIT` and `INT_TO_FLOAT`.\n"
180 "   :type fetch_mode: `str`\n"
181 );
182 static PyObject *bpygpu_VertFormat_attr_add(BPyGPUVertFormat *self, PyObject *args, PyObject *kwds)
183 {
184         struct {
185                 const char *id;
186                 GPUVertCompType comp_type;
187                 uint len;
188                 GPUVertFetchMode fetch_mode;
189         } params;
190
191         if (self->fmt.attr_len == GPU_VERT_ATTR_MAX_LEN) {
192                 PyErr_SetString(PyExc_ValueError, "Maxumum attr reached " STRINGIFY(GPU_VERT_ATTR_MAX_LEN));
193                 return NULL;
194         }
195
196         static const char *_keywords[] = {"id", "comp_type", "len", "fetch_mode", NULL};
197         static _PyArg_Parser _parser = {"$sO&IO&:attr_add", _keywords, 0};
198         if (!_PyArg_ParseTupleAndKeywordsFast(
199                 args, kwds, &_parser,
200                 &params.id,
201                 bpygpu_ParseVertCompType, &params.comp_type,
202                 &params.len,
203                 bpygpu_ParseVertFetchMode, &params.fetch_mode))
204         {
205                 return NULL;
206         }
207
208         uint attr_id = GPU_vertformat_attr_add(&self->fmt, params.id, params.comp_type, params.len, params.fetch_mode);
209         return PyLong_FromLong(attr_id);
210 }
211
212 static struct PyMethodDef bpygpu_VertFormat_methods[] = {
213         {"attr_add", (PyCFunction)bpygpu_VertFormat_attr_add,
214          METH_VARARGS | METH_KEYWORDS, bpygpu_VertFormat_attr_add_doc},
215         {NULL, NULL, 0, NULL}
216 };
217
218
219 static void bpygpu_VertFormat_dealloc(BPyGPUVertFormat *self)
220 {
221         Py_TYPE(self)->tp_free(self);
222 }
223
224 PyDoc_STRVAR(bpygpu_VertFormat_doc,
225 ".. class:: GPUVertFormat()\n"
226 "\n"
227 "   This object contains information about the structure of a vertex buffer.\n"
228 );
229 PyTypeObject BPyGPUVertFormat_Type = {
230         PyVarObject_HEAD_INIT(NULL, 0)
231         .tp_name = "GPUVertFormat",
232         .tp_basicsize = sizeof(BPyGPUVertFormat),
233         .tp_dealloc = (destructor)bpygpu_VertFormat_dealloc,
234         .tp_flags = Py_TPFLAGS_DEFAULT,
235         .tp_doc = bpygpu_VertFormat_doc,
236         .tp_methods = bpygpu_VertFormat_methods,
237         .tp_new = bpygpu_VertFormat_new,
238 };
239
240 /** \} */
241
242
243 /* -------------------------------------------------------------------- */
244
245 /** \name Public API
246  * \{ */
247
248 PyObject *BPyGPUVertFormat_CreatePyObject(GPUVertFormat *fmt)
249 {
250         BPyGPUVertFormat *self;
251
252         self = PyObject_New(BPyGPUVertFormat, &BPyGPUVertFormat_Type);
253         if (fmt) {
254                 self->fmt = *fmt;
255         }
256         else {
257                 memset(&self->fmt, 0, sizeof(self->fmt));
258         }
259
260         return (PyObject *)self;
261 }
262
263 /** \} */