Alex Rivera | Logout

Iterating through THREE lists at once in Python?

Asked 2011-06-27T19:22:00.530
11

This might be a rather complex question since it's possible a lot of you don't know the software that I'm writing it for: Autodesk Maya 2011. I am trying to speed up a tedious slow process (rigging: giving 3d characters the ability to move) by writing a script that does it automatically.

I'll try my best to explain the situation.

I have a script that takes an object, iterates through the children of that object, stores them in a list, then puts the initial object at the end of the list, reverses the list because it's the wrong way around, then puts the initial object at the front.

Issue: There are three different lists all of the same object TYPE but with different names and they are actually different objects. My goal is to connect them together by generating nodes called 'blendcolors'. But if I have a loop to generate them for each object in list A, then I need loops that also connect them to the objects in other lists and I can not figure this out.

Here is my code, it has been played with so is more incomplete than before as far as the actual loop goes.

    import maya.cmds as cmds

    def crBC(IKJoint, FKJoint, bindJoint, xQuan, switch):

        # gets children joints of the selected joint
        chHipIK = cmds.listRelatives(IKJoint, ad = True, type = 'joint')
        chHipFK = cmds.listRelatives(FKJoint, ad = True, type = 'joint')
        chHipBind = cmds.listRelatives(bindJoint, ad = True, type = 'joint')
        # list is built backwards, this reverses the list
        chHipIK.reverse()
        chHipFK.reverse()
        chHipBind.reverse()
        # appends the initial joint to the list
        chHipIK.append(IKJoint)
        chHipFK.append(FKJoint)
        chHipBind.append(bindJoint)
        # puts the last joint at the start of the list because the initial joint
        # was added to the end
        chHipIK.insert(0, chHipIK.pop())
        chHipFK.insert(0, chHipFK.pop())
        chHipBind.insert(0, chH
Edit
Report

1 Answer

1

Why are there "There are three different lists all of the same object TYPE"? Why can't this be fixed to create one list, where all three objects are matched correctly?

Odds are good that a simple mapping would be better than three parallel lists.

Specifically, you need to fix chHipIK = cmds.listRelatives(IKJoint, ad = True, type = 'joint') to work like this.

chHipIK = [ { 'IK': ik } for ik in mds.listRelatives(IKJoint, ad = True, type = 'joint') ]
for i, fk in enumerate(cmds.listRelatives(FKJoint, ad = True, type = 'joint')):
    chHipIK[i]['FK']= fk
for i, bind in enumerate(cmds.listRelatives(bindJoint, ad = True, type = 'joint')):
    chHipIK[i]['FK']= bind

So that chHipIK is a list of mappings that has all three pieces of information.

answered 2011-06-27T19:30:30.430

Your Answer