adding treeclimber and unifrac pieces

[mothur.git] / treemap.h

diff --git a/treemap.h b/treemap.h
index d9617f5455e03f0f5fc702fffaf3516335351b15..60f75b388571aff503ac95b96df5946e6fcfa89f 100644 (file)
--- a/treemap.h
+++ b/treemap.h
@@ -26,6 +26,7 @@ struct GroupIndex {
 
 class TreeMap {
 public:
+       TreeMap() {};
        TreeMap(string);
        ~TreeMap();
        void readMap();
@@ -36,6 +37,8 @@ public:
        string getGroup(string);
        vector<string> namesOfGroups;
        vector<string> namesOfSeqs;
+    map<string,int> seqsPerGroup;      //groupname, number of seqs in that group.
+       map<string, GroupIndex> treemap; //sequence name and groupname
        void print(ostream&);
        
 private:
@@ -43,8 +46,9 @@ private:
        string groupFileName;
        int numGroups;
        map<string, GroupIndex>::iterator it;
+       map<string, int>::iterator it2;
        void setNamesOfGroups(string); 
-       map<string, GroupIndex> treemap; //sequence name and groupname
+       
 };
 
 #endif