X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=reversecommand.cpp;h=cde88b71203ac3c5fe364c2d7a7bd112bc8da8f8;hb=8e67e9de1b200106bea5a468ac02125954656499;hp=5eceb3049f0d6cbc41025c875843adce306a745c;hpb=74d3aaf0f85080ea3cfb1f1871e706bab9f48392;p=mothur.git diff --git a/reversecommand.cpp b/reversecommand.cpp index 5eceb30..cde88b7 100644 --- a/reversecommand.cpp +++ b/reversecommand.cpp @@ -12,56 +12,73 @@ #include "qualityscores.h" //********************************************************************************************************************** -vector ReverseSeqsCommand::getValidParameters(){ +vector ReverseSeqsCommand::setParameters(){ try { - string Array[] = {"fasta", "qfile", "outputdir", "inputdir"}; - vector myArray (Array, Array+(sizeof(Array)/sizeof(string))); + CommandParameter pfasta("fasta", "InputTypes", "", "", "none", "fastaQual", "none",false,false); parameters.push_back(pfasta); + CommandParameter pqfile("qfile", "InputTypes", "", "", "none", "fastaQual", "none",false,false); parameters.push_back(pqfile); + 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, "ReverseSeqsCommand", "getValidParameters"); + m->errorOut(e, "ReverseSeqsCommand", "setParameters"); exit(1); } } //********************************************************************************************************************** -ReverseSeqsCommand::ReverseSeqsCommand(){ +string ReverseSeqsCommand::getHelpString(){ try { - abort = true; calledHelp = true; - vector tempOutNames; - outputTypes["fasta"] = tempOutNames; - outputTypes["qfile"] = tempOutNames; + string helpString = ""; + helpString += "The reverse.seqs command reads a fastafile and outputs a fasta file containing the reverse compliment.\n"; + helpString += "The reverse.seqs command parameters fasta or qfile are required.\n"; + helpString += "The reverse.seqs command should be in the following format: \n"; + helpString += "reverse.seqs(fasta=yourFastaFile) \n"; + return helpString; } catch(exception& e) { - m->errorOut(e, "ReverseSeqsCommand", "ReverseSeqsCommand"); + m->errorOut(e, "ReverseSeqsCommand", "getHelpString"); exit(1); } } //********************************************************************************************************************** - -vector ReverseSeqsCommand::getRequiredParameters(){ +string ReverseSeqsCommand::getOutputFileNameTag(string type, string inputName=""){ try { - string Array[] = {"fasta", "qfile"}; - vector myArray (Array, Array+(sizeof(Array)/sizeof(string))); - return myArray; + string outputFileName = ""; + map >::iterator it; + + //is this a type this command creates + it = outputTypes.find(type); + if (it == outputTypes.end()) { m->mothurOut("[ERROR]: this command doesn't create a " + type + " output file.\n"); } + else { + if (type == "fasta") { outputFileName = "rc" + m->getExtension(inputName); } + else if (type == "qfile") { outputFileName = "rc" + m->getExtension(inputName); } + else { m->mothurOut("[ERROR]: No definition for type " + type + " output file tag.\n"); m->control_pressed = true; } + } + return outputFileName; } catch(exception& e) { - m->errorOut(e, "ReverseSeqsCommand", "getRequiredParameters"); + m->errorOut(e, "ReverseSeqsCommand", "getOutputFileNameTag"); exit(1); } } - //********************************************************************************************************************** - -vector ReverseSeqsCommand::getRequiredFiles(){ +ReverseSeqsCommand::ReverseSeqsCommand(){ try { - vector myArray; - return myArray; + abort = true; calledHelp = true; + setParameters(); + vector tempOutNames; + outputTypes["fasta"] = tempOutNames; + outputTypes["qfile"] = tempOutNames; } catch(exception& e) { - m->errorOut(e, "ReverseSeqsCommand", "getRequiredFiles"); + m->errorOut(e, "ReverseSeqsCommand", "ReverseSeqsCommand"); exit(1); } -}//*************************************************************************************************************** +} +//*************************************************************************************************************** ReverseSeqsCommand::ReverseSeqsCommand(string option) { try { @@ -69,11 +86,10 @@ ReverseSeqsCommand::ReverseSeqsCommand(string option) { //allow user to run help 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", "qfile", "outputdir","inputdir"}; - vector myArray (Array, Array+(sizeof(Array)/sizeof(string))); + vector myArray = setParameters(); OptionParser parser(option); map parameters = parser.getParameters(); @@ -103,27 +119,43 @@ ReverseSeqsCommand::ReverseSeqsCommand(string option) { //if the user has not given a path then, add inputdir. else leave path alone. if (path == "") { parameters["fasta"] = inputDir + it->second; } } + + it = parameters.find("qfile"); + //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["qfile"] = inputDir + it->second; } + } } //check for required parameters fastaFileName = validParameter.validFile(parameters, "fasta", true); if (fastaFileName == "not open") { abort = true; } else if (fastaFileName == "not found") { fastaFileName = "";}// m->mothurOut("fasta is a required parameter for the reverse.seqs command."); m->mothurOutEndLine(); abort = true; } + else { m->setFastaFile(fastaFileName); } qualFileName = validParameter.validFile(parameters, "qfile", true); if (qualFileName == "not open") { abort = true; } else if (qualFileName == "not found") { qualFileName = ""; }//m->mothurOut("fasta is a required parameter for the reverse.seqs command."); m->mothurOutEndLine(); abort = true; } - - if(fastaFileName == "not open" && qualFileName == "not open"){ - m->mothurOut("fasta or qfile is a required parameter for the reverse.seqs command."); - m->mothurOutEndLine(); - abort = true; + else { m->setQualFile(qualFileName); } + + if((fastaFileName == "") && (qualFileName == "")){ + fastaFileName = m->getFastaFile(); + if (fastaFileName != "") { m->mothurOut("Using " + fastaFileName + " as input file for the fasta parameter."); m->mothurOutEndLine(); } + else { + qualFileName = m->getQualFile(); + if (qualFileName != "") { m->mothurOut("Using " + qualFileName + " as input file for the qfile parameter."); m->mothurOutEndLine(); } + else { + m->mothurOut("You have no current files for fasta or qfile, and fasta or qfile is a required parameter for the reverse.seqs command."); m->mothurOutEndLine(); + abort = true; + } + } } //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 = ""; - outputDir += m->hasPath(fastaFileName); //if user entered a file with a path then preserve it } } @@ -133,25 +165,6 @@ ReverseSeqsCommand::ReverseSeqsCommand(string option) { exit(1); } } -//********************************************************************************************************************** - -void ReverseSeqsCommand::help(){ - try { - m->mothurOut("The reverse.seqs command reads a fastafile and outputs a fasta file containing the reverse compliment.\n"); - m->mothurOut("The reverse.seqs command parameters fasta or qfile are required.\n"); - m->mothurOut("The reverse.seqs command should be in the following format: \n"); - m->mothurOut("reverse.seqs(fasta=yourFastaFile) \n"); - } - catch(exception& e) { - m->errorOut(e, "ReverseSeqsCommand", "help"); - exit(1); - } -} - -//*************************************************************************************************************** - -ReverseSeqsCommand::~ReverseSeqsCommand(){ /* do nothing */ } - //*************************************************************************************************************** @@ -167,11 +180,13 @@ int ReverseSeqsCommand::execute(){ m->openInputFile(fastaFileName, inFASTA); ofstream outFASTA; - fastaReverseFileName = outputDir + m->getRootName(m->getSimpleName(fastaFileName)) + "rc" + m->getExtension(fastaFileName); + string tempOutputDir = outputDir; + if (outputDir == "") { tempOutputDir += m->hasPath(fastaFileName); } //if user entered a file with a path then preserve it + fastaReverseFileName = tempOutputDir + m->getRootName(m->getSimpleName(fastaFileName)) + getOutputFileNameTag("fasta", fastaFileName); m->openOutputFile(fastaReverseFileName, outFASTA); while(!inFASTA.eof()){ - if (m->control_pressed) { inFASTA.close(); outFASTA.close(); remove(fastaReverseFileName.c_str()); return 0; } + if (m->control_pressed) { inFASTA.close(); outFASTA.close(); m->mothurRemove(fastaReverseFileName); return 0; } Sequence currSeq(inFASTA); m->gobble(inFASTA); if (currSeq.getName() != "") { @@ -181,7 +196,7 @@ int ReverseSeqsCommand::execute(){ } inFASTA.close(); outFASTA.close(); - outputNames.push_back(fastaReverseFileName); + outputNames.push_back(fastaReverseFileName); outputTypes["fasta"].push_back(fastaReverseFileName); } string qualReverseFileName; @@ -193,20 +208,37 @@ int ReverseSeqsCommand::execute(){ m->openInputFile(qualFileName, inQual); ofstream outQual; - string qualReverseFileName = outputDir + m->getRootName(m->getSimpleName(qualFileName)) + "rc" + m->getExtension(qualFileName); - m->openOutputFile(qualReverseFileName, outQual); + string tempOutputDir = outputDir; + if (outputDir == "") { tempOutputDir += m->hasPath(qualFileName); } //if user entered a file with a path then preserve it + string qualReverseFileName = tempOutputDir + m->getRootName(m->getSimpleName(qualFileName)) + getOutputFileNameTag("qfile", qualFileName); + m->openOutputFile(qualReverseFileName, outQual); while(!inQual.eof()){ - if (m->control_pressed) { inQual.close(); outQual.close(); remove(qualReverseFileName.c_str()); return 0; } + if (m->control_pressed) { inQual.close(); outQual.close(); m->mothurRemove(qualReverseFileName); return 0; } currQual = QualityScores(inQual); m->gobble(inQual); currQual.flipQScores(); currQual.printQScores(outQual); } inQual.close(); outQual.close(); - outputNames.push_back(qualReverseFileName); + outputNames.push_back(qualReverseFileName); outputTypes["qfile"].push_back(qualReverseFileName); + } + + if (m->control_pressed) { m->mothurRemove(qualReverseFileName); m->mothurRemove(fastaReverseFileName); return 0; } + + //set fasta file as new current fastafile + string current = ""; + itTypes = outputTypes.find("fasta"); + if (itTypes != outputTypes.end()) { + if ((itTypes->second).size() != 0) { current = (itTypes->second)[0]; m->setFastaFile(current); } + } + + itTypes = outputTypes.find("qfile"); + if (itTypes != outputTypes.end()) { + if ((itTypes->second).size() != 0) { current = (itTypes->second)[0]; m->setQualFile(current); } } + m->mothurOutEndLine(); m->mothurOut("Output File Name: "); m->mothurOutEndLine(); for(int i=0;imothurOutEndLine(); } + + return 0; } catch(exception& e) {