source/blender/python/intern/bpy_interface.c

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