#!/bin/bash

for d in work/*
do
    echo $d
    pushd $d >/dev/null

    if [ -d encoder ]; then
        xcrun coremlcompiler compile encoder/chunked_pipeline.mlpackage .
        rm -rf encoder.mlmodelc
        mv chunked_pipeline.mlmodelc encoder.mlmodelc
    else
        xcrun coremlcompiler compile encoder.mlpackage .
    fi
    xcrun coremlcompiler compile decoder_first.mlpackage .
    xcrun coremlcompiler compile decoder_second.mlpackage .

    popd >/dev/null
done

mkdir -p output
for d in work/*
do
    out=${d/work/output}
    mkdir -p $out
    mv $d/*.mlmodelc $d/model_dims.json $out/
done

mkdir -p index
for d in output/*
do
    model=${d##*/}
    (cd $d && find * -type f) > index/$model
done