X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=mothur.h;h=74d71caed4df23c2472b7b9b8c229e315721a424;hb=f89b6501b7d80e705b6c50ff2c8bdd8bb5d2edf7;hp=78ccf1105795ec6dd018ae5820c5ceca2110c505;hpb=4ba6188707e67c638ea5ee3c787d5d07ccd07e3e;p=mothur.git diff --git a/mothur.h b/mothur.h index 78ccf11..74d71ca 100644 --- a/mothur.h +++ b/mothur.h @@ -23,6 +23,7 @@ #include #include + //exception #include #include @@ -45,16 +46,23 @@ #include #include #include -#include -#include /***********************************************************************/ #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) #include #include + + #ifdef USE_READLINE + #include + #include + #endif + + //#include + //#include #else #include //allows unbuffered screen capture from stdin + #include //get cwd #endif using namespace std; @@ -66,7 +74,6 @@ using namespace std; #define isnan(x) ((x) != (x)) #define isinf(x) (fabs(x) == std::numeric_limits::infinity()) - typedef unsigned long ull; struct IntNode { @@ -94,7 +101,20 @@ struct clusterNode { int smallChild; //used to make linkTable work with list and rabund. represents bin number of this cluster node clusterNode(int num, int par, int kid) : numSeq(num), parent(par), smallChild(kid) {}; }; - +/************************************************************/ +struct seqDist { + int seq1; + int seq2; + float dist; + seqDist() {} + seqDist(int s1, int s2, float d) : seq1(s1), seq2(s2), dist(d) {} + ~seqDist() {} +}; +//******************************************************************************************************************** +//sorts lowest to highest +inline bool compareSequenceDistance(seqDist left, seqDist right){ + return (left.dist < right.dist); +} /***********************************************************************/ // snagged from http://www.parashift.com/c++-faq-lite/misc-technical-issues.html#faq-39.2 @@ -195,7 +215,6 @@ inline void gobble(istream& f){ f.putback(d); } - /***********************************************************************/ inline string getline(ifstream& fileHandle) { @@ -374,11 +393,18 @@ inline string getSimpleName(string longName){ string simpleName = longName; - if(longName.find_last_of("/") != longName.npos){ - int pos = longName.find_last_of('/')+1; - simpleName = longName.substr(pos, longName.length()); - } + size_t found; + found=longName.find_last_of("/\\"); + if(found != longName.npos){ + simpleName = longName.substr(found+1); + } + + //if(longName.find_last_of("/") != longName.npos){ + // int pos = longName.find_last_of('/')+1; + // simpleName = longName.substr(pos, longName.length()); + //} + return simpleName; } @@ -407,13 +433,28 @@ inline string getPathName(string longName){ string rootPathName = longName; - if(longName.find_last_of('/') != longName.npos){ - int pos = longName.find_last_of('/')+1; + if(longName.find_last_of("/\\") != longName.npos){ + int pos = longName.find_last_of("/\\")+1; rootPathName = longName.substr(0, pos); } - + return rootPathName; } +/***********************************************************************/ + +inline string hasPath(string longName){ + + string path = ""; + + size_t found; + found=longName.find_last_of("/\\"); + + if(found != longName.npos){ + path = longName.substr(0, found+1); + } + + return path; +} /***********************************************************************/ @@ -445,30 +486,173 @@ inline bool isBlank(string fileName){ } /***********************************************************************/ -inline int openInputFile(string fileName, ifstream& fileHandle){ +inline string getFullPathName(string fileName){ + + string path = hasPath(fileName); + string newFileName; + int pos; + + if (path == "") { return fileName; } //its a simple name + else { //we need to complete the pathname + // ex. ../../../filename + // cwd = /user/work/desktop + + string cwd; + //get current working directory + #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) + if (path.rfind("./") == -1) { return fileName; } //already complete name + else { newFileName = fileName.substr(fileName.rfind("./")+2); } //save the complete part of the name + + char* cwdpath; + size_t size; + cwdpath=getcwd(cwdpath,size); + cwd = cwdpath; + + //rip off first '/' + string simpleCWD; + if (cwd.length() > 0) { simpleCWD = cwd.substr(1); } + + //break apart the current working directory + vector dirs; + while (simpleCWD.find_first_of('/') != -1) { + string dir = simpleCWD.substr(0,simpleCWD.find_first_of('/')); + simpleCWD = simpleCWD.substr(simpleCWD.find_first_of('/')+1, simpleCWD.length()); + dirs.push_back(dir); + } + //get last one // ex. ../../../filename = /user/work/desktop/filename + dirs.push_back(simpleCWD); //ex. dirs[0] = user, dirs[1] = work, dirs[2] = desktop + + int index = dirs.size()-1; + + while((pos = path.rfind("./")) != -1) { //while you don't have a complete path + if (path[(pos-1)] == '.') { //you want your parent directory ../ + path = path.substr(0, pos-1); + index--; + if (index == 0) { break; } + }else if (path[(pos-1)] == '/') { //you want the current working dir ./ + path = path.substr(0, pos); + }else if (pos == 1) { break; + }else { mothurOut("cannot resolve path for " + fileName); mothurOutEndLine(); return fileName; } + } + + for (int i = index; i >= 0; i--) { + newFileName = dirs[i] + "/" + newFileName; + } + + newFileName = "/" + newFileName; + return newFileName; + + #else + if (path.rfind(".\\") == -1) { return fileName; } //already complete name + else { newFileName = fileName.substr(fileName.rfind(".\\")+2); } //save the complete part of the name + + char *cwdpath = NULL; + cwdpath = getcwd(NULL, 0); // or _getcwd + if ( cwdpath != NULL) { cwd = cwdpath; } + else { cwd = ""; } + + //break apart the current working directory + vector dirs; + while (cwd.find_first_of('\\') != -1) { + string dir = cwd.substr(0,cwd.find_first_of('\\')); + cwd = cwd.substr(cwd.find_first_of('\\')+1, cwd.length()); + dirs.push_back(dir); + + } + //get last one + dirs.push_back(cwd); //ex. dirs[0] = user, dirs[1] = work, dirs[2] = desktop + + int index = dirs.size()-1; + + while((pos = path.rfind(".\\")) != -1) { //while you don't have a complete path + if (path[(pos-1)] == '.') { //you want your parent directory ../ + path = path.substr(0, pos-1); + index--; + if (index == 0) { break; } + }else if (path[(pos-1)] == '\\') { //you want the current working dir ./ + path = path.substr(0, pos); + }else if (pos == 1) { break; + }else { mothurOut("cannot resolve path for " + fileName); mothurOutEndLine(); return fileName; } + } + + for (int i = index; i >= 0; i--) { + newFileName = dirs[i] + "\\" + newFileName; + } + + return newFileName; + + #endif + } + +} +/***********************************************************************/ - fileHandle.open(fileName.c_str()); +inline int openInputFile(string fileName, ifstream& fileHandle, string m){ + + //get full path name + string completeFileName = getFullPathName(fileName); + + fileHandle.open(completeFileName.c_str()); if(!fileHandle) { - mothurOut("Error: Could not open " + fileName); mothurOutEndLine(); + return 1; + }else { + //check for blank file + gobble(fileHandle); + return 0; + } +} +/***********************************************************************/ + +inline int openInputFile(string fileName, ifstream& fileHandle){ + //get full path name + string completeFileName = getFullPathName(fileName); + + fileHandle.open(completeFileName.c_str()); + if(!fileHandle) { + mothurOut("Error: Could not open " + completeFileName); mothurOutEndLine(); return 1; } else { //check for blank file gobble(fileHandle); - if (fileHandle.eof()) { mothurOut(fileName + " is blank. Please correct."); mothurOutEndLine(); return 1; } + if (fileHandle.eof()) { mothurOut(completeFileName + " is blank. Please correct."); mothurOutEndLine(); return 1; } return 0; } } +/***********************************************************************/ + +inline int renameFile(string oldName, string newName){ + + ifstream inTest; + int exist = openInputFile(newName, inTest, ""); + +#if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) + if (exist == 0) { //you could open it so you want to delete it + inTest.close(); + string command = "rm " + newName; + system(command.c_str()); + } + + string command = "mv " + oldName + " " + newName; + system(command.c_str()); +#else + remove(newName.c_str()); + int renameOk = rename(oldName.c_str(), newName.c_str()); +#endif + return 0; +} /***********************************************************************/ inline int openOutputFile(string fileName, ofstream& fileHandle){ + + string completeFileName = getFullPathName(fileName); - fileHandle.open(fileName.c_str(), ios::trunc); + fileHandle.open(completeFileName.c_str(), ios::trunc); if(!fileHandle) { - mothurOut("Error: Could not open " + fileName); mothurOutEndLine(); + mothurOut("Error: Could not open " + completeFileName); mothurOutEndLine(); return 1; } else { @@ -496,7 +680,6 @@ inline bool inVector(string member, vector group){ return false; } - /***********************************************************************/ //This function parses the estimator options and puts them in a vector @@ -771,6 +954,92 @@ inline bool anyLabelsToProcess(string label, set& userLabels, string err } } +/**************************************************************************************************/ +inline void appendFiles(string temp, string filename) { + try{ + ofstream output; + ifstream input; + + //open output file in append mode + openOutputFileAppend(filename, output); + openInputFile(temp, input); + + while(char c = input.get()){ + if(input.eof()) { break; } + else { output << c; } + } + + input.close(); + output.close(); + } + catch(exception& e) { + errorOut(e, "mothur", "appendFiles"); + exit(1); + } +} + +/**************************************************************************************************/ +inline string sortFile(string distFile){ + try { + string outfile = getRootName(distFile) + "sorted.dist"; + + //if you can, use the unix sort since its been optimized for years + #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) + string command = "sort -n -k +3 " + distFile + " -o " + outfile; + system(command.c_str()); + #else //you are stuck with my best attempt... + //windows sort does not have a way to specify a column, only a character in the line + //since we cannot assume that the distance will always be at the the same character location on each line + //due to variable sequence name lengths, I chose to force the distance into first position, then sort and then put it back. + + //read in file line by file and put distance first + string tempDistFile = distFile + ".temp"; + ifstream input; + ofstream output; + openInputFile(distFile, input); + openOutputFile(tempDistFile, output); + + string firstName, secondName; + float dist; + while (input) { + input >> firstName >> secondName >> dist; + output << dist << '\t' << firstName << '\t' << secondName << endl; + gobble(input); + } + input.close(); + output.close(); + + + //sort using windows sort + string tempOutfile = outfile + ".temp"; + string command = "sort " + tempDistFile + " /O " + tempOutfile; + system(command.c_str()); + + //read in sorted file and put distance at end again + ifstream input2; + openInputFile(tempOutfile, input2); + openOutputFile(outfile, output); + + while (input2) { + input2 >> dist >> firstName >> secondName; + output << firstName << '\t' << secondName << '\t' << dist << endl; + gobble(input2); + } + input2.close(); + output.close(); + + //remove temp files + remove(tempDistFile.c_str()); + remove(tempOutfile.c_str()); + #endif + + return outfile; + } + catch(exception& e) { + errorOut(e, "mothur", "sortFile"); + exit(1); + } +} /**************************************************************************************************/ #endif