#!/bin/bash

#CLASSES=(infl normal)

#DATADIR=../data/data_1024/infl_split/

#DATADIR=/repos/data/data_128_subsample_8x/fullsplit/

#CLASSES=(glom normal)

#DATADIR=../data/data_1024/glom_split/

DATADIR="$1"

cd $DATADIR;

DATADIR=$(pwd -P)

cd -

CLASSES=(lymph norm)

echo "CLASSES:"

for CLASS in ${CLASSES[@]}

do

    echo -e "\t$CLASS"

done

#val_ids=(70bb3032750d09e7549928c0dbf79afc30d7cb68 a1fc67fbb21f43b9e8904b9b46bd94f83493b37a f7f931a5cf3185a385e9aa34e6e9a566fc88000)

#trainids=(c886827fe8c10b4699a0f1616331e36b46a05617 dfe3ee768f72bddd289c7d5bb88b15cbb89be7e6)

SETFILE=~/repos/tables/splits-2019-03-24.csv

readarray train_ids < <(grep 'train' $SETFILE | cut -f1 -d',' )

readarray test_ids < <(grep 'test' $SETFILE | cut -f1 -d',' )

readarray val_ids < <(grep 'val' $SETFILE | cut -f1 -d',' )

echo -e "test\t${test_ids[1]}"

#val_ids=(01c02ba0782776e0aee49c751d170fb5a7f3e189 1613eab8267591c8aad6db6c18308c970bdb0229 4aeba5a0fd1fd39eb494ee759c8d3d566a30bf63 56356da1a0fa39a7394f83e45ff50bd0f54d9461 8250cc470d1320502cfdd7022d410de161885945 88f73323af6f45440a7a532dc4f5aca51c624f5c)

realpath() {

    [[ $1 = /* ]] && echo "$1" || echo "$PWD/${1#./}"

}

for CLASS in ${CLASSES[@]}

do

    mkdir -p ${DATADIR}/val/${CLASS}

    mkdir -p ${DATADIR}/train/${CLASS}

    mkdir -p ${DATADIR}/test/${CLASS}

    for id in ${test_ids[@]}

    do

        echo ""

        echo -e "ID:\t$id"

        #ln -s ${DATADIR}/all/${CLASS}/${id}* ${DATADIR}/train/${CLASS}/

        #find ${DATADIR}/all/${CLASS}/ -name "${id}*" -exec sh -c 'echo $PWD/$(readlink $1) ' _ {} \;

        #find ${DATADIR}/all/${CLASS}/ -name "${id}*" -exec sh -c 'ln -s $PWD/$(readlink $1) '${DATADIR}/train/${CLASS}/'' _ {} \;

        find ${DATADIR}/all/${CLASS}/ -name "${id}*" -exec sh -c 'ln -s $(readlink -f $1) '${DATADIR}/test/${CLASS}/'' _ {} \;

    done

    echo -e "see\t${DATADIR}/test/"

    for id in ${train_ids[@]}

    do

        #echo ""

        #ln -s ${DATADIR}/all/${CLASS}/${id}* ${DATADIR}/train/${CLASS}/

        #find ${DATADIR}/all/${CLASS}/ -name "${id}*" -exec sh -c 'echo $PWD/$(readlink $1) ' _ {} \;

        #find ${DATADIR}/all/${CLASS}/ -name "${id}*" -exec sh -c 'ln -s $PWD/$(readlink $1) '${DATADIR}/train/${CLASS}/'' _ {} \;

        find ${DATADIR}/all/${CLASS}/ -name "${id}*" -exec sh -c 'ln -s $(readlink -f $1) '${DATADIR}/train/${CLASS}/'' _ {} \;

    done

    echo -e "see\t${DATADIR}/train/"

    for id in ${val_ids[@]}

    do

        #ln -s ${DATADIR}/all/${CLASS}/${id}* ${DATADIR}/val/${CLASS}/

        find ${DATADIR}/all/${CLASS}/ -name "${id}*" -exec sh -c 'ln -s $(readlink -f $1) '${DATADIR}/val/${CLASS}/'' _ {} \;

        #find ${DATADIR}/all/${CLASS}/ -name "${id}*" -exec sh -c 'ln -s $PWD/$(readlink $1) '${DATADIR}/val/${CLASS}/'' _ {} \;

    done

    echo -e "see\t${DATADIR}/val/"

done