def get_joint_group_idx()

in src/eval.py [0:0]

• 54 lines of code
• 2 McCabe index (conditional complexity)

def get_joint_group_idx(c: Train_Config):
    I = c.skeleton.Idx
    if c.skeleton == definitions.Skeleton:
        return {
            "neck": [definitions.Skeleton.Idx.neck],
            "shoulder": [definitions.Skeleton.Idx.lupperarm, definitions.Skeleton.Idx.rupperarm],
            "elbow": [definitions.Skeleton.Idx.llowerarm, definitions.Skeleton.Idx.rlowerarm],
        }
    return {
        "shoulder": [I.lupperarm, I.lscap, I.lshoulder, I.rupperarm, I.rscap, I.rshoulder],
        "elbow": [I.llowerarm, I.rlowerarm],
        "wrist": [I.lwrist, I.rwrist],
        "finger": [
            I.lindex1, I.lindex2, I.lindex3,
            I.lmiddle1, I.lmiddle2, I.lmiddle3,
            I.lring1, I.lring2, I.lring3,
            I.lpinky1, I.lpinky2, I.lpinky3,
            I.lthumb0, I.lthumb1, I.lthumb2, I.lthumb3,
            I.rindex1, I.rindex2, I.rindex3,
            I.rmiddle1, I.rmiddle2, I.rmiddle3,
            I.rring1, I.rring2, I.rring3,
            I.rpinky1, I.rpinky2, I.rpinky3,
            I.rthumb0, I.rthumb1, I.rthumb2, I.rthumb3,
        ],
        "finger_wo_thumb": [
            I.lindex1, I.lindex2, I.lindex3,
            I.lmiddle1, I.lmiddle2, I.lmiddle3,
            I.lring1, I.lring2, I.lring3,
            I.lpinky1, I.lpinky2, I.lpinky3,
            I.rindex1, I.rindex2, I.rindex3,
            I.rmiddle1, I.rmiddle2, I.rmiddle3,
            I.rring1, I.rring2, I.rring3,
            I.rpinky1, I.rpinky2, I.rpinky3,
        ],
        "thumb": [
            I.lthumb0, I.lthumb1, I.lthumb2, I.lthumb3,
            I.rthumb0, I.rthumb1, I.rthumb2, I.rthumb3,
        ],
        "wrist_local_finger": [
            I.lindex1, I.lindex2, I.lindex3,
            I.lmiddle1, I.lmiddle2, I.lmiddle3,
            I.lring1, I.lring2, I.lring3,
            I.lpinky1, I.lpinky2, I.lpinky3,
            I.lthumb0, I.lthumb1, I.lthumb2, I.lthumb3,
            I.rindex1, I.rindex2, I.rindex3,
            I.rmiddle1, I.rmiddle2, I.rmiddle3,
            I.rring1, I.rring2, I.rring3,
            I.rpinky1, I.rpinky2, I.rpinky3,
            I.rthumb0, I.rthumb1, I.rthumb2, I.rthumb3,
        ],
        "hip": [I.lupperleg, I.rupperleg],
        "knee": [I.llowerleg, I.rlowerleg],
        "foot": [I.lfoot, I.lfootball, I.rfoot, I.rfootball]
    }