pds' small change to sffinfo

[mothur.git] / treecalculator.h

diff --git a/treecalculator.h b/treecalculator.h
index 16663bf161afbb61c92722c27facdd8885a0dad5..86cb52798a291990ea7d0d20adfdf9c14f0ea569 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,9 +23,9 @@ typedef vector<double> EstOutput;
 class TreeCalculator {
 
 public:
-       TreeCalculator(){};
+       TreeCalculator(){ m = MothurOut::getInstance(); }
        TreeCalculator(string n) : name(n) {};
-       ~TreeCalculator(){};
+       virtual ~TreeCalculator(){};
        virtual EstOutput getValues(Tree*) = 0; 
        virtual EstOutput getValues(Tree*, string, string) = 0;
        
@@ -33,9 +34,10 @@ public:
 protected:
        EstOutput data;
        string name;
+       MothurOut* m;
 
 };
 
 /***********************************************************************/
 
-#endif
\ No newline at end of file
+#endif