Bugfix for #5000
Setup for Armature weak ref list was missing from some places where we execute py code. This confused the interpreter and gave random attribute/tuple parse errors. Changed name of weak ref list to "__arm_weakrefs" to avoid name collision with user variables.
This commit is contained in:
@@ -1348,36 +1348,38 @@ PyObject *Armature_RebuildBones(PyObject *pyarmature)
|
||||
PyObject *PyArmature_FromArmature(struct bArmature *armature)
|
||||
{
|
||||
BPy_Armature *py_armature = NULL;
|
||||
PyObject *maindict = NULL, *armdict = NULL, *weakref = NULL;
|
||||
PyObject *maindict = NULL, *weakref = NULL;
|
||||
PyObject *armlist = NULL; /* list of armature weak refs */
|
||||
char *list_name = ARM_WEAKREF_LIST_NAME;
|
||||
|
||||
//create armature type
|
||||
py_armature = (BPy_Armature*)PyObject_New(BPy_Armature, &Armature_Type); //Armature_Type.tp_alloc(&Armature_Type, 0); //*new*
|
||||
py_armature->weaklist = NULL; //init the weaklist
|
||||
if (!py_armature)
|
||||
py_armature = (BPy_Armature*)Armature_Type.tp_alloc(&Armature_Type, 0); /*new*/
|
||||
if (!py_armature){
|
||||
printf("Oops - can't create py armature\n");
|
||||
goto RuntimeError;
|
||||
}
|
||||
|
||||
py_armature->weaklist = NULL; //init the weaklist
|
||||
py_armature->armature = armature;
|
||||
|
||||
//create armature.bones
|
||||
py_armature->Bones = (BPy_BonesDict*)PyBonesDict_FromPyArmature(py_armature);
|
||||
if (!py_armature->Bones)
|
||||
goto RuntimeError;
|
||||
|
||||
//put a weakreference in __main__
|
||||
maindict= PyModule_GetDict(PyImport_AddModule( "__main__"));
|
||||
|
||||
armdict = PyDict_GetItemString(maindict, "armatures");
|
||||
/*armature list doesn't exist for pydrivers. . .so check.
|
||||
by the way, why is the var called armatures? what if a script author has
|
||||
a similar var in his script? won't they conflict?.
|
||||
|
||||
joeedh*/
|
||||
if (!armdict) return (PyObject *) py_armature;
|
||||
|
||||
weakref = PyWeakref_NewProxy((PyObject*)py_armature, Py_None);
|
||||
if (PyList_Append(armdict, weakref) == -1){
|
||||
if (!py_armature->Bones){
|
||||
printf("Oops - creating armature.bones\n");
|
||||
goto RuntimeError;
|
||||
}
|
||||
|
||||
//put a weakreference in __main__
|
||||
maindict= PyModule_GetDict(PyImport_AddModule( "__main__"));
|
||||
|
||||
armlist = PyDict_GetItemString(maindict, list_name);
|
||||
if( armlist){
|
||||
weakref = PyWeakref_NewProxy((PyObject*)py_armature, Py_None);
|
||||
if (PyList_Append(armlist, weakref) == -1){
|
||||
printf("Oops - list-append failed\n");
|
||||
goto RuntimeError;
|
||||
}
|
||||
}
|
||||
return (PyObject *) py_armature;
|
||||
|
||||
RuntimeError:
|
||||
|
||||
@@ -884,18 +884,42 @@ static PyObject *M_Window_GetPerspMatrix( PyObject * self )
|
||||
return perspmat;
|
||||
}
|
||||
|
||||
|
||||
/* update_armature_weakrefs()
|
||||
* helper function used in M_Window_EditMode.
|
||||
* rebuilds list of Armature weakrefs in __main__
|
||||
*/
|
||||
|
||||
static int update_armature_weakrefs()
|
||||
{
|
||||
/* stuff for armature weak refs */
|
||||
char *list_name = ARM_WEAKREF_LIST_NAME;
|
||||
PyObject *maindict = NULL, *armlist = NULL;
|
||||
PyObject *pyarmature = NULL;
|
||||
int x;
|
||||
|
||||
maindict= PyModule_GetDict(PyImport_AddModule( "__main__"));
|
||||
armlist = PyDict_GetItemString(maindict, list_name);
|
||||
if( !armlist){
|
||||
printf("Oops - update_armature_weakrefs()\n");
|
||||
return 0;
|
||||
}
|
||||
|
||||
for (x = 0; x < PySequence_Size(armlist); x++){
|
||||
pyarmature = PyWeakref_GetObject(PySequence_GetItem( armlist, x));
|
||||
if (pyarmature != Py_None)
|
||||
Armature_RebuildEditbones(pyarmature);
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
|
||||
static PyObject *M_Window_EditMode( PyObject * self, PyObject * args )
|
||||
{
|
||||
short status = -1;
|
||||
char *undo_str = "From script";
|
||||
int undo_str_len = 11;
|
||||
int do_undo = 1;
|
||||
#if 0
|
||||
/* bug 5000 */
|
||||
PyObject *maindict = NULL, *armlist = NULL;
|
||||
PyObject *pyarmature = NULL;
|
||||
int x;
|
||||
#endif
|
||||
|
||||
if( !PyArg_ParseTuple( args,
|
||||
"|hs#i", &status, &undo_str, &undo_str_len, &do_undo ) )
|
||||
@@ -905,17 +929,14 @@ static PyObject *M_Window_EditMode( PyObject * self, PyObject * args )
|
||||
if( status >= 0 ) {
|
||||
if( status ) {
|
||||
if( !G.obedit ){
|
||||
#if 0
|
||||
/* bug 5000 */
|
||||
|
||||
//update armatures
|
||||
maindict= PyModule_GetDict(PyImport_AddModule( "__main__"));
|
||||
armlist = PyDict_GetItemString(maindict, "armatures");
|
||||
for (x = 0; x < PySequence_Size(armlist); x++){
|
||||
pyarmature = PyWeakref_GetObject(PySequence_GetItem( armlist, x));
|
||||
if (pyarmature != Py_None)
|
||||
Armature_RebuildEditbones(pyarmature);
|
||||
if(! update_armature_weakrefs()){
|
||||
return EXPP_ReturnPyObjError(
|
||||
PyExc_RuntimeError,
|
||||
"internal error - update_armature_weakrefs");
|
||||
}
|
||||
#endif
|
||||
|
||||
//enter editmode
|
||||
enter_editmode(0);
|
||||
}
|
||||
@@ -925,17 +946,13 @@ static PyObject *M_Window_EditMode( PyObject * self, PyObject * args )
|
||||
BIF_undo_push( undo_str ); /* This checks user undo settings */
|
||||
exit_editmode( EM_FREEDATA );
|
||||
|
||||
#if 0
|
||||
/* bug 5000 */
|
||||
//update armatures
|
||||
maindict= PyModule_GetDict(PyImport_AddModule( "__main__"));
|
||||
armlist = PyDict_GetItemString(maindict, "armatures");
|
||||
for (x = 0; x < PySequence_Size(armlist); x++){
|
||||
pyarmature = PyWeakref_GetObject(PySequence_GetItem( armlist, x));
|
||||
if (pyarmature != Py_None)
|
||||
Armature_RebuildBones(pyarmature);
|
||||
if(! update_armature_weakrefs()){
|
||||
return EXPP_ReturnPyObjError(
|
||||
PyExc_RuntimeError,
|
||||
"internal error - update_armature_weakrefs");
|
||||
}
|
||||
#endif
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -59,6 +59,9 @@
|
||||
#define Py_RETURN_TRUE return PyBool_FromLong(1)
|
||||
#endif
|
||||
|
||||
/* name of list of Armature weak refs built into __main__ */
|
||||
#define ARM_WEAKREF_LIST_NAME "__arm_weakrefs"
|
||||
|
||||
int EXPP_FloatsAreEqual(float A, float B, int floatSteps);
|
||||
int EXPP_VectorsAreEqual(float *vecA, float *vecB, int size, int floatSteps);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user