X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=unweighted.h;h=c6c13bb3a7abd14c824dfe6b495871790809833d;hb=529ec122f7cac4af987e121d150b878d7c7a0d5d;hp=f41326115020013fb2e59df755bff1e961001c8d;hpb=a98eb683e17d8e49583bf2d215ab7562a4cdca75;p=mothur.git

diff --git a/unweighted.h b/unweighted.h
index f413261..c6c13bb 100644
--- 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>);
 };
 
 /***********************************************************************/