X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=treemap.cpp;fp=treemap.cpp;h=450d8fba67b8beafe7c824b6a18b479705b3875c;hb=03dca3b32a903c3f29fbcf5b410b19d6ab6dae63;hp=1fc5c01b796a67bd948a70147fa4dead9320ea08;hpb=22e2ebf2ee6462b0a828deed0499124975ede6e5;p=mothur.git

diff --git a/treemap.cpp b/treemap.cpp
index 1fc5c01..450d8fb 100644
--- a/treemap.cpp
+++ b/treemap.cpp
@@ -229,6 +229,60 @@ void TreeMap::makeSim(ListVector* list) {
 		exit(1);
 	}
 }
+/************************************************************/
+int TreeMap::getCopy(TreeMap* copy){
+	try {
+         
+        namesOfGroups = copy->getNamesOfGroups();
+		numGroups = copy->getNumGroups();
+        namesOfSeqs = copy->namesOfSeqs;
+        seqsPerGroup = copy->seqsPerGroup;
+        treemap = copy->treemap;
+        
+        return 0;
+	}
+	catch(exception& e) {
+		m->errorOut(e, "TreeMap", "getCopy");
+		exit(1);
+	}
+}
+/************************************************************/
+vector<string> TreeMap::getNamesSeqs(){
+	try {
+        
+		vector<string> names;
+		
+        for(it = treemap.begin(); it != treemap.end(); it++){
+            names.push_back(it->first);
+		}
+		
+		return names;
+	}
+	catch(exception& e) {
+		m->errorOut(e, "TreeMap", "getNamesSeqs");
+		exit(1);
+	}
+}
+/************************************************************/
+vector<string> TreeMap::getNamesSeqs(vector<string> picked){
+	try {
+		
+		vector<string> names;
+		
+		for(it = treemap.begin(); it != treemap.end(); it++){
+			//if you are belong to one the the groups in the picked vector add you
+			if (m->inUsersGroups(it->second.groupname, picked)) {
+				names.push_back(it->first);
+			}
+		}
+		
+		return names;
+	}
+	catch(exception& e) {
+		m->errorOut(e, "TreeMap", "getNamesSeqs");
+		exit(1);
+	}
+}
 
 /************************************************************/