added oligos, pdiffs, bdiffs, ldiffs, sdiffs, tiffs parameters to sffinfo to allow...

[mothur.git] / unweighted.h

diff --git a/unweighted.h b/unweighted.h
index f41326115020013fb2e59df755bff1e961001c8d..c6c13bb3a7abd14c824dfe6b495871790809833d 100644 (file)
--- a/unweighted.h
+++ b/unweighted.h
@@ -19,7 +19,7 @@
 class Unweighted : public TreeCalculator  {
        
        public:
-               Unweighted(TreeMap* t) : tmap(t) {};
+        Unweighted(bool r) : includeRoot(r) {};
                ~Unweighted() {};
                EstOutput getValues(Tree*, int, string);
                EstOutput getValues(Tree*, string, string, int, string);
@@ -32,17 +32,17 @@ class Unweighted : public TreeCalculator  {
                };
                vector<linePair> lines;
                
-               GlobalData* globaldata;
                EstOutput data;
-               TreeMap* tmap;
                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> >);
-               EstOutput driver(Tree*, vector< vector<string> >, int, int, bool); 
-               EstOutput createProcesses(Tree*, vector< vector<string> >, bool);
-
+               EstOutput driver(Tree*, vector< vector<string> >, int, int, TreeMap*); 
+               EstOutput createProcesses(Tree*, vector< vector<string> >, TreeMap*);
+               EstOutput driver(Tree*, vector< vector<string> >, int, int, bool, TreeMap*); 
+               EstOutput createProcesses(Tree*, vector< vector<string> >, bool, TreeMap*);
+               int getRoot(Tree*, int, vector<string>);
 };
 
 /***********************************************************************/