fixed bug in unifrac commands with unrooted trees

[mothur.git] / unweighted.h

diff --git a/unweighted.h b/unweighted.h
index f41326115020013fb2e59df755bff1e961001c8d..598af1a9a561fb51a5861c9e26a4c1994b1d2ad3 100644 (file)
--- a/unweighted.h
+++ b/unweighted.h
@@ -37,12 +37,13 @@ class Unweighted : public TreeCalculator  {
                TreeMap* tmap;
                int processors;
                string outputDir;
+               map< vector<string>, set<int> > rootForGrouping;  //maps a grouping combo to the roots for that combo
                
                EstOutput driver(Tree*, vector< vector<string> >, int, int); 
                EstOutput createProcesses(Tree*, vector< vector<string> >);
                EstOutput driver(Tree*, vector< vector<string> >, int, int, bool); 
                EstOutput createProcesses(Tree*, vector< vector<string> >, bool);
-
+               int getRoot(Tree*, int, vector<string>);
 };
 
 /***********************************************************************/