import numpy as np

rotate = {
    'HumanAct12': [1., -1., -1.],
    'CMU_Mocap': [0.05, 0.05, 0.05],
    'UTD_MHAD': [-1., 1., -1.],
    'Human3.6M': [-0.001, -0.001, 0.001]
}


def transform(name, arr: np.ndarray):
    for i in range(arr.shape[0]):
        origin = arr[i][0].copy()
        for j in range(arr.shape[1]):
            arr[i][j] -= origin
            for k in range(3):
                arr[i][j][k] *= rotate[name][k]
    return arr