X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=groupmap.cpp;h=481fd1decfc4516eb5e1cf618c6b4d40cffbc8a9;hb=2bb9267aa4b4ecdf8488b06605cc9f3f36fa4332;hp=b627a6bda6e3fa63de073a4b3f6c20bee7a2a680;hpb=9354b70bb84f6dd52ff4a1955754bcbf7edeedbf;p=mothur.git

diff --git a/groupmap.cpp b/groupmap.cpp
index b627a6b..481fd1d 100644
--- a/groupmap.cpp
+++ b/groupmap.cpp
@@ -12,8 +12,9 @@
 /************************************************************/
 
  GroupMap::GroupMap(string filename) {
+	m = MothurOut::getInstance();
 	groupFileName = filename;
-	openInputFile(filename, fileHandle);
+	m->openInputFile(filename, fileHandle);
 	index = 0;
 }
 
@@ -21,22 +22,58 @@
  GroupMap::~GroupMap(){}
 
 /************************************************************/
-void GroupMap::readMap() {
+int GroupMap::readMap() {
 		string seqName, seqGroup;
+		int error = 0;
+
+		while(fileHandle){
+			fileHandle >> seqName;	m->gobble(fileHandle);		//read from first column
+			fileHandle >> seqGroup;			//read from second column
+			
+			if (m->control_pressed) {  fileHandle.close();  return 1; }
 	
+			setNamesOfGroups(seqGroup);
+			
+			it = groupmap.find(seqName);
+			
+			if (it != groupmap.end()) { error = 1; m->mothurOut("Your groupfile contains more than 1 sequence named " + seqName + ", sequence names must be unique. Please correct."); m->mothurOutEndLine();  }
+			else {
+				groupmap[seqName] = seqGroup;	//store data in map
+				seqsPerGroup[seqGroup]++;  //increment number of seqs in that group
+			}
+			m->gobble(fileHandle);
+		}
+		fileHandle.close();
+		m->setAllGroups(namesOfGroups);
+		return error;
+}
+/************************************************************/
+int GroupMap::readDesignMap() {
+		string seqName, seqGroup;
+		int error = 0;
+
 		while(fileHandle){
-			fileHandle >> seqName;			//read from first column
+			fileHandle >> seqName;	m->gobble(fileHandle);		//read from first column
 			fileHandle >> seqGroup;			//read from second column
 			
+			if (m->control_pressed) {  fileHandle.close();  return 1; }
+	
 			setNamesOfGroups(seqGroup);
 			
-			groupmap[seqName] = seqGroup;	//store data in map
-			seqsPerGroup[seqGroup]++;  //increment number of seqs in that group
-		
-			gobble(fileHandle);
+			it = groupmap.find(seqName);
+			
+			if (it != groupmap.end()) { error = 1; m->mothurOut("Your designfile contains more than 1 group named " + seqName + ", group names must be unique. Please correct."); m->mothurOutEndLine();  }
+			else {
+				groupmap[seqName] = seqGroup;	//store data in map
+				seqsPerGroup[seqGroup]++;  //increment number of seqs in that group
+			}
+			m->gobble(fileHandle);
 		}
 		fileHandle.close();
+		m->setAllGroups(namesOfGroups);
+		return error;
 }
+
 /************************************************************/
 int GroupMap::getNumGroups() { return namesOfGroups.size();	}
 /************************************************************/
@@ -52,26 +89,36 @@ string GroupMap::getGroup(string sequenceName) {
 }
 
 /************************************************************/
+
 void GroupMap::setGroup(string sequenceName, string groupN) {
-	groupmap[sequenceName] = groupN;
+	setNamesOfGroups(groupN);
+	
+	it = groupmap.find(sequenceName);
+	
+	if (it != groupmap.end()) {  m->mothurOut("Your groupfile contains more than 1 sequence named " + sequenceName + ", sequence names must be unique. Please correct."); m->mothurOutEndLine();  }
+	else {
+		groupmap[sequenceName] = groupN;	//store data in map
+		seqsPerGroup[groupN]++;  //increment number of seqs in that group
+	}
 }
+
 /************************************************************/
 void GroupMap::setNamesOfGroups(string seqGroup) {
-			int i, count;
-			count = 0;
-			for (i=0; i<namesOfGroups.size(); i++) {
-				if (namesOfGroups[i] != seqGroup) {
-					count++; //you have not found this group
-				}else {
-					break; //you already have it
-				}
-			}
-			if (count == namesOfGroups.size()) {
-				namesOfGroups.push_back(seqGroup); //new group
-				seqsPerGroup[seqGroup] = 0;
-				groupIndex[seqGroup] = index;
-				index++;
-			}
+	int i, count;
+	count = 0;
+	for (i=0; i<namesOfGroups.size(); i++) {
+		if (namesOfGroups[i] != seqGroup) {
+			count++; //you have not found this group
+		}else {
+			break; //you already have it
+		}
+	}
+	if (count == namesOfGroups.size()) {
+		namesOfGroups.push_back(seqGroup); //new group
+		seqsPerGroup[seqGroup] = 0;
+		groupIndex[seqGroup] = index;
+		index++;
+	}
 }
 /************************************************************/
 bool GroupMap::isValidGroup(string groupname) {
@@ -83,7 +130,7 @@ bool GroupMap::isValidGroup(string groupname) {
 		return false;
 	}
 	catch(exception& e) {
-		errorOut(e, "GroupMap", "isValidGroup");
+		m->errorOut(e, "GroupMap", "isValidGroup");
 		exit(1);
 	}
 }
@@ -101,7 +148,7 @@ int GroupMap::getNumSeqs(string group) {
 		
 	}
 	catch(exception& e) {
-		errorOut(e, "GroupMap", "getNumSeqs");
+		m->errorOut(e, "GroupMap", "getNumSeqs");
 		exit(1);
 	}
 }
@@ -119,9 +166,30 @@ vector<string> GroupMap::getNamesSeqs(){
 		return names;
 	}
 	catch(exception& e) {
-		errorOut(e, "GroupMap", "getNamesSeqs");
+		m->errorOut(e, "GroupMap", "getNamesSeqs");
 		exit(1);
 	}
 }
 /************************************************************/
+vector<string> GroupMap::getNamesSeqs(vector<string> picked){
+	try {
+		
+		vector<string> names;
+		
+		for (it = groupmap.begin(); it != groupmap.end(); it++) {
+			//if you are belong to one the the groups in the picked vector add you
+			if (m->inUsersGroups(it->second, picked)) {
+				names.push_back(it->first);
+			}
+		}
+		
+		return names;
+	}
+	catch(exception& e) {
+		m->errorOut(e, "GroupMap", "getNamesSeqs");
+		exit(1);
+	}
+}
+
+/************************************************************/