X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=weighted.h;h=d78edf4d49ffd17e24f8eb559747a43952ad522a;hb=fca3f55d5ded10c3dc77856f3cc4a1c53b02bb6f;hp=8a8ad8f2642c35083cba53d6d93e2210f113f7e0;hpb=f099fdc1e3a0d7b75d780a164e5bdb93496a7a1d;p=mothur.git

diff --git a/weighted.h b/weighted.h
index 8a8ad8f..d78edf4 100644
--- a/weighted.h
+++ b/weighted.h
@@ -19,21 +19,19 @@
 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, vector<double>&);
+		EstOutput getValues(Tree*, string, string);
 		EstOutput getValues(Tree*, int, string);
 		
-		vector<double> getBranchLengthSums(Tree*);
-		
 	private:
 		struct linePair {
 			int start;
 			int num;
 			linePair(int i, int j) : start(i), num(j) {}
 		};
-		vector<linePair*> lines;
+		vector<linePair> lines;
 
 		GlobalData* globaldata;
 		EstOutput data;
@@ -42,9 +40,12 @@ class Weighted : public TreeCalculator  {
 		map<string, double> WScore; //a score for each group combination i.e. AB, AC, BC.
 		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, vector<double>&); 
-		EstOutput createProcesses(Tree*, vector< vector<string> >, vector<double>&);
+		EstOutput driver(Tree*, vector< vector<string> >, int, int); 
+		EstOutput createProcesses(Tree*, vector< vector<string> >);
+		double getLengthToRoot(Tree*, int, string, string);
 };
 
 /***********************************************************************/