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

[mothur.git] / unweighted.h

diff --git a/unweighted.h b/unweighted.h
index 598af1a9a561fb51a5861c9e26a4c1994b1d2ad3..0ccfab55ad9500f46f33d74e1c1c19f95ec525ba 100644 (file)
--- a/unweighted.h
+++ b/unweighted.h
@@ -19,7 +19,7 @@
 class Unweighted : public TreeCalculator  {
        
        public:
-               Unweighted(TreeMap* t) : tmap(t) {};
+               Unweighted(TreeMap* t, bool r) : tmap(t), includeRoot(r) {};
                ~Unweighted() {};
                EstOutput getValues(Tree*, int, string);
                EstOutput getValues(Tree*, string, string, int, string);
@@ -38,6 +38,7 @@ class Unweighted : public TreeCalculator  {
                int processors;
                string outputDir;
                map< vector<string>, set<int> > rootForGrouping;  //maps a grouping combo to the roots for that combo
+               bool includeRoot;
                
                EstOutput driver(Tree*, vector< vector<string> >, int, int); 
                EstOutput createProcesses(Tree*, vector< vector<string> >);