X-Git-Url: https://git.donarmstrong.com/?p=mothur.git;a=blobdiff_plain;f=listotulabelscommand.cpp;h=512dd6da763241d47d727dd10efcce3a68eab2b8;hp=93b1510d22bc04cc19fec975df447381285b7788;hb=b206f634aae1b4ce13978d203247fb64757d5482;hpb=e0b6040c7b6ad4ecdad4aa90b4326c0528379178 diff --git a/listotulabelscommand.cpp b/listotulabelscommand.cpp index 93b1510..512dd6d 100644 --- a/listotulabelscommand.cpp +++ b/listotulabelscommand.cpp @@ -12,13 +12,14 @@ //********************************************************************************************************************** vector ListOtuLabelsCommand::setParameters(){ try { - CommandParameter pshared("shared", "InputTypes", "", "", "SharedRel", "SharedRel", "none",false,false); parameters.push_back(pshared); - CommandParameter prelabund("relabund", "InputTypes", "", "", "SharedRel", "SharedRel", "none",false,false); parameters.push_back(prelabund); - CommandParameter pgroups("groups", "String", "", "", "", "", "",false,false); parameters.push_back(pgroups); - CommandParameter plabel("label", "String", "", "", "", "", "",false,false); parameters.push_back(plabel); + CommandParameter pshared("shared", "InputTypes", "", "", "SharedRel", "SharedRel", "none","otulabels",false,false,true); parameters.push_back(pshared); + CommandParameter prelabund("relabund", "InputTypes", "", "", "SharedRel", "SharedRel", "none","otulabels",false,false); parameters.push_back(prelabund); + CommandParameter plist("list", "InputTypes", "", "", "SharedRel", "SharedRel", "none","otulabels",false,false); parameters.push_back(plist); + CommandParameter pgroups("groups", "String", "", "", "", "", "","",false,false); parameters.push_back(pgroups); + CommandParameter plabel("label", "String", "", "", "", "", "","",false,false); parameters.push_back(plabel); //every command must have inputdir and outputdir. This allows mothur users to redirect input and output files. - CommandParameter pinputdir("inputdir", "String", "", "", "", "", "",false,false); parameters.push_back(pinputdir); - CommandParameter poutputdir("outputdir", "String", "", "", "", "", "",false,false); parameters.push_back(poutputdir); + CommandParameter pinputdir("inputdir", "String", "", "", "", "", "","",false,false); parameters.push_back(pinputdir); + CommandParameter poutputdir("outputdir", "String", "", "", "", "", "","",false,false); parameters.push_back(poutputdir); vector myArray; for (int i = 0; i < parameters.size(); i++) { myArray.push_back(parameters[i].name); } @@ -33,7 +34,7 @@ vector ListOtuLabelsCommand::setParameters(){ string ListOtuLabelsCommand::getHelpString(){ try { string helpString = ""; - helpString += "The list.otulabels lists otu labels from shared or relabund file. The results can be used by the get.otulabels to select specific otus with the output from classify.otu, otu.association, or corr.axes.\n"; + helpString += "The list.otulabels lists otu labels from shared, relabund or list file. The results can be used by the get.otulabels to select specific otus with the output from classify.otu, otu.association, or corr.axes.\n"; helpString += "The list.otulabels parameters are: shared, relabund, label and groups.\n"; helpString += "The label parameter is used to analyze specific labels in your input.\n"; helpString += "The groups parameter allows you to specify which of the groups you would like analyzed.\n"; @@ -47,6 +48,21 @@ string ListOtuLabelsCommand::getHelpString(){ } } //********************************************************************************************************************** +string ListOtuLabelsCommand::getOutputPattern(string type) { + try { + string pattern = ""; + + if (type == "otulabels") { pattern = "[filename],[distance],otulabels"; } + else { m->mothurOut("[ERROR]: No definition for type " + type + " output pattern.\n"); m->control_pressed = true; } + + return pattern; + } + catch(exception& e) { + m->errorOut(e, "ListOtuLabelsCommand", "getOutputPattern"); + exit(1); + } +} +//********************************************************************************************************************** ListOtuLabelsCommand::ListOtuLabelsCommand(){ try { abort = true; calledHelp = true; @@ -107,6 +123,14 @@ ListOtuLabelsCommand::ListOtuLabelsCommand(string option) { //if the user has not given a path then, add inputdir. else leave path alone. if (path == "") { parameters["shared"] = inputDir + it->second; } } + + it = parameters.find("list"); + //user has given a template file + if(it != parameters.end()){ + path = m->hasPath(it->second); + //if the user has not given a path then, add inputdir. else leave path alone. + if (path == "") { parameters["list"] = inputDir + it->second; } + } } vector tempOutNames; @@ -123,7 +147,13 @@ ListOtuLabelsCommand::ListOtuLabelsCommand(string option) { else if (relabundfile == "not found") { relabundfile = ""; } else { inputFileName = relabundfile; format = "relabund"; m->setRelAbundFile(relabundfile); } - if ((relabundfile == "") && (sharedfile == "")) { + listfile = validParameter.validFile(parameters, "list", true); + if (listfile == "not open") { abort = true; } + else if (listfile == "not found") { listfile = ""; } + else { inputFileName = listfile; format = "list"; m->setListFile(listfile); } + + + if ((relabundfile == "") && (sharedfile == "") && (listfile== "")) { //is there are current file available for either of these? //give priority to shared, then relabund sharedfile = m->getSharedFile(); @@ -132,8 +162,12 @@ ListOtuLabelsCommand::ListOtuLabelsCommand(string option) { relabundfile = m->getRelAbundFile(); if (relabundfile != "") { inputFileName = relabundfile; format="relabund"; m->mothurOut("Using " + relabundfile + " as input file for the relabund parameter."); m->mothurOutEndLine(); } else { - m->mothurOut("No valid current files. You must provide a shared or relabund."); m->mothurOutEndLine(); - abort = true; + listfile = m->getListFile(); + if (listfile != "") { inputFileName = listfile; format="list"; m->mothurOut("Using " + listfile + " as input file for the list parameter."); m->mothurOutEndLine(); } + else { + m->mothurOut("No valid current files. You must provide a shared, list or relabund."); m->mothurOutEndLine(); + abort = true; + } } } } @@ -246,7 +280,7 @@ int ListOtuLabelsCommand::execute(){ for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } } - }else { + }else if (format == "sharedfile") { vector lookup = input.getSharedRAbundVectors(); string lastLabel = lookup[0]->getLabel(); @@ -322,6 +356,81 @@ int ListOtuLabelsCommand::execute(){ for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } } + }else { + ListVector* list = input.getListVector(); + string lastLabel = list->getLabel(); + + //if the users enters label "0.06" and there is no "0.06" in their file use the next lowest label. + set processedLabels; + set userLabels = labels; + + //as long as you are not at the end of the file or done wih the lines you want + while((list != NULL) && ((allLines == 1) || (userLabels.size() != 0))) { + + if (m->control_pressed) { delete list; for (int i = 0; i < outputNames.size(); i++) { m->mothurRemove(outputNames[i]); } return 0; } + + if(allLines == 1 || labels.count(list->getLabel()) == 1){ + + m->mothurOut(list->getLabel()); m->mothurOutEndLine(); + + createList(list); + + processedLabels.insert(list->getLabel()); + userLabels.erase(list->getLabel()); + } + + if ((m->anyLabelsToProcess(list->getLabel(), userLabels, "") == true) && (processedLabels.count(lastLabel) != 1)) { + string saveLabel = list->getLabel(); + + delete list; + list = input.getListVector(lastLabel); + m->mothurOut(list->getLabel()); m->mothurOutEndLine(); + + createList(list); + + processedLabels.insert(list->getLabel()); + userLabels.erase(list->getLabel()); + + //restore real lastlabel to save below + list->setLabel(saveLabel); + } + + lastLabel = list->getLabel(); + //prevent memory leak + delete list; list = NULL; + + if (m->control_pressed) { for (int i = 0; i < outputNames.size(); i++) { m->mothurRemove(outputNames[i]); } return 0; } + + //get next line to process + list = input.getListVector(); + } + + if (m->control_pressed) { for (int i = 0; i < outputNames.size(); i++) { m->mothurRemove(outputNames[i]); } return 0; } + + //output error messages about any remaining user labels + set::iterator it; + bool needToRun = false; + for (it = userLabels.begin(); it != userLabels.end(); it++) { + m->mothurOut("Your file does not include the label " + *it); + if (processedLabels.count(lastLabel) != 1) { + m->mothurOut(". I will use " + lastLabel + "."); m->mothurOutEndLine(); + needToRun = true; + }else { + m->mothurOut(". Please refer to " + lastLabel + "."); m->mothurOutEndLine(); + } + } + + //run last label if you need to + if (needToRun == true) { + delete list; + list = input.getListVector(lastLabel); + + m->mothurOut(list->getLabel()); m->mothurOutEndLine(); + + createList(list); + + delete list; + } } if (m->control_pressed) { for (int i = 0; i < outputNames.size(); i++) { m->mothurRemove(outputNames[i]); } return 0; } @@ -344,19 +453,22 @@ int ListOtuLabelsCommand::execute(){ int ListOtuLabelsCommand::createList(vector& lookup){ try { - string outputFileName = outputDir + m->getRootName(m->getSimpleName(inputFileName)) + lookup[0]->getLabel() + ".otu.labels"; + map variables; + variables["[filename]"] = outputDir + m->getRootName(m->getSimpleName(inputFileName)); + variables["[distance]"] = lookup[0]->getLabel(); + string outputFileName = getOutputFileName("otulabels",variables); outputNames.push_back(outputFileName); outputTypes["otulabels"].push_back(outputFileName); ofstream out; m->openOutputFile(outputFileName, out); - for (int i = 0; i < m->currentBinLabels.size(); i++) { out << m->currentBinLabels[i] << endl; } + for (int i = 0; i < m->currentSharedBinLabels.size(); i++) { out << m->currentSharedBinLabels[i] << endl; } out.close(); return 0; } catch(exception& e) { - m->errorOut(e, "ListOtuLabelsCommand", "createTable"); + m->errorOut(e, "ListOtuLabelsCommand", "createList"); exit(1); } } @@ -365,20 +477,45 @@ int ListOtuLabelsCommand::createList(vector& lookup){ int ListOtuLabelsCommand::createList(vector& lookup){ try { - - string outputFileName = outputDir + m->getRootName(m->getSimpleName(inputFileName)) + lookup[0]->getLabel() + ".otu.labels"; + map variables; + variables["[filename]"] = outputDir + m->getRootName(m->getSimpleName(inputFileName)); + variables["[distance]"] = lookup[0]->getLabel(); + string outputFileName = getOutputFileName("otulabels",variables); outputNames.push_back(outputFileName); outputTypes["accnos"].push_back(outputFileName); ofstream out; m->openOutputFile(outputFileName, out); - for (int i = 0; i < m->currentBinLabels.size(); i++) { out << m->currentBinLabels[i] << endl; } + for (int i = 0; i < m->currentSharedBinLabels.size(); i++) { out << m->currentSharedBinLabels[i] << endl; } + + out.close(); + + return 0; + } + catch(exception& e) { + m->errorOut(e, "ListOtuLabelsCommand", "createList"); + exit(1); + } +} +//********************************************************************************************************************** +int ListOtuLabelsCommand::createList(ListVector*& list){ + try { + map variables; + variables["[filename]"] = outputDir + m->getRootName(m->getSimpleName(inputFileName)); + variables["[distance]"] = list->getLabel(); + string outputFileName = getOutputFileName("otulabels",variables); + outputNames.push_back(outputFileName); outputTypes["accnos"].push_back(outputFileName); + ofstream out; + m->openOutputFile(outputFileName, out); + vector binLabels = list->getLabels(); + for (int i = 0; i < binLabels.size(); i++) { out << binLabels[i] << endl; } + out.close(); return 0; } catch(exception& e) { - m->errorOut(e, "ListOtuLabelsCommand", "createTable"); + m->errorOut(e, "ListOtuLabelsCommand", "createList"); exit(1); } }