added checks for ^C to quit command instead of program

[mothur.git] / bayesian.cpp

diff --git a/bayesian.cpp b/bayesian.cpp
index 42fb0232ad758fed402d6a2db2fe1ca10752b85a..af1fa27f2202de76086e562c775229f4f241e387 100644 (file)
--- a/bayesian.cpp
+++ b/bayesian.cpp
@@ -56,6 +56,7 @@ Classify(tfile, tempFile, method, ksize, 0.0, 0.0, 0.0, 0.0), kmerSize(ksize), c
                        
                        //for each word
                        for (int i = 0; i < numKmers; i++) {
+                               if (m->control_pressed) { break; }
                                
                                out << i << '\t';
                                
@@ -114,6 +115,8 @@ string Bayesian::getTaxonomy(Sequence* seq) {
                }
        
                int index = getMostProbableTaxonomy(queryKmers);
+               
+               if (m->control_pressed) { return tax; }
                                        
                //bootstrap - to set confidenceScore
                int numToSelect = queryKmers.size() / 8;
@@ -143,6 +146,8 @@ string Bayesian::bootstrapResults(vector<int> kmers, int tax, int numToSelect) {
                map<int, int>::iterator itConvert;
                
                for (int i = 0; i < iters; i++) {
+                       if (m->control_pressed) { return "control"; }
+                       
                        vector<int> temp;
                                                
                        for (int j = 0; j < numToSelect; j++) {