From 3d77cb5b70850da88b24da4b1b8deb56826d4e04 Mon Sep 17 00:00:00 2001 From: Venkat Malladi <venkat.malladi@utsouthwestern.edu> Date: Sat, 7 Oct 2017 10:46:54 -0500 Subject: [PATCH] update the naming of the reads. --- workflow/scripts/map_reads.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/workflow/scripts/map_reads.py b/workflow/scripts/map_reads.py index ebce796..972a816 100644 --- a/workflow/scripts/map_reads.py +++ b/workflow/scripts/map_reads.py @@ -112,24 +112,27 @@ def align_se(fastq, sai, reference, fastq_basename): '''Use BWA to align SE data.''' sam_filename = "%s.sam" % (fastq_basename) + bam_filename = '%s.bam' % (fastq_basename) steps = [ "bwa samse %s %s %s" % (reference, fastq[0], sai[0]), "samtools view -@%d -Su -" % (cpu_count()), "samtools sort -@%d -o %s" - % (cpu_count(), fastq_basename)] + % (cpu_count(), bam_filename)] out, err = utils.run_pipe(steps) if err: logger.error("samse/samtools error: %s" % (err)) + return bam_filename def align_pe(fastq, sai, reference, fastq_basename): '''Use BWA to align PE data.''' sam_filename = "%s.sam" % (fastq_basename) badcigar_filename = "%s.badReads" % (fastq_basename) + bam_filename = '%s.bam' % (fastq_basename) # Remove read pairs with bad CIGAR strings and sort by position steps = [ @@ -150,12 +153,14 @@ def align_pe(fastq, sai, reference, fastq_basename): "grep -v -F -f %s" % (badcigar_filename), "samtools view -@%d -Su -" % (cpu_count()), "samtools sort -@%d -o %s" - % (cpu_count(), fastq_basename)] + % (cpu_count(), bam_filename)] out, err = utils.run_pipe(steps) if err: logger.error("samtools error: %s" % (err)) + return bam_filename + def main(): args = get_args() @@ -184,18 +189,18 @@ def main(): strip_extensions(fastq[1], STRIP_EXTENSIONS)) fastq_basename = fastq_r1_basename + fastq_r2_basename - align_pe(fastq, sai, reference, fastq_basename) + bam_filename = align_pe(fastq, sai, reference, fastq_basename) else: fastq_basename = os.path.basename( strip_extensions(fastq[0], STRIP_EXTENSIONS)) - align_se(fastq, sai, reference, fastq_basename) + bam_filename = align_se(fastq, sai, reference, fastq_basename) bam_mapstats_filename = '%s.raw.srt.bam.flagstat.qc' % (fastq_basename) with open(bam_mapstats_filename, 'w') as fh: subprocess.check_call( - shlex.split("samtools flagstat %s" % (bam_mapstats_filename)), + shlex.split("samtools flagstat %s" % (bam_filename)), stdout=fh) # Remove sai files -- GitLab