#!/bin/bash
#trimgalore.sh

usage() {
    echo "-h Help documentation for hisat.sh"
    echo "-r  --Reference Genome: GRCh38 or GRCm38"
    echo "-b  --BAM File"
    echo "-n  --NucType"
    echo "-p  --Prefix for output file name"
    echo "-c  --Capture Bedfile"
    echo "-d  --RemoveDuplicates 1=yes, 0=no default=no"
    echo "Example: bash bamqc.sh -p prefix -r /project/shared/bicf_workflow_ref/human/GRCh38 -b SRR1551047.bam  -n dna -c target.bed"
    exit 1
}
OPTIND=1 # Reset OPTIND
while getopts :r:b:c:n:p:u:e:s:d:h opt
do
    case $opt in
        r) index_path=$OPTARG;;
        b) sbam=$OPTARG;;
        c) bed=$OPTARG;;
        n) nuctype=$OPTARG;;
        p) pair_id=$OPTARG;;
	d) dedup=$OPTARG;;
	e) version=$OPTARG;;
	s) skiplc=1;;
	u) user=$OPTARG;;
        h) usage;;
    esac
done

shift $(($OPTIND -1))

# Check for mandatory options
#if [[ -z $pair_id ]] || [[ -z $sbam ]]; then
#    usage
#fi

if [[ -z $version ]]
then
    version='NA'
fi
NPROC=$SLURM_CPUS_ON_NODE
if [[ -z $NPROC ]]
then
    NPROC=`nproc`
fi
threads=`expr $NPROC - 10`

if [[ -n $user ]]
then
    USER=$user
else
    USER=$(whoami)
fi

parseopt=" -u $USER"
if [[ -n $version ]]
then
    parseopt="$parseopt -e $version"
fi
if [[ -f $index_path/reference_info.txt ]]
then
    parseopt="$parseopt -r $index_path"
fi
tmpdir=`pwd`
source /etc/profile.d/modules.sh
module load samtools/gcc/1.10 fastqc/0.11.8 bedtools/2.29.0 picard/2.10.3

samtools flagstat ${sbam} > ${pair_id}.flagstat.txt
fastqc -f bam ${sbam}
baseDir="`dirname \"$0\"`"

if [[ $dedup == 1 ]]
then
    mv $sbam ori.bam
    samtools view -@ $threads -F 1024 -b -o ${sbam} ori.bam
fi

if [[ $nuctype == 'dna' ]]
then
    bedtools coverage -a ${bed} -b ${sbam} -hist > ${pair_id}.covhist.txt
    grep ^all ${pair_id}.covhist.txt >  ${pair_id}.genomecov.txt
    perl $baseDir/calculate_depthcov.pl ${pair_id}.covhist.txt
    if [[ -z $skiplc ]]
    then
	samtools view -@ $threads -b -L ${bed} -o ${pair_id}.ontarget.bam ${sbam}
	samtools index -@ $threads ${pair_id}.ontarget.bam
	samtools flagstat  ${pair_id}.ontarget.bam > ${pair_id}.ontarget.flagstat.txt
	samtools view  -@ $threads -b -q 1 ${sbam} | bedtools coverage -hist -b stdin -a ${bed} > ${pair_id}.mapqualcov.txt
	java -Xmx64g -Djava.io.tmpdir=${tmpdir} -XX:ParallelGCThreads=$threads -jar $PICARD/picard.jar EstimateLibraryComplexity BARCODE_TAG=RG I=${sbam} OUTPUT=${pair_id}.libcomplex.txt TMP_DIR=${tmpdir}
	#java -Xmx64g -Djava.io.tmpdir=${tmpdir} -jar $PICARD/picard.jar CollectAlignmentSummaryMetrics R=${index_path}/genome.fa I=${pair_id}.ontarget.bam OUTPUT=${pair_id}.alignmentsummarymetrics.txt TMP_DIR=${tmpdir}
	#samtools view  -@ $threads ${sbam} | awk '{sum+=$5} END { print "Mean MAPQ =",sum/NR}' > ${pair_id}.meanmap.txt
    fi
    #java -Xmx64g -Djava.io.tmpdir=${tmpdir} -jar $PICARD/picard.jar CollectInsertSizeMetrics INPUT=${sbam} HISTOGRAM_FILE=${pair_id}.hist.ps REFERENCE_SEQUENCE=${index_path}/genome.fa OUTPUT=${pair_id}.hist.txt TMP_DIR=${tmpdir}
    perl $baseDir/sequenceqc_dna.pl $parseopt ${pair_id}.genomecov.txt
else
    perl $baseDir/sequenceqc_rna.pl $parseopt ${pair_id}.flagstat.txt
fi