X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=consensusseqscommand.cpp;h=223e5dbb86600907fafaafdf5af48ee665adfc5b;hb=2c97dd48b8e27ee0a6a86c7a082f4c504c3357c6;hp=9c447c217b30247bedbddb7b3351b71952de4c55;hpb=fd5f17bcbe5fadab2c525f3d1fcfd1d79a60ae72;p=mothur.git diff --git a/consensusseqscommand.cpp b/consensusseqscommand.cpp index 9c447c2..223e5db 100644 --- a/consensusseqscommand.cpp +++ b/consensusseqscommand.cpp @@ -12,68 +12,76 @@ #include "inputdata.h" //********************************************************************************************************************** -vector ConsensusSeqsCommand::getValidParameters(){ +vector ConsensusSeqsCommand::setParameters(){ try { - string Array[] = {"fasta", "list", "name", "label","outputdir","inputdir"}; - vector myArray (Array, Array+(sizeof(Array)/sizeof(string))); + CommandParameter pfasta("fasta", "InputTypes", "", "", "none", "none", "none",false,true); parameters.push_back(pfasta); + CommandParameter pname("name", "InputTypes", "", "", "none", "none", "none",false,false); parameters.push_back(pname); + CommandParameter plist("list", "InputTypes", "", "", "none", "none", "none",false,false); parameters.push_back(plist); + CommandParameter plabel("label", "String", "", "", "", "", "",false,false); parameters.push_back(plabel); + CommandParameter pcutoff("cutoff", "Number", "", "100", "", "", "",false,false); parameters.push_back(pcutoff); + 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); } return myArray; } catch(exception& e) { - m->errorOut(e, "ConsensusSeqsCommand", "getValidParameters"); + m->errorOut(e, "ConsensusSeqsCommand", "setParameters"); exit(1); } } //********************************************************************************************************************** -ConsensusSeqsCommand::ConsensusSeqsCommand(){ +string ConsensusSeqsCommand::getHelpString(){ try { - abort = true; - //initialize outputTypes - vector tempOutNames; - outputTypes["fasta"] = tempOutNames; - outputTypes["name"] = tempOutNames; - outputTypes["summary"] = tempOutNames; + string helpString = ""; + helpString += "The consensus.seqs command can be used in 2 ways: create a consensus sequence from a fastafile, or with a listfile create a consensus sequence for each otu. Sequences must be aligned.\n"; + helpString += "The consensus.seqs command parameters are fasta, list, name, cutoff and label.\n"; + helpString += "The fasta parameter allows you to enter the fasta file containing your sequences, and is required, unless you have a valid current fasta file. \n"; + helpString += "The list parameter allows you to enter a your list file. \n"; + helpString += "The name parameter allows you to enter a names file associated with the fasta file. \n"; + helpString += "The label parameter allows you to select what distance levels you would like output files for, and are separated by dashes.\n"; + helpString += "The cutoff parameter allows you set a percentage of sequences that support the base. For example: cutoff=97 would only return a sequence that only showed ambiguities for bases that were not supported by at least 97% of sequences.\n"; + helpString += "The consensus.seqs command should be in the following format: \n"; + helpString += "consensus.seqs(fasta=yourFastaFile, list=yourListFile) \n"; + helpString += "Example: consensus.seqs(fasta=abrecovery.align, list=abrecovery.fn.list) \n"; + helpString += "Note: No spaces between parameter labels (i.e. fasta), '=' and parameters (i.e.yourFastaFile).\n"; + return helpString; } catch(exception& e) { - m->errorOut(e, "ConsensusSeqsCommand", "ConsensusSeqsCommand"); - exit(1); - } -} -//********************************************************************************************************************** -vector ConsensusSeqsCommand::getRequiredParameters(){ - try { - string Array[] = {"fasta", "list"}; - vector myArray (Array, Array+(sizeof(Array)/sizeof(string))); - return myArray; - } - catch(exception& e) { - m->errorOut(e, "ConsensusSeqsCommand", "getRequiredParameters"); + m->errorOut(e, "ConsensusSeqsCommand", "getHelpString"); exit(1); } } + //********************************************************************************************************************** -vector ConsensusSeqsCommand::getRequiredFiles(){ +ConsensusSeqsCommand::ConsensusSeqsCommand(){ try { - vector myArray; - return myArray; + abort = true; calledHelp = true; + setParameters(); + vector tempOutNames; + outputTypes["fasta"] = tempOutNames; + outputTypes["name"] = tempOutNames; + outputTypes["summary"] = tempOutNames; } catch(exception& e) { - m->errorOut(e, "DegapSeqsCommand", "getRequiredFiles"); + m->errorOut(e, "ConsensusSeqsCommand", "ConsensusSeqsCommand"); exit(1); } } //*************************************************************************************************************** ConsensusSeqsCommand::ConsensusSeqsCommand(string option) { try { - abort = false; + abort = false; calledHelp = false; allLines = 1; //allow user to run help - if(option == "help") { help(); abort = true; } + if(option == "help") { help(); abort = true; calledHelp = true; } + else if(option == "citation") { citation(); abort = true; calledHelp = true;} else { - //valid paramters for this command - string Array[] = {"fasta","list","name","label", "outputdir","inputdir"}; - vector myArray (Array, Array+(sizeof(Array)/sizeof(string))); + + vector myArray = setParameters(); OptionParser parser(option); map parameters = parser.getParameters(); @@ -127,15 +135,21 @@ ConsensusSeqsCommand::ConsensusSeqsCommand(string option) { //check for parameters fastafile = validParameter.validFile(parameters, "fasta", true); if (fastafile == "not open") { abort = true; } - else if (fastafile == "not found") { fastafile = ""; m->mothurOut("fasta is a required parameter for the consensus.seqs command."); m->mothurOutEndLine(); abort = true; } + else if (fastafile == "not found") { + fastafile = m->getFastaFile(); + if (fastafile != "") { m->mothurOut("Using " + fastafile + " as input file for the fasta parameter."); m->mothurOutEndLine(); } + else { m->mothurOut("You have no current fastafile and the fasta parameter is required."); m->mothurOutEndLine(); abort = true; } + }else { m->setFastaFile(fastafile); } namefile = validParameter.validFile(parameters, "name", true); - if (namefile == "not open") { abort = true; } - else if (namefile == "not found") { namefile = ""; } + if (namefile == "not open") { namefile = ""; abort = true; } + else if (namefile == "not found") { namefile = ""; } + else { m->setNameFile(namefile); } listfile = validParameter.validFile(parameters, "list", true); if (listfile == "not open") { abort = true; } else if (listfile == "not found") { listfile = ""; } + else { m->setListFile(listfile); } label = validParameter.validFile(parameters, "label", false); if (label == "not found") { label = ""; } @@ -144,9 +158,16 @@ ConsensusSeqsCommand::ConsensusSeqsCommand(string option) { else { allLines = 1; } } + string temp = validParameter.validFile(parameters, "cutoff", false); if (temp == "not found") { temp = "100"; } + m->mothurConvert(temp, cutoff); + //if the user changes the output directory command factory will send this info to us in the output parameter outputDir = validParameter.validFile(parameters, "outputdir", false); if (outputDir == "not found"){ outputDir = m->hasPath(fastafile); } - + + if (namefile == ""){ + vector files; files.push_back(fastafile); + parser.getNameFile(files); + } } } catch(exception& e) { @@ -154,37 +175,12 @@ ConsensusSeqsCommand::ConsensusSeqsCommand(string option) { exit(1); } } -//********************************************************************************************************************** - -void ConsensusSeqsCommand::help(){ - try { - m->mothurOut("The consensus.seqs command can be used in 2 ways: create a consensus sequence from a fastafile, or with a listfile create a consensus sequence for each otu. Sequences must be aligned.\n"); - m->mothurOut("The consensus.seqs command parameters are fasta, list, name and label.\n"); - m->mothurOut("The fasta parameter allows you to enter the fasta file containing your sequences, and is required. \n"); - m->mothurOut("The list parameter allows you to enter a your list file. \n"); - m->mothurOut("The name parameter allows you to enter a names file associated with the fasta file. \n"); - m->mothurOut("The label parameter allows you to select what distance levels you would like output files for, and are separated by dashes.\n"); - m->mothurOut("The consensus.seqs command should be in the following format: \n"); - m->mothurOut("consensus.seqs(fasta=yourFastaFile, list=yourListFile) \n"); - m->mothurOut("Example: consensus.seqs(fasta=abrecovery.align, list=abrecovery.fn.list) \n"); - m->mothurOut("Note: No spaces between parameter labels (i.e. fasta), '=' and parameters (i.e.yourFastaFile).\n\n"); - } - catch(exception& e) { - m->errorOut(e, "ConsensusSeqsCommand", "help"); - exit(1); - } -} - -//*************************************************************************************************************** - -ConsensusSeqsCommand::~ConsensusSeqsCommand(){ /* do nothing */ } - //*************************************************************************************************************** int ConsensusSeqsCommand::execute(){ try{ - if (abort == true) { return 0; } + if (abort == true) { if (calledHelp) { return 0; } return 2; } readFasta(); @@ -203,6 +199,8 @@ int ConsensusSeqsCommand::execute(){ outSummary.setf(ios::fixed, ios::floatfield); outSummary.setf(ios::showpoint); outputNames.push_back(outputSummaryFile); outputTypes["summary"].push_back(outputSummaryFile); + outSummary << "PositioninAlignment\tA\tT\tG\tC\tGap\tNumberofSeqs\tConsensusBase" << endl; + ofstream outFasta; string outputFastaFile = outputDir + m->getRootName(m->getSimpleName(fastafile)) + "cons.fasta"; m->openOutputFile(outputFastaFile, outFasta); @@ -212,7 +210,7 @@ int ConsensusSeqsCommand::execute(){ int seqLength = 0; for (map::iterator it = nameMap.begin(); it != nameMap.end(); it++) { - if (m->control_pressed) { outSummary.close(); outFasta.close(); for (int i = 0; i < outputNames.size(); i++) { remove(outputNames[i].c_str()); } return 0; } + if (m->control_pressed) { outSummary.close(); outFasta.close(); for (int i = 0; i < outputNames.size(); i++) { m->mothurRemove(outputNames[i]); } return 0; } string seq = fastaMap[it->second]; seqs.push_back(seq); @@ -229,7 +227,7 @@ int ConsensusSeqsCommand::execute(){ //get counts for (int j = 0; j < seqLength; j++) { - if (m->control_pressed) { outSummary.close(); outFasta.close(); for (int i = 0; i < outputNames.size(); i++) { remove(outputNames[i].c_str()); } return 0; } + if (m->control_pressed) { outSummary.close(); outFasta.close(); for (int i = 0; i < outputNames.size(); i++) { m->mothurRemove(outputNames[i]); } return 0; } vector counts; counts.resize(5, 0); //A,T,G,C,Gap int numDots = 0; @@ -247,7 +245,7 @@ int ConsensusSeqsCommand::execute(){ } char conBase = '.'; - if (numDots != seqs.size()) { conBase = getBase(counts); } + if (numDots != seqs.size()) { conBase = getBase(counts, seqs.size()); } consSeq += conBase; @@ -259,21 +257,9 @@ int ConsensusSeqsCommand::execute(){ } - outSummary << "A" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[0][j] << '\t'; } - outSummary << endl; - outSummary << "T" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[1][j] << '\t'; } - outSummary << endl; - outSummary << "G" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[2][j] << '\t'; } - outSummary << endl; - outSummary << "C" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[3][j] << '\t'; } - outSummary << endl; - outSummary << "Gap" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[4][j] << '\t'; } - outSummary << endl; + for (int j = 0; j < seqLength; j++) { + outSummary << (j+1) << '\t' << percentages[0][j] << '\t'<< percentages[1][j] << '\t'<< percentages[2][j] << '\t' << percentages[3][j] << '\t' << percentages[4][j] << '\t' << seqs.size() << '\t' << consSeq[j] << endl; + } outFasta << ">conseq" << endl << consSeq << endl; @@ -294,7 +280,7 @@ int ConsensusSeqsCommand::execute(){ //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) { for (int i = 0; i < outputNames.size(); i++) { remove(outputNames[i].c_str()); } delete list; delete input; return 0; } + if (m->control_pressed) { for (int i = 0; i < outputNames.size(); i++) { m->mothurRemove(outputNames[i]); } delete list; delete input; return 0; } if(allLines == 1 || labels.count(list->getLabel()) == 1){ @@ -332,7 +318,7 @@ int ConsensusSeqsCommand::execute(){ } - if (m->control_pressed) { for (int i = 0; i < outputNames.size(); i++) { remove(outputNames[i].c_str()); } if (list != NULL) { delete list; } delete input; return 0; } + if (m->control_pressed) { for (int i = 0; i < outputNames.size(); i++) { m->mothurRemove(outputNames[i]); } if (list != NULL) { delete list; } delete input; return 0; } //output error messages about any remaining user labels set::iterator it; @@ -399,6 +385,8 @@ int ConsensusSeqsCommand::processList(ListVector*& list){ m->openOutputFile(outputFastaFile, outFasta); outputNames.push_back(outputFastaFile); outputTypes["fasta"].push_back(outputFastaFile); + outSummary << "OTU#\tPositioninAlignment\tA\tT\tG\tC\tGap\tNumberofSeqs\tConsensusBase" << endl; + for (int i = 0; i < list->getNumBins(); i++) { if (m->control_pressed) { outSummary.close(); outName.close(); outFasta.close(); return 0; } @@ -500,7 +488,7 @@ string ConsensusSeqsCommand::getConsSeq(string bin, ofstream& outSummary, string } char conBase = '.'; - if (numDots != seqs.size()) { conBase = getBase(counts); } + if (numDots != seqs.size()) { conBase = getBase(counts, seqs.size()); } consSeq += conBase; @@ -512,23 +500,9 @@ string ConsensusSeqsCommand::getConsSeq(string bin, ofstream& outSummary, string } - outSummary << ">seq" << (binNumber + 1) << endl; - outSummary << "A" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[0][j] << '\t'; } - outSummary << endl; - outSummary << "T" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[1][j] << '\t'; } - outSummary << endl; - outSummary << "G" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[2][j] << '\t'; } - outSummary << endl; - outSummary << "C" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[3][j] << '\t'; } - outSummary << endl; - outSummary << "Gap" << '\t'; - for (int j = 0; j < seqLength; j++) { outSummary << percentages[4][j] << '\t'; } - outSummary << endl; - + for (int j = 0; j < seqLength; j++) { + outSummary << (binNumber + 1) << '\t' << (j+1) << '\t' << percentages[0][j] << '\t'<< percentages[1][j] << '\t'<< percentages[2][j] << '\t' << percentages[3][j] << '\t' << percentages[4][j] << '\t' << seqs.size() << '\t' << consSeq[j] << endl; + } return consSeq; @@ -540,7 +514,7 @@ string ConsensusSeqsCommand::getConsSeq(string bin, ofstream& outSummary, string } //*************************************************************************************************************** -char ConsensusSeqsCommand::getBase(vector counts){ //A,T,G,C,Gap +char ConsensusSeqsCommand::getBase(vector counts, int size){ //A,T,G,C,Gap try{ /* A = adenine * C = cytosine @@ -560,6 +534,14 @@ char ConsensusSeqsCommand::getBase(vector counts){ //A,T,G,C,Gap char conBase = 'N'; + //zero out counts that don't make the cutoff + float percentage = (100.0 - cutoff) / 100.0; + int zeroCutoff = percentage * size; + + for (int i = 0; i < counts.size(); i++) { + if (counts[i] < zeroCutoff) { counts[i] = 0; } + } + //any if ((counts[0] != 0) && (counts[1] != 0) && (counts[2] != 0) && (counts[3] != 0) && (counts[4] != 0)) { conBase = 'n'; } //any no gap @@ -622,6 +604,8 @@ char ConsensusSeqsCommand::getBase(vector counts){ //A,T,G,C,Gap else if ((counts[0] == 0) && (counts[1] == 0) && (counts[2] == 0) && (counts[3] != 0) && (counts[4] == 0)) { conBase = 'C'; } //only gap else if ((counts[0] == 0) && (counts[1] == 0) && (counts[2] == 0) && (counts[3] == 0) && (counts[4] != 0)) { conBase = '-'; } + //cutoff removed all counts + else if ((counts[0] == 0) && (counts[1] == 0) && (counts[2] == 0) && (counts[3] == 0) && (counts[4] == 0)) { conBase = 'N'; } else{ m->mothurOut("[ERROR]: cannot find consensus base."); m->mothurOutEndLine(); } return conBase;