X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=treereader.h;fp=treereader.h;h=fb9c791959f96cd04ce95fae0385f02c9740f546;hb=f55cf350ca6643f8eb070d8336e1957699a3f109;hp=0000000000000000000000000000000000000000;hpb=eb4bb52af987ff4df7354d95d60cd763aec537e1;p=mothur.git

diff --git a/treereader.h b/treereader.h
new file mode 100644
index 0000000..fb9c791
--- /dev/null
+++ b/treereader.h
@@ -0,0 +1,44 @@
+#ifndef Mothur_treereader_h
+#define Mothur_treereader_h
+
+//
+//  treereader.h
+//  Mothur
+//
+//  Created by Sarah Westcott on 4/11/12.
+//  Copyright (c) 2012 Schloss Lab. All rights reserved.
+//
+
+#include "mothurout.h"
+#include "tree.h"
+
+class TreeReader {
+    
+public:
+    
+    TreeReader(string tf);
+	TreeReader(string tf, string gf);
+    TreeReader(string tf, string gf, string nf);
+	~TreeReader() {}	
+    
+    vector<Tree*> getTrees()            { return trees;     }
+    map<string, string> getNames()      { return nameMap;   } //dups -> unique
+    map<string, string> getNameMap()    { return names;     } //unique -> dups list
+    
+    
+private:
+    MothurOut* m;
+	vector<Tree*> trees;
+    TreeMap* tmap;
+    map<string, string> nameMap; //dupName -> uniqueName
+    map<string, string> names;
+    
+    string treefile, groupfile, namefile;
+    
+    bool readTrees();
+    int readNamesFile();
+};
+
+
+
+#endif