added checks for ^C to quit command instead of program

[mothur.git] / rarefact.cpp

diff --git a/rarefact.cpp b/rarefact.cpp
index 5cd9b99c5051be03b00757df6a99bfb479adabfe..f5e7952688cdbb1dd182ca44df9c152ef46879d9 100644 (file)
--- a/rarefact.cpp
+++ b/rarefact.cpp
@@ -12,7 +12,7 @@
 
 /***********************************************************************/
 
-void Rarefact::getCurve(int increment = 1, int nIters = 1000){
+int Rarefact::getCurve(int increment = 1, int nIters = 1000){
        try {
                RarefactionCurveData* rcd = new RarefactionCurveData();
                for(int i=0;i<displays.size();i++){
@@ -31,6 +31,8 @@ void Rarefact::getCurve(int increment = 1, int nIters = 1000){
                
                        for(int i=0;i<numSeqs;i++){
                        
+                               if (m->control_pressed) { delete lookup; delete rank; delete rcd; return 0;  }
+                       
                                int binNumber = order->get(i);
                                int abundance = lookup->get(binNumber);
                        
@@ -61,16 +63,17 @@ void Rarefact::getCurve(int increment = 1, int nIters = 1000){
                        displays[i]->close();
                }
                delete rcd;
+               return 0;
        }
        catch(exception& e) {
-               errorOut(e, "Rarefact", "getCurve");
+               m->errorOut(e, "Rarefact", "getCurve");
                exit(1);
        }
 }
 
 /***********************************************************************/
 
-void Rarefact::getSharedCurve(int increment = 1, int nIters = 1000){
+int Rarefact::getSharedCurve(int increment = 1, int nIters = 1000){
 try {
                SharedRarefactionCurveData* rcd = new SharedRarefactionCurveData();
                
@@ -106,6 +109,8 @@ try {
                        vector<SharedRAbundVector*> subset;
                        //send each group one at a time
                        for (int k = 0; k < lookup.size(); k++) { 
+                               if (m->control_pressed) {  delete merge; delete rcd; return 0;  }
+                               
                                subset.clear(); //clears out old pair of sharedrabunds
                                //add in new pair of sharedrabunds
                                subset.push_back(merge); subset.push_back(lookup[k]);
@@ -127,9 +132,10 @@ try {
                }
                
                delete rcd;
+               return 0;
        }
        catch(exception& e) {
-               errorOut(e, "Rarefact", "getSharedCurve");
+               m->errorOut(e, "Rarefact", "getSharedCurve");
                exit(1);
        }
 }
@@ -143,7 +149,7 @@ void Rarefact::mergeVectors(SharedRAbundVector* shared1, SharedRAbundVector* sha
                }
        }
        catch(exception& e) {
-               errorOut(e, "Rarefact", "mergeVectors");
+               m->errorOut(e, "Rarefact", "mergeVectors");
                exit(1);
        }
 }