Merge with 2.5 -r 21619:21756.
[blender.git] / source / blender / python / intern / bpy_operator.c
index 0da0782c2d1c73a87066c6ae6269861ed5ce9bd7..240ce2030bae0cb18fb386e06f52920481b091f4 100644 (file)
  * ***** END GPL LICENSE BLOCK *****
  */
 
+/* Note, this module is not to be used directly by the user.
+ * its accessed from blender with bpy.__ops__
+ * */
+
 #include "bpy_operator.h"
+#include "bpy_operator_wrap.h"
+#include "bpy_rna.h" /* for setting arg props only - pyrna_py_to_prop() */
 #include "bpy_compat.h"
-#include "bpy_idprop.h"
+#include "bpy_util.h"
 
-//#include "blendef.h"
-#include "BLI_dynstr.h"
 #include "WM_api.h"
 #include "WM_types.h"
 
 #include "MEM_guardedalloc.h"
-#include "BKE_idprop.h"
-
-extern ListBase global_ops; /* evil, temp use */
-
-/* floats bigger then this are displayed as inf in the docstrings */
-#define MAXFLOAT_DOC 10000000
-
-static int pyop_func_compare( BPy_OperatorFunc * a, BPy_OperatorFunc * b )
-{
-       return (strcmp(a->name, b->name)==0) ? 0 : -1;
-}
+#include "BKE_report.h"
 
-/*----------------------repr--------------------------------------------*/
-static PyObject *pyop_base_repr( BPy_OperatorBase * self )
-{
-       return PyUnicode_FromFormat( "[BPy_OperatorBase]");
-}
 
-static PyObject *pyop_func_repr( BPy_OperatorFunc * self )
+/* 'self' stores the operator string */
+static PyObject *pyop_call( PyObject * self, PyObject * args)
 {
-       return PyUnicode_FromFormat( "[BPy_OperatorFunc \"%s\"]", self->name);
-}
-
-//---------------getattr--------------------------------------------
-static PyObject *pyop_base_getattro( BPy_OperatorBase * self, PyObject *pyname )
-{
-       char *name = _PyUnicode_AsString(pyname);
-       PyObject *ret;
        wmOperatorType *ot;
+       int error_val = 0;
+       PointerRNA ptr;
+       
+       char *opname;
+       PyObject *kw= NULL;
 
-       if( strcmp( name, "__members__" ) == 0 ) {
-               PyObject *item;
+       // XXX Todo, work out a better solution for passing on context, could make a tuple from self and pack the name and Context into it...
+       bContext *C = BPy_GetContext();
+       
 
-               ret = PyList_New(0);
+       switch(PyTuple_Size(args)) {
+       case 2:
+               kw = PyTuple_GET_ITEM(args, 1);
 
-               for(ot= WM_operatortype_first(); ot; ot= ot->next) {
-                       item = PyUnicode_FromString( ot->idname );
-                       PyList_Append(ret, item);
-                       Py_DECREF(item);
+               if(!PyDict_Check(kw)) {
+                       PyErr_SetString( PyExc_AttributeError, "bpy.__ops__.call: expected second arg to be a dict");
+                       return NULL;
                }
-       } else {
-               ot = WM_operatortype_find(name);
+               /* pass through */
+       case 1:
+               opname = _PyUnicode_AsString(PyTuple_GET_ITEM(args, 0));
 
-               if (ot) {
-                       return pyop_func_CreatePyObject(self->C, name);
-               }
-               else {
-                       PyErr_Format( PyExc_AttributeError, "Operator \"%s\" not found", name);
+               if(opname==NULL) {
+                       PyErr_SetString( PyExc_AttributeError, "bpy.__ops__.call: expected the first arg to be a string");
+                       return NULL;
                }
-       }
-       
-       return ret;
-}
-
-static PyObject * pyop_func_call(BPy_OperatorFunc * self, PyObject *args, PyObject *kw)
-{
-       IDProperty *properties = NULL;
-
-       if (PyTuple_Size(args)) {
-               PyErr_SetString( PyExc_AttributeError, "All operator args must be keywords");
+               break;
+       default:
+               PyErr_SetString( PyExc_AttributeError, "bpy.__ops__.call: expected a string and optional dict");
                return NULL;
        }
 
-       if (kw && PyDict_Size(kw) > 0) {
-               IDPropertyTemplate val;
-               val.i = 0; /* silence MSVC warning about uninitialized var when debugging */
 
-               properties= IDP_New(IDP_GROUP, val, "property");
-               BPy_IDGroup_Update(properties, kw);
+       ot= WM_operatortype_find(opname, 1);
 
-               if (PyErr_Occurred()) {
-                       IDP_FreeProperty(properties);
-                       MEM_freeN(properties);
-                       return NULL;
-               }
+       if (ot == NULL) {
+               PyErr_Format( PyExc_SystemError, "bpy.__ops__.call: operator \"%s\"could not be found", opname);
+               return NULL;
        }
        
-       WM_operator_name_call(self->C, self->name, WM_OP_DEFAULT, properties);
-
-       if (properties) {
-               IDP_FreeProperty(properties);
-               MEM_freeN(properties);
+       if(ot->poll && (ot->poll(C) == 0)) {
+               PyErr_SetString( PyExc_SystemError, "bpy.__ops__.call: operator poll() function failed, context is incorrect");
+               return NULL;
        }
-       Py_RETURN_NONE;
-}
-
-/*-----------------------BPy_OperatorBase method def------------------------------*/
-PyTypeObject pyop_base_Type = {
-#if (PY_VERSION_HEX >= 0x02060000)
-       PyVarObject_HEAD_INIT(&PyType_Type, 0)
-#else
-       /* python 2.5 and below */
-       PyObject_HEAD_INIT( NULL )  /* required py macro */
-       0,                          /* ob_size */
-#endif
-
-       "Operator",             /* tp_name */
-       sizeof( BPy_OperatorBase ),                     /* tp_basicsize */
-       0,                      /* tp_itemsize */
-       /* methods */
-       NULL,                                           /* tp_dealloc */
-       NULL,                       /* printfunc tp_print; */
-       NULL,                                           /* getattrfunc tp_getattr; */
-       NULL,                       /* setattrfunc tp_setattr; */
-       NULL,                                           /* tp_compare */
-       ( reprfunc ) pyop_base_repr,    /* tp_repr */
-
-       /* Method suites for standard classes */
-
-       NULL,                       /* PyNumberMethods *tp_as_number; */
-       NULL,                                           /* PySequenceMethods *tp_as_sequence; */
-       NULL,                                           /* PyMappingMethods *tp_as_mapping; */
-
-       /* More standard operations (here for binary compatibility) */
-
-       NULL,                                           /* hashfunc tp_hash; */
-       NULL,                       /* ternaryfunc tp_call; */
-       NULL,                       /* reprfunc tp_str; */
-       ( getattrofunc )pyop_base_getattro, /*PyObject_GenericGetAttr - MINGW Complains, assign later */        /* getattrofunc tp_getattro; */
-       NULL, /*PyObject_GenericSetAttr - MINGW Complains, assign later */      /* setattrofunc tp_setattro; */
-
-       /* Functions to access object as input/output buffer */
-       NULL,                       /* PyBufferProcs *tp_as_buffer; */
+       
+       /* WM_operator_properties_create(&ptr, opname); */
+       /* Save another lookup */
+       RNA_pointer_create(NULL, ot->srna, NULL, &ptr);
+       
+       if(kw && PyDict_Size(kw))
+               error_val= pyrna_pydict_to_props(&ptr, kw, "Converting py args to operator properties: ");
 
-  /*** Flags to define presence of optional/expanded features ***/
-       Py_TPFLAGS_DEFAULT,         /* long tp_flags; */
+       
+       if (error_val==0) {
+               ReportList reports;
 
-       NULL,                                           /*  char *tp_doc;  Documentation string */
-  /*** Assigned meaning in release 2.0 ***/
-       /* call function for all accessible objects */
-       NULL,                       /* traverseproc tp_traverse; */
+               BKE_reports_init(&reports, RPT_STORE);
 
-       /* delete references to contained objects */
-       NULL,                       /* inquiry tp_clear; */
+               WM_operator_call_py(C, ot, &ptr, &reports);
 
-  /***  Assigned meaning in release 2.1 ***/
-  /*** rich comparisons ***/
-       NULL,                       /* richcmpfunc tp_richcompare; */
+               if(BPy_reports_to_error(&reports))
+                       error_val = -1;
 
-  /***  weak reference enabler ***/
-       0,                          /* long tp_weaklistoffset; */
+               BKE_reports_clear(&reports);
+       }
 
-  /*** Added in release 2.2 ***/
-       /*   Iterators */
-       NULL,                                           /* getiterfunc tp_iter; */
-       NULL,                       /* iternextfunc tp_iternext; */
+       WM_operator_properties_free(&ptr);
 
-  /*** Attribute descriptor and subclassing stuff ***/
-       NULL,                                           /* struct PyMethodDef *tp_methods; */
-       NULL,                       /* struct PyMemberDef *tp_members; */
-       NULL,                                           /* struct PyGetSetDef *tp_getset; */
-       NULL,                       /* struct _typeobject *tp_base; */
-       NULL,                       /* PyObject *tp_dict; */
-       NULL,                       /* descrgetfunc tp_descr_get; */
-       NULL,                       /* descrsetfunc tp_descr_set; */
-       0,                          /* long tp_dictoffset; */
-       NULL,                       /* initproc tp_init; */
-       NULL,                       /* allocfunc tp_alloc; */
-       NULL,                                           /* newfunc tp_new; */
-       /*  Low-level free-memory routine */
-       NULL,                       /* freefunc tp_free;  */
-       /* For PyObject_IS_GC */
-       NULL,                       /* inquiry tp_is_gc;  */
-       NULL,                       /* PyObject *tp_bases; */
-       /* method resolution order */
-       NULL,                       /* PyObject *tp_mro;  */
-       NULL,                       /* PyObject *tp_cache; */
-       NULL,                       /* PyObject *tp_subclasses; */
-       NULL,                       /* PyObject *tp_weaklist; */
-       NULL
-};
+#if 0
+       /* if there is some way to know an operator takes args we should use this */
+       {
+               /* no props */
+               if (kw != NULL) {
+                       PyErr_Format(PyExc_AttributeError, "Operator \"%s\" does not take any args", opname);
+                       return NULL;
+               }
 
-/*-----------------------BPy_OperatorBase method def------------------------------*/
-PyTypeObject pyop_func_Type = {
-#if (PY_VERSION_HEX >= 0x02060000)
-       PyVarObject_HEAD_INIT(&PyType_Type, 0)
-#else
-       /* python 2.5 and below */
-       PyObject_HEAD_INIT( NULL )  /* required py macro */
-       0,                          /* ob_size */
+               WM_operator_name_call(C, opname, WM_OP_EXEC_DEFAULT, NULL);
+       }
 #endif
 
-       "OperatorFunc",         /* tp_name */
-       sizeof( BPy_OperatorFunc ),                     /* tp_basicsize */
-       0,                      /* tp_itemsize */
-       /* methods */
-       NULL,                                           /* tp_dealloc */
-       NULL,                       /* printfunc tp_print; */
-       NULL,                                           /* getattrfunc tp_getattr; */
-       NULL,                       /* setattrfunc tp_setattr; */
-       ( cmpfunc ) pyop_func_compare,  /* tp_compare */
-       ( reprfunc ) pyop_func_repr,    /* tp_repr */
-
-       /* Method suites for standard classes */
-
-       NULL,                       /* PyNumberMethods *tp_as_number; */
-       NULL,                                           /* PySequenceMethods *tp_as_sequence; */
-       NULL,                                           /* PyMappingMethods *tp_as_mapping; */
-
-       /* More standard operations (here for binary compatibility) */
-
-       NULL,                                           /* hashfunc tp_hash; */
-       (ternaryfunc)pyop_func_call,                       /* ternaryfunc tp_call; */
-       NULL,                       /* reprfunc tp_str; */
-       NULL, /*PyObject_GenericGetAttr - MINGW Complains, assign later */      /* getattrofunc tp_getattro; */
-       NULL, /*PyObject_GenericSetAttr - MINGW Complains, assign later */      /* setattrofunc tp_setattro; */
-
-       /* Functions to access object as input/output buffer */
-       NULL,                       /* PyBufferProcs *tp_as_buffer; */
-
-  /*** Flags to define presence of optional/expanded features ***/
-       Py_TPFLAGS_DEFAULT,         /* long tp_flags; */
-
-       NULL,                                           /*  char *tp_doc;  Documentation string */
-  /*** Assigned meaning in release 2.0 ***/
-       /* call function for all accessible objects */
-       NULL,                       /* traverseproc tp_traverse; */
-
-       /* delete references to contained objects */
-       NULL,                       /* inquiry tp_clear; */
-
-  /***  Assigned meaning in release 2.1 ***/
-  /*** rich comparisons ***/
-       NULL,                       /* richcmpfunc tp_richcompare; */
-
-  /***  weak reference enabler ***/
-       0,                          /* long tp_weaklistoffset; */
-
-  /*** Added in release 2.2 ***/
-       /*   Iterators */
-       NULL,                                           /* getiterfunc tp_iter; */
-       NULL,                       /* iternextfunc tp_iternext; */
-
-  /*** Attribute descriptor and subclassing stuff ***/
-       NULL,                                           /* struct PyMethodDef *tp_methods; */
-       NULL,                       /* struct PyMemberDef *tp_members; */
-       NULL,                                           /* struct PyGetSetDef *tp_getset; */
-       NULL,                       /* struct _typeobject *tp_base; */
-       NULL,                       /* PyObject *tp_dict; */
-       NULL,                       /* descrgetfunc tp_descr_get; */
-       NULL,                       /* descrsetfunc tp_descr_set; */
-       0,                          /* long tp_dictoffset; */
-       NULL,                       /* initproc tp_init; */
-       NULL,                       /* allocfunc tp_alloc; */
-       NULL,                                           /* newfunc tp_new; */
-       /*  Low-level free-memory routine */
-       NULL,                       /* freefunc tp_free;  */
-       /* For PyObject_IS_GC */
-       NULL,                       /* inquiry tp_is_gc;  */
-       NULL,                       /* PyObject *tp_bases; */
-       /* method resolution order */
-       NULL,                       /* PyObject *tp_mro;  */
-       NULL,                       /* PyObject *tp_cache; */
-       NULL,                       /* PyObject *tp_subclasses; */
-       NULL,                       /* PyObject *tp_weaklist; */
-       NULL
-};
-
-PyObject *pyop_base_CreatePyObject( bContext *C )
-{
-       BPy_OperatorBase *pyop;
-
-       pyop = ( BPy_OperatorBase * ) PyObject_NEW( BPy_OperatorBase, &pyop_base_Type );
-
-       if( !pyop ) {
-               PyErr_SetString( PyExc_MemoryError, "couldn't create BPy_OperatorBase object" );
+       if (error_val==-1) {
                return NULL;
        }
 
-       pyop->C = C; /* TODO - copy this? */
-
-       return ( PyObject * ) pyop;
+       Py_RETURN_NONE;
 }
 
-PyObject *pyop_func_CreatePyObject( bContext *C, char *name )
+static PyObject *pyop_dir(PyObject *self)
 {
-       BPy_OperatorFunc *pyop;
-
-       pyop = ( BPy_OperatorFunc * ) PyObject_NEW( BPy_OperatorFunc, &pyop_func_Type );
-
-       if( !pyop ) {
-               PyErr_SetString( PyExc_MemoryError, "couldn't create BPy_OperatorFunc object" );
-               return NULL;
+       PyObject *list = PyList_New(0), *name;
+       wmOperatorType *ot;
+       
+       for(ot= WM_operatortype_first(); ot; ot= ot->next) {
+               name = PyUnicode_FromString(ot->idname);
+               PyList_Append(list, name);
+               Py_DECREF(name);
        }
-
-       strcpy(pyop->name, name);
-       pyop->C= C; /* TODO - how should contexts be dealt with? */
-
-       return ( PyObject * ) pyop;
+       
+       return list;
 }
 
-PyObject *BPY_operator_module( bContext *C )
+PyObject *BPY_operator_module( void )
 {
-       if( PyType_Ready( &pyop_base_Type ) < 0 )
-               return NULL;
-
-       if( PyType_Ready( &pyop_func_Type ) < 0 )
-               return NULL;
-
-       //submodule = Py_InitModule3( "operator", M_rna_methods, "rna module" );
-       return pyop_base_CreatePyObject(C);
-}
+       static PyMethodDef pyop_call_meth =             {"call", (PyCFunction) pyop_call, METH_VARARGS, NULL};
+       static PyMethodDef pyop_dir_meth =              {"dir", (PyCFunction) pyop_dir, METH_NOARGS, NULL};
+       static PyMethodDef pyop_add_meth =              {"add", (PyCFunction) PYOP_wrap_add, METH_O, NULL};
+       static PyMethodDef pyop_remove_meth =   {"remove", (PyCFunction) PYOP_wrap_remove, METH_O, NULL};
 
+       PyObject *submodule = PyModule_New("bpy.__ops__");
+       PyDict_SetItemString(PySys_GetObject("modules"), "bpy.__ops__", submodule);
 
+       PyModule_AddObject( submodule, "call",  PyCFunction_New(&pyop_call_meth,        NULL) );
+       PyModule_AddObject( submodule, "dir",           PyCFunction_New(&pyop_dir_meth,         NULL) );
+       PyModule_AddObject( submodule, "add",           PyCFunction_New(&pyop_add_meth,         NULL) );
+       PyModule_AddObject( submodule, "remove",        PyCFunction_New(&pyop_remove_meth,      NULL) );
 
+       return submodule;
+}