SMPLX_KEYPOINTS = [
    'pelvis',
    'left_hip',
    'right_hip',
    'spine_1',
    'left_knee',
    'right_knee',
    'spine_2',
    'left_ankle',
    'right_ankle',
    'spine_3',
    'left_foot',
    'right_foot',
    'neck',
    'left_collar',
    'right_collar',
    'head',
    'left_shoulder',
    'right_shoulder',
    'left_elbow',
    'right_elbow',
    'left_wrist',
    'right_wrist',
    'jaw',
    'left_eyeball',
    'right_eyeball',
    'left_index_1',
    'left_index_2',
    'left_index_3',
    'left_middle_1',
    'left_middle_2',
    'left_middle_3',
    'left_pinky_1',
    'left_pinky_2',
    'left_pinky_3',
    'left_ring_1',
    'left_ring_2',
    'left_ring_3',
    'left_thumb_1',
    'left_thumb_2',
    'left_thumb_3',
    'right_index_1',
    'right_index_2',
    'right_index_3',
    'right_middle_1',
    'right_middle_2',
    'right_middle_3',
    'right_pinky_1',
    'right_pinky_2',
    'right_pinky_3',
    'right_ring_1',
    'right_ring_2',
    'right_ring_3',
    'right_thumb_1',
    'right_thumb_2',
    'right_thumb_3',
    'nose',
    'right_eye',
    'left_eye',
    'right_ear',
    'left_ear',
    'left_bigtoe',
    'left_smalltoe',
    'left_heel',
    'right_bigtoe',
    'right_smalltoe',
    'right_heel',
    'left_thumb',
    'left_index',
    'left_middle',
    'left_ring',
    'left_pinky',
    'right_thumb',
    'right_index',
    'right_middle',
    'right_ring',
    'right_pinky',
    'right_eyebrow_1',
    'right_eyebrow_2',
    'right_eyebrow_3',
    'right_eyebrow_4',
    'right_eyebrow_5',
    'left_eyebrow_5',
    'left_eyebrow_4',
    'left_eyebrow_3',
    'left_eyebrow_2',
    'left_eyebrow_1',
    'nosebridge_1',
    'nosebridge_2',
    'nosebridge_3',
    'nosebridge_4',
    'right_nose_2',  # original name: nose_1
    'right_nose_1',  # original name: nose_2
    'nose_middle',  # original name: nose_3
    'left_nose_1',  # original name: nose_4
    'left_nose_2',  # original name: nose_5
    'right_eye_1',
    'right_eye_2',
    'right_eye_3',
    'right_eye_4',
    'right_eye_5',
    'right_eye_6',
    'left_eye_4',
    'left_eye_3',
    'left_eye_2',
    'left_eye_1',
    'left_eye_6',
    'left_eye_5',
    'right_mouth_1',  # original name: mouth_1
    'right_mouth_2',  # original name: mouth_2
    'right_mouth_3',  # original name: mouth_3
    'mouth_top',  # original name: mouth_4
    'left_mouth_3',  # original name: mouth_5
    'left_mouth_2',  # original name: mouth_6
    'left_mouth_1',  # original name: mouth_7
    'left_mouth_5',  # original name: mouth_8
    'left_mouth_4',  # original name: mouth_9
    'mouth_bottom',  # original name: mouth_10
    'right_mouth_4',  # original name: mouth_11
    'right_mouth_5',  # original name: mouth_12
    'right_lip_1',  # original name: lip_1
    'right_lip_2',  # original name: lip_2
    'lip_top',  # original name: lip_3
    'left_lip_2',  # original name: lip_4
    'left_lip_1',  # original name: lip_5
    'left_lip_3',  # original name: lip_6
    'lip_bottom',  # original name: lip_7
    'right_lip_3',  # original name: lip_8
    'right_contour_1',  # original name: face_contour_1
    'right_contour_2',  # original name: face_contour_2
    'right_contour_3',  # original name: face_contour_3
    'right_contour_4',  # original name: face_contour_4
    'right_contour_5',  # original name: face_contour_5
    'right_contour_6',  # original name: face_contour_6
    'right_contour_7',  # original name: face_contour_7
    'right_contour_8',  # original name: face_contour_8
    'contour_middle',  # original name: face_contour_9
    'left_contour_8',  # original name: face_contour_10
    'left_contour_7',  # original name: face_contour_11
    'left_contour_6',  # original name: face_contour_12
    'left_contour_5',  # original name: face_contour_13
    'left_contour_4',  # original name: face_contour_14
    'left_contour_3',  # original name: face_contour_15
    'left_contour_2',  # original name: face_contour_16
    'left_contour_1',  # original name: face_contour_17
]

SMPLX_LIMBS = {
    'body': [['pelvis', 'left_hip'], ['pelvis', 'right_hip'],
             ['left_hip', 'right_hip'], ['left_shoulder', 'right_shoulder'],
             ['pelvis', 'spine_1'], ['spine_1', 'spine_2'],
             ['spine_2', 'spine_3'], ['spine_3', 'neck'], ['neck', 'head'],
             ['left_ankle', 'left_knee'], ['left_knee', 'left_hip'],
             ['right_ankle', 'right_knee'], ['right_knee', 'right_hip'],
             ['right_ankle', 'right_foot'], ['left_ankle', 'left_foot'],
             ['left_hip', 'right_hip'], ['left_shoulder', 'left_hip'],
             ['right_shoulder', 'right_hip'], ['left_collar', 'spine_3'],
             ['right_collar', 'spine_3'], ['right_collar', 'right_shoulder'],
             ['left_collar', 'left_shoulder'],
             ['left_shoulder', 'right_shoulder'],
             ['left_shoulder',
              'left_elbow'], ['right_shoulder', 'right_elbow'],
             ['left_elbow', 'left_wrist'], ['right_elbow', 'right_wrist'],
             ['left_ankle', 'left_bigtoe'], ['left_ankle', 'left_smalltoe'],
             ['left_ankle', 'left_heel'], ['right_ankle', 'right_bigtoe'],
             ['right_ankle', 'right_smalltoe'], ['right_ankle', 'right_heel'],
             ['left_shoulder', 'left_ear'], ['right_shoulder', 'right_ear'],
             ['right_ear', 'right_eye'], ['right_eye', 'nose'],
             ['nose', 'left_eye'], ['left_eye', 'left_ear'], ['nose', 'jaw'],
             ['jaw', 'neck']],
    'face': [['right_contour_1', 'right_contour_2'],
             ['right_contour_2', 'right_contour_3'],
             ['right_contour_3', 'right_contour_4'],
             ['right_contour_4', 'right_contour_5'],
             ['right_contour_5', 'right_contour_6'],
             ['right_contour_6', 'right_contour_7'],
             ['right_contour_7', 'right_contour_8'],
             ['right_contour_8', 'contour_middle'],
             ['contour_middle', 'left_contour_8'],
             ['left_contour_8', 'left_contour_7'],
             ['left_contour_7', 'left_contour_6'],
             ['left_contour_6', 'left_contour_5'],
             ['left_contour_5', 'left_contour_4'],
             ['left_contour_4', 'left_contour_3'],
             ['left_contour_3', 'left_contour_2'],
             ['left_contour_2', 'left_contour_1']],
    'left_hand':
    [['left_wrist', 'left_thumb_1'], ['left_thumb_1', 'left_thumb_2'],
     ['left_thumb_2', 'left_thumb_3'], ['left_thumb_3', 'left_thumb'],
     ['left_wrist', 'left_index_1'], ['left_index_1', 'left_index_2'],
     ['left_index_2', 'left_index_3'], ['left_index_3', 'left_index'],
     ['left_wrist', 'left_middle_1'], ['left_middle_1', 'left_middle_2'],
     ['left_middle_2', 'left_middle_3'], ['left_middle_3', 'left_middle'],
     ['left_wrist', 'left_ring_1'], ['left_ring_1', 'left_ring_2'],
     ['left_ring_2', 'left_ring_3'], ['left_ring_3', 'left_ring'],
     ['left_wrist', 'left_pinky_1'], ['left_pinky_1', 'left_pinky_2'],
     ['left_pinky_2', 'left_pinky_3'], ['left_pinky_3', 'left_pinky']],
    'right_hand': [['right_wrist', 'right_thumb_1'],
                   ['right_thumb_1', 'right_thumb_2'],
                   ['right_thumb_2', 'right_thumb_3'],
                   ['right_thumb_3', 'right_thumb'],
                   ['right_wrist', 'right_index_1'],
                   ['right_index_1', 'right_index_2'],
                   ['right_index_2', 'right_index_3'],
                   ['right_index_3', 'right_index'],
                   ['right_wrist', 'right_middle_1'],
                   ['right_middle_1', 'right_middle_2'],
                   ['right_middle_2', 'right_middle_3'],
                   ['right_middle_3', 'right_middle'],
                   ['right_wrist', 'right_ring_1'],
                   ['right_ring_1', 'right_ring_2'],
                   ['right_ring_2', 'right_ring_3'],
                   ['right_ring_3', 'right_ring'],
                   ['right_wrist', 'right_pinky_1'],
                   ['right_pinky_1', 'right_pinky_2'],
                   ['right_pinky_2', 'right_pinky_3'],
                   ['right_pinky_3', 'right_pinky']],
    'right_eye':
    [['right_eye_1', 'right_eye_2'], ['right_eye_2', 'right_eye_3'],
     ['right_eye_3', 'right_eye_4'], ['right_eye_4', 'right_eye_5'],
     ['right_eye_5', 'right_eye_6'], ['right_eye_6', 'right_eye_1'],
     ['right_eyebrow_1', 'right_eyebrow_2'],
     ['right_eyebrow_2', 'right_eyebrow_3'],
     ['right_eyebrow_3', 'right_eyebrow_4'],
     ['right_eyebrow_4', 'right_eyebrow_5']],
    'left_eye': [['left_eye_4', 'left_eye_3'], ['left_eye_3', 'left_eye_2'],
                 ['left_eye_2', 'left_eye_1'], ['left_eye_1', 'left_eye_6'],
                 ['left_eye_6', 'left_eye_5'], ['left_eye_5', 'left_eye_4'],
                 ['left_eyebrow_1', 'left_eyebrow_2'],
                 ['left_eyebrow_2', 'left_eyebrow_3'],
                 ['left_eyebrow_3', 'left_eyebrow_4'],
                 ['left_eyebrow_4', 'left_eyebrow_5']],
    'mouth':
    [['right_mouth_1', 'right_mouth_2'], ['right_mouth_2', 'right_mouth_3'],
     ['right_mouth_3', 'mouth_top'], ['mouth_top', 'left_mouth_3'],
     ['left_mouth_3', 'left_mouth_2'], ['left_mouth_2', 'left_mouth_1'],
     ['left_mouth_1', 'left_mouth_5'], ['left_mouth_5', 'left_mouth_4'],
     ['left_mouth_4', 'mouth_bottom'], ['mouth_bottom', 'right_mouth_4'],
     ['right_mouth_4', 'right_mouth_5'], ['right_mouth_5', 'right_mouth_1'],
     ['right_lip_1', 'right_lip_2'], ['right_lip_2', 'lip_top'],
     ['lip_top', 'left_lip_2'], ['left_lip_2', 'left_lip_1'],
     ['left_lip_1', 'left_lip_3'], ['left_lip_3', 'lip_bottom'],
     ['lip_bottom', 'right_lip_3'], ['right_lip_3', 'right_lip_1']],
    'nose': [
        ['nosebridge_1', 'nosebridge_2'],
        ['nosebridge_2', 'nosebridge_3'],
        ['nosebridge_3', 'nosebridge_4'],
        ['right_nose_2', 'right_nose_1'],
        ['right_nose_1', 'nose_middle'],
        ['nose_middle', 'left_nose_1'],
        ['left_nose_1', 'left_nose_2'],
    ]
}

SMPLX_LIMBS_INDEX = {}
for k in SMPLX_LIMBS:
    SMPLX_LIMBS_INDEX[k] = [[
        SMPLX_KEYPOINTS.index(limb[0]),
        SMPLX_KEYPOINTS.index(limb[1])
    ] for limb in SMPLX_LIMBS[k]]

SMPLX_PALETTE = {
    'left_eye': [[0, 0, 0]],
    'right_eye': [[0, 0, 0]],
    'nose': [[0, 0, 255]],
    'mouth': [[0, 255, 255]],
    'face': [[255, 0, 0]],
    'left_hand': [[0, 0, 0]],
    'right_hand': [[0, 0, 0]]
}


joint_idx = \
        (0,1,2,4,5,7,8,12,16,17,18,19,20,21,60,61,62,63,64,65,59,58,57,56,55, # body joints
        37,38,39,66,25,26,27,67,28,29,30,68,34,35,36,69,31,32,33,70, # left hand joints
        52,53,54,71,40,41,42,72,43,44,45,73,49,50,51,74,46,47,48,75, # right hand joints
        22,15, # jaw, head
        57,56, # eyeballs
        76,77,78,79,80,81,82,83,84,85, # eyebrow
        86,87,88,89, # nose
        90,91,92,93,94, # below nose
        95,96,97,98,99,100,101,102,103,104,105,106, # eyes
        107, # right mouth
        108,109,110,111,112, # upper mouth
        113, # left mouth
        114,115,116,117,118, # lower mouth
        119, # right lip
        120,121,122, # upper lip
        123, # left lip
        124,125,126, # lower lip
        127,128,129,130,131,132,133,134,135,136,137,138,139,140,141,142,143 # face contour
        )

SMPLX_137_KEYPOINTS = []
for idx in joint_idx:
    SMPLX_137_KEYPOINTS.append(SMPLX_KEYPOINTS[idx])


SMPLX_LHAND = [
    # 'left_thumb_2',
    'left_wrist',
    'left_thumb',
    # 'left_index_1',
    'left_index',
    # 'left_middle_1',
    'left_middle',
    # 'left_ring_1',
    'left_ring',
    # 'left_pinky_1',
    'left_pinky',
]
SMPLX_RHAND = [
    # 'right_thumb_2',
    'right_wrist',
    'right_thumb',
    # 'right_index_1',
    'right_index',
    # 'right_middle_1',
    'right_middle',
    # 'right_ring_1',
    'right_ring',
    # 'right_pinky_1',
    'right_pinky',
]

SMPLX_FACE = [
    'nose',
    'mouth_top',
    'jaw',
    'right_contour_1',
    'contour_middle',
    'left_contour_1'
]

AiOS_35_KEYPOINTS = [
    'nose',
    'left_eye',
    'right_eye',
    'left_ear',
    'right_ear',
    'left_shoulder',
    'right_shoulder',
    'left_elbow',
    'right_elbow',
    'left_wrist',
    'right_wrist',
    'left_hip_extra',
    'right_hip_extra',
    'left_knee',
    'right_knee',
    'left_ankle',
    'right_ankle',
    'left_wrist',
    'left_thumb',
    # 'left_index_1',
    'left_index',
    # 'left_middle_1',
    'left_middle',
    # 'left_ring_1',
    'left_ring',
    # 'left_pinky_1',
    'left_pinky',

    'right_wrist',
    'right_thumb',
    # 'right_index_1',
    'right_index',
    # 'right_middle_1',
    'right_middle',
    # 'right_ring_1',
    'right_ring',
    # 'right_pinky_1',
    'right_pinky',

    'nose',
    'mouth_top',
    'jaw',
    'right_contour_1',
    'contour_middle',
    'left_contour_1'
    
]