|
18 | 18 | #ifndef Rcpp_api_meat_Rcpp_eval_h |
19 | 19 | #define Rcpp_api_meat_Rcpp_eval_h |
20 | 20 |
|
| 21 | +#include <Rcpp/Interrupt.h> |
| 22 | + |
21 | 23 | namespace Rcpp{ |
22 | 24 |
|
23 | | - inline SEXP Rcpp_eval(SEXP expr_, SEXP env) { |
24 | | - Shield<SEXP> expr( expr_) ; |
| 25 | + struct EvalCall { |
| 26 | + SEXP expr; |
| 27 | + SEXP env; |
| 28 | + SEXP result; |
| 29 | + std::vector<std::string> warnings; |
| 30 | + std::string error_message; |
| 31 | + }; |
| 32 | + |
| 33 | + inline void Rcpp_eval(void* data) { |
| 34 | + |
| 35 | + EvalCall* evalCall = (EvalCall*)data; |
| 36 | + SEXP env = evalCall->env; |
| 37 | + |
| 38 | + Shield<SEXP> expr(evalCall->expr) ; |
25 | 39 |
|
26 | 40 | reset_current_error() ; |
27 | 41 |
|
28 | 42 | Environment RCPP = Environment::Rcpp_namespace(); |
| 43 | + SEXP withCallingHandlersSym = ::Rf_install("withCallingHandlers"); |
29 | 44 | SEXP tryCatchSym = ::Rf_install("tryCatch"); |
30 | 45 | SEXP evalqSym = ::Rf_install("evalq"); |
31 | 46 | SEXP conditionMessageSym = ::Rf_install("conditionMessage"); |
32 | 47 | SEXP errorRecorderSym = ::Rf_install(".rcpp_error_recorder"); |
| 48 | + SEXP warningRecorderSym = ::Rf_install(".rcpp_warning_recorder"); |
| 49 | + SEXP collectWarningsSym = ::Rf_install(".rcpp_collect_warnings"); |
33 | 50 | SEXP errorSym = ::Rf_install("error"); |
| 51 | + SEXP warningSym = ::Rf_install("warning"); |
34 | 52 |
|
35 | | - Shield<SEXP> call( Rf_lang3( |
| 53 | + // define the tryCatchCall |
| 54 | + Shield<SEXP> tryCatchCall( Rf_lang3( |
36 | 55 | tryCatchSym, |
37 | 56 | Rf_lang3( evalqSym, expr, env ), |
38 | 57 | errorRecorderSym |
39 | 58 | ) ) ; |
40 | | - SET_TAG( CDDR(call), errorSym ) ; |
41 | | - /* call the tryCatch call */ |
| 59 | + SET_TAG( CDDR(tryCatchCall), errorSym ) ; |
| 60 | + |
| 61 | + // encose it in withCallingHandlers |
| 62 | + Shield<SEXP> call( Rf_lang3( |
| 63 | + withCallingHandlersSym, |
| 64 | + tryCatchCall, |
| 65 | + warningRecorderSym |
| 66 | + ) ) ; |
| 67 | + SET_TAG( CDDR(call), warningSym ) ; |
| 68 | + |
| 69 | + // execute the call |
42 | 70 | Shield<SEXP> res(::Rf_eval( call, RCPP ) ); |
43 | 71 |
|
| 72 | + // collect warnings |
| 73 | + Shield<SEXP> warningCall(Rf_lang1(collectWarningsSym)); |
| 74 | + Shield<SEXP> warnings(::Rf_eval(warningCall, RCPP)); |
| 75 | + |
| 76 | + evalCall->warnings = Rcpp::as<std::vector<std::string> >(warnings); |
| 77 | + |
| 78 | + // check for error |
44 | 79 | if( error_occured() ) { |
45 | 80 | Shield<SEXP> current_error ( rcpp_get_current_error() ) ; |
46 | 81 | Shield<SEXP> conditionMessageCall (::Rf_lang2(conditionMessageSym, current_error)) ; |
47 | 82 | Shield<SEXP> condition_message (::Rf_eval(conditionMessageCall, R_GlobalEnv)) ; |
48 | | - std::string message(CHAR(::Rf_asChar(condition_message))); |
49 | | - throw eval_error(message) ; |
| 83 | + evalCall->error_message = std::string(CHAR(::Rf_asChar(condition_message))); |
| 84 | + } else { |
| 85 | + evalCall->result = res; |
50 | 86 | } |
| 87 | + } |
| 88 | + |
| 89 | + inline SEXP Rcpp_eval(SEXP expr_, SEXP env) { |
51 | 90 |
|
52 | | - return res ; |
| 91 | + // create the call object |
| 92 | + EvalCall call; |
| 93 | + call.expr = expr_; |
| 94 | + call.env = env; |
| 95 | + |
| 96 | + // execute it |
| 97 | + Rboolean completed = R_ToplevelExec(Rcpp_eval, (void*)&call); |
| 98 | + |
| 99 | + // print warnings |
| 100 | + for (size_t i = 0; i<call.warnings.size(); i++) |
| 101 | + Rf_warning(call.warnings[i].c_str()); |
| 102 | + |
| 103 | + // handle error or result if it completed, else throw interrupt |
| 104 | + if (completed) { |
| 105 | + if (!call.error_message.empty()) |
| 106 | + throw eval_error(call.error_message); |
| 107 | + else |
| 108 | + return call.result; |
| 109 | + } else { |
| 110 | + throw internal::InterruptedException(); |
| 111 | + } |
53 | 112 | } |
54 | 113 |
|
55 | 114 | } |
|
0 commit comments