X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=unweighted.h;h=b136b007b517e56645be544ee6556f83a45de649;hb=b740fb182372143e152143c6cc5b9ac8591f09b7;hp=87514699b35b74e3b580f4d26dfbc90ca948f6c8;hpb=fc7cf3aac8fd6106fd725b43baa8ab5ca6f836f8;p=mothur.git

diff --git a/unweighted.h b/unweighted.h
index 8751469..b136b00 100644
--- a/unweighted.h
+++ b/unweighted.h
@@ -12,28 +12,39 @@
  */
 
 #include "treecalculator.h"
-#include "treemap.h"
+#include "counttable.h"
 
 /***********************************************************************/
 
 class Unweighted : public TreeCalculator  {
 	
 	public:
-		Unweighted(TreeMap* t) : tmap(t) {};
+        Unweighted(bool r) : includeRoot(r) {};
 		~Unweighted() {};
-		EstOutput getValues(Tree*);
-		EstOutput getValues(Tree*, string, string);
+		EstOutput getValues(Tree*, int, string);
+		EstOutput getValues(Tree*, string, string, int, string);
 		
 	private:
-		GlobalData* globaldata;
-		Tree* copyTree;
+		struct linePair {
+			int start;
+			int num;
+			linePair(int i, int j) : start(i), num(j) {}
+		};
+		vector<linePair> lines;
+		
 		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, CountTable*); 
+		EstOutput createProcesses(Tree*, vector< vector<string> >, CountTable*);
+		EstOutput driver(Tree*, vector< vector<string> >, int, int, bool, CountTable*); 
+		EstOutput createProcesses(Tree*, vector< vector<string> >, bool, CountTable*);
+		int getRoot(Tree*, int, vector<string>);
 };
 
 /***********************************************************************/
 
-
-
-#endif
\ No newline at end of file
+#endif