Merge branch 'blender2.7'
[blender.git] / source / blender / python / generic / py_capi_utils.h
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  */
16
17 /** \file \ingroup pygen
18  */
19
20 #ifndef __PY_CAPI_UTILS_H__
21 #define __PY_CAPI_UTILS_H__
22
23 #include "BLI_sys_types.h"
24 #include "BLI_utildefines_variadic.h"
25
26 void      PyC_ObSpit(const char *name, PyObject *var);
27 void      PyC_ObSpitStr(char *result, size_t result_len, PyObject *var);
28 void      PyC_LineSpit(void);
29 void      PyC_StackSpit(void);
30 PyObject *PyC_ExceptionBuffer(void);
31 PyObject *PyC_ExceptionBuffer_Simple(void);
32 PyObject *PyC_Object_GetAttrStringArgs(PyObject *o, Py_ssize_t n, ...);
33 PyObject *PyC_FrozenSetFromStrings(const char **strings);
34
35 PyObject *PyC_Err_Format_Prefix(PyObject *exception_type_prefix, const char *format, ...);
36 PyObject *PyC_Err_SetString_Prefix(PyObject *exception_type_prefix, const char *str);
37
38 void      PyC_Err_PrintWithFunc(PyObject *py_func);
39
40 void    PyC_FileAndNum(const char **filename, int *lineno);
41 void    PyC_FileAndNum_Safe(const char **filename, int *lineno); /* checks python is running */
42 int             PyC_AsArray_FAST(
43         void *array, PyObject *value_fast, const Py_ssize_t length,
44         const PyTypeObject *type, const bool is_double, const char *error_prefix);
45 int             PyC_AsArray(
46         void *array, PyObject *value, const Py_ssize_t length,
47         const PyTypeObject *type, const bool is_double, const char *error_prefix);
48
49 PyObject       *PyC_Tuple_PackArray_F32(const float *array, uint len);
50 PyObject       *PyC_Tuple_PackArray_F64(const double *array, uint len);
51 PyObject       *PyC_Tuple_PackArray_I32(const int *array, uint len);
52 PyObject       *PyC_Tuple_PackArray_I32FromBool(const int *array, uint len);
53 PyObject       *PyC_Tuple_PackArray_Bool(const bool *array, uint len);
54
55 #define PyC_Tuple_Pack_F32(...) \
56         PyC_Tuple_PackArray_F32(((const float []){__VA_ARGS__}), VA_NARGS_COUNT(__VA_ARGS__))
57 #define PyC_Tuple_Pack_F64(...) \
58         PyC_Tuple_PackArray_F64(((const double []){__VA_ARGS__}), VA_NARGS_COUNT(__VA_ARGS__))
59 #define PyC_Tuple_Pack_I32(...) \
60         PyC_Tuple_PackArray_I32(((const int []){__VA_ARGS__}), VA_NARGS_COUNT(__VA_ARGS__))
61 #define PyC_Tuple_Pack_I32FromBool(...) \
62         PyC_Tuple_PackArray_I32FromBool(((const int []){__VA_ARGS__}), VA_NARGS_COUNT(__VA_ARGS__))
63 #define PyC_Tuple_Pack_Bool(...) \
64         PyC_Tuple_PackArray_Bool(((const bool []){__VA_ARGS__}), VA_NARGS_COUNT(__VA_ARGS__))
65
66 void            PyC_Tuple_Fill(PyObject *tuple, PyObject *value);
67 void            PyC_List_Fill(PyObject *list, PyObject *value);
68
69 /* follow http://www.python.org/dev/peps/pep-0383/ */
70 PyObject   *PyC_UnicodeFromByte(const char *str);
71 PyObject   *PyC_UnicodeFromByteAndSize(const char *str, Py_ssize_t size);
72 const char *PyC_UnicodeAsByte(PyObject *py_str, PyObject **coerce); /* coerce must be NULL */
73 const char *PyC_UnicodeAsByteAndSize(PyObject *py_str, Py_ssize_t *size, PyObject **coerce);
74
75 /* name namespace function for bpy */
76 PyObject *PyC_DefaultNameSpace(const char *filename);
77 void PyC_RunQuicky(const char *filepath, int n, ...);
78 bool PyC_NameSpace_ImportArray(PyObject *py_dict, const char *imports[]);
79
80 void PyC_MainModule_Backup(PyObject **main_mod);
81 void PyC_MainModule_Restore(PyObject *main_mod);
82
83 void PyC_SetHomePath(const char *py_path_bundle);
84
85 bool PyC_IsInterpreterActive(void);
86
87 void *PyC_RNA_AsPointer(PyObject *value, const char *type_name);
88
89 /* flag / set --- interchange */
90 typedef struct PyC_FlagSet {
91         int value;
92         const char *identifier;
93 } PyC_FlagSet;
94
95 char     *PyC_FlagSet_AsString(PyC_FlagSet *item);
96 int       PyC_FlagSet_ValueFromID_int(PyC_FlagSet *item, const char *identifier, int *r_value);
97 int       PyC_FlagSet_ValueFromID(PyC_FlagSet *item, const char *identifier, int *r_value, const char *error_prefix);
98 int       PyC_FlagSet_ToBitfield(PyC_FlagSet *items, PyObject *value, int *r_value, const char *error_prefix);
99 PyObject *PyC_FlagSet_FromBitfield(PyC_FlagSet *items, int flag);
100
101 bool PyC_RunString_AsNumber(const char **imports, const char *expr, const char *filename, double *r_value);
102 bool PyC_RunString_AsIntPtr(const char **imports, const char *expr, const char *filename, intptr_t *r_value);
103 bool PyC_RunString_AsString(const char **imports, const char *expr, const char *filename, char **r_value);
104
105 int PyC_ParseBool(PyObject *o, void *p);
106
107 int PyC_CheckArgs_DeepCopy(PyObject *args);
108
109 /* Integer parsing (with overflow checks), -1 on error. */
110 int     PyC_Long_AsBool(PyObject *value);
111 int8_t  PyC_Long_AsI8(PyObject *value);
112 int16_t PyC_Long_AsI16(PyObject *value);
113 #if 0 /* inline */
114 int32_t PyC_Long_AsI32(PyObject *value);
115 int64_t PyC_Long_AsI64(PyObject *value);
116 #endif
117
118 uint8_t  PyC_Long_AsU8(PyObject *value);
119 uint16_t PyC_Long_AsU16(PyObject *value);
120 uint32_t PyC_Long_AsU32(PyObject *value);
121 #if 0 /* inline */
122 uint64_t PyC_Long_AsU64(PyObject *value);
123 #endif
124
125 /* inline so type signatures match as expected */
126 Py_LOCAL_INLINE(int32_t)  PyC_Long_AsI32(PyObject *value) { return (int32_t)_PyLong_AsInt(value); }
127 Py_LOCAL_INLINE(int64_t)  PyC_Long_AsI64(PyObject *value) { return (int64_t)PyLong_AsLongLong(value); }
128 Py_LOCAL_INLINE(uint64_t) PyC_Long_AsU64(PyObject *value) { return (uint64_t)PyLong_AsUnsignedLongLong(value); }
129
130 /* utils for format string in `struct` module style syntax */
131 char PyC_StructFmt_type_from_str(const char *typestr);
132 bool PyC_StructFmt_type_is_float_any(char format);
133 bool PyC_StructFmt_type_is_int_any(char format);
134 bool PyC_StructFmt_type_is_byte(char format);
135 bool PyC_StructFmt_type_is_bool(char format);
136
137 #endif  /* __PY_CAPI_UTILS_H__ */