bugs fixes while testing for 1.5 release

[mothur.git] / treemap.cpp

diff --git a/treemap.cpp b/treemap.cpp
index 044e87ca6c603f8762edc863e5adf827ede73bad..e472570895414a4ec535e10c5f02bee10dfea982 100644 (file)
--- a/treemap.cpp
+++ b/treemap.cpp
@@ -44,6 +44,27 @@ void TreeMap::readMap() {
                fileHandle.close();
 }
 /************************************************************/
+void TreeMap::removeSeq(string seqName) {
+       
+       //erase name from namesOfSeqs
+       for (int i = 0; i < namesOfSeqs.size(); i++) {
+               if (namesOfSeqs[i] == seqName)  {
+                       namesOfSeqs.erase (namesOfSeqs.begin()+i);
+                       break;
+               }
+       }
+       
+       //decrement sequences in this group
+       string group = treemap[seqName].groupname;
+       seqsPerGroup[group]--;
+       
+       //remove seq from treemap
+       it = treemap.find(seqName);
+       treemap.erase(it);
+       
+
+}
+/************************************************************/
 
 int TreeMap::getNumGroups() {