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:
		@@ -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
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user