X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=treecalculator.h;fp=treecalculator.h;h=06abdb8e278e5dd4c04536444745f5f71679f092;hb=0caf3fbabaa3ece404f8ce77f4c883dc5b1bf1dc;hp=0000000000000000000000000000000000000000;hpb=1b73ff67c83892a025e597dabd9df6fe7b58206a;p=mothur.git

diff --git a/treecalculator.h b/treecalculator.h
new file mode 100644
index 0000000..06abdb8
--- /dev/null
+++ b/treecalculator.h
@@ -0,0 +1,46 @@
+#ifndef TREECALCULATOR_H
+#define TREECALCULATOR_H
+
+/*
+ *  treecalculator.h
+ *  Mothur
+ *
+ *  Created by Sarah Westcott on 1/26/09.
+ *  Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
+ *
+ */
+
+#include "mothur.h"
+#include "tree.h"
+#include "mothurout.h"
+
+/* The tree calculator class is the parent class for tree calculators in mothur. */ 
+
+typedef vector<double> EstOutput;
+
+/***********************************************************************/
+
+class TreeCalculator {
+
+public:
+	TreeCalculator(){ m = MothurOut::getInstance(); }
+	TreeCalculator(string n) : name(n) {};
+	
+	virtual ~TreeCalculator(){};
+	virtual EstOutput getValues(Tree*) { return data; }	
+	virtual EstOutput getValues(Tree*, int, string) { return data; }	
+	virtual EstOutput getValues(Tree*, string, string) { return data; }
+	virtual EstOutput getValues(Tree*, string, string, vector<double>&) { return data; }
+	
+	virtual string getName()		{	return name;	}
+		
+protected:
+	EstOutput data;
+	string name;
+	MothurOut* m;
+
+};
+
+/***********************************************************************/
+
+#endif