]> git.donarmstrong.com Git - mothur.git/blobdiff - inputdata.cpp
changes while testing
[mothur.git] / inputdata.cpp
index cbf44ed11a96a03c251fcc3025b185c2020c2830..8b1829a3db697df21b3d0f64bf6be6cfea15c536 100644 (file)
 /***********************************************************************/
 
 InputData::InputData(string fName, string f) : format(f){
-       
-       openInputFile(fName, fileHandle);
+       m = MothurOut::getInstance();
+       m->openInputFile(fName, fileHandle);
        filename = fName;
-       
+       m->saveNextLabel = "";
 }
-
 /***********************************************************************/
 
-
 InputData::~InputData(){
        fileHandle.close();
-//     delete output;
-       
+       m->saveNextLabel = "";
 }
 
 /***********************************************************************/
 
 InputData::InputData(string fName, string orderFileName, string f) : format(f){
        try {
-               
+               m = MothurOut::getInstance();
                ifstream ofHandle;
-               openInputFile(orderFileName, ofHandle);
+               m->openInputFile(orderFileName, ofHandle);
                string name;
 
                int count = 0;
@@ -45,14 +42,16 @@ InputData::InputData(string fName, string orderFileName, string f) : format(f){
                        ofHandle >> name;
                        orderMap[name] = count;
                        count++;
-                       gobble(ofHandle);
+                       m->gobble(ofHandle);
                }
                ofHandle.close();
        
-               openInputFile(fName, fileHandle);
+               m->openInputFile(fName, fileHandle);
+               m->saveNextLabel = "";
+               
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "InputData");
+               m->errorOut(e, "InputData", "InputData");
                exit(1);
        }
 }
@@ -60,12 +59,12 @@ InputData::InputData(string fName, string orderFileName, string f) : format(f){
 
 ListVector* InputData::getListVector(){
        try {
-               if(fileHandle){
+               if(!fileHandle.eof()){
                        if(format == "list") {
                                list = new ListVector(fileHandle);
-                       }
+                       }else{ list = NULL;  }
                                        
-                       gobble(fileHandle);
+                       m->gobble(fileHandle);
                        return list;
                }
                else{
@@ -73,18 +72,17 @@ ListVector* InputData::getListVector(){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getListVector");
+               m->errorOut(e, "InputData", "getListVector");
                exit(1);
        }
 }
 
 /***********************************************************************/
-
 ListVector* InputData::getListVector(string label){
        try {
                ifstream in;
                string  thisLabel;
-               openInputFile(filename, in);
+               m->openInputFile(filename, in);
                
                if(in){
 
@@ -99,9 +97,9 @@ ListVector* InputData::getListVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete list;    }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
-                       }
+                       }else{ list = NULL;  }
                        
                        in.close();
                        return list;
@@ -111,11 +109,44 @@ ListVector* InputData::getListVector(string label){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getListVector");
+               m->errorOut(e, "InputData", "getListVector");
                exit(1);
        }
 }
+/***********************************************************************/
+ListVector* InputData::getListVector(string label, bool resetFP){
+       try {
+               string  thisLabel;
+               fileHandle.clear();
+               fileHandle.seekg(0);
+               
+               if(fileHandle){
 
+                       if (format == "list") {
+                       
+                               while (fileHandle.eof() != true) {
+                                       
+                                       list = new ListVector(fileHandle); m->gobble(fileHandle);
+                                       thisLabel = list->getLabel();
+                                       
+                                       //if you are at the last label
+                                       if (thisLabel == label) {  break;  }
+                                       //so you don't loose this memory
+                                       else {  delete list;    }
+                               }
+                       }else{ list = NULL;  }
+               
+                       return list;
+               }
+               else{
+                       return NULL;
+               }
+       }
+       catch(exception& e) {
+               m->errorOut(e, "InputData", "getListVector");
+               exit(1);
+       }
+}
 
 /***********************************************************************/
 
@@ -124,9 +155,9 @@ SharedListVector* InputData::getSharedListVector(){
                if(fileHandle){
                        if (format == "shared")  {
                                SharedList = new SharedListVector(fileHandle);
-                       }
+                       }else{ SharedList = NULL;  }
                                        
-                       gobble(fileHandle);
+                       m->gobble(fileHandle);
                        return SharedList;
                }
                else{
@@ -134,7 +165,7 @@ SharedListVector* InputData::getSharedListVector(){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getSharedListVector");
+               m->errorOut(e, "InputData", "getSharedListVector");
                exit(1);
        }
 }
@@ -144,7 +175,7 @@ SharedListVector* InputData::getSharedListVector(string label){
        try {
                ifstream in;
                string  thisLabel;
-               openInputFile(filename, in);
+               m->openInputFile(filename, in);
                
                if(in){
 
@@ -159,10 +190,10 @@ SharedListVector* InputData::getSharedListVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete SharedList;      }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
-                       }
+                       }else{ SharedList = NULL;  }
                                
                        in.close();
                        return SharedList;
@@ -172,7 +203,7 @@ SharedListVector* InputData::getSharedListVector(string label){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getSharedListVector");
+               m->errorOut(e, "InputData", "getSharedListVector");
                exit(1);
        }
 }
@@ -186,9 +217,9 @@ SharedOrderVector* InputData::getSharedOrderVector(){
                if(fileHandle){
                        if (format == "sharedfile")  {
                                SharedOrder = new SharedOrderVector(fileHandle);
-                       }
+                       }else{ SharedOrder = NULL;  }
                                
-                       gobble(fileHandle);
+                       m->gobble(fileHandle);
                        return SharedOrder;
                        
                }else{
@@ -196,7 +227,7 @@ SharedOrderVector* InputData::getSharedOrderVector(){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getSharedOrderVector");
+               m->errorOut(e, "InputData", "getSharedOrderVector");
                exit(1);
        }
 }
@@ -207,7 +238,7 @@ SharedOrderVector* InputData::getSharedOrderVector(string label){
        try {
                ifstream in;
                string  thisLabel;
-               openInputFile(filename, in);
+               m->openInputFile(filename, in);
                
                if(in){
 
@@ -222,10 +253,10 @@ SharedOrderVector* InputData::getSharedOrderVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete SharedOrder;     }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
-                       }
+                       }else{ SharedOrder = NULL;  }
                                
                        in.close();
                        return SharedOrder;
@@ -235,7 +266,7 @@ SharedOrderVector* InputData::getSharedOrderVector(string label){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getSharedOrderVector");
+               m->errorOut(e, "InputData", "getSharedOrderVector");
                exit(1);
        }
 }
@@ -262,11 +293,12 @@ OrderVector* InputData::getOrderVector(){
                        else if(format == "sabund"){
                                input = new SAbundVector(fileHandle);
                        }
-                                       
-                       gobble(fileHandle);
-                       output = new OrderVector();
-                       *output = (input->getOrderVector());
+                                               
+                       m->gobble(fileHandle);
                        
+                       output = new OrderVector();     
+                       *output = (input->getOrderVector());
+               
                        return output;
                }
                else{
@@ -274,7 +306,7 @@ OrderVector* InputData::getOrderVector(){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getOrderVector");
+               m->errorOut(e, "InputData", "getOrderVector");
                exit(1);
        }
 }
@@ -285,7 +317,7 @@ OrderVector* InputData::getOrderVector(string label){
        
                ifstream in;
                string  thisLabel;
-               openInputFile(filename, in);
+               m->openInputFile(filename, in);
                
                if(in){
                        if((format == "list") || (format == "listorder")) {
@@ -299,7 +331,7 @@ OrderVector* InputData::getOrderVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
                        }
                        else if (format == "shared")  {
@@ -313,7 +345,7 @@ OrderVector* InputData::getOrderVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
                        }
@@ -328,7 +360,7 @@ OrderVector* InputData::getOrderVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
                        }
@@ -343,7 +375,7 @@ OrderVector* InputData::getOrderVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
                        }
@@ -358,12 +390,12 @@ OrderVector* InputData::getOrderVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                        
                                }
 
                        }
-                       
+                                               
                        in.close();             
 
                        output = new OrderVector();
@@ -377,7 +409,7 @@ OrderVector* InputData::getOrderVector(string label){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getOrderVector");
+               m->errorOut(e, "InputData", "getOrderVector");
                exit(1);
        }
 }
@@ -394,11 +426,12 @@ vector<SharedRAbundVector*> InputData::getSharedRAbundVectors(){
                                }
                        }else if (format == "shared") {
                                SharedList = new SharedListVector(fileHandle);
+                               
                                if (SharedList != NULL) {
                                        return SharedList->getSharedRAbundVector();
                                }
                        }
-                       gobble(fileHandle);
+                       m->gobble(fileHandle);
                }
                                
                //this is created to signal to calling function that the input file is at eof
@@ -407,7 +440,7 @@ vector<SharedRAbundVector*> InputData::getSharedRAbundVectors(){
                
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getSharedRAbundVectors");
+               m->errorOut(e, "InputData", "getSharedRAbundVectors");
                exit(1);
        }
 }
@@ -417,8 +450,9 @@ vector<SharedRAbundVector*> InputData::getSharedRAbundVectors(string label){
                ifstream in;
                string  thisLabel;
                
-               openInputFile(filename, in);
-               
+               m->openInputFile(filename, in);
+               m->saveNextLabel = "";
+       
                if(in){
                        if (format == "sharedfile")  {
                                while (in.eof() != true) {
@@ -426,6 +460,7 @@ vector<SharedRAbundVector*> InputData::getSharedRAbundVectors(string label){
                                        SharedRAbundVector* SharedRAbund = new SharedRAbundVector(in);
                                        if (SharedRAbund != NULL) {
                                                thisLabel = SharedRAbund->getLabel();
+                                       
                                                //if you are at the last label
                                                if (thisLabel == label) {  in.close(); return SharedRAbund->getSharedRAbundVectors();  }
                                                else {
@@ -435,13 +470,14 @@ vector<SharedRAbundVector*> InputData::getSharedRAbundVectors(string label){
                                                        delete SharedRAbund;
                                                }
                                        }else{  break;  }
-                                       gobble(in);
+                                       m->gobble(in);
                                        
                                }
                        }else if (format == "shared") {
                                while (in.eof() != true) {
                                        
                                        SharedList = new SharedListVector(in);
+                                       
                                        if (SharedList != NULL) {
                                                thisLabel = SharedList->getLabel();
                                                //if you are at the last label
@@ -451,7 +487,7 @@ vector<SharedRAbundVector*> InputData::getSharedRAbundVectors(string label){
                                                        delete SharedList;
                                                }
                                        }else{  break;  }
-                                       gobble(in);
+                                       m->gobble(in);
                                        
                                }
                        
@@ -465,12 +501,109 @@ vector<SharedRAbundVector*> InputData::getSharedRAbundVectors(string label){
        
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getSharedRAbundVectors");
+               m->errorOut(e, "InputData", "getSharedRAbundVectors");
                exit(1);
        }
 }
 
-
+/***********************************************************************/
+//this is used when you don't need the order vector
+vector<SharedRAbundFloatVector*> InputData::getSharedRAbundFloatVectors(){
+       try {
+               if(fileHandle){
+                       if (format == "relabund")  {
+                               SharedRAbundFloatVector* SharedRelAbund = new SharedRAbundFloatVector(fileHandle);
+                               if (SharedRelAbund != NULL) {
+                                       return SharedRelAbund->getSharedRAbundFloatVectors();
+                               }
+                       }else if (format == "sharedfile")  {
+                               SharedRAbundVector* SharedRAbund = new SharedRAbundVector(fileHandle);
+                               if (SharedRAbund != NULL) {
+                                       vector<SharedRAbundVector*> lookup = SharedRAbund->getSharedRAbundVectors(); 
+                                       vector<SharedRAbundFloatVector*> lookupFloat = SharedRAbund->getSharedRAbundFloatVectors(lookup); 
+                                       for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } lookup.clear();
+                                       return lookupFloat;  
+                               }
+                                               
+                       }
+                       m->gobble(fileHandle);
+               }
+                               
+               //this is created to signal to calling function that the input file is at eof
+               vector<SharedRAbundFloatVector*> null;  null.push_back(NULL);
+               return null;
+               
+       }
+       catch(exception& e) {
+               m->errorOut(e, "InputData", "getSharedRAbundFloatVectors");
+               exit(1);
+       }
+}
+/***********************************************************************/
+vector<SharedRAbundFloatVector*> InputData::getSharedRAbundFloatVectors(string label){
+       try {
+               ifstream in;
+               string  thisLabel;
+               
+               m->openInputFile(filename, in);
+               m->saveNextLabel = "";
+               
+               if(in){
+                       if (format == "relabund")  {
+                               while (in.eof() != true) {
+                                       
+                                       SharedRAbundFloatVector* SharedRelAbund = new SharedRAbundFloatVector(in);
+                                       if (SharedRelAbund != NULL) {
+                                               thisLabel = SharedRelAbund->getLabel();
+                                               //if you are at the last label
+                                               if (thisLabel == label) {  in.close(); return SharedRelAbund->getSharedRAbundFloatVectors();  }
+                                               else {
+                                                       //so you don't loose this memory
+                                                       vector<SharedRAbundFloatVector*> lookup = SharedRelAbund->getSharedRAbundFloatVectors(); 
+                                                       for (int i = 0; i < lookup.size(); i++) {  delete lookup[i];  }
+                                                       delete SharedRelAbund;
+                                               }
+                                       }else{  break;  }
+                                       m->gobble(in);
+                               }
+                       }else if (format == "sharedfile")  {
+                               while (in.eof() != true) {
+                                       
+                                       SharedRAbundVector* SharedRAbund = new SharedRAbundVector(in);
+                                       if (SharedRAbund != NULL) {
+                                               thisLabel = SharedRAbund->getLabel();
+                                               
+                                               //if you are at the last label
+                                               if (thisLabel == label) {  
+                                                       in.close(); 
+                                                       vector<SharedRAbundVector*> lookup = SharedRAbund->getSharedRAbundVectors(); 
+                                                       vector<SharedRAbundFloatVector*> lookupFloat = SharedRAbund->getSharedRAbundFloatVectors(lookup); 
+                                                       for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } lookup.clear();
+                                                       return lookupFloat;  
+                                               }else {
+                                                       //so you don't loose this memory
+                                                       vector<SharedRAbundVector*> lookup = SharedRAbund->getSharedRAbundVectors(); 
+                                                       for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } lookup.clear();
+                                                       delete SharedRAbund;
+                                               }
+                                       }else{  break;  }
+                                       m->gobble(in);
+                               }
+                       }       
+               }
+               
+                               
+               //this is created to signal to calling function that the input file is at eof
+               vector<SharedRAbundFloatVector*> null;  null.push_back(NULL);
+               in.close();
+               return null;
+       
+       }
+       catch(exception& e) {
+               m->errorOut(e, "InputData", "getSharedRAbundFloatVectors");
+               exit(1);
+       }
+}
 /***********************************************************************/
 
 SAbundVector* InputData::getSAbundVector(){
@@ -490,9 +623,8 @@ SAbundVector* InputData::getSAbundVector(){
                        }
                        else if(format == "sabund"){
                                input = new SAbundVector(fileHandle);
-                       }
-                                       
-                       gobble(fileHandle);
+                       }                                       
+                       m->gobble(fileHandle);
 
                        sabund = new SAbundVector();
                        *sabund = (input->getSAbundVector());
@@ -504,7 +636,7 @@ SAbundVector* InputData::getSAbundVector(){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getSAbundVector");
+               m->errorOut(e, "InputData", "getSAbundVector");
                exit(1);
        }
 }
@@ -514,7 +646,7 @@ SAbundVector* InputData::getSAbundVector(string label){
        
                ifstream in;
                string  thisLabel;
-               openInputFile(filename, in);
+               m->openInputFile(filename, in);
                
                if(in){
                        if (format == "list") {
@@ -528,7 +660,7 @@ SAbundVector* InputData::getSAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
                        }
                        else if (format == "shared")  {
@@ -542,7 +674,7 @@ SAbundVector* InputData::getSAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
                        }
@@ -557,7 +689,7 @@ SAbundVector* InputData::getSAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
                        }
@@ -572,7 +704,7 @@ SAbundVector* InputData::getSAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
                        }
@@ -587,7 +719,7 @@ SAbundVector* InputData::getSAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                        
                                }
 
@@ -606,7 +738,7 @@ SAbundVector* InputData::getSAbundVector(string label){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getSAbundVector");
+               m->errorOut(e, "InputData", "getSAbundVector");
                exit(1);
        }
 }
@@ -630,8 +762,8 @@ RAbundVector* InputData::getRAbundVector(){
                        else if(format == "sabund"){
                                input = new SAbundVector(fileHandle);
                        }
-                                       
-                       gobble(fileHandle);
+                       
+                       m->gobble(fileHandle);
 
                        rabund = new RAbundVector();
                        *rabund = (input->getRAbundVector());
@@ -643,7 +775,7 @@ RAbundVector* InputData::getRAbundVector(){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getRAbundVector");
+               m->errorOut(e, "InputData", "getRAbundVector");
                exit(1);
        }
 }
@@ -653,7 +785,7 @@ RAbundVector* InputData::getRAbundVector(string label){
        
                ifstream in;
                string  thisLabel;
-               openInputFile(filename, in);
+               m->openInputFile(filename, in);
                
                if(in){
                        if (format == "list") {
@@ -667,7 +799,7 @@ RAbundVector* InputData::getRAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
                        }
                        else if (format == "shared")  {
@@ -681,7 +813,7 @@ RAbundVector* InputData::getRAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
                        }
@@ -696,7 +828,7 @@ RAbundVector* InputData::getRAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
                        }
@@ -711,7 +843,7 @@ RAbundVector* InputData::getRAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                }
 
                        }
@@ -726,11 +858,12 @@ RAbundVector* InputData::getRAbundVector(string label){
                                        if (thisLabel == label) {  break;  }
                                        //so you don't loose this memory
                                        else {  delete input;   }
-                                       gobble(in);
+                                       m->gobble(in);
                                        
                                }
 
-                       }
+                       }                       
+                       
                        
                        in.close();             
 
@@ -744,7 +877,7 @@ RAbundVector* InputData::getRAbundVector(string label){
                }
        }
        catch(exception& e) {
-               errorOut(e, "InputData", "getRAbundVector");
+               m->errorOut(e, "InputData", "getRAbundVector");
                exit(1);
        }
 }