*/
#include "sharedcommand.h"
-
+//********************************************************************************************************************
+//sorts lowest to highest
+inline bool compareSharedRabunds(SharedRAbundVector* left, SharedRAbundVector* right){
+ return (left->getGroup() < right->getGroup());
+}
//**********************************************************************************************************************
SharedCommand::SharedCommand(string o) : outputDir(o) {
//clears file before we start to write to it below
for (int i=0; i<groups.size(); i++) {
remove((fileroot + groups[i] + ".rabund").c_str());
+ outputNames.push_back((fileroot + groups[i] + ".rabund"));
}
}
string errorOff = "no error";
//errorOff = "";
+ //if user provided an order file containing the order the shared file should be in read it
+ if (globaldata->getOrderGroupFile() != "") { readOrderFile(); }
+
//read in listfile
read = new ReadOTUFile(globaldata->inputFileName);
read->read(&*globaldata);
string lastLabel = SharedList->getLabel();
vector<SharedRAbundVector*> lookup;
+ if (m->control_pressed) {
+ delete input; delete SharedList; globaldata->ginput = NULL; globaldata->gSharedList = NULL;
+ for (it3 = filehandles.begin(); it3 != filehandles.end(); it3++) { delete it3->second; }
+ out.close(); remove(filename.c_str());
+ for (int i=0; i<groups.size(); i++) { remove((fileroot + groups[i] + ".rabund").c_str()); }
+ return 1;
+ }
+
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
m->mothurOut("Your group file contains " + toString(groupMap->getNumSeqs()) + " sequences and list file contains " + toString(SharedList->getNumSeqs()) + " sequences. Please correct."); m->mothurOutEndLine();
for (it3 = filehandles.begin(); it3 != filehandles.end(); it3++) {
delete it3->second;
}
+ delete input;
+ globaldata->ginput = NULL;
delete SharedList;
globaldata->gSharedList = NULL;
set<string> userLabels = globaldata->labels;
while((SharedList != NULL) && ((globaldata->allLines == 1) || (userLabels.size() != 0))) {
+ if (m->control_pressed) {
+ delete input; delete SharedList; globaldata->ginput = NULL; globaldata->gSharedList = NULL;
+ for (it3 = filehandles.begin(); it3 != filehandles.end(); it3++) { delete it3->second; }
+ out.close(); remove(filename.c_str());
+ for (int i=0; i<groups.size(); i++) { remove((fileroot + groups[i] + ".rabund").c_str()); }
+ return 1;
+ }
if(globaldata->allLines == 1 || globaldata->labels.count(SharedList->getLabel()) == 1){
-
+
lookup = SharedList->getSharedRAbundVector();
+
+ m->mothurOut(lookup[0]->getLabel()); m->mothurOutEndLine();
if (pickedGroups) { //check for otus with no seqs in them
eliminateZeroOTUS(lookup);
}
- m->mothurOut(lookup[0]->getLabel()); m->mothurOutEndLine();
+
+ if (m->control_pressed) {
+ delete input; delete SharedList; globaldata->ginput = NULL; globaldata->gSharedList = NULL;
+ for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; }
+ for (it3 = filehandles.begin(); it3 != filehandles.end(); it3++) { delete it3->second; }
+ out.close(); remove(filename.c_str());
+ for (int i=0; i<groups.size(); i++) { remove((fileroot + groups[i] + ".rabund").c_str()); }
+ return 1;
+ }
printSharedData(lookup); //prints info to the .shared file
for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; }
SharedList = input->getSharedListVector(lastLabel); //get new list vector to process
lookup = SharedList->getSharedRAbundVector();
+ m->mothurOut(lookup[0]->getLabel()); m->mothurOutEndLine();
if (pickedGroups) { //check for otus with no seqs in them
eliminateZeroOTUS(lookup);
}
- m->mothurOut(lookup[0]->getLabel()); m->mothurOutEndLine();
+
+
+ if (m->control_pressed) {
+ delete input; delete SharedList; globaldata->ginput = NULL; globaldata->gSharedList = NULL;
+ for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; }
+ for (it3 = filehandles.begin(); it3 != filehandles.end(); it3++) { delete it3->second; }
+ out.close(); remove(filename.c_str());
+ for (int i=0; i<groups.size(); i++) { remove((fileroot + groups[i] + ".rabund").c_str()); }
+ return 1;
+ }
printSharedData(lookup); //prints info to the .shared file
for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; }
SharedList = input->getSharedListVector(lastLabel); //get new list vector to process
lookup = SharedList->getSharedRAbundVector();
+ m->mothurOut(lookup[0]->getLabel()); m->mothurOutEndLine();
if (pickedGroups) { //check for otus with no seqs in them
eliminateZeroOTUS(lookup);
}
- m->mothurOut(lookup[0]->getLabel()); m->mothurOutEndLine();
+
+ if (m->control_pressed) {
+ delete input; globaldata->ginput = NULL;
+ for (it3 = filehandles.begin(); it3 != filehandles.end(); it3++) { delete it3->second; }
+ out.close(); remove(filename.c_str());
+ for (int i=0; i<groups.size(); i++) { remove((fileroot + groups[i] + ".rabund").c_str()); }
+ return 1;
+ }
printSharedData(lookup); //prints info to the .shared file
for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; }
globaldata->setListFile("");
globaldata->setGroupFile("");
globaldata->setSharedFile(filename);
-
+
+ if (m->control_pressed) {
+ delete input; globaldata->ginput = NULL;
+ remove(filename.c_str());
+ for (int i=0; i<groups.size(); i++) { remove((fileroot + groups[i] + ".rabund").c_str()); }
+ return 1;
+ }
+
+ m->mothurOutEndLine();
+ m->mothurOut("Output File Names: "); m->mothurOutEndLine();
+ for (int i = 0; i < outputNames.size(); i++) { m->mothurOut(outputNames[i]); m->mothurOutEndLine(); }
+ m->mothurOut(filename); m->mothurOutEndLine();
+ m->mothurOutEndLine();
return 0;
}
void SharedCommand::printSharedData(vector<SharedRAbundVector*> thislookup) {
try {
- //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);
+ if (order.size() == 0) { //user has not specified an order so do aplabetically
+ sort(thislookup.begin(), thislookup.end(), compareSharedRabunds);
+
+ //initialize bin values
+ for (int i = 0; i < thislookup.size(); i++) {
+ out << thislookup[i]->getLabel() << '\t' << thislookup[i]->getGroup() << '\t';
+ thislookup[i]->print(out);
+
+ RAbundVector rav = thislookup[i]->getRAbundVector();
+ openOutputFileAppend(fileroot + thislookup[i]->getGroup() + ".rabund", *(filehandles[thislookup[i]->getGroup()]));
+ rav.print(*(filehandles[thislookup[i]->getGroup()]));
+ (*(filehandles[thislookup[i]->getGroup()])).close();
+ }
+ }else{
+ //create a map from groupName to each sharedrabund
+ map<string, SharedRAbundVector*> myMap;
+ map<string, SharedRAbundVector*>::iterator myIt;
- RAbundVector rav = thislookup[i]->getRAbundVector();
- openOutputFileAppend(fileroot + thislookup[i]->getGroup() + ".rabund", *(filehandles[thislookup[i]->getGroup()]));
- rav.print(*(filehandles[thislookup[i]->getGroup()]));
- (*(filehandles[thislookup[i]->getGroup()])).close();
+ for (int i = 0; i < thislookup.size(); i++) {
+ myMap[thislookup[i]->getGroup()] = thislookup[i];
+ }
+
+
+ //loop through ordered list and print the rabund
+ for (int i = 0; i < order.size(); i++) {
+ myIt = myMap.find(order[i]);
+
+ if(myIt != myMap.end()) { //we found it
+ out << (myIt->second)->getLabel() << '\t' << (myIt->second)->getGroup() << '\t';
+ (myIt->second)->print(out);
+
+ RAbundVector rav = (myIt->second)->getRAbundVector();
+ openOutputFileAppend(fileroot + (myIt->second)->getGroup() + ".rabund", *(filehandles[(myIt->second)->getGroup()]));
+ rav.print(*(filehandles[(myIt->second)->getGroup()]));
+ (*(filehandles[(myIt->second)->getGroup()])).close();
+ }else{
+ m->mothurOut("Can't find shared info for " + order[i] + ", skipping."); m->mothurOutEndLine();
+ }
+ }
+
}
}
}
}
//**********************************************************************************************************************
-void SharedCommand::eliminateZeroOTUS(vector<SharedRAbundVector*>& thislookup) {
+int SharedCommand::eliminateZeroOTUS(vector<SharedRAbundVector*>& thislookup) {
try {
vector<SharedRAbundVector*> newLookup;
//for each bin
for (int i = 0; i < thislookup[0]->getNumBins(); i++) {
+ if (m->control_pressed) { for (int j = 0; j < newLookup.size(); j++) { delete newLookup[j]; } return 0; }
//look at each sharedRabund and make sure they are not all zero
bool allZero = true;
for (int j = 0; j < thislookup.size(); j++) { delete thislookup[j]; }
thislookup = newLookup;
-
+
+ return 0;
}
catch(exception& e) {
}
}
//**********************************************************************************************************************
-void SharedCommand::createMisMatchFile() {
+int SharedCommand::createMisMatchFile() {
try {
ofstream outMisMatch;
string outputMisMatchName = outputDir + getRootName(getSimpleName(globaldata->inputFileName));
//go through list and if group returns "not found" output it
for (int i = 0; i < SharedList->getNumBins(); i++) {
+ if (m->control_pressed) { outMisMatch.close(); remove(outputMisMatchName.c_str()); return 0; }
string names = SharedList->get(i);
//go through listfile and get names
for (int i = 0; i < SharedList->getNumBins(); i++) {
+ if (m->control_pressed) { return 0; }
+
string names = SharedList->get(i);
//loop through names in seqNames and if they aren't in namesIn list output them
for (int i = 0; i < seqNames.size(); i++) {
+ if (m->control_pressed) { outMisMatch.close(); remove(outputMisMatchName.c_str()); return 0; }
itMatch = namesInList.find(seqNames[i]);
}
outMisMatch.close();
}
-
+
+ return 0;
}
catch(exception& e) {
m->errorOut(e, "SharedCommand", "createMisMatchFile");
}
-
+//**********************************************************************************************************************
+int SharedCommand::readOrderFile() {
+ try {
+ //remove old names
+ order.clear();
+
+ ifstream in;
+ openInputFile(globaldata->getOrderGroupFile(), in);
+ string thisGroup;
+
+ while(!in.eof()){
+ in >> thisGroup; gobble(in);
+
+ order.push_back(thisGroup);
+
+ if (m->control_pressed) { order.clear(); break; }
+ }
+ in.close();
+
+ return 0;
+ }
+ catch(exception& e) {
+ m->errorOut(e, "SharedCommand", "readOrderFile");
+ exit(1);
+ }
+}
//**********************************************************************************************************************
bool SharedCommand::isValidGroup(string groupname, vector<string> groups) {
projects / mothur.git / blobdiff