Changes a C++ exit to Rcpp:stop

jkrijthe · Sep 30, 2018 · f2edaf0 · f2edaf0
1 parent 9c95888
commit f2edaf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tsne.cpp b/src/tsne.cpp
@@ -741,7 +741,7 @@ void TSNE<NDims>::zeroMean(double* X, int N, int D) {
 
 	// Compute data mean
 	double* mean = (double*) calloc(D, sizeof(double));
-  if(mean == NULL) { printf("Memory allocation failed!\n"); exit(1); }
+  if(mean == NULL) {  Rcpp::stop("Memory allocation failed!\n"); }
   int nD = 0;
   for(int n = 0; n < N; n++) {
     for(int d = 0; d < D; d++) {