#!/usr/bin/env bash



if [ $# != 2 ]; then
  echo "Usage: $0 <data-dir> <num-to-split>"
  echo "E.g.: $0 data/train_cifar10 5"
  echo "This script creates split-up versions of images.scp and labels.txt in"
  echo "(in the example) data/train_cifar10/split5/{1,2,3,4,5}/{images.scp,labels.txt}."
  exit 1
fi

data=$1
numsplit=$2


if ! [ "$numsplit" -gt 0 ]; then
  echo "Invalid num-split argument $numsplit";
  exit 1;
fi

n=0;


s1=$data/split${numsplit}/1
if [ ! -d $s1 ]; then
  need_to_split=true
else
  need_to_split=false
  for f in images.scp labels.txt; do
    if [[ -f $data/$f && ( ! -f $s1/$f || $s1/$f -ot $data/$f ) ]]; then
      need_to_split=true
    fi
  done
fi

if ! $need_to_split; then
  exit 0;
fi

labels=$(for n in `seq $numsplit`; do echo $data/split${numsplit}${utt}/$n/labels.txt; done)

images=$(for n in `seq $numsplit`; do echo $data/split${numsplit}${utt}/$n/images.scp; done)

directories=$(for n in `seq $numsplit`; do echo $data/split${numsplit}${utt}/$n; done)

# if this mkdir fails due to argument-list being too long, iterate.
if ! mkdir -p $directories >&/dev/null; then
  for n in `seq $numsplit`; do
    mkdir -p $data/split${numsplit}${utt}/$n
  done
fi

# If lockfile is not installed, just don't lock it.  It's not a big deal.
which lockfile >&/dev/null && lockfile -l 60 $data/.split_lock
trap 'rm -f $data/.split_lock' EXIT HUP INT PIPE TERM

utils/split_scp.pl $data/labels.txt $labels || exit 1

utils/split_scp.pl $data/images.scp $images || exit 1


exit 0