This works like Python's -c argument, handy to be able to avoid writing small scripts to disk.
		
			
				
	
	
		
			914 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			914 lines
		
	
	
		
			23 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>
 | |
| 
 | |
| #ifdef WIN32
 | |
| #  include "BLI_math_base.h"  /* finite */
 | |
| #endif
 | |
| 
 | |
| #include "MEM_guardedalloc.h"
 | |
| 
 | |
| #include "BLI_utildefines.h"
 | |
| #include "BLI_path_util.h"
 | |
| #include "BLI_fileops.h"
 | |
| #include "BLI_listbase.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 "bpy_app_translations.h"
 | |
| 
 | |
| #include "DNA_text_types.h"
 | |
| 
 | |
| #include "BKE_appdir.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_IsInterpreterActive();
 | |
| 
 | |
| 		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[] = {
 | |
| 	{"mathutils", PyInit_mathutils},
 | |
| #if 0
 | |
| 	{"mathutils.geometry", PyInit_mathutils_geometry},
 | |
| 	{"mathutils.noise", PyInit_mathutils_noise},
 | |
| 	{"mathutils.kdtree", PyInit_mathutils_kdtree},
 | |
| #endif
 | |
| 	{"_bpy_path", BPyInit__bpy_path},
 | |
| 	{"bgl", BPyInit_bgl},
 | |
| 	{"blf", BPyInit_blf},
 | |
| 	{"bmesh", BPyInit_bmesh},
 | |
| #if 0
 | |
| 	{"bmesh.types", BPyInit_bmesh_types},
 | |
| 	{"bmesh.utils", BPyInit_bmesh_utils},
 | |
| 	{"bmesh.utils", BPyInit_bmesh_geometry},
 | |
| #endif
 | |
| #ifdef WITH_AUDASPACE
 | |
| 	{"aud", AUD_initPython},
 | |
| #endif
 | |
| #ifdef WITH_CYCLES
 | |
| 	{"_cycles", CCL_initPython},
 | |
| #endif
 | |
| 	{"gpu", GPU_initPython},
 | |
| 	{"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 = BKE_appdir_folder_id(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, BKE_appdir_program_path(), ARRAY_SIZE(program_path_wchar));
 | |
| 	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 */
 | |
| 	Py_SetStandardStreamEncoding("utf-8", "surrogateescape");
 | |
| 
 | |
| 	/* Update, Py3.3 resolves attempting to parse non-existing header */
 | |
| #if 0
 | |
| 	/* 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;
 | |
| 	}
 | |
| #endif
 | |
| 
 | |
| 	Py_FrozenFlag = 1;
 | |
| 
 | |
| 	Py_Initialize();
 | |
| 
 | |
| 	// 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_STATE gilstate;
 | |
| 
 | |
| 	/* finalizing, no need to grab the state, except when we are a module */
 | |
| 	gilstate = PyGILState_Ensure();
 | |
| 	
 | |
| 	/* free other python data. */
 | |
| 	pyrna_free_types();
 | |
| 
 | |
| 	/* clear all python data from structs */
 | |
| 
 | |
| 	bpy_intern_string_exit();
 | |
| 
 | |
| 	/* bpy.app modules that need cleanup */
 | |
| 	BPY_app_translations_end();
 | |
| 
 | |
| #ifndef WITH_PYTHON_MODULE
 | |
| 	BPY_atexit_unregister(); /* without this we get recursive calls to WM_exit */
 | |
| 
 | |
| 	Py_Finalize();
 | |
| 
 | |
| 	(void)gilstate;
 | |
| #else
 | |
| 	PyGILState_Release(gilstate);
 | |
| #endif
 | |
| 
 | |
| #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
 | |
| 
 | |
| }
 | |
| 
 | |
| void BPY_python_reset(bContext *C)
 | |
| {
 | |
| 	/* unrelated security stuff */
 | |
| 	G.f &= ~(G_SCRIPT_AUTOEXEC_FAIL | G_SCRIPT_AUTOEXEC_FAIL_QUIET);
 | |
| 	G.autoexec_fail[0] = '\0';
 | |
| 
 | |
| 	BPY_driver_reset();
 | |
| 	BPY_app_handlers_reset(false);
 | |
| 	BPY_modules_load_user(C);
 | |
| }
 | |
| 
 | |
| 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;
 | |
| 	/* omit 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;
 | |
| 			PyObject *fn_dummy_py;
 | |
| 
 | |
| 			fn_dummy_py = PyC_UnicodeFromByte(fn_dummy);
 | |
| 
 | |
| 			buf = txt_to_buf(text);
 | |
| 			text->compiled = Py_CompileStringObject(buf, fn_dummy_py, Py_file_input, NULL, -1);
 | |
| 			MEM_freeN(buf);
 | |
| 
 | |
| 			Py_DECREF(fn_dummy_py);
 | |
| 
 | |
| 			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 */
 | |
| 			{
 | |
| 				const char *pystring = "with open(__file__, 'r') as f: exec(f.read())";
 | |
| 
 | |
| 				fclose(fp);
 | |
| 
 | |
| 				py_result = PyRun_String(pystring, Py_file_input, py_dict, py_dict);
 | |
| 			}
 | |
| #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 bool verbose)
 | |
| {
 | |
| 	PyGILState_STATE gilstate;
 | |
| 	int error_ret = 0;
 | |
| 
 | |
| 	if (!value || !expr) return -1;
 | |
| 
 | |
| 	if (expr[0] == '\0') {
 | |
| 		*value = 0.0;
 | |
| 		return error_ret;
 | |
| 	}
 | |
| 
 | |
| 	bpy_context_set(C, &gilstate);
 | |
| 
 | |
| 	error_ret = PyC_RunString_AsNumber(expr, value, "<blender button>");
 | |
| 
 | |
| 	if (error_ret) {
 | |
| 		if (verbose) {
 | |
| 			BPy_errors_to_report_ex(CTX_wm_reports(C), false, false);
 | |
| 		}
 | |
| 		else {
 | |
| 			PyErr_Clear();
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	bpy_context_clear(C, &gilstate);
 | |
| 
 | |
| 	return error_ret;
 | |
| }
 | |
| 
 | |
| int BPY_string_exec_ex(bContext *C, const char *expr, bool use_eval)
 | |
| {
 | |
| 	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, use_eval ? Py_eval_input : Py_file_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;
 | |
| }
 | |
| 
 | |
| int BPY_string_exec(bContext *C, const char *expr)
 | |
| {
 | |
| 	return BPY_string_exec_ex(C, expr, true);
 | |
| }
 | |
| 
 | |
| 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)) {
 | |
| 				if (!(G.f & G_SCRIPT_AUTOEXEC_FAIL_QUIET)) {
 | |
| 					G.f |= G_SCRIPT_AUTOEXEC_FAIL;
 | |
| 					BLI_snprintf(G.autoexec_fail, sizeof(G.autoexec_fail), "Text '%s'", text->id.name + 2);
 | |
| 
 | |
| 					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_IsInterpreterActive();
 | |
| 
 | |
| 	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) {
 | |
| 		done = true;
 | |
| 	}
 | |
| 	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;
 | |
| 
 | |
| /* 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);
 | |
| }
 |