closes #151

brentp · May 18, 2021 · 09595b1 · 09595b1
1 parent 2a0c183
commit 09595b1
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 10 deletions.
diff --git a/HISTORY.md b/HISTORY.md
@@ -2,6 +2,7 @@ v0.2.7
 ======
 + use csi index for bams to support larger organisms
 + fix ordering of error-checking (#146)
++ support for longer file-lists (#151)
 
 v0.2.6
 ======

diff --git a/paste/paste.go b/paste/paste.go
@@ -98,17 +98,27 @@ func Main() {
 
 	cli := cliargs{OutDir: "./"}
 	arg.MustParse(&cli)
-
-	wg := &sync.WaitGroup{}
-	wg.Add(len(cli.VCFs))
-	procs := 5
-
-	go count(wg, procs, cli.VCFs)
-
 	outvcf := fmt.Sprintf(filepath.Join(cli.OutDir, cli.Name) + ".smoove.square.vcf.gz")
-	shared.Slogger.Printf("squaring %d files to %s", len(cli.VCFs), outvcf)
+	var wg *sync.WaitGroup
+
+	// TODO: check files in list
+	if !strings.HasSuffix(cli.VCFs[0], ".list") {
+		wg = &sync.WaitGroup{}
+		wg.Add(len(cli.VCFs))
+		procs := 5
+
+		go count(wg, procs, cli.VCFs)
+		shared.Slogger.Printf("squaring %d files to %s", len(cli.VCFs), outvcf)
+	} else {
+		shared.Slogger.Printf("squaring files from %s to %s", cli.VCFs[0], outvcf)
+	}
+
 	args := []string{"merge", "-o", outvcf, "-O", "z", "--threads", "3"}
-	args = append(args, cli.VCFs...)
+	if strings.HasSuffix(cli.VCFs[0], ".list") && len(cli.VCFs) == 1 {
+		args = append(args, []string{"-l", cli.VCFs[0]}...)
+	} else {
+		args = append(args, cli.VCFs...)
+	}
 
 	p := exec.Command("bcftools", args...)
 	p.Stderr = shared.Slogger
@@ -118,5 +128,7 @@ func Main() {
 		log.Fatal(err)
 	}
 	shared.Slogger.Printf("wrote squared file to %s", outvcf)
-	wg.Wait()
+	if wg != nil {
+		wg.Wait()
+	}
 }