pyrna - add own callable function type rather then using a standard python method, gives small speedup drawing buttons since every layout.prop/col/operator/menu etc creates and throws away one of these.
This commit is contained in:
parent
db319f8544
commit
85fe36ab61
@ -908,6 +908,13 @@ static PyObject *pyrna_prop_repr(BPy_PropertyRNA *self)
|
|||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static PyObject *pyrna_func_repr(BPy_FunctionRNA *self)
|
||||||
|
{
|
||||||
|
return PyUnicode_FromFormat("<%.200s %.200s.%.200s()>", Py_TYPE(self)->tp_name, RNA_struct_identifier(self->ptr.type), RNA_function_identifier(self->func));
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
static long pyrna_struct_hash(BPy_StructRNA *self)
|
static long pyrna_struct_hash(BPy_StructRNA *self)
|
||||||
{
|
{
|
||||||
return _Py_HashPointer(self->ptr.data);
|
return _Py_HashPointer(self->ptr.data);
|
||||||
@ -1344,36 +1351,16 @@ int pyrna_pydict_to_props(PointerRNA *ptr, PyObject *kw, int all_args, const cha
|
|||||||
return error_val;
|
return error_val;
|
||||||
}
|
}
|
||||||
|
|
||||||
static PyObject *pyrna_func_call(PyObject *self, PyObject *args, PyObject *kw);
|
|
||||||
|
|
||||||
static PyObject *pyrna_func_to_py(BPy_DummyPointerRNA *pyrna, FunctionRNA *func)
|
static PyObject *pyrna_func_to_py(PointerRNA *ptr, FunctionRNA *func)
|
||||||
{
|
{
|
||||||
static PyMethodDef func_meth= {"<generic rna function>", (PyCFunction)pyrna_func_call, METH_VARARGS|METH_KEYWORDS, "python rna function"};
|
BPy_FunctionRNA* pyfunc= (BPy_FunctionRNA *) PyObject_NEW(BPy_FunctionRNA, &pyrna_func_Type);
|
||||||
PyObject *self;
|
pyfunc->ptr= *ptr;
|
||||||
PyObject *ret;
|
pyfunc->func= func;
|
||||||
|
return (PyObject *)pyfunc;
|
||||||
if(func==NULL) {
|
|
||||||
PyErr_Format(PyExc_RuntimeError,
|
|
||||||
"%.200s: type attempted to get NULL function",
|
|
||||||
RNA_struct_identifier(pyrna->ptr.type));
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
self= PyTuple_New(2);
|
|
||||||
|
|
||||||
PyTuple_SET_ITEM(self, 0, (PyObject *)pyrna);
|
|
||||||
Py_INCREF(pyrna);
|
|
||||||
|
|
||||||
PyTuple_SET_ITEM(self, 1, PyCapsule_New((void *)func, NULL, NULL));
|
|
||||||
|
|
||||||
ret= PyCFunction_New(&func_meth, self);
|
|
||||||
Py_DECREF(self);
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
static int pyrna_py_to_prop(PointerRNA *ptr, PropertyRNA *prop, void *data, PyObject *value, const char *error_prefix)
|
static int pyrna_py_to_prop(PointerRNA *ptr, PropertyRNA *prop, void *data, PyObject *value, const char *error_prefix)
|
||||||
{
|
{
|
||||||
/* XXX hard limits should be checked here */
|
/* XXX hard limits should be checked here */
|
||||||
@ -3001,7 +2988,7 @@ static PyObject *pyrna_struct_getattro(BPy_StructRNA *self, PyObject *pyname)
|
|||||||
}
|
}
|
||||||
/* RNA function only if callback is declared (no optional functions) */
|
/* RNA function only if callback is declared (no optional functions) */
|
||||||
else if ((func= RNA_struct_find_function(&self->ptr, name)) && RNA_function_defined(func)) {
|
else if ((func= RNA_struct_find_function(&self->ptr, name)) && RNA_function_defined(func)) {
|
||||||
ret= pyrna_func_to_py((BPy_DummyPointerRNA *)self, func);
|
ret= pyrna_func_to_py(&self->ptr, func);
|
||||||
}
|
}
|
||||||
else if (self->ptr.type == &RNA_Context) {
|
else if (self->ptr.type == &RNA_Context) {
|
||||||
bContext *C= self->ptr.data;
|
bContext *C= self->ptr.data;
|
||||||
@ -3303,7 +3290,7 @@ static PyObject *pyrna_prop_collection_getattro(BPy_PropertyRNA *self, PyObject
|
|||||||
}
|
}
|
||||||
else if ((func= RNA_struct_find_function(&r_ptr, name))) {
|
else if ((func= RNA_struct_find_function(&r_ptr, name))) {
|
||||||
PyObject *self_collection= pyrna_struct_CreatePyObject(&r_ptr);
|
PyObject *self_collection= pyrna_struct_CreatePyObject(&r_ptr);
|
||||||
ret= pyrna_func_to_py((BPy_DummyPointerRNA *)self_collection, func);
|
ret= pyrna_func_to_py(&((BPy_DummyPointerRNA *)self_collection)->ptr, func);
|
||||||
Py_DECREF(self_collection);
|
Py_DECREF(self_collection);
|
||||||
|
|
||||||
return ret;
|
return ret;
|
||||||
@ -4257,11 +4244,11 @@ static PyObject *pyrna_param_to_py(PointerRNA *ptr, PropertyRNA *prop, void *dat
|
|||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
static PyObject *pyrna_func_call(PyObject *self, PyObject *args, PyObject *kw)
|
static PyObject *pyrna_func_call(BPy_FunctionRNA *self, PyObject *args, PyObject *kw)
|
||||||
{
|
{
|
||||||
/* Note, both BPy_StructRNA and BPy_PropertyRNA can be used here */
|
/* Note, both BPy_StructRNA and BPy_PropertyRNA can be used here */
|
||||||
PointerRNA *self_ptr= &(((BPy_DummyPointerRNA *)PyTuple_GET_ITEM(self, 0))->ptr);
|
PointerRNA *self_ptr= &self->ptr;
|
||||||
FunctionRNA *self_func= PyCapsule_GetPointer(PyTuple_GET_ITEM(self, 1), NULL);
|
FunctionRNA *self_func= self->func;
|
||||||
|
|
||||||
PointerRNA funcptr;
|
PointerRNA funcptr;
|
||||||
ParameterList parms;
|
ParameterList parms;
|
||||||
@ -5045,6 +5032,91 @@ static PyTypeObject pyrna_prop_collection_idprop_Type= {
|
|||||||
NULL
|
NULL
|
||||||
};
|
};
|
||||||
|
|
||||||
|
/*-----------------------BPy_PropertyRNA method def------------------------------*/
|
||||||
|
PyTypeObject pyrna_func_Type= {
|
||||||
|
PyVarObject_HEAD_INIT(NULL, 0)
|
||||||
|
"bpy_func", /* tp_name */
|
||||||
|
sizeof(BPy_FunctionRNA), /* tp_basicsize */
|
||||||
|
0, /* tp_itemsize */
|
||||||
|
/* methods */
|
||||||
|
NULL, /* tp_dealloc */
|
||||||
|
NULL, /* printfunc tp_print; */
|
||||||
|
NULL, /* getattrfunc tp_getattr; */
|
||||||
|
NULL, /* setattrfunc tp_setattr; */
|
||||||
|
NULL, /* tp_compare */ /* DEPRECATED in python 3.0! */
|
||||||
|
(reprfunc) pyrna_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)pyrna_func_call, /* ternaryfunc tp_call; */
|
||||||
|
NULL, /* reprfunc tp_str; */
|
||||||
|
|
||||||
|
/* will only use these if this is a subtype of a py class */
|
||||||
|
NULL, /* getattrofunc tp_getattro; */
|
||||||
|
NULL, /* 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 ***/
|
||||||
|
#ifdef USE_WEAKREFS
|
||||||
|
offsetof(BPy_PropertyRNA, in_weakreflist), /* long tp_weaklistoffset; */
|
||||||
|
#else
|
||||||
|
0,
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/*** 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
|
||||||
|
};
|
||||||
|
|
||||||
#ifdef USE_PYRNA_ITER
|
#ifdef USE_PYRNA_ITER
|
||||||
/* --- collection iterator: start --- */
|
/* --- collection iterator: start --- */
|
||||||
/* wrap rna collection iterator functions */
|
/* wrap rna collection iterator functions */
|
||||||
@ -5516,6 +5588,9 @@ void BPY_rna_init(void)
|
|||||||
if(PyType_Ready(&pyrna_prop_collection_idprop_Type) < 0)
|
if(PyType_Ready(&pyrna_prop_collection_idprop_Type) < 0)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
|
if(PyType_Ready(&pyrna_func_Type) < 0)
|
||||||
|
return;
|
||||||
|
|
||||||
#ifdef USE_PYRNA_ITER
|
#ifdef USE_PYRNA_ITER
|
||||||
if(PyType_Ready(&pyrna_prop_collection_iter_Type) < 0)
|
if(PyType_Ready(&pyrna_prop_collection_iter_Type) < 0)
|
||||||
return;
|
return;
|
||||||
|
@ -71,6 +71,7 @@ extern PyTypeObject pyrna_struct_Type;
|
|||||||
extern PyTypeObject pyrna_prop_Type;
|
extern PyTypeObject pyrna_prop_Type;
|
||||||
extern PyTypeObject pyrna_prop_array_Type;
|
extern PyTypeObject pyrna_prop_array_Type;
|
||||||
extern PyTypeObject pyrna_prop_collection_Type;
|
extern PyTypeObject pyrna_prop_collection_Type;
|
||||||
|
extern PyTypeObject pyrna_func_Type;
|
||||||
|
|
||||||
#define BPy_StructRNA_Check(v) (PyObject_TypeCheck(v, &pyrna_struct_Type))
|
#define BPy_StructRNA_Check(v) (PyObject_TypeCheck(v, &pyrna_struct_Type))
|
||||||
#define BPy_StructRNA_CheckExact(v) (Py_TYPE(v) == &pyrna_struct_Type)
|
#define BPy_StructRNA_CheckExact(v) (Py_TYPE(v) == &pyrna_struct_Type)
|
||||||
@ -142,6 +143,15 @@ typedef struct {
|
|||||||
CollectionPropertyIterator iter;
|
CollectionPropertyIterator iter;
|
||||||
} BPy_PropertyCollectionIterRNA;
|
} BPy_PropertyCollectionIterRNA;
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
PyObject_HEAD /* required python macro */
|
||||||
|
#ifdef USE_WEAKREFS
|
||||||
|
PyObject *in_weakreflist;
|
||||||
|
#endif
|
||||||
|
PointerRNA ptr;
|
||||||
|
FunctionRNA *func;
|
||||||
|
} BPy_FunctionRNA;
|
||||||
|
|
||||||
/* cheap trick */
|
/* cheap trick */
|
||||||
#define BPy_BaseTypeRNA BPy_PropertyRNA
|
#define BPy_BaseTypeRNA BPy_PropertyRNA
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user