added query nucleotide- and file count

fethalen · Apr 23, 2024 · 8e2d6ad · 8e2d6ad
1 parent 207f037
commit 8e2d6ad
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/fasta.jl b/src/fasta.jl
@@ -248,13 +248,14 @@ function fasta_countrecords(file::AbstractString)
     end
 
     recordcount = 0
-    for _ in reader
+    for record in reader
         recordcount += 1
+        nucleotides += length(FASTX.sequence(record))
     end
 
     close(reader)
 
-    return recordcount
+    return recordcount, nucleotides
 end
 
 
diff --git a/src/fastq.jl b/src/fastq.jl
@@ -145,11 +145,13 @@ function fastq_countrecords(file::AbstractString)
     end
 
     recordcount = 0
-    for _ in reader
+    nucleotides = 0
+    for record in reader
         recordcount += 1
+        nucleotides += length(FASTX.sequence(record))
     end
 
     close(reader)
 
-    return recordcount
+    return recordcount, nucleotides
 end