added root parameter to the unifrac commands so you can choose to include the entire...

[mothur.git] / weighted.h

diff --git a/weighted.h b/weighted.h
index 11cd26b2d9642cb2679a0f3dfc06d930c15127f0..d78edf4d49ffd17e24f8eb559747a43952ad522a 100644 (file)
--- a/weighted.h
+++ b/weighted.h
@@ -19,7 +19,7 @@
 class Weighted : public TreeCalculator  {
        
        public:
-               Weighted(TreeMap* t) : tmap(t) {};
+               Weighted(TreeMap* t, bool r) : tmap(t), includeRoot(r) {};
                ~Weighted() {};
                
                EstOutput getValues(Tree*, string, string);
@@ -41,6 +41,7 @@ class Weighted : public TreeCalculator  {
                int processors;
                string outputDir;
                map< vector<string>, set<int> > rootForGrouping;  //maps a grouping combo to the root for that combo
+               bool includeRoot;
                
                EstOutput driver(Tree*, vector< vector<string> >, int, int); 
                EstOutput createProcesses(Tree*, vector< vector<string> >);