Checking for a user interrupt during registrations

jonclayden · Apr 5, 2016 · fcd30ea · fcd30ea
1 parent d27d545
commit fcd30ea
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/reg-lib/_reg_aladin.cpp b/src/reg-lib/_reg_aladin.cpp
@@ -602,6 +602,9 @@ void reg_aladin<T>::resolveMatrix(unsigned int iterations, const unsigned int op
 		this->UpdateTransformationMatrix(optimizationFlag);
 
 		iteration++;
+#ifdef RNIFTYREG
+        Rcpp::checkUserInterrupt();
+#endif
 	}
 
 #ifdef RNIFTYREG

diff --git a/src/reg-lib/_reg_base.cpp b/src/reg-lib/_reg_base.cpp
@@ -1272,6 +1272,10 @@ void reg_base<T>::Run()
 
             // Update the obecjtive function variables and print some information
             this->PrintCurrentObjFunctionValue(currentSize);
+
+#ifdef RNIFTYREG
+            Rcpp::checkUserInterrupt();
+#endif
          } // while
 
 #ifdef RNIFTYREG