]> git.donarmstrong.com Git - mothur.git/blobdiff - distancecommand.cpp
fixed Thomas' calculators
[mothur.git] / distancecommand.cpp
index c20ddb91aecf7a4c0f7460cce4b11601c64ac597..7d1f6a0308b5358b2ef39ef70579d866fafca6da 100644 (file)
@@ -21,10 +21,8 @@ DistanceCommand::DistanceCommand(){
                globaldata = GlobalData::getInstance();
                validCalculator = new ValidCalculators();
                ends = globaldata->getEnds();
-               seqDB = globaldata->gSequenceDB;
                convert(globaldata->getProcessors(), processors);
                convert(globaldata->getCutOff(), cutoff);
-               distFile = getRootName(globaldata->getFastaFile()) + "dist";
                
                int i;
                if (ends != "T") {
@@ -52,11 +50,8 @@ DistanceCommand::DistanceCommand(){
                        }
                }
                                
-                               
                //reset calc for next command
                globaldata->setCalc("");
-
-               
        }
        catch(exception& e) {
                cout << "Standard Error: " << e.what() << " has occurred in the DistanceCommand class Function DistanceCommand. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
@@ -71,101 +66,112 @@ DistanceCommand::DistanceCommand(){
 
 int DistanceCommand::execute(){
        try {
+               
+               //read file
+               string filename = globaldata->inputFileName;
+               
+               if(globaldata->getFastaFile() != "") {
+                       readSeqs =  new ReadFasta(filename); }
+               else if(globaldata->getNexusFile() != "") {
+                       readSeqs = new ReadNexus(filename); }
+               else if(globaldata->getClustalFile() != "") {
+                       readSeqs = new ReadClustal(filename); }
+               else if(globaldata->getPhylipFile() != "") {
+                       readSeqs = new ReadPhylip(filename); }
+                       
+               readSeqs->read();
+               seqDB = readSeqs->getDB();
+       
                int numSeqs = seqDB->getNumSeqs();
                
-               system(("rm "+distFile).c_str() );
-               if(processors == 1){
-                       driver(distCalculator, seqDB, 0, numSeqs, distFile, cutoff);    
-               }       
-               else if(processors == 2){
+               string distFile = getRootName(globaldata->getFastaFile()) + "dist";
                
-                       int pid = fork();
-                       if(pid > 0){
-                               driver(distCalculator, seqDB, 0, (numSeqs/sqrt(2)), distFile + "tempa", cutoff);        
-//                             system(("cat " + distFile + "tempa" + " >> " + distFile).c_str());
-//                             system(("rm " + distFile + "tempa").c_str());                           
-                       }
-                       else{
-                               driver(distCalculator, seqDB, (numSeqs/sqrt(2)), numSeqs, distFile + "tempb", cutoff);  
-//                             system(("cat " + distFile + "tempb" + " >> " + distFile).c_str());
-//                             system(("rm " + distFile + "tempb").c_str());                           
-                       }
-                       wait(NULL);
-
-               }
-               else if(processors == 3){
-                       int pid1 = fork();
-                       if(pid1 > 0){
-                               int pid2 = fork();
-                               if(pid2 > 0){
-                                       driver(distCalculator, seqDB, 0, sqrt(3) * numSeqs / 3, distFile + "tempa", cutoff);
-                                       #ifdef HAVE_CAT
-                                               system(("cat " + distFile + "tempa" + " >> " + distFile).c_str());
-                                       #else
-                                               #ifdef HAVE_COPY
-//get system call from pat system(("copy " + distFile + "tempa").c_str());
-                                               #else
-                                                       cout << "Sorry but I can't continue because this operating system doesn't appear to support the cat() or copy() system calls." << endl;
-                                               #endif
-                                       #endif
-                                       
-                                       #ifdef HAVE_RM
-                                               system(("rm " + distFile + "tempa").c_str());   
-                                       #else
-                                               #ifdef HAVE_ERASE
-                                                       system(("erase " + distFile + "tempa").c_str());
-                                               #else
-                                                       cout << "Sorry but I can't remove the required files because this operating system doesn't appear to support the rm() or erase() system calls." << endl;
-                                               #endif
-                                       #endif
+               remove(distFile.c_str());
+               
+               //#     if defined (_WIN32)
+                       //figure out how to implement the fork and wait commands in windows
+               //      driver(distCalculator, seqDB, 0, numSeqs, distFile, cutoff);
+               //#     endif
+               
+               #       if defined (__APPLE__) || (__MACH__) || (linux) || (__linux)
+                       if(processors == 1){
+                               driver(distCalculator, seqDB, 0, numSeqs, distFile, cutoff);    
+                       }       
+                       else if(processors == 2){
+               
+                               int pid = fork();
+                               if(pid > 0){
+                                       driver(distCalculator, seqDB, 0, (numSeqs/sqrt(2)), distFile + "tempa", cutoff);        
+                                       appendFiles((distFile+"tempa"), distFile);
+                                       remove((distFile + "tempa").c_str());   
                                }
                                else{
-                                       driver(distCalculator, seqDB, sqrt(3) * numSeqs / 3, sqrt(6) * numSeqs / 3, distFile + "tempb", cutoff);        
-                                       system(("cat " + distFile + "tempb" + " >> " + distFile).c_str());
-                                       system(("rm " + distFile + "tempb").c_str());                           
+                                       driver(distCalculator, seqDB, (numSeqs/sqrt(2)), numSeqs, distFile + "tempb", cutoff);  
+                                       appendFiles((distFile+"tempb"), distFile);
+                                       remove((distFile + "tempb").c_str());   
                                }
                                wait(NULL);
+
                        }
-                       else{
-                               driver(distCalculator, seqDB, sqrt(6) * numSeqs / 3, numSeqs, distFile + "tempc", cutoff);      
-                               system(("cat " + distFile + "tempc" + " >> " + distFile).c_str());
-                               system(("rm " + distFile + "tempc").c_str());                           
-                       }
-                       wait(NULL);
-               }
-               else if(processors == 4){
-                       int pid1 = fork();
-                       if(pid1 > 0){
-                               int pid2 = fork();
-                               if(pid2 > 0){
-                                       driver(distCalculator, seqDB, 0, numSeqs / 2, distFile + "tempa", cutoff);      
-                                       system(("cat " + distFile + "tempa" + " >> " + distFile).c_str());
-                                       system(("rm " + distFile + "tempa").c_str());                           
+                       else if(processors == 3){
+                               int pid1 = fork();
+                               if(pid1 > 0){
+                                       int pid2 = fork();
+                                       if(pid2 > 0){
+                                               driver(distCalculator, seqDB, 0, sqrt(3) * numSeqs / 3, distFile + "tempa", cutoff);
+                                               appendFiles(distFile+"tempa", distFile);
+                                               remove((distFile + "tempa").c_str());   
+                                       }
+                                       else{
+                                               driver(distCalculator, seqDB, sqrt(3) * numSeqs / 3, sqrt(6) * numSeqs / 3, distFile + "tempb", cutoff);        
+                                               appendFiles(distFile+"tempb", distFile);
+                                               remove((distFile + "tempb").c_str());                           
+                                       }
+                                       wait(NULL);
                                }
                                else{
-                                       driver(distCalculator, seqDB, numSeqs / 2, (numSeqs/sqrt(2)), distFile + "tempb", cutoff);      
-                                       system(("cat " + distFile + "tempb" + " >> " + distFile).c_str());
-                                       system(("rm " + distFile + "tempb").c_str());                           
+                                       driver(distCalculator, seqDB, sqrt(6) * numSeqs / 3, numSeqs, distFile + "tempc", cutoff);      
+                                       appendFiles(distFile+"tempc", distFile);
+                                       remove((distFile + "tempc").c_str());                   
                                }
                                wait(NULL);
                        }
-                       else{
-                               int pid3 = fork();
-                               if(pid3 > 0){
-                                       driver(distCalculator, seqDB, (numSeqs/sqrt(2)), (sqrt(3) * numSeqs / 2), distFile + "tempc", cutoff);  
-                                       system(("cat " + distFile + "tempc" + " >> " + distFile).c_str());
-                                       system(("rm " + distFile + "tempc").c_str());                           
+                       else if(processors == 4){
+                               int pid1 = fork();
+                               if(pid1 > 0){
+                                       int pid2 = fork();
+                                       if(pid2 > 0){
+                                               driver(distCalculator, seqDB, 0, numSeqs / 2, distFile + "tempa", cutoff);      
+                                               appendFiles(distFile+"tempa", distFile);
+                                               remove((distFile + "tempa").c_str());                   
+                                       }
+                                       else{
+                                               driver(distCalculator, seqDB, numSeqs / 2, (numSeqs/sqrt(2)), distFile + "tempb", cutoff);      
+                                               appendFiles(distFile+"tempb", distFile);
+                                               remove((distFile + "tempb").c_str());                           
+                                       }
+                                       wait(NULL);
                                }
                                else{
-                                       driver(distCalculator, seqDB, (sqrt(3) * numSeqs / 2), numSeqs, distFile + "tempd", cutoff);    
-                                       system(("cat " + distFile + "tempd" + " >> " + distFile).c_str());
-                                       system(("rm " + distFile + "tempd").c_str());                           
+                                       int pid3 = fork();
+                                       if(pid3 > 0){
+                                               driver(distCalculator, seqDB, (numSeqs/sqrt(2)), (sqrt(3) * numSeqs / 2), distFile + "tempc", cutoff);  
+                                               appendFiles(distFile+"tempc", distFile);
+                                               remove((distFile + "tempc").c_str());                           
+                                       }
+                                       else{
+                                               driver(distCalculator, seqDB, (sqrt(3) * numSeqs / 2), numSeqs, distFile + "tempd", cutoff);    
+                                               appendFiles(distFile+"tempd", distFile);
+                                               remove((distFile + "tempd").c_str());                           
+                                       }
+                                       wait(NULL);
                                }
                                wait(NULL);
                        }
                        wait(NULL);
-               }
-               wait(NULL);
+               #       else
+                       driver(distCalculator, seqDB, 0, numSeqs, distFile, cutoff);
+               #       endif
        
                delete distCalculator;
        
@@ -195,11 +201,15 @@ int DistanceCommand::driver(Dist* distCalculator, SequenceDB* align, int startLi
                for(int i=startLine;i<endLine;i++){
                
                        for(int j=0;j<i;j++){
-                       
+//cout << "unaligned" << endl;
+//cout << align->get(i).getUnaligned() << "  " << align->get(j).getUnaligned() << endl;
+//cout << "aligned" << endl;
+//cout << align->get(i).getAligned() << "  " << align->get(j).getAligned() << endl;
                                distCalculator->calcDist(align->get(i), align->get(j));
                                double dist = distCalculator->getDist();
+
                                if(dist <= cutoff){
-                               distFile << align->get(i).getName() << ' ' << align->get(j).getName() << ' ' << dist << endl;
+                                       distFile << align->get(i).getName() << ' ' << align->get(j).getName() << ' ' << dist << endl;
                                }
                        
                        }
@@ -224,4 +234,36 @@ int DistanceCommand::driver(Dist* distCalculator, SequenceDB* align, int startLi
 }
 
 /**************************************************************************************************/
-
+void DistanceCommand::appendFiles(string temp, string filename) {
+       try{
+               ofstream output;
+               ifstream input;
+               
+               //open output file in append mode
+               openOutputFileAppend(filename, output);
+               
+               //open temp file for reading
+               openInputFile(temp, input);
+               
+               string line;
+               //read input file and write to output file
+               while(input.eof() != true) {
+                       getline(input, line); //getline removes the newline char
+                       if (line != "") {
+                               output << line << endl;   // Appending back newline char 
+                       }
+               }       
+                       
+               input.close();
+               output.close();
+       }
+       catch(exception& e) {
+               cout << "Standard Error: " << e.what() << " has occurred in the DistanceCommand class Function appendFiles. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
+               exit(1);
+       }
+       catch(...) {
+               cout << "An unknown error has occurred in the DistanceCommand class function appendFiles. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
+               exit(1);
+       }       
+}
+/**************************************************************************************************/
\ No newline at end of file