]> git.donarmstrong.com Git - mothur.git/blobdiff - setdircommand.cpp
changes while testing
[mothur.git] / setdircommand.cpp
index 081a306d712a8ee10bb67cf1d72bbf0663d0ff07..3e50f4a508d2608ef1b51b9445fd3a388d1f4dec 100644 (file)
 //**********************************************************************************************************************
 vector<string> SetDirectoryCommand::setParameters(){   
        try {
-               CommandParameter ptempdefault("tempdefault", "String", "", "", "", "", "",false,false); parameters.push_back(ptempdefault);     
-               CommandParameter pinput("input", "String", "", "", "", "", "",false,false); parameters.push_back(pinput);
-               CommandParameter poutput("output", "String", "", "", "", "", "",false,false); parameters.push_back(poutput);
-               CommandParameter pinputdir("inputdir", "String", "", "", "", "", "",false,false); parameters.push_back(pinputdir);
-               CommandParameter poutputdir("outputdir", "String", "", "", "", "", "",false,false); parameters.push_back(poutputdir);
+               CommandParameter ptempdefault("tempdefault", "String", "", "", "", "", "","",false,false); parameters.push_back(ptempdefault);
+        CommandParameter pdebug("debug", "Boolean", "", "F", "", "", "","",false,false); parameters.push_back(pdebug);
+        CommandParameter pmodnames("modifynames", "Boolean", "", "T", "", "", "","",false,false); parameters.push_back(pmodnames);
+               CommandParameter pinput("input", "String", "", "", "", "", "","",false,false,true); parameters.push_back(pinput);
+               CommandParameter poutput("output", "String", "", "", "", "", "","",false,false,true); parameters.push_back(poutput);
+               CommandParameter pinputdir("inputdir", "String", "", "", "", "", "","",false,false); parameters.push_back(pinputdir);
+               CommandParameter poutputdir("outputdir", "String", "", "", "", "", "","",false,false); parameters.push_back(poutputdir);
                
                vector<string> myArray;
                for (int i = 0; i < parameters.size(); i++) {   myArray.push_back(parameters[i].name);          }
@@ -34,7 +36,10 @@ string SetDirectoryCommand::getHelpString(){
                helpString += "The set.dir command can be used to direct the output files generated by mothur to a specific place, the directory must exist.\n";
                helpString += "The set.dir command can also be used to specify the directory where your input files are located, the directory must exist.\n";
                helpString += "The set.dir command can also be used to override or set the default location mothur will look for files if it is unable to find them, the directory must exist.\n";
-               helpString += "The set.dir command parameters are input, output and tempdefault and one is required.\n";
+        helpString += "The set.dir command can also be used to run mothur in debug mode.\n";
+        helpString += "The set.dir command can also be used to set the modifynames parameter. Default=t, meaning if your sequence names contain ':' change them to '_' to avoid issues while making trees.  modifynames=F will leave sequence names as they are.\n";
+               helpString += "The set.dir command parameters are input, output, tempdefault and debug and one is required.\n";
+        helpString += "To run mothur in debug mode set debug=true. Default debug=false.\n";
                helpString += "To return the output to the same directory as the input files you may enter: output=clear.\n";
                helpString += "To return the input to the current working directory you may enter: input=clear.\n";
                helpString += "To set the output to the directory where mothur.exe is located you may enter: output=default.\n";
@@ -81,10 +86,27 @@ SetDirectoryCommand::SetDirectoryCommand(string option)  {
                        
                        tempdefault = validParameter.validFile(parameters, "tempdefault", false);                       
                        if (tempdefault == "not found") {  tempdefault = "";  }
+            
+            bool debug = false;
+            bool nodebug = false;
+            debugOnly = false;
+            string temp = validParameter.validFile(parameters, "debug", false);                        
+                       if (temp == "not found") {  debug = false;  nodebug=true; }
+            else {  debug = m->isTrue(temp); }
+            m->debug = debug;
+            
+            bool nomod = false;
+            temp = validParameter.validFile(parameters, "modifynames", false);
+                       if (temp == "not found") {  modifyNames = true;  nomod=true; }
+            else {  modifyNames = m->isTrue(temp); }
+            m->modifyNames = modifyNames;
+            
+            if (debug) { m->mothurOut("Setting [DEBUG] flag.\n"); }
+            
                                
-                       if ((input == "") && (output == "") && (tempdefault == "")) {   
-                               m->mothurOut("You must provide either an input, output or tempdefault for the set.outdir command."); m->mothurOutEndLine(); abort = true;
-                       }
+                       if ((input == "") && (output == "") && (tempdefault == "") && nodebug && nomod) {
+                               m->mothurOut("You must provide either an input, output, tempdefault, debug or modifynames for the set.outdir command."); m->mothurOutEndLine(); abort = true;
+                       }else if((input == "") && (output == "") && (tempdefault == "")) { debugOnly = true; }
                }
        }
        catch(exception& e) {
@@ -99,16 +121,10 @@ int SetDirectoryCommand::execute(){
                
                if (abort == true) { if (calledHelp) { return 0; }  return 2;   }
                
+        if (debugOnly) { return 0; }
+        
                commandFactory = CommandFactory::getInstance();
                
-               string tag = "";
-#ifdef USE_MPI
-               int pid; 
-               MPI_Comm_rank(MPI_COMM_WORLD, &pid); //find out who we are
-               
-               tag = toString(pid);
-#endif
-               
                m->mothurOut("Mothur's directories:"); m->mothurOutEndLine();
                
                //redirect output
@@ -120,27 +136,10 @@ int SetDirectoryCommand::execute(){
                        m->mothurOut("outputDir=" + output); m->mothurOutEndLine();  
                        commandFactory->setOutputDirectory(output);
                }else {
-                       //add / to name if needed
-                       string lastChar = output.substr(output.length()-1);
-                       #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
-                               if (lastChar != "/") { output += "/"; }
-                       #else
-                               if (lastChar != "\\") { output += "\\"; }       
-                       #endif
-                       
-                       //test to make sure directory exists
-                       output = m->getFullPathName(output);
-                       string outTemp = output + tag + "temp";
-                       ofstream out;
-                       out.open(outTemp.c_str(), ios::trunc);
-                       if(!out) {
-                               m->mothurOut(output + " directory does not exist or is not writable."); m->mothurOutEndLine(); 
-                       }else{
-                               out.close();
-                               m->mothurRemove(outTemp);
-                               m->mothurOut("outputDir=" + output); m->mothurOutEndLine();  
+            if (m->dirCheck(output)) {
+                m->mothurOut("outputDir=" + output); m->mothurOutEndLine();  
                                commandFactory->setOutputDirectory(output);
-                       }
+            }
                }
                
                //redirect input
@@ -152,28 +151,11 @@ int SetDirectoryCommand::execute(){
                        m->mothurOut("inputDir=" + input); m->mothurOutEndLine();  
                        commandFactory->setInputDirectory(input);
                }else {
-                       //add / to name if needed
-                       string lastChar = input.substr(input.length()-1);
-                       #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
-                               if (lastChar != "/") { input += "/"; }
-                       #else
-                               if (lastChar != "\\") { input += "\\"; }        
-                       #endif
-                       
-                       //test to make sure directory exists
-                       input = m->getFullPathName(input);
-                       string inTemp = input + tag + "temp";
-                       ofstream in;
-                       in.open(inTemp.c_str(), ios::trunc);
-                       if(!in) {
-                               m->mothurOut(input + " directory does not exist or is not writable."); m->mothurOutEndLine(); 
-                       }else{
-                               in.close();
-                               m->mothurRemove(inTemp);
-                               m->mothurOut("inputDir=" + input); m->mothurOutEndLine();  
+            if (m->dirCheck(input)) {
+                m->mothurOut("inputDir=" + input); m->mothurOutEndLine();  
                                commandFactory->setInputDirectory(input); 
-                       }
-               }
+            }
+        }
                
                //set default
                if (tempdefault == "clear") {  
@@ -194,28 +176,11 @@ int SetDirectoryCommand::execute(){
                        m->mothurOut("tempDefault=" + tempdefault); m->mothurOutEndLine();  
                        m->setDefaultPath(tempdefault);
                }else {
-                       //add / to name if needed
-                       string lastChar = tempdefault.substr(tempdefault.length()-1);
-                       #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
-                               if (lastChar != "/") { tempdefault += "/"; }
-                       #else
-                               if (lastChar != "\\") { tempdefault += "\\"; }  
-                       #endif
-                       
-                       //test to make sure directory exists
-                       tempdefault = m->getFullPathName(tempdefault);
-                       string inTemp = tempdefault + tag + "temp";
-                       ofstream in;
-                       in.open(inTemp.c_str(), ios::trunc);
-                       if(!in) {
-                               m->mothurOut(tempdefault + " directory does not exist or is not writable."); m->mothurOutEndLine(); 
-                       }else{
-                               in.close();
-                               m->mothurRemove(inTemp);
-                               m->mothurOut("tempDefault=" + tempdefault); m->mothurOutEndLine();  
-                               m->setDefaultPath(tempdefault);
-                       }
-               }
+            if (m->dirCheck(tempdefault)) {
+                m->mothurOut("tempDefault=" + tempdefault); m->mothurOutEndLine();  
+                               m->setDefaultPath(tempdefault); 
+            }
+        }
 
                return 0;
        }