"""
This script is adapted from: https://github.com/yufu-wang/tram/blob/main/scripts/download_models.sh
"""

mkdir -p ./smpl
mkdir -p ./smpl_raw

#!/bin/bash
urle () { [[ "${1}" ]] || return 1; local LANG=C i x; for (( i = 0; i < ${#1}; i++ )); do x="${1:i:1}"; [[ "${x}" == [a-zA-Z0-9.~-] ]] && echo -n "${x}" || printf '%%%02X' "'${x}"; done; echo; }

# SMPL Neutral model
echo -e "\nYou need to register at https://smplify.is.tue.mpg.de"
read -p "Username (SMPLify):" username
read -p "Password (SMPLify):" password
username=$(urle $username)
password=$(urle $password)

wget --post-data "username=$username&password=$password" 'https://download.is.tue.mpg.de/download.php?domain=smplify&resume=1&sfile=mpips_smplify_public_v2.zip' -O 'smpl_raw/smplify.zip' --no-check-certificate --continue
unzip smpl_raw/smplify.zip -d smpl_raw
mv smpl_raw/smplify/smplify_public/code/models/basicModel_neutral_lbs_10_207_0_v1.0.0.pkl ./smpl/SMPL_NEUTRAL.pkl

# SMPL Male and Female model
echo -e "\nYou need to register at https://smpl.is.tue.mpg.de"
read -p "Username (SMPL):" username
read -p "Password (SMPL):" password
username=$(urle $username)
password=$(urle $password)

wget --post-data "username=$username&password=$password" 'https://download.is.tue.mpg.de/download.php?domain=smpl&sfile=SMPL_python_v.1.0.0.zip' -O 'smpl_raw/smpl.zip' --no-check-certificate --continue
unzip smpl_raw/smpl.zip -d smpl_raw
mv smpl_raw/smpl/models/basicModel_f_lbs_10_207_0_v1.0.0.pkl ./smpl/SMPL_FEMALE.pkl
mv smpl_raw/smpl/models/basicmodel_m_lbs_10_207_0_v1.0.0.pkl ./smpl/SMPL_MALE.pkl

rm -rf smpl_raw