Code cleanup: use 'const' for arrays (python)
[blender.git] / source / blender / python / intern / bpy_util.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  * Contributor(s): Campbell Barton
19  *
20  * ***** END GPL LICENSE BLOCK *****
21  */
22
23 /** \file blender/python/intern/bpy_util.c
24  *  \ingroup pythonintern
25  *
26  * This file contains blender/python utility functions for the api's internal
27  * use (unrelated to 'bpy.utils')
28  */
29
30 #include <Python.h>
31
32 #include "BLI_utildefines.h"
33 #include "BLI_dynstr.h"
34
35 #include "bpy_util.h"
36
37 #include "MEM_guardedalloc.h"
38
39 #include "BKE_report.h"
40 #include "BKE_context.h"
41
42 #include "BLF_translation.h"
43
44 #include "../generic/py_capi_utils.h"
45
46 static bContext *__py_context = NULL;
47 bContext   *BPy_GetContext(void) { return __py_context; }
48 void        BPy_SetContext(bContext *C) { __py_context = C; }
49
50 char *BPy_enum_as_string(EnumPropertyItem *item)
51 {
52         DynStr *dynstr = BLI_dynstr_new();
53         EnumPropertyItem *e;
54         char *cstring;
55
56         for (e = item; item->identifier; item++) {
57                 if (item->identifier[0])
58                         BLI_dynstr_appendf(dynstr, (e == item) ? "'%s'" : ", '%s'", item->identifier);
59         }
60
61         cstring = BLI_dynstr_get_cstring(dynstr);
62         BLI_dynstr_free(dynstr);
63         return cstring;
64 }
65
66 short BPy_reports_to_error(ReportList *reports, PyObject *exception, const bool clear)
67 {
68         char *report_str;
69
70         report_str = BKE_reports_string(reports, RPT_ERROR);
71
72         if (clear == true) {
73                 BKE_reports_clear(reports);
74         }
75
76         if (report_str) {
77                 PyErr_SetString(exception, report_str);
78                 MEM_freeN(report_str);
79         }
80
81         return (report_str == NULL) ? 0 : -1;
82 }
83
84
85 short BPy_errors_to_report(ReportList *reports)
86 {
87         PyObject *pystring;
88         PyObject *pystring_format = NULL;  /* workaround, see below */
89         const char *cstring;
90
91         const char *filename;
92         int lineno;
93
94         if (!PyErr_Occurred())
95                 return 1;
96         
97         /* less hassle if we allow NULL */
98         if (reports == NULL) {
99                 PyErr_Print();
100                 PyErr_Clear();
101                 return 1;
102         }
103         
104         pystring = PyC_ExceptionBuffer();
105         
106         if (pystring == NULL) {
107                 BKE_report(reports, RPT_ERROR, "Unknown py-exception, could not convert");
108                 return 0;
109         }
110         
111         PyC_FileAndNum(&filename, &lineno);
112         if (filename == NULL)
113                 filename = "<unknown location>";
114         
115         cstring = _PyUnicode_AsString(pystring);
116
117 #if 0 /* ARG!. workaround for a bug in blenders use of vsnprintf */
118         BKE_reportf(reports, RPT_ERROR, "%s\nlocation: %s:%d\n", cstring, filename, lineno);
119 #else
120         pystring_format = PyUnicode_FromFormat(TIP_("%s\nlocation: %s:%d\n"), cstring, filename, lineno);
121         cstring = _PyUnicode_AsString(pystring_format);
122         BKE_report(reports, RPT_ERROR, cstring);
123 #endif
124
125         /* not exactly needed. just for testing */
126         fprintf(stderr, TIP_("%s\nlocation: %s:%d\n"), cstring, filename, lineno);
127
128         Py_DECREF(pystring);
129         Py_DECREF(pystring_format);  /* workaround */
130         return 1;
131 }