some changes while testing 1.9

[mothur.git] / kmerdb.hpp

diff --git a/kmerdb.hpp b/kmerdb.hpp
index 5019858dcf477c2e27bcc9ad3007617e26d3ce96..513f3f07d64bbd267e11bd16777fae5aadb97a5c 100644 (file)
--- a/kmerdb.hpp
+++ b/kmerdb.hpp
@@ -20,19 +20,33 @@
  */
 
 #include "mothur.h"
+#include "database.hpp"
 
 class KmerDB : public Database {
        
 public:
        KmerDB(string, int);
+       KmerDB();
        ~KmerDB();
-       Sequence findClosestSequence(Sequence*);
-
+       
+       void generateDB();
+       void addSequence(Sequence);
+       vector<int> findClosestSequences(Sequence*, int);
+       void readKmerDB(ifstream&);
+       int getCount(int);  //returns number of sequences with that kmer number
+       vector<int> getSequencesWithKmer(int);  //returns vector of sequences that contain kmer passed in
+       int getMaxKmer() { return maxKmer; }
+       
+       #ifdef USE_MPI  
+       int MPISend(int); //just sends kmersize
+       int MPIRecv(int);
+       #endif
+       
 private:
-       void generateKmerDB(string);
-       void readKmerDB(string, ifstream&);
+       
        int kmerSize;
-       int maxKmer;
+       int maxKmer, count;
+       string kmerDBName;
        vector<vector<int> > kmerLocations;
 };