finished work on classify.seqs bayesian method and various bug fixes

[mothur.git] / kmerdb.cpp

diff --git a/kmerdb.cpp b/kmerdb.cpp
index bf8679e57a932af0567df8a456f038b76fbf0fae..2560a443044016d8096f7ce2f7cf0636291e97c1 100644 (file)
--- a/kmerdb.cpp
+++ b/kmerdb.cpp
@@ -176,3 +176,36 @@ void KmerDB::readKmerDB(ifstream& kmerDBFile){
 }
 
 /**************************************************************************************************/
+int KmerDB::getCount(int kmer) {
+       try {
+               if (kmer < 0) { return 0; }  //if user gives negative number
+               else if (kmer > maxKmer) {      return 0;       }  //or a kmer that is bigger than maxkmer
+               else {  return kmerLocations[kmer].size();      }  // kmer is in vector range
+       }
+       catch(exception& e) {
+               errorOut(e, "KmerDB", "getCount");
+               exit(1);
+       }       
+}
+/**************************************************************************************************/
+vector<int> KmerDB::getSequencesWithKmer(int kmer) {
+       try {
+               
+               vector<int> seqs;
+       
+               if (kmer < 0) { }  //if user gives negative number
+               else if (kmer > maxKmer) {      }  //or a kmer that is bigger than maxkmer
+               else {  seqs = kmerLocations[kmer];     }
+               
+               return seqs;
+       }
+       catch(exception& e) {
+               errorOut(e, "KmerDB", "getSequencesWithKmer");
+               exit(1);
+       }       
+}
+
+
+/**************************************************************************************************/
+
+