changes while testing

[mothur.git] / phylosummary.h

diff --git a/phylosummary.h b/phylosummary.h
index 04ba65f8ad6c93a5db879a323a2a6b6ed890a3c5..bd31173a48647e1bf2ed7b2f7ec95d25e0bfe388 100644 (file)
--- a/phylosummary.h
+++ b/phylosummary.h
@@ -13,6 +13,7 @@
 #include "mothur.h"
 #include "mothurout.h"
 #include "groupmap.h"
+#include "counttable.h"
 
 /**************************************************************************************************/
 
@@ -32,23 +33,28 @@ struct rawTaxNode {
 class PhyloSummary {
 
 public:
-       PhyloSummary(string);
-       PhyloSummary(string, string);
-       ~PhyloSummary() { if (groupmap != NULL)  {  delete groupmap;  }  }
+       PhyloSummary(GroupMap*);
+       PhyloSummary(string, GroupMap*);
+    PhyloSummary(CountTable*);
+       PhyloSummary(string, CountTable*);
+       ~PhyloSummary() {}
        
-       void summarize(string);  //pass it a taxonomy file and a group file and it makes the tree
+       int summarize(string);  //pass it a taxonomy file and a group file and it makes the tree
        int addSeqToTree(string, string);
-       int addSeqToTree(string, vector<string>);
+       int addSeqToTree(string, map<string, bool>);
        void print(ofstream&);
+    void print(ofstream&, bool);
        int getMaxLevel() { return maxLevel; }
        
 private:
        string getNextTaxon(string&);
        vector<rawTaxNode> tree;
        void print(int, ofstream&);
+    void print(int, ofstream&, bool);
        void assignRank(int);
        void readTreeStruct(ifstream&);
        GroupMap* groupmap;
+    CountTable* ct;
        bool ignore;
        
        int numNodes;