import copy


def get_social_config(env):
    """
    Arguments:
        env: Arena-Blowblow-Sparse-2T2P-Discrete

    Returns:
        [[0,1],[2,3]]
    """

    xTxP = env.split("-")[-2]

    T = int(xTxP.split("T")[0])
    P = int(xTxP.split("T")[1].split("P")[0])

    policy_i = 0
    all_list = []
    for t in range(T):
        t_list = []
        for p in range(P):
            t_list += [copy.deepcopy(policy_i)]
            policy_i += 1
        all_list += [copy.deepcopy(t_list)]

    return all_list