missing bpy_compat.h for <3.0 py
[blender.git] / source / blender / python / intern / bpy_util.c
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19  *
20  * Contributor(s): Campbell Barton
21  *
22  * ***** END GPL LICENSE BLOCK *****
23  */
24
25 #include "bpy_compat.h"
26 #include "bpy_util.h"
27 #include "BLI_dynstr.h"
28 #include "MEM_guardedalloc.h"
29
30 PyObject *BPY_flag_to_list(struct BPY_flag_def *flagdef, int flag)
31 {
32         PyObject *list = PyList_New(0);
33         
34         PyObject *item;
35         BPY_flag_def *fd;
36
37         fd= flagdef;
38         while(fd->name) {
39                 if (fd->flag & flag) {
40                         item = PyUnicode_FromString(fd->name);
41                         PyList_Append(list, item);
42                         Py_DECREF(item);
43                 }
44                 fd++;
45         }
46         
47         return list;
48
49 }
50
51 static char *bpy_flag_error_str(BPY_flag_def *flagdef)
52 {
53         BPY_flag_def *fd= flagdef;
54         DynStr *dynstr= BLI_dynstr_new();
55         char *cstring;
56
57         BLI_dynstr_append(dynstr, "Error converting a sequence of strings into a flag.\n\tExpected only these strings...\n\t");
58
59         while(fd->name) {
60                 BLI_dynstr_appendf(dynstr, fd!=flagdef?", '%s'":"'%s'", fd->name);
61                 fd++;
62         }
63         
64         cstring = BLI_dynstr_get_cstring(dynstr);
65         BLI_dynstr_free(dynstr);
66         return cstring;
67 }
68
69 int BPY_flag_from_seq(BPY_flag_def *flagdef, PyObject *seq, int *flag)
70 {
71         int i, error_val= 0;
72         char *cstring;
73         PyObject *item;
74         BPY_flag_def *fd;
75
76         if (PySequence_Check(seq)) {
77                 i= PySequence_Length(seq);
78
79                 while(i--) {
80                         item = PySequence_ITEM(seq, i);
81                         cstring= _PyUnicode_AsString(item);
82                         if(cstring) {
83                                 fd= flagdef;
84                                 while(fd->name) {
85                                         if (strcmp(cstring, fd->name) == 0)
86                                                 (*flag) |= fd->flag;
87                                         fd++;
88                                 }
89                                 if (fd==NULL) { /* could not find a match */
90                                         error_val= 1;
91                                 }
92                         } else {
93                                 error_val= 1;
94                         }
95                         Py_DECREF(item);
96                 }
97         }
98         else {
99                 error_val= 1;
100         }
101
102         if (error_val) {
103                 char *buf = bpy_flag_error_str(flagdef);
104                 PyErr_SetString(PyExc_AttributeError, buf);
105                 MEM_freeN(buf);
106                 return -1; /* error value */
107         }
108
109         return 0; /* ok */
110 }
111
112 /* for debugging */
113 void PyObSpit(char *name, PyObject *var) {
114         fprintf(stderr, "<%s> : ", name);
115         if (var==NULL) {
116                 fprintf(stderr, "<NIL>");
117         }
118         else {
119                 PyObject_Print(var, stderr, 0);
120         }
121         fprintf(stderr, "\n");
122 }