fix kmer count

sharkLoc · Jun 14, 2024 · d7f8d98 · d7f8d98
1 parent c73d90a
commit d7f8d98
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/cli/kmer.rs b/src/cli/kmer.rs
@@ -22,12 +22,13 @@ pub fn kmer_count(
 
     let mut writer = file_writer(output, compression_level)?;
     let mut kmers = HashMap::new();
-    let (mut sidx, mut eidx) = (0, kmer_len);
 
     for rec in reader.records().flatten() {
+        let (mut sidx, mut eidx) = (0, kmer_len);
         let khash = nthash(rec.seq(), kmer_len);
-
-        while eidx <= rec.seq().len() {
+        let len = rec.seq().len();
+
+        while eidx <= len {
             let kseq = &rec.seq()[sidx..eidx];
             let khash_this = nthash(kseq, kmer_len)[0];
             if khash.contains(&khash_this) {