fixed bug in bootstrap command that was caused by globaldata's breakup, and made...

[mothur.git] / treemap.h

diff --git a/treemap.h b/treemap.h
index 60fb924278c8df36a405ca363c2584d5ae2fa42a..8f744c80a3e92873699aca320032cedeea59a007 100644 (file)
--- a/treemap.h
+++ b/treemap.h
@@ -9,11 +9,9 @@
  *
  */
 
-#include <map>
-#include <string>
-#include <iostream>
-#include <fstream>
-#include "utilities.hpp"
+#include "mothur.h"
+#include "groupmap.h"
+#include "listvector.hpp"
 
 /* This class is used by the read.tree command to build the tree container. */
 
@@ -22,7 +20,8 @@ struct GroupIndex {
        int             vectorIndex;
 };
 
-
+class GroupMap;
+class ListVector;
 
 class TreeMap {
 public:
@@ -41,6 +40,8 @@ public:
     map<string,int> seqsPerGroup;      //groupname, number of seqs in that group.
        map<string, GroupIndex> treemap; //sequence name and <groupname, vector index>
        void print(ostream&);
+       void makeSim(GroupMap*);  //takes groupmap info and fills treemap for use by tree.shared command.
+       void makeSim(ListVector*);  //takes listvector info and fills treemap for use by tree.shared command.   
        
 private:
        ifstream fileHandle;
@@ -50,6 +51,7 @@ private:
        map<string, int>::iterator it2;
        void setNamesOfGroups(string); 
        
+       
 };
 
 #endif