#!/bin/bash

set -eu

PYTHON_DIR=$(dirname $0)
ROOT_DIR=${PYTHON_DIR}/../..
SRC_DIR=${ROOT_DIR}/src/add_python
RESULTS_DIR=${ROOT_DIR}/results
CRYPTO_AEAD=${RESULTS_DIR}/crypto_aead_py

VERSION=$(${ROOT_DIR}/nist/version.sh)

mkdir -p ${CRYPTO_AEAD}/lilliputae{i,ii}{128,192,256}v${VERSION}

declare -A NAMES=([1]=lilliputaei [2]=lilliputaeii)


generate ()
{
    local mode=$1
    local keylen=$2
    local dest_dir=${CRYPTO_AEAD}/${NAMES[${mode}]}${keylen}v${VERSION}
    local param_dir=${RESULTS_DIR}/python-${mode}-${keylen}

    mkdir -p ${param_dir}
    cat <<EOF > ${param_dir}/parameters.py
MODE = ${mode}
KEYBYTES = $((keylen/8))
EOF

    echo ${mode} ${keylen}: starting

    PYTHONPATH=${param_dir}:${SRC_DIR} ${PYTHON_DIR}/genkat_aead.py ${dest_dir}

    echo ${mode} ${keylen}: finished
}


for mode in 1 2
do
    for keylen in 128 192 256
    do
        generate ${mode} ${keylen} &
    done
done

for ((i=0; i<6; i++))
do
    wait -n
done