}
/************************************************************/
- TreeMap::~TreeMap(){};
+ TreeMap::~TreeMap(){}
/************************************************************/
void TreeMap::readMap() {
while(fileHandle){
fileHandle >> seqName; //read from first column
fileHandle >> seqGroup; //read from second column
-
+
namesOfSeqs.push_back(seqName);
setNamesOfGroups(seqGroup);
-
+
treemap[seqName].groupname = seqGroup; //store data in map
it2 = seqsPerGroup.find(seqGroup);
fileHandle.close();
}
/************************************************************/
+void TreeMap::removeSeq(string seqName) {
+
+ //erase name from namesOfSeqs
+ for (int i = 0; i < namesOfSeqs.size(); i++) {
+ if (namesOfSeqs[i] == seqName) {
+ namesOfSeqs.erase(namesOfSeqs.begin()+i);
+ break;
+ }
+ }
+
+ //decrement sequences in this group
+ string group = treemap[seqName].groupname;
+ seqsPerGroup[group]--;
+
+ //remove seq from treemap
+ it = treemap.find(seqName);
+ treemap.erase(it);
+}
+/************************************************************/
int TreeMap::getNumGroups() {
}
/************************************************************/
void TreeMap::setIndex(string seq, int index) {
- treemap[seq].vectorIndex = index;
+ it = treemap.find(seq);
+ if (it != treemap.end()) { //sequence name was in group file
+ treemap[seq].vectorIndex = index;
+ }else {
+ treemap[seq].vectorIndex = index;
+ treemap[seq].groupname = "not found";
+ }
}
/************************************************************/
int TreeMap::getIndex(string seq) {
return false;
}
catch(exception& e) {
- cout << "Standard Error: " << e.what() << " has occurred in the TreeMap class Function isValidGroup. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
- exit(1);
- }
- catch(...) {
- cout << "An unknown error has occurred in the TreeMap class function isValidGroup. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
+ m->errorOut(e, "TreeMap", "isValidGroup");
exit(1);
}
}
}
}
catch(exception& e) {
- cout << "Standard Error: " << e.what() << " has occurred in the TreeMap class Function print. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
+ m->errorOut(e, "TreeMap", "print");
+ exit(1);
+ }
+}
+
+/************************************************************/
+void TreeMap::makeSim(GroupMap* groupmap) {
+ try {
+ //set names of groups
+ namesOfGroups = groupmap->namesOfGroups;
+
+ //set names of seqs to names of groups
+ namesOfSeqs = groupmap->namesOfGroups;
+
+ // make map where key and value are both the group name since that what the tree.shared command wants
+ for (int i = 0; i < namesOfGroups.size(); i++) {
+ treemap[namesOfGroups[i]].groupname = namesOfGroups[i];
+ seqsPerGroup[namesOfGroups[i]] = 1;
+ }
+
+ numGroups = namesOfGroups.size();
+
+ }
+ catch(exception& e) {
+ m->errorOut(e, "TreeMap", "makeSim");
exit(1);
}
- catch(...) {
- cout << "An unknown error has occurred in the TreeMap class function print. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
+}
+/************************************************************/
+void TreeMap::makeSim(ListVector* list) {
+ try {
+ //set names of groups
+ namesOfGroups.clear();
+ for(int i = 0; i < list->size(); i++) {
+ namesOfGroups.push_back(list->get(i));
+ }
+
+ //set names of seqs to names of groups
+ namesOfSeqs = namesOfGroups;
+
+ // make map where key and value are both the group name since that what the tree.shared command wants
+ for (int i = 0; i < namesOfGroups.size(); i++) {
+ treemap[namesOfGroups[i]].groupname = namesOfGroups[i];
+ seqsPerGroup[namesOfGroups[i]] = 1;
+ }
+
+ numGroups = namesOfGroups.size();
+
+ }
+ catch(exception& e) {
+ m->errorOut(e, "TreeMap", "makeSim");
exit(1);
}
}
/************************************************************/
+