use iterator for non zero... wayy faster

1 files changed, 8 insertions, 11 deletions

diff --git a/kmer_utils.c b/kmer_utils.c
index c46e580..bc78999 100644
--- a/kmer_utils.c
+++ b/kmer_utils.c
@@ -363,19 +363,16 @@ void print_kmer(kmer_map *counts, bool label, bool nonzero, unsigned int kmer) {
 	if(nonzero) {
 		// if labels is set, print out our labels
 		if(label) {
-			for(i = 0; i < width; i++)
-				if(counts->count(i) != 0) {
-					char *kmer_str = index_to_kmer(i, kmer);
-					fprintf(stdout, "%s\t%llu\n", kmer_str, counts->at(i));
-					free(kmer_str);
-				}
-
+			for(auto it = counts->begin(); it != counts->end(); it++ ) {
+				char *kmer_str = index_to_kmer(it->first, kmer);
+				fprintf(stdout, "%s\t%llu\n", kmer_str, it->second);
+				free(kmer_str);
+			}
 		}
 		else {
-			for(i = 0; i < width; i++)
-				if(counts->count(i) != 0) 
-					fprintf(stdout, "%zu\t%llu\n", i, counts->at(i));
-
+			for(auto it = counts->begin(); it != counts->end(); it++ ) {
+				fprintf(stdout, "%zu\t%llu\n", it->first, it->second);
+			}
 		}
 	}
 	// If we aren't printing nonzeros print everything