944 lines
24 KiB
C
944 lines
24 KiB
C
/*
|
|
* ***** BEGIN GPL LICENSE BLOCK *****
|
|
*
|
|
* 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.
|
|
*
|
|
* Contributor(s): Michel Selten, Willian P. Germano, Stephen Swaney,
|
|
* Chris Keith, Chris Want, Ken Hughes, Campbell Barton
|
|
*
|
|
* ***** END GPL LICENSE BLOCK *****
|
|
*/
|
|
|
|
/** \file blender/python/intern/bpy_interface.c
|
|
* \ingroup pythonintern
|
|
*
|
|
* This file deals with embedding the python interpreter within blender,
|
|
* starting and stopping python and exposing blender/python modules so they can
|
|
* be accesses from scripts.
|
|
*/
|
|
|
|
|
|
/* grr, python redefines */
|
|
#ifdef _POSIX_C_SOURCE
|
|
# undef _POSIX_C_SOURCE
|
|
#endif
|
|
|
|
#include <Python.h>
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
#include "BLI_utildefines.h"
|
|
#include "BLI_path_util.h"
|
|
#include "BLI_fileops.h"
|
|
#include "BLI_listbase.h"
|
|
#include "BLI_math_base.h"
|
|
#include "BLI_string.h"
|
|
#include "BLI_string_utf8.h"
|
|
#include "BLI_threads.h"
|
|
|
|
#include "RNA_types.h"
|
|
|
|
#include "bpy.h"
|
|
#include "gpu.h"
|
|
#include "bpy_rna.h"
|
|
#include "bpy_path.h"
|
|
#include "bpy_util.h"
|
|
#include "bpy_traceback.h"
|
|
#include "bpy_intern_string.h"
|
|
|
|
#include "DNA_space_types.h"
|
|
#include "DNA_text_types.h"
|
|
|
|
#include "BKE_context.h"
|
|
#include "BKE_text.h"
|
|
#include "BKE_main.h"
|
|
#include "BKE_global.h" /* only for script checking */
|
|
|
|
#include "CCL_api.h"
|
|
|
|
#include "BPY_extern.h"
|
|
|
|
#include "../generic/bpy_internal_import.h" /* our own imports */
|
|
#include "../generic/py_capi_utils.h"
|
|
|
|
/* inittab initialization functions */
|
|
#include "../generic/bgl.h"
|
|
#include "../generic/blf_py_api.h"
|
|
#include "../generic/idprop_py_api.h"
|
|
#include "../bmesh/bmesh_py_api.h"
|
|
#include "../mathutils/mathutils.h"
|
|
|
|
|
|
/* for internal use, when starting and ending python scripts */
|
|
|
|
/* in case a python script triggers another python call, stop bpy_context_clear from invalidating */
|
|
static int py_call_level = 0;
|
|
BPy_StructRNA *bpy_context_module = NULL; /* for fast access */
|
|
|
|
// #define TIME_PY_RUN // simple python tests. prints on exit.
|
|
|
|
#ifdef TIME_PY_RUN
|
|
#include "PIL_time.h"
|
|
static int bpy_timer_count = 0;
|
|
static double bpy_timer; /* time since python starts */
|
|
static double bpy_timer_run; /* time for each python script run */
|
|
static double bpy_timer_run_tot; /* accumulate python runs */
|
|
#endif
|
|
|
|
/* use for updating while a python script runs - in case of file load */
|
|
void BPY_context_update(bContext *C)
|
|
{
|
|
/* don't do this from a non-main (e.g. render) thread, it can cause a race
|
|
* condition on C->data.recursion. ideal solution would be to disable
|
|
* context entirely from non-main threads, but that's more complicated */
|
|
if (!BLI_thread_is_main())
|
|
return;
|
|
|
|
BPy_SetContext(C);
|
|
bpy_import_main_set(CTX_data_main(C));
|
|
BPY_modules_update(C); /* can give really bad results if this isn't here */
|
|
}
|
|
|
|
void bpy_context_set(bContext *C, PyGILState_STATE *gilstate)
|
|
{
|
|
py_call_level++;
|
|
|
|
if (gilstate)
|
|
*gilstate = PyGILState_Ensure();
|
|
|
|
if (py_call_level == 1) {
|
|
BPY_context_update(C);
|
|
|
|
#ifdef TIME_PY_RUN
|
|
if (bpy_timer_count == 0) {
|
|
/* record time from the beginning */
|
|
bpy_timer = PIL_check_seconds_timer();
|
|
bpy_timer_run = bpy_timer_run_tot = 0.0;
|
|
}
|
|
bpy_timer_run = PIL_check_seconds_timer();
|
|
|
|
|
|
bpy_timer_count++;
|
|
#endif
|
|
}
|
|
}
|
|
|
|
/* context should be used but not now because it causes some bugs */
|
|
void bpy_context_clear(bContext *UNUSED(C), PyGILState_STATE *gilstate)
|
|
{
|
|
py_call_level--;
|
|
|
|
if (gilstate)
|
|
PyGILState_Release(*gilstate);
|
|
|
|
if (py_call_level < 0) {
|
|
fprintf(stderr, "ERROR: Python context internal state bug. this should not happen!\n");
|
|
}
|
|
else if (py_call_level == 0) {
|
|
/* XXX - Calling classes currently wont store the context :\,
|
|
* cant set NULL because of this. but this is very flakey still. */
|
|
#if 0
|
|
BPy_SetContext(NULL);
|
|
bpy_import_main_set(NULL);
|
|
#endif
|
|
|
|
#ifdef TIME_PY_RUN
|
|
bpy_timer_run_tot += PIL_check_seconds_timer() - bpy_timer_run;
|
|
bpy_timer_count++;
|
|
#endif
|
|
|
|
}
|
|
}
|
|
|
|
void BPY_text_free_code(Text *text)
|
|
{
|
|
if (text->compiled) {
|
|
PyGILState_STATE gilstate;
|
|
bool use_gil = !PYC_INTERPRETER_ACTIVE;
|
|
|
|
if (use_gil)
|
|
gilstate = PyGILState_Ensure();
|
|
|
|
Py_DECREF((PyObject *)text->compiled);
|
|
text->compiled = NULL;
|
|
|
|
if (use_gil)
|
|
PyGILState_Release(gilstate);
|
|
}
|
|
}
|
|
|
|
void BPY_modules_update(bContext *C)
|
|
{
|
|
#if 0 /* slow, this runs all the time poll, draw etc 100's of time a sec. */
|
|
PyObject *mod = PyImport_ImportModuleLevel("bpy", NULL, NULL, NULL, 0);
|
|
PyModule_AddObject(mod, "data", BPY_rna_module());
|
|
PyModule_AddObject(mod, "types", BPY_rna_types()); /* atm this does not need updating */
|
|
#endif
|
|
|
|
/* refreshes the main struct */
|
|
BPY_update_rna_module();
|
|
if (bpy_context_module)
|
|
bpy_context_module->ptr.data = (void *)C;
|
|
}
|
|
|
|
void BPY_context_set(bContext *C)
|
|
{
|
|
BPy_SetContext(C);
|
|
}
|
|
|
|
/* defined in AUD_C-API.cpp */
|
|
extern PyObject *AUD_initPython(void);
|
|
|
|
#ifdef WITH_CYCLES
|
|
/* defined in cycles module */
|
|
static PyObject *CCL_initPython(void)
|
|
{
|
|
return (PyObject *)CCL_python_module_init();
|
|
}
|
|
#endif
|
|
|
|
static struct _inittab bpy_internal_modules[] = {
|
|
{(char *)"mathutils", PyInit_mathutils},
|
|
// {(char *)"mathutils.geometry", PyInit_mathutils_geometry},
|
|
// {(char *)"mathutils.noise", PyInit_mathutils_noise},
|
|
{(char *)"_bpy_path", BPyInit__bpy_path},
|
|
{(char *)"bgl", BPyInit_bgl},
|
|
{(char *)"blf", BPyInit_blf},
|
|
{(char *)"bmesh", BPyInit_bmesh},
|
|
// {(char *)"bmesh.types", BPyInit_bmesh_types},
|
|
// {(char *)"bmesh.utils", BPyInit_bmesh_utils},
|
|
#ifdef WITH_AUDASPACE
|
|
{(char *)"aud", AUD_initPython},
|
|
#endif
|
|
#ifdef WITH_CYCLES
|
|
{(char *)"_cycles", CCL_initPython},
|
|
#endif
|
|
{(char *)"gpu", GPU_initPython},
|
|
{(char *)"idprop", BPyInit_idprop},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
/* call BPY_context_set first */
|
|
void BPY_python_start(int argc, const char **argv)
|
|
{
|
|
#ifndef WITH_PYTHON_MODULE
|
|
PyThreadState *py_tstate = NULL;
|
|
const char *py_path_bundle = BLI_get_folder(BLENDER_SYSTEM_PYTHON, NULL);
|
|
|
|
/* not essential but nice to set our name */
|
|
static wchar_t program_path_wchar[FILE_MAX]; /* python holds a reference */
|
|
BLI_strncpy_wchar_from_utf8(program_path_wchar, BLI_program_path(), sizeof(program_path_wchar) / sizeof(wchar_t));
|
|
Py_SetProgramName(program_path_wchar);
|
|
|
|
/* must run before python initializes */
|
|
PyImport_ExtendInittab(bpy_internal_modules);
|
|
|
|
/* allow to use our own included python */
|
|
PyC_SetHomePath(py_path_bundle);
|
|
|
|
/* without this the sys.stdout may be set to 'ascii'
|
|
* (it is on my system at least), where printing unicode values will raise
|
|
* an error, this is highly annoying, another stumbling block for devs,
|
|
* so use a more relaxed error handler and enforce utf-8 since the rest of
|
|
* blender is utf-8 too - campbell */
|
|
|
|
/* XXX, update: this is unreliable! 'PYTHONIOENCODING' is ignored in MS-Windows
|
|
* when dynamically linked, see: [#31555] for details.
|
|
* Python doesn't expose a good way to set this. */
|
|
BLI_setenv("PYTHONIOENCODING", "utf-8:surrogateescape");
|
|
|
|
/* Python 3.2 now looks for '2.xx/python/include/python3.2d/pyconfig.h' to
|
|
* parse from the 'sysconfig' module which is used by 'site',
|
|
* so for now disable site. alternatively we could copy the file. */
|
|
if (py_path_bundle) {
|
|
Py_NoSiteFlag = 1;
|
|
}
|
|
|
|
Py_FrozenFlag = 1;
|
|
|
|
Py_Initialize();
|
|
|
|
/* THIS IS BAD: see http://bugs.python.org/issue16129 */
|
|
/* this clobbers the stdout on exit (no 'MEM_printmemlist_stats') */
|
|
#if 0
|
|
/* until python provides a reliable way to set the env var */
|
|
PyRun_SimpleString("import sys, io\n"
|
|
"sys.__backup_stdio__ = sys.__stdout__, sys.__stderr__\n" /* else we loose the FD's [#32720] */
|
|
"sys.__stdout__ = sys.stdout = io.TextIOWrapper(io.open(sys.stdout.fileno(), 'wb', -1), "
|
|
"encoding='utf-8', errors='surrogateescape', newline='\\n', line_buffering=True)\n"
|
|
"sys.__stderr__ = sys.stderr = io.TextIOWrapper(io.open(sys.stderr.fileno(), 'wb', -1), "
|
|
"encoding='utf-8', errors='surrogateescape', newline='\\n', line_buffering=True)\n");
|
|
if (PyErr_Occurred()) {
|
|
PyErr_Print();
|
|
PyErr_Clear();
|
|
}
|
|
#endif
|
|
/* end the baddness */
|
|
|
|
|
|
// PySys_SetArgv(argc, argv); /* broken in py3, not a huge deal */
|
|
/* sigh, why do python guys not have a (char **) version anymore? */
|
|
{
|
|
int i;
|
|
PyObject *py_argv = PyList_New(argc);
|
|
for (i = 0; i < argc; i++) {
|
|
/* should fix bug #20021 - utf path name problems, by replacing
|
|
* PyUnicode_FromString, with this one */
|
|
PyList_SET_ITEM(py_argv, i, PyC_UnicodeFromByte(argv[i]));
|
|
}
|
|
|
|
PySys_SetObject("argv", py_argv);
|
|
Py_DECREF(py_argv);
|
|
}
|
|
|
|
/* Initialize thread support (also acquires lock) */
|
|
PyEval_InitThreads();
|
|
#else
|
|
(void)argc;
|
|
(void)argv;
|
|
|
|
/* must run before python initializes */
|
|
/* broken in py3.3, load explicitly below */
|
|
// PyImport_ExtendInittab(bpy_internal_modules);
|
|
#endif
|
|
|
|
bpy_intern_string_init();
|
|
|
|
|
|
#ifdef WITH_PYTHON_MODULE
|
|
{
|
|
/* Manually load all modules */
|
|
struct _inittab *inittab_item;
|
|
PyObject *sys_modules = PyImport_GetModuleDict();
|
|
|
|
for (inittab_item = bpy_internal_modules; inittab_item->name; inittab_item++) {
|
|
PyObject *mod = inittab_item->initfunc();
|
|
if (mod) {
|
|
PyDict_SetItemString(sys_modules, inittab_item->name, mod);
|
|
}
|
|
else {
|
|
PyErr_Print();
|
|
PyErr_Clear();
|
|
}
|
|
// Py_DECREF(mod); /* ideally would decref, but in this case we never want to free */
|
|
}
|
|
}
|
|
#endif
|
|
|
|
/* bpy.* and lets us import it */
|
|
BPy_init_modules();
|
|
|
|
bpy_import_init(PyEval_GetBuiltins());
|
|
|
|
pyrna_alloc_types();
|
|
|
|
#ifndef WITH_PYTHON_MODULE
|
|
/* py module runs atexit when bpy is freed */
|
|
BPY_atexit_register(); /* this can init any time */
|
|
|
|
py_tstate = PyGILState_GetThisThreadState();
|
|
PyEval_ReleaseThread(py_tstate);
|
|
#endif
|
|
}
|
|
|
|
void BPY_python_end(void)
|
|
{
|
|
// fprintf(stderr, "Ending Python!\n");
|
|
|
|
PyGILState_Ensure(); /* finalizing, no need to grab the state */
|
|
|
|
/* free other python data. */
|
|
pyrna_free_types();
|
|
|
|
/* clear all python data from structs */
|
|
|
|
bpy_intern_string_exit();
|
|
|
|
#ifndef WITH_PYTHON_MODULE
|
|
BPY_atexit_unregister(); /* without this we get recursive calls to WM_exit */
|
|
#endif
|
|
|
|
Py_Finalize();
|
|
|
|
#ifdef TIME_PY_RUN
|
|
/* measure time since py started */
|
|
bpy_timer = PIL_check_seconds_timer() - bpy_timer;
|
|
|
|
printf("*bpy stats* - ");
|
|
printf("tot exec: %d, ", bpy_timer_count);
|
|
printf("tot run: %.4fsec, ", bpy_timer_run_tot);
|
|
if (bpy_timer_count > 0)
|
|
printf("average run: %.6fsec, ", (bpy_timer_run_tot / bpy_timer_count));
|
|
|
|
if (bpy_timer > 0.0)
|
|
printf("tot usage %.4f%%", (bpy_timer_run_tot / bpy_timer) * 100.0);
|
|
|
|
printf("\n");
|
|
|
|
// fprintf(stderr, "Ending Python Done!\n");
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
static void python_script_error_jump_text(struct Text *text)
|
|
{
|
|
int lineno;
|
|
int offset;
|
|
python_script_error_jump(text->id.name + 2, &lineno, &offset);
|
|
if (lineno != -1) {
|
|
/* select the line with the error */
|
|
txt_move_to(text, lineno - 1, INT_MAX, false);
|
|
txt_move_to(text, lineno - 1, offset, true);
|
|
}
|
|
}
|
|
|
|
/* super annoying, undo _PyModule_Clear(), bug [#23871] */
|
|
#define PYMODULE_CLEAR_WORKAROUND
|
|
|
|
#ifdef PYMODULE_CLEAR_WORKAROUND
|
|
/* bad!, we should never do this, but currently only safe way I could find to keep namespace.
|
|
* from being cleared. - campbell */
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *md_dict;
|
|
/* ommit other values, we only want the dict. */
|
|
} PyModuleObject;
|
|
#endif
|
|
|
|
static int python_script_exec(bContext *C, const char *fn, struct Text *text,
|
|
struct ReportList *reports, const bool do_jump)
|
|
{
|
|
Main *bmain_old = CTX_data_main(C);
|
|
PyObject *main_mod = NULL;
|
|
PyObject *py_dict = NULL, *py_result = NULL;
|
|
PyGILState_STATE gilstate;
|
|
|
|
BLI_assert(fn || text);
|
|
|
|
if (fn == NULL && text == NULL) {
|
|
return 0;
|
|
}
|
|
|
|
bpy_context_set(C, &gilstate);
|
|
|
|
PyC_MainModule_Backup(&main_mod);
|
|
|
|
if (text) {
|
|
char fn_dummy[FILE_MAXDIR];
|
|
bpy_text_filename_get(fn_dummy, sizeof(fn_dummy), text);
|
|
|
|
if (text->compiled == NULL) { /* if it wasn't already compiled, do it now */
|
|
char *buf = txt_to_buf(text);
|
|
|
|
text->compiled = Py_CompileString(buf, fn_dummy, Py_file_input);
|
|
|
|
MEM_freeN(buf);
|
|
|
|
if (PyErr_Occurred()) {
|
|
if (do_jump) {
|
|
python_script_error_jump_text(text);
|
|
}
|
|
BPY_text_free_code(text);
|
|
}
|
|
}
|
|
|
|
if (text->compiled) {
|
|
py_dict = PyC_DefaultNameSpace(fn_dummy);
|
|
py_result = PyEval_EvalCode(text->compiled, py_dict, py_dict);
|
|
}
|
|
|
|
}
|
|
else {
|
|
FILE *fp = BLI_fopen(fn, "r");
|
|
|
|
if (fp) {
|
|
py_dict = PyC_DefaultNameSpace(fn);
|
|
|
|
#ifdef _WIN32
|
|
/* Previously we used PyRun_File to run directly the code on a FILE
|
|
* object, but as written in the Python/C API Ref Manual, chapter 2,
|
|
* 'FILE structs for different C libraries can be different and
|
|
* incompatible'.
|
|
* So now we load the script file data to a buffer */
|
|
{
|
|
char *pystring;
|
|
|
|
fclose(fp);
|
|
|
|
pystring = MEM_mallocN(strlen(fn) + 37, "pystring");
|
|
pystring[0] = '\0';
|
|
sprintf(pystring, "f=open(r'%s');exec(f.read());f.close()", fn);
|
|
py_result = PyRun_String(pystring, Py_file_input, py_dict, py_dict);
|
|
MEM_freeN(pystring);
|
|
}
|
|
#else
|
|
py_result = PyRun_File(fp, fn, Py_file_input, py_dict, py_dict);
|
|
fclose(fp);
|
|
#endif
|
|
}
|
|
else {
|
|
PyErr_Format(PyExc_IOError,
|
|
"Python file \"%s\" could not be opened: %s",
|
|
fn, strerror(errno));
|
|
py_result = NULL;
|
|
}
|
|
}
|
|
|
|
if (!py_result) {
|
|
if (text) {
|
|
if (do_jump) {
|
|
/* ensure text is valid before use, the script may have freed its self */
|
|
Main *bmain_new = CTX_data_main(C);
|
|
if ((bmain_old == bmain_new) && (BLI_findindex(&bmain_new->text, text) != -1)) {
|
|
python_script_error_jump_text(text);
|
|
}
|
|
}
|
|
}
|
|
BPy_errors_to_report(reports);
|
|
}
|
|
else {
|
|
Py_DECREF(py_result);
|
|
}
|
|
|
|
if (py_dict) {
|
|
#ifdef PYMODULE_CLEAR_WORKAROUND
|
|
PyModuleObject *mmod = (PyModuleObject *)PyDict_GetItemString(PyThreadState_GET()->interp->modules, "__main__");
|
|
PyObject *dict_back = mmod->md_dict;
|
|
/* freeing the module will clear the namespace,
|
|
* gives problems running classes defined in this namespace being used later. */
|
|
mmod->md_dict = NULL;
|
|
Py_DECREF(dict_back);
|
|
#endif
|
|
|
|
#undef PYMODULE_CLEAR_WORKAROUND
|
|
}
|
|
|
|
PyC_MainModule_Restore(main_mod);
|
|
|
|
bpy_context_clear(C, &gilstate);
|
|
|
|
return (py_result != NULL);
|
|
}
|
|
|
|
/* Can run a file or text block */
|
|
int BPY_filepath_exec(bContext *C, const char *filepath, struct ReportList *reports)
|
|
{
|
|
return python_script_exec(C, filepath, NULL, reports, false);
|
|
}
|
|
|
|
|
|
int BPY_text_exec(bContext *C, struct Text *text, struct ReportList *reports, const bool do_jump)
|
|
{
|
|
return python_script_exec(C, NULL, text, reports, do_jump);
|
|
}
|
|
|
|
void BPY_DECREF(void *pyob_ptr)
|
|
{
|
|
PyGILState_STATE gilstate = PyGILState_Ensure();
|
|
Py_DECREF((PyObject *)pyob_ptr);
|
|
PyGILState_Release(gilstate);
|
|
}
|
|
|
|
void BPY_DECREF_RNA_INVALIDATE(void *pyob_ptr)
|
|
{
|
|
PyGILState_STATE gilstate = PyGILState_Ensure();
|
|
const int do_invalidate = (Py_REFCNT((PyObject *)pyob_ptr) > 1);
|
|
Py_DECREF((PyObject *)pyob_ptr);
|
|
if (do_invalidate) {
|
|
pyrna_invalidate(pyob_ptr);
|
|
}
|
|
PyGILState_Release(gilstate);
|
|
}
|
|
|
|
|
|
/* return -1 on error, else 0 */
|
|
int BPY_button_exec(bContext *C, const char *expr, double *value, const short verbose)
|
|
{
|
|
PyGILState_STATE gilstate;
|
|
PyObject *py_dict, *mod, *retval;
|
|
int error_ret = 0;
|
|
PyObject *main_mod = NULL;
|
|
|
|
if (!value || !expr) return -1;
|
|
|
|
if (expr[0] == '\0') {
|
|
*value = 0.0;
|
|
return error_ret;
|
|
}
|
|
|
|
bpy_context_set(C, &gilstate);
|
|
|
|
PyC_MainModule_Backup(&main_mod);
|
|
|
|
py_dict = PyC_DefaultNameSpace("<blender button>");
|
|
|
|
mod = PyImport_ImportModule("math");
|
|
if (mod) {
|
|
PyDict_Merge(py_dict, PyModule_GetDict(mod), 0); /* 0 - don't overwrite existing values */
|
|
Py_DECREF(mod);
|
|
}
|
|
else { /* highly unlikely but possibly */
|
|
PyErr_Print();
|
|
PyErr_Clear();
|
|
}
|
|
|
|
retval = PyRun_String(expr, Py_eval_input, py_dict, py_dict);
|
|
|
|
if (retval == NULL) {
|
|
error_ret = -1;
|
|
}
|
|
else {
|
|
double val;
|
|
|
|
if (PyTuple_Check(retval)) {
|
|
/* Users my have typed in 10km, 2m
|
|
* add up all values */
|
|
int i;
|
|
val = 0.0;
|
|
|
|
for (i = 0; i < PyTuple_GET_SIZE(retval); i++) {
|
|
const double val_item = PyFloat_AsDouble(PyTuple_GET_ITEM(retval, i));
|
|
if (val_item == -1 && PyErr_Occurred()) {
|
|
val = -1;
|
|
break;
|
|
}
|
|
val += val_item;
|
|
}
|
|
}
|
|
else {
|
|
val = PyFloat_AsDouble(retval);
|
|
}
|
|
Py_DECREF(retval);
|
|
|
|
if (val == -1 && PyErr_Occurred()) {
|
|
error_ret = -1;
|
|
}
|
|
else if (!finite(val)) {
|
|
*value = 0.0;
|
|
}
|
|
else {
|
|
*value = val;
|
|
}
|
|
}
|
|
|
|
if (error_ret) {
|
|
if (verbose) {
|
|
BPy_errors_to_report(CTX_wm_reports(C));
|
|
}
|
|
else {
|
|
PyErr_Clear();
|
|
}
|
|
}
|
|
|
|
PyC_MainModule_Restore(main_mod);
|
|
|
|
bpy_context_clear(C, &gilstate);
|
|
|
|
return error_ret;
|
|
}
|
|
|
|
int BPY_string_exec(bContext *C, const char *expr)
|
|
{
|
|
PyGILState_STATE gilstate;
|
|
PyObject *main_mod = NULL;
|
|
PyObject *py_dict, *retval;
|
|
int error_ret = 0;
|
|
Main *bmain_back; /* XXX, quick fix for release (Copy Settings crash), needs further investigation */
|
|
|
|
if (!expr) return -1;
|
|
|
|
if (expr[0] == '\0') {
|
|
return error_ret;
|
|
}
|
|
|
|
bpy_context_set(C, &gilstate);
|
|
|
|
PyC_MainModule_Backup(&main_mod);
|
|
|
|
py_dict = PyC_DefaultNameSpace("<blender string>");
|
|
|
|
bmain_back = bpy_import_main_get();
|
|
bpy_import_main_set(CTX_data_main(C));
|
|
|
|
retval = PyRun_String(expr, Py_eval_input, py_dict, py_dict);
|
|
|
|
bpy_import_main_set(bmain_back);
|
|
|
|
if (retval == NULL) {
|
|
error_ret = -1;
|
|
|
|
BPy_errors_to_report(CTX_wm_reports(C));
|
|
}
|
|
else {
|
|
Py_DECREF(retval);
|
|
}
|
|
|
|
PyC_MainModule_Restore(main_mod);
|
|
|
|
bpy_context_clear(C, &gilstate);
|
|
|
|
return error_ret;
|
|
}
|
|
|
|
|
|
void BPY_modules_load_user(bContext *C)
|
|
{
|
|
PyGILState_STATE gilstate;
|
|
Main *bmain = CTX_data_main(C);
|
|
Text *text;
|
|
|
|
/* can happen on file load */
|
|
if (bmain == NULL)
|
|
return;
|
|
|
|
/* update pointers since this can run from a nested script
|
|
* on file load */
|
|
if (py_call_level) {
|
|
BPY_context_update(C);
|
|
}
|
|
|
|
bpy_context_set(C, &gilstate);
|
|
|
|
for (text = bmain->text.first; text; text = text->id.next) {
|
|
if (text->flags & TXT_ISSCRIPT && BLI_testextensie(text->id.name + 2, ".py")) {
|
|
if (!(G.f & G_SCRIPT_AUTOEXEC)) {
|
|
printf("scripts disabled for \"%s\", skipping '%s'\n", bmain->name, text->id.name + 2);
|
|
}
|
|
else {
|
|
PyObject *module = bpy_text_import(text);
|
|
|
|
if (module == NULL) {
|
|
PyErr_Print();
|
|
PyErr_Clear();
|
|
}
|
|
else {
|
|
Py_DECREF(module);
|
|
}
|
|
|
|
/* check if the script loaded a new file */
|
|
if (bmain != CTX_data_main(C)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
bpy_context_clear(C, &gilstate);
|
|
}
|
|
|
|
int BPY_context_member_get(bContext *C, const char *member, bContextDataResult *result)
|
|
{
|
|
PyGILState_STATE gilstate;
|
|
bool use_gil = !PYC_INTERPRETER_ACTIVE;
|
|
|
|
PyObject *pyctx;
|
|
PyObject *item;
|
|
PointerRNA *ptr = NULL;
|
|
bool done = false;
|
|
|
|
if (use_gil)
|
|
gilstate = PyGILState_Ensure();
|
|
|
|
pyctx = (PyObject *)CTX_py_dict_get(C);
|
|
item = PyDict_GetItemString(pyctx, member);
|
|
|
|
if (item == NULL) {
|
|
/* pass */
|
|
}
|
|
else if (item == Py_None) {
|
|
/* pass */
|
|
}
|
|
else if (BPy_StructRNA_Check(item)) {
|
|
ptr = &(((BPy_StructRNA *)item)->ptr);
|
|
|
|
//result->ptr = ((BPy_StructRNA *)item)->ptr;
|
|
CTX_data_pointer_set(result, ptr->id.data, ptr->type, ptr->data);
|
|
CTX_data_type_set(result, CTX_DATA_TYPE_POINTER);
|
|
done = true;
|
|
}
|
|
else if (PySequence_Check(item)) {
|
|
PyObject *seq_fast = PySequence_Fast(item, "bpy_context_get sequence conversion");
|
|
if (seq_fast == NULL) {
|
|
PyErr_Print();
|
|
PyErr_Clear();
|
|
}
|
|
else {
|
|
int len = PySequence_Fast_GET_SIZE(seq_fast);
|
|
int i;
|
|
for (i = 0; i < len; i++) {
|
|
PyObject *list_item = PySequence_Fast_GET_ITEM(seq_fast, i);
|
|
|
|
if (BPy_StructRNA_Check(list_item)) {
|
|
#if 0
|
|
CollectionPointerLink *link = MEM_callocN(sizeof(CollectionPointerLink), "bpy_context_get");
|
|
link->ptr = ((BPy_StructRNA *)item)->ptr;
|
|
BLI_addtail(&result->list, link);
|
|
#endif
|
|
ptr = &(((BPy_StructRNA *)list_item)->ptr);
|
|
CTX_data_list_add(result, ptr->id.data, ptr->type, ptr->data);
|
|
}
|
|
else {
|
|
printf("PyContext: '%s' list item not a valid type in sequece type '%s'\n",
|
|
member, Py_TYPE(item)->tp_name);
|
|
}
|
|
|
|
}
|
|
Py_DECREF(seq_fast);
|
|
CTX_data_type_set(result, CTX_DATA_TYPE_COLLECTION);
|
|
done = true;
|
|
}
|
|
}
|
|
|
|
if (done == false) {
|
|
if (item) printf("PyContext '%s' not a valid type\n", member);
|
|
else printf("PyContext '%s' not found\n", member);
|
|
}
|
|
else {
|
|
if (G.debug & G_DEBUG_PYTHON) {
|
|
printf("PyContext '%s' found\n", member);
|
|
}
|
|
}
|
|
|
|
if (use_gil)
|
|
PyGILState_Release(gilstate);
|
|
|
|
return done;
|
|
}
|
|
|
|
#ifdef WITH_PYTHON_MODULE
|
|
#include "BLI_fileops.h"
|
|
/* TODO, reloading the module isn't functional at the moment. */
|
|
|
|
static void bpy_module_free(void *mod);
|
|
extern int main_python_enter(int argc, const char **argv);
|
|
extern void main_python_exit(void);
|
|
static struct PyModuleDef bpy_proxy_def = {
|
|
PyModuleDef_HEAD_INIT,
|
|
"bpy", /* m_name */
|
|
NULL, /* m_doc */
|
|
0, /* m_size */
|
|
NULL, /* m_methods */
|
|
NULL, /* m_reload */
|
|
NULL, /* m_traverse */
|
|
NULL, /* m_clear */
|
|
bpy_module_free, /* m_free */
|
|
};
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
/* Type-specific fields go here. */
|
|
PyObject *mod;
|
|
} dealloc_obj;
|
|
|
|
/* call once __file__ is set */
|
|
static void bpy_module_delay_init(PyObject *bpy_proxy)
|
|
{
|
|
const int argc = 1;
|
|
const char *argv[2];
|
|
|
|
/* updating the module dict below will loose the reference to __file__ */
|
|
PyObject *filename_obj = PyModule_GetFilenameObject(bpy_proxy);
|
|
|
|
const char *filename_rel = _PyUnicode_AsString(filename_obj); /* can be relative */
|
|
char filename_abs[1024];
|
|
|
|
BLI_strncpy(filename_abs, filename_rel, sizeof(filename_abs));
|
|
BLI_path_cwd(filename_abs);
|
|
|
|
argv[0] = filename_abs;
|
|
argv[1] = NULL;
|
|
|
|
// printf("module found %s\n", argv[0]);
|
|
|
|
main_python_enter(argc, argv);
|
|
|
|
/* initialized in BPy_init_modules() */
|
|
PyDict_Update(PyModule_GetDict(bpy_proxy), PyModule_GetDict(bpy_package_py));
|
|
}
|
|
|
|
static void dealloc_obj_dealloc(PyObject *self);
|
|
|
|
static PyTypeObject dealloc_obj_Type = {{{0}}};
|
|
|
|
/* use our own dealloc so we can free a property if we use one */
|
|
static void dealloc_obj_dealloc(PyObject *self)
|
|
{
|
|
bpy_module_delay_init(((dealloc_obj *)self)->mod);
|
|
|
|
/* Note, for subclassed PyObjects we cant just call PyObject_DEL() directly or it will crash */
|
|
dealloc_obj_Type.tp_free(self);
|
|
}
|
|
|
|
PyMODINIT_FUNC
|
|
PyInit_bpy(void);
|
|
|
|
PyMODINIT_FUNC
|
|
PyInit_bpy(void)
|
|
{
|
|
PyObject *bpy_proxy = PyModule_Create(&bpy_proxy_def);
|
|
|
|
/* Problem:
|
|
* 1) this init function is expected to have a private member defined - 'md_def'
|
|
* but this is only set for C defined modules (not py packages)
|
|
* so we cant return 'bpy_package_py' as is.
|
|
*
|
|
* 2) there is a 'bpy' C module for python to load which is basically all of blender,
|
|
* and there is scripts/bpy/__init__.py,
|
|
* we may end up having to rename this module so there is no naming conflict here eg:
|
|
* 'from blender import bpy'
|
|
*
|
|
* 3) we don't know the filename at this point, workaround by assigning a dummy value
|
|
* which calls back when its freed so the real loading can take place.
|
|
*/
|
|
|
|
/* assign an object which is freed after __file__ is assigned */
|
|
dealloc_obj *dob;
|
|
|
|
/* assign dummy type */
|
|
dealloc_obj_Type.tp_name = "dealloc_obj";
|
|
dealloc_obj_Type.tp_basicsize = sizeof(dealloc_obj);
|
|
dealloc_obj_Type.tp_dealloc = dealloc_obj_dealloc;
|
|
dealloc_obj_Type.tp_flags = Py_TPFLAGS_DEFAULT;
|
|
|
|
if (PyType_Ready(&dealloc_obj_Type) < 0)
|
|
return NULL;
|
|
|
|
dob = (dealloc_obj *) dealloc_obj_Type.tp_alloc(&dealloc_obj_Type, 0);
|
|
dob->mod = bpy_proxy; /* borrow */
|
|
PyModule_AddObject(bpy_proxy, "__file__", (PyObject *)dob); /* borrow */
|
|
|
|
return bpy_proxy;
|
|
}
|
|
|
|
static void bpy_module_free(void *UNUSED(mod))
|
|
{
|
|
main_python_exit();
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
/* EVIL, define text.c functions here... */
|
|
/* BKE_text.h */
|
|
int text_check_identifier_unicode(const unsigned int ch)
|
|
{
|
|
return (ch < 255 && text_check_identifier((char)ch)) || Py_UNICODE_ISALNUM(ch);
|
|
}
|
|
|
|
int text_check_identifier_nodigit_unicode(const unsigned int ch)
|
|
{
|
|
return (ch < 255 && text_check_identifier_nodigit((char)ch)) || Py_UNICODE_ISALPHA(ch);
|
|
}
|