added versioning info to all shortcut files mothur makes.

[mothur.git] / treemap.h

diff --git a/treemap.h b/treemap.h
index 8f744c80a3e92873699aca320032cedeea59a007..bda6926c20ea79800e9f9cc8a06b87ce6f44c68c 100644 (file)
--- a/treemap.h
+++ b/treemap.h
@@ -25,7 +25,7 @@ class ListVector;
 
 class TreeMap {
 public:
-       TreeMap() {};
+       TreeMap() { m = MothurOut::getInstance(); }
        TreeMap(string);
        ~TreeMap();
        void readMap();
@@ -34,6 +34,7 @@ public:
        void setIndex(string, int);  //sequencename, index
        int getIndex(string);           //returns vector index of sequence
        bool isValidGroup(string);  //return true if string is a valid group
+       void removeSeq(string);  //removes a sequence, this is to accomadate trees that do not contain all the seqs in your groupfile
        string getGroup(string);
        vector<string> namesOfGroups;
        vector<string> namesOfSeqs;
@@ -50,6 +51,7 @@ private:
        map<string, GroupIndex>::iterator it;
        map<string, int>::iterator it2;
        void setNamesOfGroups(string); 
+       MothurOut* m;
        
        
 };