fix for crash when a python operator or render engine was freed in the C code and...
[blender.git] / source / blender / python / intern / bpy_interface.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): Michel Selten, Willian P. Germano, Stephen Swaney,
19  * Chris Keith, Chris Want, Ken Hughes, Campbell Barton
20  *
21  * ***** END GPL LICENSE BLOCK *****
22  */
23
24 /** \file blender/python/intern/bpy_interface.c
25  *  \ingroup pythonintern
26  *
27  * This file deals with embedding the python interpreter within blender,
28  * starting and stopping python and exposing blender/python modules so they can
29  * be accesses from scripts.
30  */
31
32  
33 /* grr, python redefines */
34 #ifdef _POSIX_C_SOURCE
35 #  undef _POSIX_C_SOURCE
36 #endif
37
38 #include <Python.h>
39
40 #include "MEM_guardedalloc.h"
41
42 #include "RNA_types.h"
43
44 #include "bpy.h"
45 #include "gpu.h"
46 #include "bpy_rna.h"
47 #include "bpy_util.h"
48 #include "bpy_traceback.h"
49 #include "bpy_intern_string.h"
50
51 #include "DNA_space_types.h"
52 #include "DNA_text_types.h"
53
54 #include "BLI_path_util.h"
55 #include "BLI_fileops.h"
56 #include "BLI_listbase.h"
57 #include "BLI_math_base.h"
58 #include "BLI_string.h"
59 #include "BLI_string_utf8.h"
60 #include "BLI_utildefines.h"
61 #include "BLI_threads.h"
62
63 #include "BKE_context.h"
64 #include "BKE_text.h"
65 #include "BKE_main.h"
66 #include "BKE_global.h" /* only for script checking */
67
68 #include "CCL_api.h"
69
70 #include "BPY_extern.h"
71
72 #include "../generic/bpy_internal_import.h"  /* our own imports */
73 #include "../generic/py_capi_utils.h"
74
75 /* inittab initialization functions */
76 #include "../generic/bgl.h"
77 #include "../generic/blf_py_api.h"
78 #include "../generic/idprop_py_api.h"
79 #include "../bmesh/bmesh_py_api.h"
80 #include "../mathutils/mathutils.h"
81
82
83 /* for internal use, when starting and ending python scripts */
84
85 /* in case a python script triggers another python call, stop bpy_context_clear from invalidating */
86 static int py_call_level = 0;
87 BPy_StructRNA *bpy_context_module = NULL; /* for fast access */
88
89 // #define TIME_PY_RUN // simple python tests. prints on exit.
90
91 #ifdef TIME_PY_RUN
92 #include "PIL_time.h"
93 static int     bpy_timer_count = 0;
94 static double  bpy_timer;   /* time since python starts */
95 static double  bpy_timer_run;   /* time for each python script run */
96 static double  bpy_timer_run_tot;   /* accumulate python runs */
97 #endif
98
99 /* use for updating while a python script runs - in case of file load */
100 void BPY_context_update(bContext *C)
101 {
102         /* don't do this from a non-main (e.g. render) thread, it can cause a race
103          * condition on C->data.recursion. ideal solution would be to disable
104          * context entirely from non-main threads, but that's more complicated */
105         if (!BLI_thread_is_main())
106                 return;
107
108         BPy_SetContext(C);
109         bpy_import_main_set(CTX_data_main(C));
110         BPY_modules_update(C); /* can give really bad results if this isn't here */
111 }
112
113 void bpy_context_set(bContext *C, PyGILState_STATE *gilstate)
114 {
115         py_call_level++;
116
117         if (gilstate)
118                 *gilstate = PyGILState_Ensure();
119
120         if (py_call_level == 1) {
121                 BPY_context_update(C);
122
123 #ifdef TIME_PY_RUN
124                 if (bpy_timer_count == 0) {
125                         /* record time from the beginning */
126                         bpy_timer = PIL_check_seconds_timer();
127                         bpy_timer_run = bpy_timer_run_tot = 0.0;
128                 }
129                 bpy_timer_run = PIL_check_seconds_timer();
130
131
132                 bpy_timer_count++;
133 #endif
134         }
135 }
136
137 /* context should be used but not now because it causes some bugs */
138 void bpy_context_clear(bContext *UNUSED(C), PyGILState_STATE *gilstate)
139 {
140         py_call_level--;
141
142         if (gilstate)
143                 PyGILState_Release(*gilstate);
144
145         if (py_call_level < 0) {
146                 fprintf(stderr, "ERROR: Python context internal state bug. this should not happen!\n");
147         }
148         else if (py_call_level == 0) {
149                 /* XXX - Calling classes currently wont store the context :\,
150                  * cant set NULL because of this. but this is very flakey still. */
151 #if 0
152                 BPy_SetContext(NULL);
153                 bpy_import_main_set(NULL);
154 #endif
155
156 #ifdef TIME_PY_RUN
157                 bpy_timer_run_tot += PIL_check_seconds_timer() - bpy_timer_run;
158                 bpy_timer_count++;
159 #endif
160
161         }
162 }
163
164 void BPY_text_free_code(Text *text)
165 {
166         if (text->compiled) {
167                 PyGILState_STATE gilstate;
168                 int use_gil = !PYC_INTERPRETER_ACTIVE;
169
170                 if (use_gil)
171                         gilstate = PyGILState_Ensure();
172
173                 Py_DECREF((PyObject *)text->compiled);
174                 text->compiled = NULL;
175
176                 if (use_gil)
177                         PyGILState_Release(gilstate);
178         }
179 }
180
181 void BPY_modules_update(bContext *C)
182 {
183 #if 0  /* slow, this runs all the time poll, draw etc 100's of time a sec. */
184         PyObject *mod = PyImport_ImportModuleLevel("bpy", NULL, NULL, NULL, 0);
185         PyModule_AddObject(mod, "data", BPY_rna_module());
186         PyModule_AddObject(mod, "types", BPY_rna_types());  /* atm this does not need updating */
187 #endif
188
189         /* refreshes the main struct */
190         BPY_update_rna_module();
191         if (bpy_context_module)
192                 bpy_context_module->ptr.data = (void *)C;
193 }
194
195 void BPY_context_set(bContext *C)
196 {
197         BPy_SetContext(C);
198 }
199
200 /* defined in AUD_C-API.cpp */
201 extern PyObject *AUD_initPython(void);
202
203 #ifdef WITH_CYCLES
204 /* defined in cycles module */
205 static PyObject *CCL_initPython(void)
206 {
207         return (PyObject *)CCL_python_module_init();
208 }
209 #endif
210
211 static struct _inittab bpy_internal_modules[] = {
212         {(char *)"mathutils", PyInit_mathutils},
213 //      {(char *)"mathutils.geometry", PyInit_mathutils_geometry},
214 //      {(char *)"mathutils.noise", PyInit_mathutils_noise},
215         {(char *)"bgl", BPyInit_bgl},
216         {(char *)"blf", BPyInit_blf},
217         {(char *)"bmesh", BPyInit_bmesh},
218         // {(char *)"bmesh.types", BPyInit_bmesh_types},
219         // {(char *)"bmesh.utils", BPyInit_bmesh_utils},
220 #ifdef WITH_AUDASPACE
221         {(char *)"aud", AUD_initPython},
222 #endif
223 #ifdef WITH_CYCLES
224         {(char *)"_cycles", CCL_initPython},
225 #endif
226         {(char *)"gpu", GPU_initPython},
227         {(char *)"idprop", BPyInit_idprop},
228         {NULL, NULL}
229 };
230
231 /* call BPY_context_set first */
232 void BPY_python_start(int argc, const char **argv)
233 {
234 #ifndef WITH_PYTHON_MODULE
235         PyThreadState *py_tstate = NULL;
236         const char *py_path_bundle = BLI_get_folder(BLENDER_SYSTEM_PYTHON, NULL);
237
238         /* not essential but nice to set our name */
239         static wchar_t program_path_wchar[FILE_MAX]; /* python holds a reference */
240         BLI_strncpy_wchar_from_utf8(program_path_wchar, BLI_program_path(), sizeof(program_path_wchar) / sizeof(wchar_t));
241         Py_SetProgramName(program_path_wchar);
242
243         /* must run before python initializes */
244         PyImport_ExtendInittab(bpy_internal_modules);
245
246         /* allow to use our own included python */
247         PyC_SetHomePath(py_path_bundle);
248
249         /* without this the sys.stdout may be set to 'ascii'
250          * (it is on my system at least), where printing unicode values will raise
251          * an error, this is highly annoying, another stumbling block for devs,
252          * so use a more relaxed error handler and enforce utf-8 since the rest of
253          * blender is utf-8 too - campbell */
254
255         /* XXX, update: this is unreliable! 'PYTHONIOENCODING' is ignored in MS-Windows
256          * when dynamically linked, see: [#31555] for details.
257          * Python doesn't expose a good way to set this. */
258         BLI_setenv("PYTHONIOENCODING", "utf-8:surrogateescape");
259
260         /* Python 3.2 now looks for '2.xx/python/include/python3.2d/pyconfig.h' to
261          * parse from the 'sysconfig' module which is used by 'site',
262          * so for now disable site. alternatively we could copy the file. */
263         if (py_path_bundle) {
264                 Py_NoSiteFlag = 1;
265         }
266
267         Py_FrozenFlag = 1;
268
269         Py_Initialize();
270
271         // PySys_SetArgv(argc, argv);  /* broken in py3, not a huge deal */
272         /* sigh, why do python guys not have a (char **) version anymore? */
273         {
274                 int i;
275                 PyObject *py_argv = PyList_New(argc);
276                 for (i = 0; i < argc; i++) {
277                         /* should fix bug #20021 - utf path name problems, by replacing
278                          * PyUnicode_FromString, with this one */
279                         PyList_SET_ITEM(py_argv, i, PyC_UnicodeFromByte(argv[i]));
280                 }
281
282                 PySys_SetObject("argv", py_argv);
283                 Py_DECREF(py_argv);
284         }
285         
286         /* Initialize thread support (also acquires lock) */
287         PyEval_InitThreads();
288 #else
289         (void)argc;
290         (void)argv;
291
292         /* must run before python initializes */
293         PyImport_ExtendInittab(bpy_internal_modules);
294 #endif
295
296         bpy_intern_string_init();
297
298         /* bpy.* and lets us import it */
299         BPy_init_modules();
300
301         bpy_import_init(PyEval_GetBuiltins());
302         
303         pyrna_alloc_types();
304
305 #ifndef WITH_PYTHON_MODULE
306         /* py module runs atexit when bpy is freed */
307         BPY_atexit_register(); /* this can init any time */
308
309         py_tstate = PyGILState_GetThisThreadState();
310         PyEval_ReleaseThread(py_tstate);
311 #endif
312 }
313
314 void BPY_python_end(void)
315 {
316         // fprintf(stderr, "Ending Python!\n");
317
318         PyGILState_Ensure(); /* finalizing, no need to grab the state */
319         
320         /* free other python data. */
321         pyrna_free_types();
322
323         /* clear all python data from structs */
324
325         bpy_intern_string_exit();
326
327 #ifndef WITH_PYTHON_MODULE
328         BPY_atexit_unregister(); /* without this we get recursive calls to WM_exit */
329 #endif
330
331         Py_Finalize();
332         
333 #ifdef TIME_PY_RUN
334         /* measure time since py started */
335         bpy_timer = PIL_check_seconds_timer() - bpy_timer;
336
337         printf("*bpy stats* - ");
338         printf("tot exec: %d,  ", bpy_timer_count);
339         printf("tot run: %.4fsec,  ", bpy_timer_run_tot);
340         if (bpy_timer_count > 0)
341                 printf("average run: %.6fsec,  ", (bpy_timer_run_tot / bpy_timer_count));
342
343         if (bpy_timer > 0.0)
344                 printf("tot usage %.4f%%", (bpy_timer_run_tot / bpy_timer) * 100.0);
345
346         printf("\n");
347
348         // fprintf(stderr, "Ending Python Done!\n");
349
350 #endif
351
352 }
353
354 static void python_script_error_jump_text(struct Text *text)
355 {
356         int lineno;
357         int offset;
358         python_script_error_jump(text->id.name + 2, &lineno, &offset);
359         if (lineno != -1) {
360                 /* select the line with the error */
361                 txt_move_to(text, lineno - 1, INT_MAX, FALSE);
362                 txt_move_to(text, lineno - 1, offset, TRUE);
363         }
364 }
365
366 /* super annoying, undo _PyModule_Clear(), bug [#23871] */
367 #define PYMODULE_CLEAR_WORKAROUND
368
369 #ifdef PYMODULE_CLEAR_WORKAROUND
370 /* bad!, we should never do this, but currently only safe way I could find to keep namespace.
371  * from being cleared. - campbell */
372 typedef struct {
373         PyObject_HEAD
374         PyObject *md_dict;
375         /* ommit other values, we only want the dict. */
376 } PyModuleObject;
377 #endif
378
379 static int python_script_exec(bContext *C, const char *fn, struct Text *text,
380                               struct ReportList *reports, const short do_jump)
381 {
382         Main *bmain_old = CTX_data_main(C);
383         PyObject *main_mod = NULL;
384         PyObject *py_dict = NULL, *py_result = NULL;
385         PyGILState_STATE gilstate;
386
387         BLI_assert(fn || text);
388
389         if (fn == NULL && text == NULL) {
390                 return 0;
391         }
392
393         bpy_context_set(C, &gilstate);
394
395         PyC_MainModule_Backup(&main_mod);
396
397         if (text) {
398                 char fn_dummy[FILE_MAXDIR];
399                 bpy_text_filename_get(fn_dummy, sizeof(fn_dummy), text);
400
401                 if (text->compiled == NULL) {   /* if it wasn't already compiled, do it now */
402                         char *buf = txt_to_buf(text);
403
404                         text->compiled = Py_CompileString(buf, fn_dummy, Py_file_input);
405
406                         MEM_freeN(buf);
407
408                         if (PyErr_Occurred()) {
409                                 if (do_jump) {
410                                         python_script_error_jump_text(text);
411                                 }
412                                 BPY_text_free_code(text);
413                         }
414                 }
415
416                 if (text->compiled) {
417                         py_dict = PyC_DefaultNameSpace(fn_dummy);
418                         py_result =  PyEval_EvalCode(text->compiled, py_dict, py_dict);
419                 }
420
421         }
422         else {
423                 FILE *fp = BLI_fopen(fn, "r");
424
425                 if (fp) {
426                         py_dict = PyC_DefaultNameSpace(fn);
427
428 #ifdef _WIN32
429                         /* Previously we used PyRun_File to run directly the code on a FILE
430                          * object, but as written in the Python/C API Ref Manual, chapter 2,
431                          * 'FILE structs for different C libraries can be different and
432                          * incompatible'.
433                          * So now we load the script file data to a buffer */
434                         {
435                                 char *pystring;
436
437                                 fclose(fp);
438
439                                 pystring = MEM_mallocN(strlen(fn) + 37, "pystring");
440                                 pystring[0] = '\0';
441                                 sprintf(pystring, "f=open(r'%s');exec(f.read());f.close()", fn);
442                                 py_result = PyRun_String(pystring, Py_file_input, py_dict, py_dict);
443                                 MEM_freeN(pystring);
444                         }
445 #else
446                         py_result = PyRun_File(fp, fn, Py_file_input, py_dict, py_dict);
447                         fclose(fp);
448 #endif
449                 }
450                 else {
451                         PyErr_Format(PyExc_IOError,
452                                      "Python file \"%s\" could not be opened: %s",
453                                      fn, strerror(errno));
454                         py_result = NULL;
455                 }
456         }
457
458         if (!py_result) {
459                 if (text) {
460                         if (do_jump) {
461                                 /* ensure text is valid before use, the script may have freed its self */
462                                 Main *bmain_new = CTX_data_main(C);
463                                 if ((bmain_old == bmain_new) && (BLI_findindex(&bmain_new->text, text) != -1)) {
464                                         python_script_error_jump_text(text);
465                                 }
466                         }
467                 }
468                 BPy_errors_to_report(reports);
469         }
470         else {
471                 Py_DECREF(py_result);
472         }
473
474         if (py_dict) {
475 #ifdef PYMODULE_CLEAR_WORKAROUND
476                 PyModuleObject *mmod = (PyModuleObject *)PyDict_GetItemString(PyThreadState_GET()->interp->modules, "__main__");
477                 PyObject *dict_back = mmod->md_dict;
478                 /* freeing the module will clear the namespace,
479                  * gives problems running classes defined in this namespace being used later. */
480                 mmod->md_dict = NULL;
481                 Py_DECREF(dict_back);
482 #endif
483
484 #undef PYMODULE_CLEAR_WORKAROUND
485         }
486
487         PyC_MainModule_Restore(main_mod);
488
489         bpy_context_clear(C, &gilstate);
490
491         return (py_result != NULL);
492 }
493
494 /* Can run a file or text block */
495 int BPY_filepath_exec(bContext *C, const char *filepath, struct ReportList *reports)
496 {
497         return python_script_exec(C, filepath, NULL, reports, FALSE);
498 }
499
500
501 int BPY_text_exec(bContext *C, struct Text *text, struct ReportList *reports, const short do_jump)
502 {
503         return python_script_exec(C, NULL, text, reports, do_jump);
504 }
505
506 void BPY_DECREF(void *pyob_ptr)
507 {
508         PyGILState_STATE gilstate = PyGILState_Ensure();
509         Py_DECREF((PyObject *)pyob_ptr);
510         PyGILState_Release(gilstate);
511 }
512
513 void BPY_RNA_DECREF_INVALIDATE(void *pyob_ptr)
514 {
515         PyGILState_STATE gilstate = PyGILState_Ensure();
516         const int do_invalidate = (Py_REFCNT((PyObject *)pyob_ptr) > 1);
517         Py_DECREF((PyObject *)pyob_ptr);
518         if (do_invalidate) {
519                 pyrna_invalidate(pyob_ptr);
520         }
521         PyGILState_Release(gilstate);
522 }
523
524
525 /* return -1 on error, else 0 */
526 int BPY_button_exec(bContext *C, const char *expr, double *value, const short verbose)
527 {
528         PyGILState_STATE gilstate;
529         PyObject *py_dict, *mod, *retval;
530         int error_ret = 0;
531         PyObject *main_mod = NULL;
532         
533         if (!value || !expr) return -1;
534
535         if (expr[0] == '\0') {
536                 *value = 0.0;
537                 return error_ret;
538         }
539
540         bpy_context_set(C, &gilstate);
541
542         PyC_MainModule_Backup(&main_mod);
543
544         py_dict = PyC_DefaultNameSpace("<blender button>");
545
546         mod = PyImport_ImportModule("math");
547         if (mod) {
548                 PyDict_Merge(py_dict, PyModule_GetDict(mod), 0); /* 0 - don't overwrite existing values */
549                 Py_DECREF(mod);
550         }
551         else { /* highly unlikely but possibly */
552                 PyErr_Print();
553                 PyErr_Clear();
554         }
555         
556         retval = PyRun_String(expr, Py_eval_input, py_dict, py_dict);
557         
558         if (retval == NULL) {
559                 error_ret = -1;
560         }
561         else {
562                 double val;
563
564                 if (PyTuple_Check(retval)) {
565                         /* Users my have typed in 10km, 2m
566                          * add up all values */
567                         int i;
568                         val = 0.0;
569
570                         for (i = 0; i < PyTuple_GET_SIZE(retval); i++) {
571                                 const double val_item = PyFloat_AsDouble(PyTuple_GET_ITEM(retval, i));
572                                 if (val_item == -1 && PyErr_Occurred()) {
573                                         val = -1;
574                                         break;
575                                 }
576                                 val += val_item;
577                         }
578                 }
579                 else {
580                         val = PyFloat_AsDouble(retval);
581                 }
582                 Py_DECREF(retval);
583                 
584                 if (val == -1 && PyErr_Occurred()) {
585                         error_ret = -1;
586                 }
587                 else if (!finite(val)) {
588                         *value = 0.0;
589                 }
590                 else {
591                         *value = val;
592                 }
593         }
594         
595         if (error_ret) {
596                 if (verbose) {
597                         BPy_errors_to_report(CTX_wm_reports(C));
598                 }
599                 else {
600                         PyErr_Clear();
601                 }
602         }
603
604         PyC_MainModule_Backup(&main_mod);
605         
606         bpy_context_clear(C, &gilstate);
607         
608         return error_ret;
609 }
610
611 int BPY_string_exec(bContext *C, const char *expr)
612 {
613         PyGILState_STATE gilstate;
614         PyObject *main_mod = NULL;
615         PyObject *py_dict, *retval;
616         int error_ret = 0;
617         Main *bmain_back; /* XXX, quick fix for release (Copy Settings crash), needs further investigation */
618
619         if (!expr) return -1;
620
621         if (expr[0] == '\0') {
622                 return error_ret;
623         }
624
625         bpy_context_set(C, &gilstate);
626
627         PyC_MainModule_Backup(&main_mod);
628
629         py_dict = PyC_DefaultNameSpace("<blender string>");
630
631         bmain_back = bpy_import_main_get();
632         bpy_import_main_set(CTX_data_main(C));
633
634         retval = PyRun_String(expr, Py_eval_input, py_dict, py_dict);
635
636         bpy_import_main_set(bmain_back);
637
638         if (retval == NULL) {
639                 error_ret = -1;
640
641                 BPy_errors_to_report(CTX_wm_reports(C));
642         }
643         else {
644                 Py_DECREF(retval);
645         }
646
647         PyC_MainModule_Restore(main_mod);
648
649         bpy_context_clear(C, &gilstate);
650         
651         return error_ret;
652 }
653
654
655 void BPY_modules_load_user(bContext *C)
656 {
657         PyGILState_STATE gilstate;
658         Main *bmain = CTX_data_main(C);
659         Text *text;
660
661         /* can happen on file load */
662         if (bmain == NULL)
663                 return;
664
665         /* update pointers since this can run from a nested script
666          * on file load */
667         if (py_call_level) {
668                 BPY_context_update(C);
669         }
670
671         bpy_context_set(C, &gilstate);
672
673         for (text = bmain->text.first; text; text = text->id.next) {
674                 if (text->flags & TXT_ISSCRIPT && BLI_testextensie(text->id.name + 2, ".py")) {
675                         if (!(G.f & G_SCRIPT_AUTOEXEC)) {
676                                 printf("scripts disabled for \"%s\", skipping '%s'\n", bmain->name, text->id.name + 2);
677                         }
678                         else {
679                                 PyObject *module = bpy_text_import(text);
680
681                                 if (module == NULL) {
682                                         PyErr_Print();
683                                         PyErr_Clear();
684                                 }
685                                 else {
686                                         Py_DECREF(module);
687                                 }
688
689                                 /* check if the script loaded a new file */
690                                 if (bmain != CTX_data_main(C)) {
691                                         break;
692                                 }
693                         }
694                 }
695         }
696         bpy_context_clear(C, &gilstate);
697 }
698
699 int BPY_context_member_get(bContext *C, const char *member, bContextDataResult *result)
700 {
701         PyGILState_STATE gilstate;
702         int use_gil = !PYC_INTERPRETER_ACTIVE;
703
704         PyObject *pyctx;
705         PyObject *item;
706         PointerRNA *ptr = NULL;
707         int done = FALSE;
708
709         if (use_gil)
710                 gilstate = PyGILState_Ensure();
711
712         pyctx = (PyObject *)CTX_py_dict_get(C);
713         item = PyDict_GetItemString(pyctx, member);
714
715         if (item == NULL) {
716                 /* pass */
717         }
718         else if (item == Py_None) {
719                 /* pass */
720         }
721         else if (BPy_StructRNA_Check(item)) {
722                 ptr = &(((BPy_StructRNA *)item)->ptr);
723
724                 //result->ptr = ((BPy_StructRNA *)item)->ptr;
725                 CTX_data_pointer_set(result, ptr->id.data, ptr->type, ptr->data);
726                 done = TRUE;
727         }
728         else if (PySequence_Check(item)) {
729                 PyObject *seq_fast = PySequence_Fast(item, "bpy_context_get sequence conversion");
730                 if (seq_fast == NULL) {
731                         PyErr_Print();
732                         PyErr_Clear();
733                 }
734                 else {
735                         int len = PySequence_Fast_GET_SIZE(seq_fast);
736                         int i;
737                         for (i = 0; i < len; i++) {
738                                 PyObject *list_item = PySequence_Fast_GET_ITEM(seq_fast, i);
739
740                                 if (BPy_StructRNA_Check(list_item)) {
741 #if 0
742                                         CollectionPointerLink *link = MEM_callocN(sizeof(CollectionPointerLink), "bpy_context_get");
743                                         link->ptr = ((BPy_StructRNA *)item)->ptr;
744                                         BLI_addtail(&result->list, link);
745 #endif
746                                         ptr = &(((BPy_StructRNA *)list_item)->ptr);
747                                         CTX_data_list_add(result, ptr->id.data, ptr->type, ptr->data);
748                                 }
749                                 else {
750                                         printf("PyContext: '%s' list item not a valid type in sequece type '%s'\n",
751                                                member, Py_TYPE(item)->tp_name);
752                                 }
753
754                         }
755                         Py_DECREF(seq_fast);
756
757                         done = TRUE;
758                 }
759         }
760
761         if (done == 0) {
762                 if (item) printf("PyContext '%s' not a valid type\n", member);
763                 else      printf("PyContext '%s' not found\n", member);
764         }
765         else {
766                 if (G.debug & G_DEBUG_PYTHON) {
767                         printf("PyContext '%s' found\n", member);
768                 }
769         }
770
771         if (use_gil)
772                 PyGILState_Release(gilstate);
773
774         return done;
775 }
776
777 #ifdef WITH_PYTHON_MODULE
778 #include "BLI_fileops.h"
779 /* TODO, reloading the module isn't functional at the moment. */
780
781 static void bpy_module_free(void *mod);
782 extern int main_python_enter(int argc, const char **argv);
783 extern void main_python_exit(void);
784 static struct PyModuleDef bpy_proxy_def = {
785         PyModuleDef_HEAD_INIT,
786         "bpy",  /* m_name */
787         NULL,  /* m_doc */
788         0,  /* m_size */
789         NULL,  /* m_methods */
790         NULL,  /* m_reload */
791         NULL,  /* m_traverse */
792         NULL,  /* m_clear */
793         bpy_module_free,  /* m_free */
794 };
795
796 typedef struct {
797         PyObject_HEAD
798         /* Type-specific fields go here. */
799         PyObject *mod;
800 } dealloc_obj;
801
802 /* call once __file__ is set */
803 void bpy_module_delay_init(PyObject *bpy_proxy)
804 {
805         const int argc = 1;
806         const char *argv[2];
807
808         /* updating the module dict below will loose the reference to __file__ */
809         PyObject *filename_obj = PyModule_GetFilenameObject(bpy_proxy);
810
811         const char *filename_rel = _PyUnicode_AsString(filename_obj); /* can be relative */
812         char filename_abs[1024];
813
814         BLI_strncpy(filename_abs, filename_rel, sizeof(filename_abs));
815         BLI_path_cwd(filename_abs);
816
817         argv[0] = filename_abs;
818         argv[1] = NULL;
819         
820         // printf("module found %s\n", argv[0]);
821
822         main_python_enter(argc, argv);
823
824         /* initialized in BPy_init_modules() */
825         PyDict_Update(PyModule_GetDict(bpy_proxy), PyModule_GetDict(bpy_package_py));
826 }
827
828 static void dealloc_obj_dealloc(PyObject *self);
829
830 static PyTypeObject dealloc_obj_Type = {{{0}}};
831
832 /* use our own dealloc so we can free a property if we use one */
833 static void dealloc_obj_dealloc(PyObject *self)
834 {
835         bpy_module_delay_init(((dealloc_obj *)self)->mod);
836
837         /* Note, for subclassed PyObjects we cant just call PyObject_DEL() directly or it will crash */
838         dealloc_obj_Type.tp_free(self);
839 }
840
841 PyMODINIT_FUNC
842 PyInit_bpy(void)
843 {
844         PyObject *bpy_proxy = PyModule_Create(&bpy_proxy_def);
845         
846         /* Problem:
847          * 1) this init function is expected to have a private member defined - 'md_def'
848          *    but this is only set for C defined modules (not py packages)
849          *    so we cant return 'bpy_package_py' as is.
850          *
851          * 2) there is a 'bpy' C module for python to load which is basically all of blender,
852          *    and there is scripts/bpy/__init__.py, 
853          *    we may end up having to rename this module so there is no naming conflict here eg:
854          *    'from blender import bpy'
855          *
856          * 3) we don't know the filename at this point, workaround by assigning a dummy value
857          *    which calls back when its freed so the real loading can take place.
858          */
859
860         /* assign an object which is freed after __file__ is assigned */
861         dealloc_obj *dob;
862         
863         /* assign dummy type */
864         dealloc_obj_Type.tp_name = "dealloc_obj";
865         dealloc_obj_Type.tp_basicsize = sizeof(dealloc_obj);
866         dealloc_obj_Type.tp_dealloc = dealloc_obj_dealloc;
867         dealloc_obj_Type.tp_flags = Py_TPFLAGS_DEFAULT;
868         
869         if (PyType_Ready(&dealloc_obj_Type) < 0)
870                 return NULL;
871
872         dob = (dealloc_obj *) dealloc_obj_Type.tp_alloc(&dealloc_obj_Type, 0);
873         dob->mod = bpy_proxy; /* borrow */
874         PyModule_AddObject(bpy_proxy, "__file__", (PyObject *)dob); /* borrow */
875
876         return bpy_proxy;
877 }
878
879 static void bpy_module_free(void *UNUSED(mod))
880 {
881         main_python_exit();
882 }
883
884 #endif