}
/************************************************************/
- GroupMap::~GroupMap(){};
+ GroupMap::~GroupMap(){}
/************************************************************/
void GroupMap::readMap() {
fileHandle >> seqGroup; //read from second column
setNamesOfGroups(seqGroup);
-
+
groupmap[seqName] = seqGroup; //store data in map
+ seqsPerGroup[seqGroup]++; //increment number of seqs in that group
gobble(fileHandle);
}
fileHandle.close();
}
/************************************************************/
-
-int GroupMap::getNumGroups() {
-
- return namesOfGroups.size();
-
-}
+int GroupMap::getNumGroups() { return namesOfGroups.size(); }
/************************************************************/
string GroupMap::getGroup(string sequenceName) {
}else {
return "not found";
}
-
}
/************************************************************/
-
+void GroupMap::setGroup(string sequenceName, string groupN) {
+ groupmap[sequenceName] = groupN;
+}
+/************************************************************/
void GroupMap::setNamesOfGroups(string seqGroup) {
int i, count;
count = 0;
}
if (count == namesOfGroups.size()) {
namesOfGroups.push_back(seqGroup); //new group
+ seqsPerGroup[seqGroup] = 0;
groupIndex[seqGroup] = index;
index++;
}
}
+/************************************************************/
+bool GroupMap::isValidGroup(string groupname) {
+ try {
+ for (int i = 0; i < namesOfGroups.size(); i++) {
+ if (groupname == namesOfGroups[i]) { return true; }
+ }
+
+ return false;
+ }
+ catch(exception& e) {
+ errorOut(e, "GroupMap", "isValidGroup");
+ exit(1);
+ }
+}
+/************************************************************/
+int GroupMap::getNumSeqs(string group) {
+ try {
+
+ map<string, int>::iterator itNum;
+
+ itNum = seqsPerGroup.find(group);
+
+ if (itNum == seqsPerGroup.end()) { return 0; }
+
+ return seqsPerGroup[group];
+
+ }
+ catch(exception& e) {
+ errorOut(e, "GroupMap", "getNumSeqs");
+ exit(1);
+ }
+}
+
+/************************************************************/
+vector<string> GroupMap::getNamesSeqs(){
+ try {
+
+ vector<string> names;
+
+ for (it = groupmap.begin(); it != groupmap.end(); it++) {
+ names.push_back(it->first);
+ }
+
+ return names;
+ }
+ catch(exception& e) {
+ errorOut(e, "GroupMap", "getNamesSeqs");
+ exit(1);
+ }
+}
+/************************************************************/
+