]> git.donarmstrong.com Git - mothur.git/blobdiff - engine.cpp
moved utilities out of mothur.h and into mothurOut class.
[mothur.git] / engine.cpp
index 371bda8fb83f2b8ceb4793c49f067ed5987f2094..5e82d1344d9fa068078e0176b1e655a8290cd42b 100644 (file)
 
 #include "engine.hpp"
 
+/***********************************************************************/
+Engine::Engine(){
+       try {
+               cFactory = CommandFactory::getInstance();
+               mout = MothurOut::getInstance();
+       }
+       catch(exception& e) {
+               mout->errorOut(e, "Engine", "Engine");
+               exit(1);
+       }
+}
+
 /***********************************************************************/
 
 InteractEngine::InteractEngine(string path){
 
        globaldata = GlobalData::getInstance();
+       
+       string temppath = path.substr(0, (path.find_last_of('m')));
+       
+       //this will happen if you set the path variable to contain mothur's exe location
+       if (temppath == "") { 
+       
+               string envPath = getenv("PATH");
+               
+               //delimiting path char
+               char delim;
+               #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux)
+                       delim = ':';
+               #else
+                       delim = ';';
+               #endif
+               
+               //break apart path variable by ':'
+               vector<string> dirs;
+               mout->splitAtChar(envPath, dirs, delim);
+               
+               //get path related to mothur
+               string mothurPath = "";
+               for (int i = 0; i < dirs.size(); i++) {
+                       //to lower so we can find it
+                       string tempLower = "";
+                       for (int j = 0; j < dirs[i].length(); j++) {  tempLower += tolower(dirs[i][j]);  }
+                       
+                       //is this mothurs path?
+                       if (tempLower.find("mothur") != -1) {  mothurPath = dirs[i]; break;  }
+               }
+               
+               //add mothur so it looks like what argv would look like
+               #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux)
+                       mothurPath += "/mothur";
+               #else
+                       mothurPath += "\\mothur";
+               #endif
+               
+               path = mothurPath;
+       }
+       
        globaldata->argv = path;
 }
 
@@ -36,89 +89,126 @@ bool InteractEngine::getInput(){
                string options = "";
                int quitCommandCalled = 0;
                
-                               
                while(quitCommandCalled != 1){
-                       
-                       mothurOutEndLine();
-                       mothurOut("mothur > ");
-                       
-                       //get input char by char so you can check for use of arrow keys
-                       //if (_kbhit()){
-                       //      _getch(); // edit : if you want to check the arrow-keys you must call getch twice because special-keys have two values
-                       //      return _getch();
-                       //}
-                       //return 0; // if no key is pressed
-                       //setbuf(stdin, NULL); //no buffering
-/*if(ch==0)
-{
-ch=getch();
-if(ch==72) cout<<"up";
-else if(ch==75) cout<<"left";
-else if(ch==77) cout<<"right";
-else if(ch==80) cout<<"down";
-cout<<endl;
-}
-else break;
-}
-delay(2000);
-return 0;
-}*/
-                       
-                       //int letter = 0;
-                       //while ((letter != 10) && (letter != 13)) {
-                       //      letter = getch();
+
+                       #ifdef USE_MPI
+                               int pid, processors;
+                               MPI_Status status;
+                               MPI_Comm_rank(MPI_COMM_WORLD, &pid); 
+                               MPI_Comm_size(MPI_COMM_WORLD, &processors);
                                
-                       //      cout << "char code = " << letter << endl;
+                               if (pid == 0) {
                                
-                       //      input += char(letter);
-                       //}
-               //      input = input.substr(0, input.length()-1); //cut off newline char
-               
-       //cout << input << endl;                
+                       #endif
                        
-                       getline(cin, input);
-                       if (cin.eof()) { input = "quit()"; }
+                       mout->mothurOutEndLine();
                        
-                       //save command
-                       //previousInputs.push_back(input);
+                       input = getCommand();   
+                       mout->mothurOutEndLine();       
                        
-                       mothurOutJustToLog(input);
-                       mothurOutEndLine();
+                       if (mout->control_pressed) { input = "quit()"; }
                        
                        //allow user to omit the () on the quit command
                        if (input == "quit") { input = "quit()"; }
+
                        
+                       #ifdef USE_MPI
+                               //send commandName
+                               for(int i = 1; i < processors; i++) { 
+                                               int length = input.length();
+                                               MPI_Send(&length, 1, MPI_INT, i, 2001, MPI_COMM_WORLD);
+                                               MPI_Send(&input[0], length, MPI_CHAR, i, 2001, MPI_COMM_WORLD);
+       
+                                       }
+                               }else {
+                                       int length;
+                                       MPI_Recv(&length, 1, MPI_INT, 0, 2001, MPI_COMM_WORLD, &status);
+                                       //recieve container
+                                       char* tempBuf = new char[length];
+                                       MPI_Recv(&tempBuf[0], length, MPI_CHAR, 0, 2001, MPI_COMM_WORLD, &status);
+                                       
+                                       input = tempBuf;
+                                       if (input.length() > length) { input = input.substr(0, length);  }
+                                       delete tempBuf; 
+                               }
+
+                       
+                       #endif
+               
                        CommandOptionParser parser(input);
                        commandName = parser.getCommandString();
-       //cout << " command = " << commandName << endl;
+       
                        options = parser.getOptionString();
                        
                        if (commandName != "") {
-                       
-                               //executes valid command
-                               Command* command = cFactory->getCommand(commandName, options);
-                               quitCommandCalled = command->execute();
-                               
-                       }else {
-                               mothurOut("Your input contains errors. Please try again."); 
-                               mothurOutEndLine();
-                       }
+                                       mout->executing = true;
+                                       #ifdef USE_MPI
+                                               int pid;
+                                               MPI_Comm_rank(MPI_COMM_WORLD, &pid); 
+                                               
+                                               if ((cFactory->MPIEnabled(commandName)) || (pid == 0)) {
+                                       //cout << pid << " is in execute " << commandName << endl;
+                                       #endif
+                                       //executes valid command
+                                       Command* command = cFactory->getCommand(commandName, options);
+                                       quitCommandCalled = command->execute();
+                                       mout->control_pressed = 0;
+                                       mout->executing = false;
+                                                                               
+                                       #ifdef USE_MPI
+                                               }
+                                       #endif
+                               }else {         
+                                       mout->mothurOut("Invalid."); 
+                                       mout->mothurOutEndLine();
+                               }
                }       
                return 1;
        }
        catch(exception& e) {
-               errorOut(e, "InteractEngine", "getInput");
+               mout->errorOut(e, "InteractEngine", "getInput");
                exit(1);
        }
 }
 /***********************************************************************/
-void Engine::terminateCommand(int dummy)  {
+string Engine::getCommand()  {
        try {
-               mothurOut("Stopping command."); mothurOutEndLine();
-               cFactory->getCommand();  //terminates command
+       
+               #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux)
+                       #ifdef USE_READLINE
+                               char* nextCommand = NULL;
+                               nextCommand = readline("mothur > ");
+                               
+                               if(nextCommand != NULL) {  add_history(nextCommand);  } 
+                               else{ //^D causes null string and we want it to quit mothur
+                                       nextCommand = "quit"; 
+                                       mout->mothurOut(nextCommand);
+                               }       
+                               
+                               mout->mothurOutJustToLog("mothur > " + toString(nextCommand));
+                               return nextCommand;
+                       #else
+                               string nextCommand = "";
+                               mout->mothurOut("mothur > ");
+                               getline(cin, nextCommand);
+                               mout->mothurOutJustToLog("mothur > " + toString(nextCommand));
+                               
+                               return nextCommand;
+                       #endif
+               #else
+                               string nextCommand = "";
+                               
+                               mout->mothurOut("mothur > ");
+                               m->getline(cin, nextCommand);
+                               mout->mothurOutJustToLog(toString(nextCommand));
+                               
+                               return nextCommand;
+               #endif
+       
+                                               
        }
        catch(exception& e) {
-               errorOut(e, "InteractEngine", "terminateCommand");
+               mout->errorOut(e, "Engine", "getCommand");
                exit(1);
        }
 }
@@ -128,12 +218,53 @@ BatchEngine::BatchEngine(string path, string batchFileName){
        try {
                globaldata = GlobalData::getInstance();
        
-               openedBatch = openInputFile(batchFileName, inputBatchFile);
+               openedBatch = mout->openInputFile(batchFileName, inputBatchFile);
+               
+               string temppath = path.substr(0, (path.find_last_of('m')));
+       
+               //this will happen if you set the path variable to contain mothur's exe location
+               if (temppath == "") { 
+               
+                       string envPath = getenv("PATH");
+                       
+                       //delimiting path char
+                       char delim;
+                       #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux)
+                               delim = ':';
+                       #else
+                               delim = ';';
+                       #endif
+                       
+                       //break apart path variable by ':'
+                       vector<string> dirs;
+                       mout->splitAtChar(envPath, dirs, delim);
+                       
+                       //get path related to mothur
+                       string mothurPath = "";
+                       for (int i = 0; i < dirs.size(); i++) {
+                               //to lower so we can find it
+                               string tempLower = "";
+                               for (int j = 0; j < dirs[i].length(); j++) {  tempLower += tolower(dirs[i][j]);  }
+                               
+                               //is this mothurs path?
+                               if (tempLower.find("mothur") != -1) {  mothurPath = dirs[i]; break;  }
+                       }
+                       
+                       //add mothur so it looks like what argv would look like
+                       #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux)
+                               mothurPath += "/mothur";
+                       #else
+                               mothurPath += "\\mothur";
+                       #endif
+                       
+                       path = mothurPath;
+               }
+
                globaldata->argv = path;
                                
        }
        catch(exception& e) {
-               errorOut(e, "BatchEngine", "BatchEngine");
+               mout->errorOut(e, "BatchEngine", "BatchEngine");
                exit(1);
        }
 }
@@ -148,8 +279,8 @@ bool BatchEngine::getInput(){
        try {
                //check if this is a valid batchfile
                if (openedBatch == 1) {  
-                       mothurOut("unable to open batchfile");  
-                       mothurOutEndLine();
+                       mout->mothurOut("unable to open batchfile");  
+                       mout->mothurOutEndLine();
                        return 1; 
                }
        
@@ -159,18 +290,54 @@ bool BatchEngine::getInput(){
                
                //CommandFactory cFactory;
                int quitCommandCalled = 0;
-       
+           int count = 0;
                while(quitCommandCalled == 0){
+                       
+                       #ifdef USE_MPI
+                               int pid, processors;
+                               MPI_Status status;
+                               MPI_Comm_rank(MPI_COMM_WORLD, &pid); 
+                               MPI_Comm_size(MPI_COMM_WORLD, &processors);
+                               
+                               if (pid == 0) {
+                               
+                       #endif
+                       
+                       input = getNextCommand(inputBatchFile);
+                       count++;
+                       
+                       #ifdef USE_MPI
+                               //send commandName
+                               for(int i = 1; i < processors; i++) { 
+                                               int length = input.length();
+                                               MPI_Send(&length, 1, MPI_INT, i, 2001, MPI_COMM_WORLD);
+                                               MPI_Send(&input[0], length, MPI_CHAR, i, 2001, MPI_COMM_WORLD);
        
-                       if (inputBatchFile.eof()) { input = "quit()"; }
-                       else { input = getline(inputBatchFile); }
+                                       }
+                               }else {
+                                       int length;
+                                       MPI_Recv(&length, 1, MPI_INT, 0, 2001, MPI_COMM_WORLD, &status);
+                                       //recieve container
+                                       char* tempBuf = new char[length];
+                                       MPI_Recv(&tempBuf[0], length, MPI_CHAR, 0, 2001, MPI_COMM_WORLD, &status);
+                                       
+                                       input = tempBuf;
+                                       if (input.length() > length) { input = input.substr(0, length);  }
+                                       delete tempBuf; 
+                               }
+
+                       
+                       #endif
+
+                       
                        
                        if (input[0] != '#') {
                                
-                               mothurOutEndLine();
-                               mothurOut("mothur > " + input);
-                               mothurOutEndLine();
-                               
+                               mout->mothurOutEndLine();
+                               mout->mothurOut("mothur > " + input);
+                               mout->mothurOutEndLine();
+                                                       
+                               if (mout->control_pressed) { input = "quit()"; }
                                
                                //allow user to omit the () on the quit command
                                if (input == "quit") { input = "quit()"; }
@@ -180,28 +347,56 @@ bool BatchEngine::getInput(){
                                options = parser.getOptionString();
                                                                                
                                if (commandName != "") {
-
+                                       mout->executing = true;
+                                       #ifdef USE_MPI
+                                               int pid;
+                                               MPI_Comm_rank(MPI_COMM_WORLD, &pid); 
+                                               
+//cout << pid << " is here " << commandName << '\t' << count << endl;
+                                               if ((cFactory->MPIEnabled(commandName)) || (pid == 0)) {
+                                       #endif
                                        //executes valid command
                                        Command* command = cFactory->getCommand(commandName, options);
                                        quitCommandCalled = command->execute();
+                                       mout->control_pressed = 0;
+                                       mout->executing = false;
+                                                                               
+                                       #ifdef USE_MPI
+                                               }
+                                       #endif
                                }else {         
-                                       mothurOut("Invalid."); 
-                                       mothurOutEndLine();
+                                       mout->mothurOut("Invalid."); 
+                                       mout->mothurOutEndLine();
                                }
                                
                        }
-                       gobble(inputBatchFile);
+                       mout->gobble(inputBatchFile);
                }
                
                inputBatchFile.close();
                return 1;
        }
        catch(exception& e) {
-               errorOut(e, "BatchEngine", "getInput");
+               mout->errorOut(e, "BatchEngine", "getInput");
+               exit(1);
+       }
+}
+/***********************************************************************/
+string BatchEngine::getNextCommand(ifstream& inputBatchFile) {
+       try {
+               
+               string nextcommand = "";
+               
+               if (inputBatchFile.eof()) { nextcommand = "quit()"; }
+               else { nextcommand = mout->getline(inputBatchFile); }
+               
+               return nextcommand;
+       }
+       catch(exception& e) {
+               mout->errorOut(e, "BatchEngine", "getNextCommand");
                exit(1);
        }
 }
-
 
 /***********************************************************************/
 /***********************************************************************/
@@ -212,12 +407,52 @@ ScriptEngine::ScriptEngine(string path, string commandString){
                
                //remove quotes
                listOfCommands = commandString.substr(1, (commandString.length()-1));
+               
+               string temppath = path.substr(0, (path.find_last_of('m')));
+       
+               //this will happen if you set the path variable to contain mothur's exe location
+               if (temppath == "") { 
+               
+                       string envPath = getenv("PATH");
+                       
+                       //delimiting path char
+                       char delim;
+                       #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux)
+                               delim = ':';
+                       #else
+                               delim = ';';
+                       #endif
+                       
+                       //break apart path variable by ':'
+                       vector<string> dirs;
+                       mout->splitAtChar(envPath, dirs, delim);
+                       
+                       //get path related to mothur
+                       string mothurPath = "";
+                       for (int i = 0; i < dirs.size(); i++) {
+                               //to lower so we can find it
+                               string tempLower = "";
+                               for (int j = 0; j < dirs[i].length(); j++) {  tempLower += tolower(dirs[i][j]);  }
+                               
+                               //is this mothurs path?
+                               if (tempLower.find("mothur") != -1) {  mothurPath = dirs[i]; break;  }
+                       }
+                       
+                       //add mothur so it looks like what argv would look like
+                       #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux)
+                               mothurPath += "/mothur";
+                       #else
+                               mothurPath += "\\mothur";
+                       #endif
+                       
+                       path = mothurPath;
+               }
 
                globaldata->argv = path;
-               
+                               
        }
        catch(exception& e) {
-               errorOut(e, "ScriptEngine", "ScriptEngine");
+               mout->errorOut(e, "ScriptEngine", "ScriptEngine");
                exit(1);
        }
 }
@@ -240,16 +475,50 @@ bool ScriptEngine::getInput(){
                int quitCommandCalled = 0;
        
                while(quitCommandCalled == 0){
-               
+                       
+                       #ifdef USE_MPI
+                               int pid, processors;
+                               MPI_Status status;
+                               MPI_Comm_rank(MPI_COMM_WORLD, &pid); 
+                               MPI_Comm_size(MPI_COMM_WORLD, &processors);
+                               
+                               if (pid == 0) {
+                               
+                       #endif
+                       
                        input = getNextCommand(listOfCommands); 
                        
                        if (input == "") { input = "quit()"; }
                        
+                       mout->mothurOutEndLine();
+                       mout->mothurOut("mothur > " + input);
+                       mout->mothurOutEndLine();
                        
-                       mothurOutEndLine();
-                       mothurOut("mothur > " + input);
-                       mothurOutEndLine();
+                       #ifdef USE_MPI
+                               //send commandName
+                               for(int i = 1; i < processors; i++) { 
+                                               int length = input.length();
+                                               MPI_Send(&length, 1, MPI_INT, i, 2001, MPI_COMM_WORLD);
+                                               MPI_Send(&input[0], length, MPI_CHAR, i, 2001, MPI_COMM_WORLD);
+       
+                                       }
+                               }else {
+                                       int length;
+                                       MPI_Recv(&length, 1, MPI_INT, 0, 2001, MPI_COMM_WORLD, &status);
+                                       //recieve container
+                                       char* tempBuf = new char[length];
+                                       MPI_Recv(&tempBuf[0], length, MPI_CHAR, 0, 2001, MPI_COMM_WORLD, &status);
+                                       
+                                       input = tempBuf;
+                                       if (input.length() > length) { input = input.substr(0, length);  }
+                                       delete tempBuf; 
+                               }
 
+                       
+                       #endif
+                       
+                       
+                       if (mout->control_pressed) { input = "quit()"; }
                                
                        //allow user to omit the () on the quit command
                        if (input == "quit") { input = "quit()"; }
@@ -259,27 +528,47 @@ bool ScriptEngine::getInput(){
                        options = parser.getOptionString();
                                                                                
                        if (commandName != "") {
+                                       mout->executing = true;
+                                       #ifdef USE_MPI
+                                               int pid, numProcesses;
+                                               MPI_Status status; 
+                                               
+                                               MPI_Comm_rank(MPI_COMM_WORLD, &pid); 
+                                               MPI_Comm_size(MPI_COMM_WORLD, &numProcesses); 
+                                       
+//cout << pid << " is here " << commandName  << endl;
+                                               if ((cFactory->MPIEnabled(commandName)) || (pid == 0)) {
+                                                       //cout << pid << " is in execute" << endl;      
+                                       #endif
+                                       //executes valid command
+                                       Command* command = cFactory->getCommand(commandName, options);
+                                       quitCommandCalled = command->execute();
+                                       mout->control_pressed = 0;
+                                       mout->executing = false;
+                                                                       
+                                       #ifdef USE_MPI
+                                       //cout << pid << " is done in execute" << endl;
+                                               }
+                                       #endif
+                               }else {         
+                                       mout->mothurOut("Invalid."); 
+                                       mout->mothurOutEndLine();
+                               }
 
-                               //executes valid command
-                               Command* command = cFactory->getCommand(commandName, options);
-                               quitCommandCalled = command->execute();
-                       }else {         
-                               mothurOut("Invalid."); 
-                               mothurOutEndLine();
-                       }
                        
                }
                
                return 1;
        }
        catch(exception& e) {
-               errorOut(e, "ScriptEngine", "getInput");
+               mout->errorOut(e, "ScriptEngine", "getInput");
                exit(1);
        }
 }
 /***********************************************************************/
 string ScriptEngine::getNextCommand(string& commandString) {
        try {
+               
                string nextcommand = "";
                int count = 0;
                
@@ -304,11 +593,11 @@ string ScriptEngine::getNextCommand(string& commandString) {
                                if (commandString.length() == 0) {  break;  }
                        }
                }
-                                       
+               
                return nextcommand;
        }
        catch(exception& e) {
-               errorOut(e, "ScriptEngine", "getNextCommand");
+               mout->errorOut(e, "ScriptEngine", "getNextCommand");
                exit(1);
        }
 }