filename = getRootName(filename);
filename = filename + "shared";
openOutputFile(filename, out);
+ pickedGroups = false;
groupMap = globaldata->gGroupmap;
+ //if hte user has not specified any groups then use them all
+ if (globaldata->Groups.size() == 0) {
+ groups = groupMap->namesOfGroups;
+ }else{ //they have specified groups
+ groups = globaldata->Groups;
+ pickedGroups = true;
+ }
+
//fill filehandles with neccessary ofstreams
int i;
ofstream* temp;
- for (i=0; i<groupMap->getNumGroups(); i++) {
+ for (i=0; i<groups.size(); i++) {
temp = new ofstream;
- filehandles[groupMap->namesOfGroups[i]] = temp;
+ filehandles[groups[i]] = temp;
}
//set fileroot
fileroot = getRootName(globaldata->getListFile());
//clears file before we start to write to it below
- for (int i=0; i<groupMap->getNumGroups(); i++) {
- remove((fileroot + groupMap->namesOfGroups[i] + ".rabund").c_str());
+ for (int i=0; i<groups.size(); i++) {
+ remove((fileroot + groups[i] + ".rabund").c_str());
}
}
try {
//lookup.clear();
- int count = 1;
string errorOff = "no error";
+ //errorOff = "";
//read in listfile
read = new ReadOTUFile(globaldata->inputFileName);
string lastLabel = SharedList->getLabel();
vector<SharedRAbundVector*> lookup;
- if (SharedList->getNumSeqs() != groupMap->getNumSeqs()) {
+ if ((globaldata->Groups.size() == 0) && (SharedList->getNumSeqs() != groupMap->getNumSeqs())) { //if the user has not specified any groups and their files don't match exit with error
mothurOut("Your group file contains " + toString(groupMap->getNumSeqs()) + " sequences and list file contains " + toString(SharedList->getNumSeqs()) + " sequences. Please correct."); mothurOutEndLine();
+ out.close();
+ remove(filename.c_str()); //remove blank shared file you made
+
createMisMatchFile();
//delete memory
return 1;
}
+ //if user has specified groups make new groupfile for them
+ if (globaldata->Groups.size() != 0) { //make new group file
+ string groups = "";
+ for (int i = 0; i < globaldata->Groups.size(); i++) {
+ groups += globaldata->Groups[i] + ".";
+ }
+
+ string newGroupFile = getRootName(globaldata->inputFileName) + groups + "groups";
+ ofstream outGroups;
+ openOutputFile(newGroupFile, outGroups);
+
+ vector<string> names = groupMap->getNamesSeqs();
+ string groupName;
+ for (int i = 0; i < names.size(); i++) {
+ groupName = groupMap->getGroup(names[i]);
+ if (isValidGroup(groupName, globaldata->Groups)) {
+ outGroups << names[i] << '\t' << groupName << endl;
+ }
+ }
+ outGroups.close();
+ }
+
//if the users enters label "0.06" and there is no "0.06" in their file use the next lowest label.
set<string> processedLabels;
- set<string> userLabels = globaldata->labels;
- set<int> userLines = globaldata->lines;
-
+ set<string> userLabels = globaldata->labels;
+
+ while((SharedList != NULL) && ((globaldata->allLines == 1) || (userLabels.size() != 0))) {
- while((SharedList != NULL) && ((globaldata->allLines == 1) || (userLabels.size() != 0) || (userLines.size() != 0))) {
+ if(globaldata->allLines == 1 || globaldata->labels.count(SharedList->getLabel()) == 1){
-
- if(globaldata->allLines == 1 || globaldata->lines.count(count) == 1 || globaldata->labels.count(SharedList->getLabel()) == 1){
-
lookup = SharedList->getSharedRAbundVector();
+ if (pickedGroups) { //check for otus with no seqs in them
+ eliminateZeroOTUS(lookup);
+ }
mothurOut(lookup[0]->getLabel()); mothurOutEndLine();
printSharedData(lookup); //prints info to the .shared file
processedLabels.insert(SharedList->getLabel());
userLabels.erase(SharedList->getLabel());
- userLines.erase(count);
}
if ((anyLabelsToProcess(SharedList->getLabel(), userLabels, errorOff) == true) && (processedLabels.count(lastLabel) != 1)) {
+ string saveLabel = SharedList->getLabel();
+
delete SharedList;
SharedList = input->getSharedListVector(lastLabel); //get new list vector to process
lookup = SharedList->getSharedRAbundVector();
+ if (pickedGroups) { //check for otus with no seqs in them
+ eliminateZeroOTUS(lookup);
+ }
mothurOut(lookup[0]->getLabel()); mothurOutEndLine();
printSharedData(lookup); //prints info to the .shared file
processedLabels.insert(SharedList->getLabel());
userLabels.erase(SharedList->getLabel());
+
+ //restore real lastlabel to save below
+ SharedList->setLabel(saveLabel);
}
delete SharedList;
SharedList = input->getSharedListVector(); //get new list vector to process
-
- count++;
}
//output error messages about any remaining user labels
}
}
- //run last line if you need to
+ //run last label if you need to
if (needToRun == true) {
if (SharedList != NULL) { delete SharedList; }
SharedList = input->getSharedListVector(lastLabel); //get new list vector to process
lookup = SharedList->getSharedRAbundVector();
+ if (pickedGroups) { //check for otus with no seqs in them
+ eliminateZeroOTUS(lookup);
+ }
mothurOut(lookup[0]->getLabel()); mothurOutEndLine();
printSharedData(lookup); //prints info to the .shared file
//initialize bin values
for (int i = 0; i < thislookup.size(); i++) {
+//cout << "in printData " << thislookup[i]->getLabel() << '\t' << thislookup[i]->getGroup() << endl;
out << thislookup[i]->getLabel() << '\t' << thislookup[i]->getGroup() << '\t';
thislookup[i]->print(out);
}
}
//**********************************************************************************************************************
+void SharedCommand::eliminateZeroOTUS(vector<SharedRAbundVector*>& thislookup) {
+ try {
+
+ vector<SharedRAbundVector*> newLookup;
+ for (int i = 0; i < thislookup.size(); i++) {
+ SharedRAbundVector* temp = new SharedRAbundVector();
+ temp->setLabel(thislookup[i]->getLabel());
+ temp->setGroup(thislookup[i]->getGroup());
+ newLookup.push_back(temp);
+ }
+
+ //for each bin
+ for (int i = 0; i < thislookup[0]->getNumBins(); i++) {
+
+ //look at each sharedRabund and make sure they are not all zero
+ bool allZero = true;
+ for (int j = 0; j < thislookup.size(); j++) {
+ if (thislookup[j]->getAbundance(i) != 0) { allZero = false; break; }
+ }
+
+ //if they are not all zero add this bin
+ if (!allZero) {
+ for (int j = 0; j < thislookup.size(); j++) {
+ newLookup[j]->push_back(thislookup[j]->getAbundance(i), thislookup[j]->getGroup());
+ }
+ }
+ //else{ cout << "bin # " << i << " is all zeros" << endl; }
+ }
+
+ for (int j = 0; j < thislookup.size(); j++) { delete thislookup[j]; }
+ thislookup = newLookup;
+
+
+ }
+ catch(exception& e) {
+ errorOut(e, "SharedCommand", "eliminateZeroOTUS");
+ exit(1);
+ }
+}
+//**********************************************************************************************************************
void SharedCommand::createMisMatchFile() {
try {
ofstream outMisMatch;
map<string, string> namesInList;
//go through listfile and get names
- for (int i = 0; i < SharedList->getNumSeqs(); i++) {
+ for (int i = 0; i < SharedList->getNumBins(); i++) {
string names = SharedList->get(i);
-
+
while (names.find_first_of(',') != -1) {
string name = names.substr(0,names.find_first_of(','));
names = names.substr(names.find_first_of(',')+1, names.length());
}
//**********************************************************************************************************************
+
+bool SharedCommand::isValidGroup(string groupname, vector<string> groups) {
+ try {
+ for (int i = 0; i < groups.size(); i++) {
+ if (groupname == groups[i]) { return true; }
+ }
+
+ return false;
+ }
+ catch(exception& e) {
+ errorOut(e, "SharedCommand", "isValidGroup");
+ exit(1);
+ }
+}
+/************************************************************/
+
+
projects / mothur.git / blobdiff