创建一个递归函数以获取Maya中的层次结构

时间:2020-10-29 16:48:06

标签: python maya pymel

嗨,我正在尝试将Maya中的场景文件的层次结构作为字典。例如 enter image description here

我希望将此数据作为字典

tree = {"group5": {"group4": {"group3": {"group2": {"group1": {"pSphere1": "pSphere1Shape"}},
                                     "pSphere2": {"pSphere2Shape"}
                                     }
                          }
               },
    "group8": {"group7": {"group6": {"pCube": {"pCube1Shape"}
                                     }
                          }
               }
    }

任何建议如何获取此数据?

1 个答案:

答案 0 :(得分:0)

我通常不编写递归函数,所以我给了它一个有趣的机会

import pymel.core as pm

def getHierarchy(start):
    dict = {}
    startNode = pm.PyNode(start)
    #print dir(startNode)
    children = startNode.getChildren()
    if children:
        for child in children:
            dict[child.name()] = getHierarchy(child.name())
    
    return dict
    

if __name__ == '__main__':
    sel = pm.selected()[0]
    hierarchyDict = { sel.name() : getHierarchy(sel.name())}
    print hierarchyDict

结果:

{u'group5': {u'group4': {u'group3': {u'group2': {u'group1': {u'pSphere1': {u'pSphereShape1': {}}}, u'pSphere2': {u'pSphereShape2': {}}}}}, u'group8': {u'group7': {u'group6': {u'pCube1': {u'pCubeShape1': {}}}}}}}

除了括号间距/换行符,我注意到在您的示例中,形状节点(树的叶子)看起来不像是适当的字典。在这个例子中,我将它们作为空字典的关键。 编辑:哎呀,它不包含起点“ group5”,我将其添加到样板中,也许有人会指出如何使其变得更好