added modify names parameter to set.dir

[mothur.git] / treecalculator.h

diff --git a/treecalculator.h b/treecalculator.h
index 797c45723f835fea908fe2829397b682edc32c6f..06abdb8e278e5dd4c04536444745f5f71679f092 100644 (file)
--- a/treecalculator.h
+++ b/treecalculator.h
@@ -12,6 +12,7 @@
 
 #include "mothur.h"
 #include "tree.h"
+#include "mothurout.h"
 
 /* The tree calculator class is the parent class for tree calculators in mothur. */ 
 
@@ -22,17 +23,21 @@ typedef vector<double> EstOutput;
 class TreeCalculator {
 
 public:
-       TreeCalculator(){};
+       TreeCalculator(){ m = MothurOut::getInstance(); }
        TreeCalculator(string n) : name(n) {};
+       
        virtual ~TreeCalculator(){};
-       virtual EstOutput getValues(Tree*) = 0; 
-       virtual EstOutput getValues(Tree*, string, string) = 0;
+       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;
 
 };