adding treeclimber and unifrac pieces

[mothur.git] / treemap.cpp

diff --git a/treemap.cpp b/treemap.cpp
index 6fb853682e7c468c2ef5227eb48b9688317a61bb..bee0d11ed21323ee9dff1bc38c860a3042231150 100644 (file)
--- a/treemap.cpp
+++ b/treemap.cpp
@@ -31,7 +31,14 @@ void TreeMap::readMap() {
                        setNamesOfGroups(seqGroup);
                                                
                        treemap[seqName].groupname = seqGroup;  //store data in map
-               
+                       
+                       it2 = seqsPerGroup.find(seqGroup);
+                       if (it2 == seqsPerGroup.end()) { //if it's a new group
+                               seqsPerGroup[seqGroup] = 1;
+                       }else {//it's a group we already have
+                               seqsPerGroup[seqGroup]++;
+                       }
+
                        gobble(fileHandle);
                }
                fileHandle.close();
@@ -40,7 +47,7 @@ void TreeMap::readMap() {
 
 int TreeMap::getNumGroups() {
                        
-       return namesOfGroups.size();    
+       return seqsPerGroup.size();     
                
 }
 /************************************************************/