17802ec3a293d5b9fcf10e602836d7e22395ac8c
[blender.git] / source / blender / python / intern / bpy_interface.c
1
2 #include <stdio.h>
3 #include <stdlib.h>
4
5 #include <Python.h>
6 #include "compile.h"            /* for the PyCodeObject */
7 #include "eval.h"               /* for PyEval_EvalCode */
8
9 #include "BKE_context.h"
10
11 #include "bpy_compat.h"
12
13 #include "bpy_rna.h"
14 #include "bpy_operator.h"
15
16
17 /*****************************************************************************
18 * Description: This function creates a new Python dictionary object.
19 *****************************************************************************/
20
21 static PyObject *CreateGlobalDictionary( bContext *C )
22 {
23         PyObject *dict = PyDict_New(  );
24         PyObject *item = PyUnicode_FromString( "__main__" );
25         PyDict_SetItemString( dict, "__builtins__", PyEval_GetBuiltins(  ) );
26         PyDict_SetItemString( dict, "__name__", item );
27         Py_DECREF(item);
28         
29         /* Add Modules */
30         item = BPY_rna_module();
31         PyDict_SetItemString( dict, "bpy", item );
32         Py_DECREF(item);
33         
34         item = BPY_rna_doc();
35         PyDict_SetItemString( dict, "bpydoc", item );
36         Py_DECREF(item);
37
38         item = BPY_operator_module(C);
39         PyDict_SetItemString( dict, "bpyoperator", item );
40         Py_DECREF(item);
41
42         
43         // XXX very experemental, consiter this a test, especiall PyCObject is not meant to be perminant
44         item = BPY_ui_module();
45         PyDict_SetItemString( dict, "bpyui", item );
46         Py_DECREF(item);
47         
48         // XXX - evil, need to access context
49         item = PyCObject_FromVoidPtr( C, NULL );
50         PyDict_SetItemString( dict, "__bpy_context__", item );
51         Py_DECREF(item);
52         
53         return dict;
54 }
55
56 void BPY_start_python( void )
57 {
58         PyThreadState *py_tstate = NULL;
59
60         Py_Initialize(  );
61         
62         //PySys_SetArgv( argc_copy, argv_copy );
63         
64         /* Initialize thread support (also acquires lock) */
65         PyEval_InitThreads();
66         
67         // todo - sys paths - our own imports
68         
69         py_tstate = PyGILState_GetThisThreadState();
70         PyEval_ReleaseThread(py_tstate);
71         
72 }
73
74 void BPY_end_python( void )
75 {
76         PyGILState_Ensure(); /* finalizing, no need to grab the state */
77         
78         // free other python data.
79         
80         Py_Finalize(  );
81         return;
82 }
83
84 void BPY_run_python_script( bContext *C, const char *fn )
85 {
86         PyObject *py_dict, *py_result;
87         char pystring[512];
88         PyGILState_STATE gilstate;
89
90         /* TODO - look into a better way to run a file */
91         sprintf(pystring, "exec(open(r'%s').read())", fn);      
92         
93         //BPY_start_python();
94         
95         gilstate = PyGILState_Ensure();
96         
97         py_dict = CreateGlobalDictionary(C);
98         
99         py_result = PyRun_String( pystring, Py_file_input, py_dict, py_dict );
100         
101         if (!py_result)
102                 PyErr_Print();
103         else
104                 Py_DECREF( py_result );
105         
106         PyGILState_Release(gilstate);
107         
108         //BPY_end_python();
109 }