CNVkit_pipeline/tasks/CNVkit.wdl

task cnvkit{
    String sample_id
    File ref_dir
    String? fasta
    File ref_flat
    File? bed_file
    String? Ploidy
    File tumor_bam
    File tumor_bam_index
    Array[File]? normal_bams
    Array[File]? normal_bam_indexs
    File? normal_bam
    File? normal_bam_index
    String docker
    String cluster_config
    String disk_size
    String sample=basename(tumor_bam,".bam")

    command <<<

       set -o pipefail
       set -e
       nt=$(nproc)
        
       echo ${sample}
                  
                  
       echo ${Ploidy}
       
       center=`awk -v vv=${Ploidy} 'BEGIN {print log(2/vv)/log(2)}'`
       
       echo $center 
                  
       if [ ${sep=" " normal_bams} ];then
                   echo ${sep=" " normal_bams}
                   /usr/local/bin/cnvkit.py target ${bed_file} --annotate ${ref_flat} --split --short-names -o my_baits.bed
                   /usr/local/bin/cnvkit.py batch ${tumor_bam} \
                      --normal ${sep=" " normal_bams} \
                      --targets my_baits.bed \
                      --fasta ${ref_dir} \
                      --annotate ${ref_flat} -p $nt \
                      --drop-low-coverage \
                      --output-dir ${sample}.reference.cnn
       fi
                  
                  
       #WES
       if [ ${bed_file} ]; then
           echo "WES"
           if [ ${normal_bam} ]; then
                   echo "WES with normal"
                   /usr/local/bin/cnvkit.py target ${bed_file} --annotate ${ref_flat} --split --short-names -o my_baits.bed
                   /usr/local/bin/cnvkit.py batch ${tumor_bam} \
                      --normal ${normal_bam} \
                      --targets my_baits.bed \
                      --fasta ${ref_dir} \
                      --annotate ${ref_flat} -p $nt \
                      --drop-low-coverage \
                      --output-dir ${sample}.reference.cnn
           else
                   echo "WES no normal"
                   /usr/local/bin/cnvkit.py access ${ref_dir} -o access.bed
                   # Prepare the target bed
                   /usr/local/bin/cnvkit.py target ${bed_file} --annotate ${ref_flat} --split --short-names -o my_baits.bed

                   /usr/local/bin/cnvkit.py autobin ${tumor_bam} -t my_baits.bed -g access.bed

                   /usr/local/bin/cnvkit.py coverage ${tumor_bam} my_baits.target.bed -o ${sample}.T.targetcoverage.cnn
                   /usr/local/bin/cnvkit.py coverage ${tumor_bam} my_baits.antitarget.bed -o ${sample}.T.antitargetcoverage.cnn

                   /usr/local/bin/cnvkit.py reference -o ${sample}.reference.cnn/reference.cnn -f ${ref_dir} -t my_baits.target.bed -a my_baits.antitarget.bed
           fi
    
       #WGS         
       else
           echo "WGS"
           if [ ${normal_bam} ]; then
                   echo "WGS with normal"
                   /usr/local/bin/cnvkit.py batch ${tumor_bam} \
                      --normal ${normal_bam} \
                      --method wgs \
                      --fasta ${ref_dir} \
                      --annotate ${ref_flat} -p $nt \
                      --drop-low-coverage \
                      --output-dir ${sample}.reference.cnn
           else
                   echo "WGS no normal"
                   /usr/local/bin/cnvkit.py access ${ref_dir} -o access.bed
                   # Prepare the target bed
                   #/usr/local/bin/cnvkit.py --annotate ${ref_flat} --split --short-names -o my_baits.bed

                   /usr/local/bin/cnvkit.py autobin ${tumor_bam} --method wgs -g access.bed

                   /usr/local/bin/cnvkit.py coverage ${tumor_bam} ${sample}.target.bed -o ${sample}.T.targetcoverage.cnn
                   /usr/local/bin/cnvkit.py coverage ${tumor_bam} ${sample}.antitarget.bed -o ${sample}.T.antitargetcoverage.cnn

                   /usr/local/bin/cnvkit.py reference -o ${sample}.reference.cnn/reference.cnn -f ${ref_dir} -t ${sample}.target.bed -a ${sample}.antitarget.bed

           fi
       fi
                  
       ls ./
       
        
       /usr/local/bin/cnvkit.py batch  ${tumor_bam} \
                  -r ${sample}.reference.cnn/reference.cnn \
                  --output-dir ${sample}.cns \
                  -p $nt
                  
       ls ./
       /usr/local/bin/cnvkit.py call ${sample}.cns/${sample}.cns --center-at $center \
                  -o ${sample}.call.cns
                  
       
           # Plot the results
    	/usr/local/bin/cnvkit.py scatter ${sample}.cns/${sample}.cnr -s ${sample}.call.cns -o ${sample}.scatter.pdf
    	/usr/local/bin/cnvkit.py diagram ${sample}.cns/${sample}.cnr -s ${sample}.call.cns -o ${sample}.diagram.pdf
    	/usr/local/bin/cnvkit.py heatmap ${sample}.cns/${sample}.cnr ${sample}.call.cns -o ${sample}.heatmap.pdf
       
         
        #gain and loss
        /usr/local/bin/cnvkit.py export vcf ${sample}.call.cns -y -i "${sample}" -o ${sample}.call.cns.vcf
                

        tar cvf ${sample}.tar ${sample}*

    >>>

    runtime{
        docker:docker
        cluster:cluster_config
        systemDisk:"cloud_ssd 40"
        dataDisk:"cloud_ssd " + disk_size + " /cromwell_root/"
        timeout:259200
    }

    output{
        File out_file = "${sample}.tar"
        File cnv_bed = "${sample}.call.cns.vcf"
    }

}