Single-sample jobs for count file generation

tarmac/src/main/scala/nl/lumc/sasc/biopet/pipelines/tarmac/Tarmac.scala

@@ -4,14 +4,22 @@ import java.io.File
 
 import nl.lumc.sasc.biopet.core.{ PedigreeQscript, PipelineCommand, Reference }
 import nl.lumc.sasc.biopet.core.summary.SummaryQScript
+import nl.lumc.sasc.biopet.extensions.Ln
+import nl.lumc.sasc.biopet.extensions.gatk.DepthOfCoverage
+import nl.lumc.sasc.biopet.extensions.wisecondor.WisecondorCount
+import nl.lumc.sasc.biopet.utils.Logging
 import nl.lumc.sasc.biopet.utils.config.Configurable
 import org.broadinstitute.gatk.queue.QScript
 
+import scalaz.{ -\/, \/, \/- }
+
 /**
  * Created by Sander Bollen on 23-3-17.
  */
 class Tarmac(val root: Configurable) extends QScript with PedigreeQscript with SummaryQScript with Reference {
   qscript =>
+
+  private val targets: File = config("targets")
   def this() = this(null)
 
   def init() = {
@@ -29,13 +37,76 @@ class Tarmac(val root: Configurable) extends QScript with PedigreeQscript with S
 
   class Sample(name: String) extends AbstractSample(name) {
 
-    val inputCountFile: Option[File] = config("count_file")
+    val inputXhmmCountFile: Option[File] = config("xhmm_count_file")
+    val inputWisecondorCountFile: Option[File] = config("wisecondor_count_file")
     val bamFile: Option[File] = config("bam")
 
+    /**
+     * Create XHMM count file or create link to input count file
+     * Precedence is given to existing count files.
+     * Returns a disjunction where right is the file, and left is
+     * a potential error message
+     */
+    lazy val outputXhmmCountFile: String \/ File = {
+      val outFile = new File(sampleDir + File.separator + s"$name.dcov")
+      (inputXhmmCountFile, bamFile) match {
+        case (Some(f), _) => {
+          val ln = new Ln(root)
+          ln.input = f
+          ln.output = outFile
+          add(ln)
+          \/-(ln.output)
+        }
+        case (None, Some(bam)) => {
+          val dcov = DepthOfCoverage(root, List(bam), outFile, List(targets))
+          add(dcov)
+          \/-(dcov.out)
+        }
+        case _ => -\/(s"Cannot find bam file or xhmm count file for sample" +
+          s" $name in config. At least one must be given.")
+      }
+
+    }
+
+    /**
+     * Create wisecondor count file or create link to input count file.
+     * Precedence is given to existing count files.
+     * Returns a disjunction where right is the file, and left is
+     * a potential error message
+     */
+    lazy val outputWisecondorCountFile: String \/ File = {
+      val outFile = new File(sampleDir + File.separator + s"$name.wisecondor.bed")
+      (inputWisecondorCountFile, bamFile) match {
+        case (Some(f), _) => {
+          val ln = new Ln(root)
+          ln.input = f
+          ln.output = outFile
+          add(ln)
+          \/-(ln.output)
+        }
+        case (None, Some(bam)) => {
+          val counter = new WisecondorCount(root)
+          counter.inputBam = bam
+          counter.output = outFile
+          counter.binFile = Some(targets)
+          add(counter)
+          \/-(counter.output)
+        }
+        case _ => -\/(s"Cannot find bam file or wisecondor count for sample" +
+          s" $name. At least one must be given.")
+      }
+
+    }
+
     /** Function to add sample jobs */
-    def addJobs(): Unit = {}
+    def addJobs(): Unit = {
+      (outputWisecondorCountFile :: outputXhmmCountFile :: Nil).foreach {
+        case -\/(error) => Logging.addError(error)
+        case _          =>
+      }
+    }
 
-    /* This is necesary for compile reasons, but library does not in fact exist for this pipeline */
+    /* This is necessary for compile reasons, but library does not in fact exist for this pipeline */
     def makeLibrary(id: String) = new Library(id)
 
     class Library(id: String) extends AbstractLibrary(id) {