changes while testing

[mothur.git] / validcalculator.h

diff --git a/validcalculator.h b/validcalculator.h
index ff93c58ccc980684ae52f48b08cac9d380cf7a88..59f6bb3488ddb03acdc9a8682b748e9347f04fb9 100644 (file)
--- a/validcalculator.h
+++ b/validcalculator.h
@@ -11,6 +11,7 @@
  */
 
 #include "mothur.h"
+#include "mothurout.h"
 
 //This class contains a list of all valid calculators in Mothur.  
 //It has a function which will tell you if your calculator is valid for the given parameter.
@@ -23,6 +24,8 @@ class ValidCalculators {
                ~ValidCalculators();
                bool isValidCalculator(string, string);
                void printCalc(string, ostream&);
+               string printCalc(string);
+               void printCitations(vector<string>);
                
        private:
                map<string, string> single;
@@ -35,9 +38,11 @@ class ValidCalculators {
                map<string, string> vennshared;
                map<string, string> treegroup;
                map<string, string> matrix;
+               map<string, string> heat;
                map<string, string> boot;
                map<string, string> distance;
                map<string, string>::iterator it;
+               set<string> allCalcs;
                
                void initialSingle();
                void initialShared();
@@ -51,6 +56,9 @@ class ValidCalculators {
                void initialMatrix();
                void initialBoot();
                void initialDistance();
+               void initialHeat();
+               
+               MothurOut* m;
 };
 
 #endif