#!/usr/bin/env bash

build=$MERQURY/build

if [ -z $1 ]; then
    echo "Usage: ./_submit_build.sh [-c] <k-size> <R1.fofn> <R2.fofn> <out_prefix> [mem=T]"
    echo -e "\t-c: OPTIONAL. homopolymer compress the sequence before counting kmers."
    echo -e "\t<k-size>: kmer size k"
    echo -e "\t<R1.fofn>: Read 1. The first 23 bases will get stripped off."
    echo -e "\t<R2.fofn>: Read 2. Will be processed as normal."
    echo -e "\t<out_prefix>: Final merged meryl db will be named as <out_prefix>.meryl"
    echo -e "\t[mem=T]: Submit memory option on sbatch [DEFAULT=TRUE]. Set it to F to turn it off."
    exit -1
fi

if [ "x$1" = "x-c" ]; then
  compress="-c"
  shift
fi

k=$1
R1=$2
R2=$3
out_prefix=$4
mem_opt=$5

mkdir -p logs

# Split files >10GB
cpus=20
if [[ "$mem_opt" = "F" ]]; then
	mem=""
else
	mem="--mem=4g"
fi
name=$out_prefix.split
partition=quick
walltime=4:00:00
path=`pwd`
log=logs/$name.%A_%a.log

wait_for=""
split=0

script=$build/split_10x.sh
LEN=`wc -l $R1 | awk '{print $1}'`

echo "R1 will be split to trim off the barcodes."
split_arrs="1-$LEN"
args="$R1"
echo "
sbatch -D $path -J $name --array=$split_arrs --partition=$partition $mem --cpus-per-task=$cpus --time=$walltime --error=$log --output=$log $script $args"
sbatch -D $path -J $name --array=$split_arrs --partition=$partition $mem --cpus-per-task=$cpus --time=$walltime --error=$log --output=$log $script $args | awk '{print $NF}' > split_jid
split_jid=`cat split_jid`
wait_for="${wait_for}afterok:$split_jid,"


#####
echo "$R2 will be split if >12G"

script=$build/split.sh
LEN2=`wc -l $R2 | awk '{print $1}'`

split_arrs=""

for i in $(seq 1 $LEN2)
do
        fq=`sed -n ${i}p $R2`
        GB=`du -k $fq  | awk '{printf "%.0f", $1/1024/1024}'`
        if [[ $GB -lt 12 ]]; then
            echo "$fq is $GB, less than 12GB. Skip splitting."
            echo $fq >> $R2.$i
        else
            echo "$fq is $GB, over 12GB. Will split and run meryl in parallel. Split files will be in $R2.$i"
	    split_arrs="$split_arrs$i," # keep the line nums $i to split
            split=1
            echo
        fi
done

if [[ $split -eq 1 ]]; then
        split_arrs=${split_arrs%,}
	args="$R2"
        echo "
        sbatch -D $path -J $name --array=$split_arrs --partition=$partition $mem --cpus-per-task=$cpus --time=$walltime --error=$log --output=$log $script $args"
        sbatch -D $path -J $name --array=$split_arrs --partition=$partition $mem --cpus-per-task=$cpus --time=$walltime --error=$log --output=$log $script $args | awk '{print $NF}' >> split_jid
        split_jid=`cat split_jid | tail -n1`
        wait_for="${wait_for}afterok:$split_jid,"
fi


cpus=2
if [[ "$mem_opt" = "F" ]]; then
        mem=""
else
        mem="--mem=1g"
fi
name=$out_prefix.concat
script=$build/concat_splits.sh
args="$compress $k $R1 $out_prefix $R2"
partition=quick
walltime=10:00
path=`pwd`
log=logs/$name.%A.log
wait_for="--dependency=${wait_for%,}"
echo "$wait_for"
echo "
sbatch -D $path -J $name --partition=$partition $mem --cpus-per-task=$cpus --time=$walltime $wait_for --error=$log --output=$log $script $args"
sbatch -D $path -J $name --partition=$partition $mem --cpus-per-task=$cpus --time=$walltime $wait_for --error=$log --output=$log $script $args

