added modify names parameter to set.dir

[mothur.git] / treenode.h

diff --git a/treenode.h b/treenode.h
index d6163a2b1275f9816a576659019d178005042e6d..029cee9419cf7adf88344d1635e5709196bd5c73 100644 (file)
--- a/treenode.h
+++ b/treenode.h
@@ -11,6 +11,7 @@
  */
 
 #include "mothur.h"
+#include "mothurout.h"
 
 /* This class represents a node on a tree. */
 
@@ -21,7 +22,7 @@ class Node  {
                ~Node() { pGroups.clear(); pcount.clear(); };
                
                void setName(string);
-               void setGroup(string);  
+               void setGroup(vector<string>);  
                void setBranchLength(float);
                void setLabel(float);
                void setParent(int);
@@ -30,7 +31,7 @@ class Node  {
                void setLengthToLeaves(float);
                
                string getName();
-               string getGroup();  
+               vector<string> getGroup();  
                float getBranchLength();
                float getLengthToLeaves();
                float getLabel();
@@ -52,12 +53,13 @@ class Node  {
                        
        private:
                string                  name;
-               string                  group;
+               vector<string>  group; 
                float                   branchLength, length2leaf, label;
                int                             parent;
                int                             lchild;
                int                             rchild;
                int                             vectorIndex;
+               MothurOut* m;
 };             
 
 #endif