sandey/blender
1
0
Fork 0
forked from bartvdbraak/blender
Code Pull Requests Activity

change bone storage class so __slots__ can be used for looping over attributes without getting functions

Browse Source
This commit is contained in:
Campbell Barton 2009-12-02 19:13:54 +00:00
parent 2a6d535d23
commit 7706261e79
1 changed files with 35 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
release/scripts/modules/rigify/__init__.py
View File

@ -24,11 +24,36 @@ from rna_prop_ui import rna_idprop_ui_get, rna_idprop_ui_prop_get
empty_layer = [False] * 32
def auto_class(slots, name="ContainerClass"):
return type(name, (object,), {"__slots__":tuple(slots)})
def auto_class(slots, name="ContainerClass", class_dict=None):
def auto_class_instance(slots, name="ContainerClass"):
return auto_class(slots, name)()
if class_dict:
class_dict = class_dict.copy()
else:
class_dict = {}
class_dict["__slots__"] = tuple(slots)
return type(name, (object,), class_dict)
def auto_class_instance(slots, name="ContainerClass", class_dict=None):
return auto_class(slots, name, class_dict)()
def _bone_class_instance_update(self):
''' Re-Assigns bones from the blender data
'''
arm = self.obj.data
bbones = arm.bones
pbones = self.obj.pose.bones
ebones = arm.edit_bones
for member in self.__slots__:
if not member[-2] == "_":
name = getattr(self, member, None)
if name is not None:
setattr(self, member + "_b", bbones.get(name, None))
setattr(self, member + "_p", pbones.get(name, None))
setattr(self, member + "_e", ebones.get(name, None))
def bone_class_instance(obj, slots, name="BoneContainer"):
slots = slots[:] # dont modify the original
@ -38,34 +63,8 @@ def bone_class_instance(obj, slots, name="BoneContainer"):
slots.append(member + "_p") # pose bone
slots.append(member + "_e") # edit bone
slots.extend(["obj", "update"])
instance = auto_class_instance(slots, name)
def update():
'''
Re-Assigns bones from the blender data
'''
arm = obj.data
bbones = arm.bones
pbones = obj.pose.bones
ebones = arm.edit_bones
for member in slots:
if member in ("update", "obj"):
continue
if not member[-2] == "_":
name = getattr(instance, member, None)
if name is not None:
setattr(instance, member + "_b", bbones.get(name, None))
setattr(instance, member + "_p", pbones.get(name, None))
setattr(instance, member + "_e", ebones.get(name, None))
instance.update = update
class_dict = {"obj":obj, "update":_bone_class_instance_update}
instance = auto_class_instance(slots, name, class_dict)
return instance
def gen_none(obj, orig_bone_name):
@ -241,7 +240,7 @@ def write_meta_rig(obj, func_name="metarig_template"):
code = []
code.append("def %s():" % func_name)
code.append(" # generated by rigify.write_meta_rig")
bpy.ops.object.mode_set(mode='EDIT')
code.append(" bpy.ops.object.mode_set(mode='EDIT')")
@ -351,8 +350,8 @@ def generate_test_all(context):
path_png = fn + ".png"
saved = graphviz_export.graph_armature(ob, path_dot, CONSTRAINTS=True, DRIVERS=True)
if saved:
os.system("dot -Tpng %s > %s; eog %s" % (path_dot, path_png, path_png))
#if saved:
# os.system("dot -Tpng %s > %s; eog %s" % (path_dot, path_png, path_png))
if __name__ == "__main__":