/* * R : A Computer Language for Statistical Data Analysis * Copyright (C) 1995, 1996, 1997 Robert Gentleman and Ross Ihaka * Copyright (C) 1998--2013 The R Core Team. * Copyright (C) 2003-4 The R Foundation * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, a copy is available at * http://www.r-project.org/Licenses/ */ #ifdef HAVE_CONFIG_H #include #endif /* interval at which to check interrupts, a guess */ #define NINTERRUPT 10000000 #ifdef __OpenBSD__ /* for definition of "struct exception" in math.h */ # define __LIBM_PRIVATE #endif #include /*-> Arith.h -> math.h */ #ifdef __OpenBSD__ # undef __LIBM_PRIVATE #endif #include #define R_MSG_NA _("NaNs produced") #define R_MSG_NONNUM_MATH _("non-numeric argument to mathematical function") #include extern double Rf_gamma_cody(double); #include #include "arithmetic.h" #include #ifdef HAVE_MATHERR /* Override the SVID matherr function: the main difference here is not to print warnings. */ #ifndef __cplusplus int matherr(struct exception *exc) { switch (exc->type) { case DOMAIN: case SING: errno = EDOM; break; case OVERFLOW: errno = ERANGE; break; case UNDERFLOW: exc->retval = 0.0; break; /* There are cases TLOSS and PLOSS which are ignored here. According to the Solaris man page, there are for trigonometric algorithms and not needed for good ones. */ } return 1; } #endif #endif typedef union { double value; unsigned int word[2]; } ieee_double; /* gcc had problems with static const on AIX and Solaris Solaris was for gcc 3.1 and 3.2 under -O2 32-bit on 64-bit kernel */ #ifdef _AIX #define CONST #elif defined(sparc) && defined (__GNUC__) && __GNUC__ == 3 #define CONST #else #define CONST const #endif #ifdef WORDS_BIGENDIAN static CONST int hw = 0; static CONST int lw = 1; #else /* !WORDS_BIGENDIAN */ static CONST int hw = 1; static CONST int lw = 0; #endif /* WORDS_BIGENDIAN */ static double R_ValueOfNA(void) { /* The gcc shipping with Fedora 9 gets this wrong without * the volatile declaration. Thanks to Marc Schwartz. */ volatile ieee_double x; x.word[hw] = 0x7ff00000; x.word[lw] = 1954; return x.value; } int R_IsNA(double x) { if (isnan(x)) { ieee_double y; y.value = x; return (y.word[lw] == 1954); } return 0; } int R_IsNaN(double x) { if (isnan(x)) { ieee_double y; y.value = x; return (y.word[lw] != 1954); } return 0; } /* ISNAN uses isnan, which is undefined by C++ headers This workaround is called only when ISNAN() is used in a user code in a file with __cplusplus defined */ int R_isnancpp(double x) { return (isnan(x)!=0); } /* Mainly for use in packages */ int R_finite(double x) { #ifdef HAVE_WORKING_ISFINITE return isfinite(x); #else return (!isnan(x) & (x != R_PosInf) & (x != R_NegInf)); #endif } /* Arithmetic Initialization */ void attribute_hidden InitArithmetic() { R_NaInt = INT_MIN; R_NaReal = R_ValueOfNA(); // we assume C99, so #ifndef OLD R_NaN = NAN; R_PosInf = INFINITY; R_NegInf = -INFINITY; #else R_NaN = 0.0/R_Zero_Hack; R_PosInf = 1.0/R_Zero_Hack; R_NegInf = -1.0/R_Zero_Hack; #endif } /* Keep these two in step */ /* FIXME: consider using tmp = (LDOUBLE)x1 - floor(q) * (LDOUBLE)x2; */ static double myfmod(double x1, double x2) { if (x2 == 0.0) return R_NaN; double q = x1 / x2, tmp = x1 - floor(q) * x2; if(R_FINITE(q) && (fabs(q) > 1/R_AccuracyInfo.eps)) warning(_("probable complete loss of accuracy in modulus")); q = floor(tmp/x2); return tmp - q * x2; } static double myfloor(double x1, double x2) { double q = x1 / x2, tmp; if (x2 == 0.0) return q; tmp = x1 - floor(q) * x2; return floor(q) + floor(tmp/x2); } /* some systems get this wrong, possibly depend on what libs are loaded */ static R_INLINE double R_log(double x) { return x > 0 ? log(x) : x < 0 ? R_NaN : R_NegInf; } double R_pow(double x, double y) /* = x ^ y */ { /* squaring is the most common of the specially handled cases so check for it first. */ if(y == 2.0) return x * x; if(x == 1. || y == 0.) return(1.); if(x == 0.) { if(y > 0.) return(0.); else if(y < 0) return(R_PosInf); else return(y); /* NA or NaN, we assert */ } if (R_FINITE(x) && R_FINITE(y)) { /* There was a special case for y == 0.5 here, but gcc 4.3.0 -g -O2 mis-compiled it. Showed up with 100^0.5 as 3.162278, example(pbirthday) failed. */ return pow(x, y); } if (ISNAN(x) || ISNAN(y)) return(x + y); if(!R_FINITE(x)) { if(x > 0) /* Inf ^ y */ return (y < 0.)? 0. : R_PosInf; else { /* (-Inf) ^ y */ if(R_FINITE(y) && y == floor(y)) /* (-Inf) ^ n */ return (y < 0.) ? 0. : (myfmod(y, 2.) ? x : -x); } } if(!R_FINITE(y)) { if(x >= 0) { if(y > 0) /* y == +Inf */ return (x >= 1) ? R_PosInf : 0.; else /* y == -Inf */ return (x < 1) ? R_PosInf : 0.; } } return R_NaN; // all other cases: (-Inf)^{+-Inf, non-int}; (neg)^{+-Inf} } static R_INLINE double R_POW(double x, double y) /* handle x ^ 2 inline */ { return y == 2.0 ? x * x : R_pow(x, y); } double R_pow_di(double x, int n) { double xn = 1.0; if (ISNAN(x)) return x; if (n == NA_INTEGER) return NA_REAL; if (n != 0) { if (!R_FINITE(x)) return R_POW(x, (double)n); Rboolean is_neg = (n < 0); if(is_neg) n = -n; for(;;) { if(n & 01) xn *= x; if(n >>= 1) x *= x; else break; } if(is_neg) xn = 1. / xn; } return xn; } /* General Base Logarithms */ /* Note that the behaviour of log(0) required is not necessarily that mandated by C99 (-HUGE_VAL), and the behaviour of log(x < 0) is optional in C99. Some systems return -Inf for log(x < 0), e.g. libsunmath on Solaris. */ static double logbase(double x, double base) { #ifdef HAVE_LOG10 if(base == 10) return x > 0 ? log10(x) : x < 0 ? R_NaN : R_NegInf; #endif #ifdef HAVE_LOG2 if(base == 2) return x > 0 ? log2(x) : x < 0 ? R_NaN : R_NegInf; #endif return R_log(x) / R_log(base); } SEXP R_unary(SEXP, SEXP, SEXP); SEXP R_binary(SEXP, SEXP, SEXP, SEXP); static SEXP logical_unary(ARITHOP_TYPE, SEXP, SEXP); static SEXP integer_unary(ARITHOP_TYPE, SEXP, SEXP); static SEXP real_unary(ARITHOP_TYPE, SEXP, SEXP); static SEXP real_binary(ARITHOP_TYPE, SEXP, SEXP); static SEXP integer_binary(ARITHOP_TYPE, SEXP, SEXP, SEXP); #if 0 static int naflag; static SEXP lcall; #endif /* Integer arithmetic support */ /* The tests using integer comparisons are a bit faster than the tests using doubles, but they depend on a two's complement representation (but that is almost universal). The tests that compare results to double's depend on being able to accurately represent all int's as double's. Since int's are almost universally 32 bit that should be OK. */ #ifndef INT_32_BITS /* configure checks whether int is 32 bits. If not this code will need to be rewritten. Since 32 bit ints are pretty much universal, we can worry about writing alternate code when the need arises. To be safe, we signal a compiler error if int is not 32 bits. */ # error code requires that int have 32 bits #endif #define INTEGER_OVERFLOW_WARNING _("NAs produced by integer overflow") #define CHECK_INTEGER_OVERFLOW(call, ans, naflag) do { \ if (naflag) { \ PROTECT(ans); \ warningcall(call, INTEGER_OVERFLOW_WARNING); \ UNPROTECT(1); \ } \ } while(0) #define R_INT_MAX INT_MAX #define R_INT_MIN -INT_MAX // .. relying on fact that NA_INTEGER is outside of these static R_INLINE int R_integer_plus(int x, int y, Rboolean *pnaflag) { if (x == NA_INTEGER || y == NA_INTEGER) return NA_INTEGER; if (((y > 0) && (x > (R_INT_MAX - y))) || ((y < 0) && (x < (R_INT_MIN - y)))) { if (pnaflag != NULL) *pnaflag = TRUE; return NA_INTEGER; } return x + y; } static R_INLINE int R_integer_minus(int x, int y, Rboolean *pnaflag) { if (x == NA_INTEGER || y == NA_INTEGER) return NA_INTEGER; if (((y < 0) && (x > (R_INT_MAX + y))) || ((y > 0) && (x < (R_INT_MIN + y)))) { if (pnaflag != NULL) *pnaflag = TRUE; return NA_INTEGER; } return x - y; } #define GOODIPROD(x, y, z) ((double) (x) * (double) (y) == (z)) static R_INLINE int R_integer_times(int x, int y, Rboolean *pnaflag) { if (x == NA_INTEGER || y == NA_INTEGER) return NA_INTEGER; else { int z = x * y; if (GOODIPROD(x, y, z) && z != NA_INTEGER) return z; else { if (pnaflag != NULL) *pnaflag = TRUE; return NA_INTEGER; } } } static R_INLINE double R_integer_divide(int x, int y) { if (x == NA_INTEGER || y == NA_INTEGER) return NA_REAL; else return (double) x / (double) y; } static R_INLINE SEXP ScalarValue1(SEXP x) { if (NO_REFERENCES(x)) return x; else return allocVector(TYPEOF(x), 1); } static R_INLINE SEXP ScalarValue2(SEXP x, SEXP y) { if (NO_REFERENCES(x)) return x; else if (NO_REFERENCES(y)) return y; else return allocVector(TYPEOF(x), 1); } /* Unary and Binary Operators */ SEXP attribute_hidden do_arith(SEXP call, SEXP op, SEXP args, SEXP env) { SEXP ans, arg1, arg2; int argc; if (args == R_NilValue) argc = 0; else if (CDR(args) == R_NilValue) argc = 1; else if (CDDR(args) == R_NilValue) argc = 2; else argc = length(args); arg1 = CAR(args); arg2 = CADR(args); if (ATTRIB(arg1) != R_NilValue || ATTRIB(arg2) != R_NilValue) { if (DispatchGroup("Ops", call, op, args, env, &ans)) return ans; } else if (argc == 2) { /* Handle some scaler operations immediately */ if (IS_SCALAR(arg1, REALSXP)) { if (IS_SCALAR(arg2, REALSXP)) { double x1 = REAL(arg1)[0]; double x2 = REAL(arg2)[0]; ans = ScalarValue2(arg1, arg2); switch (PRIMVAL(op)) { case PLUSOP: REAL(ans)[0] = x1 + x2; return ans; case MINUSOP: REAL(ans)[0] = x1 - x2; return ans; case TIMESOP: REAL(ans)[0] = x1 * x2; return ans; case DIVOP: REAL(ans)[0] = x1 / x2; return ans; } } else if (IS_SCALAR(arg2, INTSXP)) { double x1 = REAL(arg1)[0]; double x2 = INTEGER(arg2)[0] != NA_INTEGER ? (double) INTEGER(arg2)[0] : NA_REAL; ans = ScalarValue1(arg1); switch (PRIMVAL(op)) { case PLUSOP: REAL(ans)[0] = x1 + x2; return ans; case MINUSOP: REAL(ans)[0] = x1 - x2; return ans; case TIMESOP: REAL(ans)[0] = x1 * x2; return ans; case DIVOP: REAL(ans)[0] = x1 / x2; return ans; } } } else if (IS_SCALAR(arg1, INTSXP)) { if (IS_SCALAR(arg2, REALSXP)) { double x1 = INTEGER(arg1)[0] != NA_INTEGER ? (double) INTEGER(arg1)[0] : NA_REAL; double x2 = REAL(arg2)[0]; ans = ScalarValue1(arg2); switch (PRIMVAL(op)) { case PLUSOP: REAL(ans)[0] = x1 + x2; return ans; case MINUSOP: REAL(ans)[0] = x1 - x2; return ans; case TIMESOP: REAL(ans)[0] = x1 * x2; return ans; case DIVOP: REAL(ans)[0] = x1 / x2; return ans; } } else if (IS_SCALAR(arg2, INTSXP)) { Rboolean naflag = FALSE; int x1 = INTEGER(arg1)[0]; int x2 = INTEGER(arg2)[0]; switch (PRIMVAL(op)) { case PLUSOP: ans = ScalarValue2(arg1, arg2); INTEGER(ans)[0] = R_integer_plus(x1, x2, &naflag); CHECK_INTEGER_OVERFLOW(call, ans, naflag); return ans; case MINUSOP: ans = ScalarValue2(arg1, arg2); INTEGER(ans)[0] = R_integer_minus(x1, x2, &naflag); CHECK_INTEGER_OVERFLOW(call, ans, naflag); return ans; case TIMESOP: ans = ScalarValue2(arg1, arg2); INTEGER(ans)[0] = R_integer_times(x1, x2, &naflag); CHECK_INTEGER_OVERFLOW(call, ans, naflag); return ans; case DIVOP: ans = ScalarReal(R_integer_divide(x1, x2)); return ans; } } } } else if (argc == 1) { if (IS_SCALAR(arg1, REALSXP)) { switch(PRIMVAL(op)) { case PLUSOP: return(arg1); case MINUSOP: ans = ScalarValue1(arg1); REAL(ans)[0] = -REAL(arg1)[0]; return ans; } } else if (IS_SCALAR(arg1, INTSXP)) { switch(PRIMVAL(op)) { case PLUSOP: return(arg1); case MINUSOP: ans = ScalarValue1(arg1); INTEGER(ans)[0] = INTEGER(arg1)[0] == NA_INTEGER ? NA_INTEGER : -INTEGER(arg1)[0]; return ans; } } } if (argc == 2) return R_binary(call, op, arg1, arg2); else if (argc == 1) return R_unary(call, op, arg1); else errorcall(call,_("operator needs one or two arguments")); return ans; /* never used; to keep -Wall happy */ } #define COERCE_IF_NEEDED(v, tp, vpi) do { \ if (TYPEOF(v) != (tp)) { \ int __vo__ = OBJECT(v); \ REPROTECT(v = coerceVector(v, (tp)), vpi); \ if (__vo__) SET_OBJECT(v, 1); \ } \ } while (0) #define FIXUP_NULL_AND_CHECK_TYPES(v, vpi) do { \ switch (TYPEOF(v)) { \ case NILSXP: REPROTECT(v = allocVector(REALSXP,0), vpi); break; \ case CPLXSXP: case REALSXP: case INTSXP: case LGLSXP: break; \ default: errorcall(lcall, _("non-numeric argument to binary operator")); \ } \ } while (0) SEXP attribute_hidden R_binary(SEXP call, SEXP op, SEXP x, SEXP y) { SEXP klass, dims, tsp, xnames, ynames, val; R_xlen_t nx, ny, mismatch = 0; int xarray, yarray, xts, yts, xS4 = 0, yS4 = 0; int xattr, yattr; SEXP lcall = call; PROTECT_INDEX xpi, ypi; ARITHOP_TYPE oper = (ARITHOP_TYPE) PRIMVAL(op); int nprotect = 2; /* x and y */ PROTECT_WITH_INDEX(x, &xpi); PROTECT_WITH_INDEX(y, &ypi); FIXUP_NULL_AND_CHECK_TYPES(x, xpi); FIXUP_NULL_AND_CHECK_TYPES(y, ypi); nx = XLENGTH(x); if (ATTRIB(x) != R_NilValue) { xattr = TRUE; xarray = isArray(x); xts = isTs(x); xS4 = isS4(x); } else xarray = xts = xattr = FALSE; ny = XLENGTH(y); if (ATTRIB(y) != R_NilValue) { yattr = TRUE; yarray = isArray(y); yts = isTs(y); yS4 = isS4(y); } else yarray = yts = yattr = FALSE; /* If either x or y is a matrix with length 1 and the other is a vector, we want to coerce the matrix to be a vector. Do we want to? We don't do it! BDR 2004-03-06 */ /* FIXME: Danger Will Robinson. * ----- We might be trashing arguments here. */ if (xarray != yarray) { if (xarray && nx==1 && ny!=1) { REPROTECT(x = duplicate(x), xpi); setAttrib(x, R_DimSymbol, R_NilValue); } if (yarray && ny==1 && nx!=1) { REPROTECT(y = duplicate(y), ypi); setAttrib(y, R_DimSymbol, R_NilValue); } } if (xarray || yarray) { if (xarray && yarray) { if (!conformable(x, y)) errorcall(lcall, _("non-conformable arrays")); PROTECT(dims = getAttrib(x, R_DimSymbol)); } else if (xarray) { PROTECT(dims = getAttrib(x, R_DimSymbol)); } else { /* (yarray) */ PROTECT(dims = getAttrib(y, R_DimSymbol)); } nprotect++; if (xattr) { PROTECT(xnames = getAttrib(x, R_DimNamesSymbol)); nprotect++; } else xnames = R_NilValue; if (yattr) { PROTECT(ynames = getAttrib(y, R_DimNamesSymbol)); nprotect++; } else ynames = R_NilValue; } else { dims = R_NilValue; if (xattr) { PROTECT(xnames = getAttrib(x, R_NamesSymbol)); nprotect++; } else xnames = R_NilValue; if (yattr) { PROTECT(ynames = getAttrib(y, R_NamesSymbol)); nprotect++; } else ynames = R_NilValue; } if (nx == ny || nx == 1 || ny == 1) mismatch = 0; else if (nx > 0 && ny > 0) { if (nx > ny) mismatch = nx % ny; else mismatch = ny % nx; } if (xts || yts) { if (xts && yts) { if (!tsConform(x, y)) errorcall(lcall, _("non-conformable time-series")); PROTECT(tsp = getAttrib(x, R_TspSymbol)); PROTECT(klass = getAttrib(x, R_ClassSymbol)); } else if (xts) { if (nx < ny) ErrorMessage(lcall, ERROR_TSVEC_MISMATCH); PROTECT(tsp = getAttrib(x, R_TspSymbol)); PROTECT(klass = getAttrib(x, R_ClassSymbol)); } else { /* (yts) */ if (ny < nx) ErrorMessage(lcall, ERROR_TSVEC_MISMATCH); PROTECT(tsp = getAttrib(y, R_TspSymbol)); PROTECT(klass = getAttrib(y, R_ClassSymbol)); } nprotect += 2; } else klass = tsp = NULL; /* -Wall */ if (mismatch) warningcall(lcall, _("longer object length is not a multiple of shorter object length")); /* need to preserve object here, as *_binary copies class attributes */ if (TYPEOF(x) == CPLXSXP || TYPEOF(y) == CPLXSXP) { COERCE_IF_NEEDED(x, CPLXSXP, xpi); COERCE_IF_NEEDED(y, CPLXSXP, ypi); val = complex_binary(oper, x, y); } else if (TYPEOF(x) == REALSXP || TYPEOF(y) == REALSXP) { /* real_binary can handle REALSXP or INTSXP operand, but not LGLSXP. */ /* Can get a LGLSXP. In base-Ex.R on 24 Oct '06, got 8 of these. */ if (TYPEOF(x) != INTSXP) COERCE_IF_NEEDED(x, REALSXP, xpi); if (TYPEOF(y) != INTSXP) COERCE_IF_NEEDED(y, REALSXP, ypi); val = real_binary(oper, x, y); } else val = integer_binary(oper, x, y, lcall); /* quick return if there are no attributes */ if (! xattr && ! yattr) { UNPROTECT(nprotect); return val; } PROTECT(val); nprotect++; /* Don't set the dims if one argument is an array of size 0 and the other isn't of size zero, cos they're wrong */ /* Not if the other argument is a scalar (PR#1979) */ if (dims != R_NilValue) { if (!((xarray && (nx == 0) && (ny > 1)) || (yarray && (ny == 0) && (nx > 1)))){ setAttrib(val, R_DimSymbol, dims); if (xnames != R_NilValue) setAttrib(val, R_DimNamesSymbol, xnames); else if (ynames != R_NilValue) setAttrib(val, R_DimNamesSymbol, ynames); } } else { if (XLENGTH(val) == xlength(xnames)) setAttrib(val, R_NamesSymbol, xnames); else if (XLENGTH(val) == xlength(ynames)) setAttrib(val, R_NamesSymbol, ynames); } if (xts || yts) { /* must set *after* dims! */ setAttrib(val, R_TspSymbol, tsp); setAttrib(val, R_ClassSymbol, klass); } if(xS4 || yS4) { /* Only set the bit: no method defined! */ val = asS4(val, TRUE, TRUE); } UNPROTECT(nprotect); return val; } SEXP attribute_hidden R_unary(SEXP call, SEXP op, SEXP s1) { ARITHOP_TYPE operation = (ARITHOP_TYPE) PRIMVAL(op); switch (TYPEOF(s1)) { case LGLSXP: return logical_unary(operation, s1, call); case INTSXP: return integer_unary(operation, s1, call); case REALSXP: return real_unary(operation, s1, call); case CPLXSXP: return complex_unary(operation, s1, call); default: errorcall(call, _("invalid argument to unary operator")); } return s1; /* never used; to keep -Wall happy */ } static SEXP logical_unary(ARITHOP_TYPE code, SEXP s1, SEXP call) { R_xlen_t n = XLENGTH(s1); SEXP ans = PROTECT(allocVector(INTSXP, n)); SEXP names = PROTECT(getAttrib(s1, R_NamesSymbol)); SEXP dim = PROTECT(getAttrib(s1, R_DimSymbol)); SEXP dimnames = PROTECT(getAttrib(s1, R_DimNamesSymbol)); if(names != R_NilValue) setAttrib(ans, R_NamesSymbol, names); if(dim != R_NilValue) setAttrib(ans, R_DimSymbol, dim); if(dimnames != R_NilValue) setAttrib(ans, R_DimNamesSymbol, dimnames); UNPROTECT(3); switch (code) { case PLUSOP: for (R_xlen_t i = 0; i < n; i++) INTEGER(ans)[i] = LOGICAL(s1)[i]; break; case MINUSOP: for (R_xlen_t i = 0; i < n; i++) { int x = LOGICAL(s1)[i]; INTEGER(ans)[i] = (x == NA_INTEGER) ? NA_INTEGER : ((x == 0.0) ? 0 : -x); } break; default: errorcall(call, _("invalid unary operator")); } UNPROTECT(1); return ans; } static SEXP integer_unary(ARITHOP_TYPE code, SEXP s1, SEXP call) { R_xlen_t i, n; int x; SEXP ans; switch (code) { case PLUSOP: return s1; case MINUSOP: ans = NO_REFERENCES(s1) ? s1 : duplicate(s1); n = XLENGTH(s1); for (i = 0; i < n; i++) { x = INTEGER(s1)[i]; INTEGER(ans)[i] = (x == NA_INTEGER) ? NA_INTEGER : ((x == 0.0) ? 0 : -x); } return ans; default: errorcall(call, _("invalid unary operator")); } return s1; /* never used; to keep -Wall happy */ } static SEXP real_unary(ARITHOP_TYPE code, SEXP s1, SEXP lcall) { R_xlen_t i, n; SEXP ans; switch (code) { case PLUSOP: return s1; case MINUSOP: ans = NO_REFERENCES(s1) ? s1 : duplicate(s1); n = XLENGTH(s1); for (i = 0; i < n; i++) REAL(ans)[i] = -REAL(s1)[i]; return ans; default: errorcall(lcall, _("invalid unary operator")); } return s1; /* never used; to keep -Wall happy */ } /* i1 = i % n1; i2 = i % n2; * this macro is quite a bit faster than having real modulo calls * in the loop (tested on Intel and Sparc) */ #define mod_iterate(n1,n2,i1,i2) for (i=i1=i2=0; i n2) ? n1 : n2; if (code == DIVOP || code == POWOP) ans = allocVector(REALSXP, n); else ans = R_allocOrReuseVector(s1, s2, INTSXP, n); if (n == 0) return(ans); PROTECT(ans); switch (code) { case PLUSOP: MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, { x1 = INTEGER(s1)[i1]; x2 = INTEGER(s2)[i2]; INTEGER(ans)[i] = R_integer_plus(x1, x2, &naflag); }); if (naflag) warningcall(lcall, INTEGER_OVERFLOW_WARNING); break; case MINUSOP: MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, { x1 = INTEGER(s1)[i1]; x2 = INTEGER(s2)[i2]; INTEGER(ans)[i] = R_integer_minus(x1, x2, &naflag); }); if (naflag) warningcall(lcall, INTEGER_OVERFLOW_WARNING); break; case TIMESOP: MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, { x1 = INTEGER(s1)[i1]; x2 = INTEGER(s2)[i2]; INTEGER(ans)[i] = R_integer_times(x1, x2, &naflag); }); if (naflag) warningcall(lcall, INTEGER_OVERFLOW_WARNING); break; case DIVOP: MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, { x1 = INTEGER(s1)[i1]; x2 = INTEGER(s2)[i2]; REAL(ans)[i] = R_integer_divide(x1, x2); }); break; case POWOP: MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, { if((x1 = INTEGER(s1)[i1]) == 1 || (x2 = INTEGER(s2)[i2]) == 0) REAL(ans)[i] = 1.; else if (x1 == NA_INTEGER || x2 == NA_INTEGER) REAL(ans)[i] = NA_REAL; else REAL(ans)[i] = R_POW((double) x1, (double) x2); }); break; case MODOP: MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, { x1 = INTEGER(s1)[i1]; x2 = INTEGER(s2)[i2]; if (x1 == NA_INTEGER || x2 == NA_INTEGER || x2 == 0) INTEGER(ans)[i] = NA_INTEGER; else { INTEGER(ans)[i] = /* till 0.63.2: x1 % x2 */ (x1 >= 0 && x2 > 0) ? x1 % x2 : (int)myfmod((double)x1,(double)x2); } }); break; case IDIVOP: MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, { x1 = INTEGER(s1)[i1]; x2 = INTEGER(s2)[i2]; /* This had x %/% 0 == 0 prior to 2.14.1, but it seems conventionally to be undefined */ if (x1 == NA_INTEGER || x2 == NA_INTEGER || x2 == 0) INTEGER(ans)[i] = NA_INTEGER; else INTEGER(ans)[i] = (int) floor((double)x1 / (double)x2); }); break; } UNPROTECT(1); /* quick return if there are no attributes */ if (ATTRIB(s1) == R_NilValue && ATTRIB(s2) == R_NilValue) return ans; /* Copy attributes from longer argument. */ if (ans != s2 && n == n2 && ATTRIB(s2) != R_NilValue) copyMostAttrib(s2, ans); if (ans != s1 && n == n1 && ATTRIB(s1) != R_NilValue) copyMostAttrib(s1, ans); /* Done 2nd so s1's attrs overwrite s2's */ return ans; } #define R_INTEGER(robj, i) (double) (INTEGER(robj)[i] == NA_INTEGER ? NA_REAL : INTEGER(robj)[i]) static SEXP real_binary(ARITHOP_TYPE code, SEXP s1, SEXP s2) { R_xlen_t i, i1, i2, n, n1, n2; SEXP ans; /* Note: "s1" and "s2" are protected above. */ n1 = XLENGTH(s1); n2 = XLENGTH(s2); /* S4-compatibility change: if n1 or n2 is 0, result is of length 0 */ if (n1 == 0 || n2 == 0) return(allocVector(REALSXP, 0)); n = (n1 > n2) ? n1 : n2; PROTECT(ans = R_allocOrReuseVector(s1, s2, REALSXP, n)); switch (code) { case PLUSOP: if(TYPEOF(s1) == REALSXP && TYPEOF(s2) == REALSXP) { double *da = REAL(ans); double *dx = REAL(s1); double *dy = REAL(s2); if (n2 == 1) { double tmp = dy[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = dx[i] + tmp;); } else if (n1 == 1) { double tmp = dx[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = tmp + dy[i];); } else if (n1 == n2) R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = dx[i] + dy[i];); else MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, da[i] = dx[i1] + dy[i2];); } else if(TYPEOF(s1) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = R_INTEGER(s1, i1) + REAL(s2)[i2];); else if(TYPEOF(s2) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = REAL(s1)[i1] + R_INTEGER(s2, i2);); break; case MINUSOP: if(TYPEOF(s1) == REALSXP && TYPEOF(s2) == REALSXP) { double *da = REAL(ans); double *dx = REAL(s1); double *dy = REAL(s2); if (n2 == 1) { double tmp = dy[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = dx[i] - tmp;); } else if (n1 == 1) { double tmp = dx[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = tmp - dy[i];); } else if (n1 == n2) R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = dx[i] - dy[i];); else MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, da[i] = dx[i1] - dy[i2];); } else if(TYPEOF(s1) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = R_INTEGER(s1, i1) - REAL(s2)[i2];); else if(TYPEOF(s2) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = REAL(s1)[i1] - R_INTEGER(s2, i2);); break; case TIMESOP: if(TYPEOF(s1) == REALSXP && TYPEOF(s2) == REALSXP) { double *da = REAL(ans); double *dx = REAL(s1); double *dy = REAL(s2); if (n2 == 1) { double tmp = dy[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = dx[i] * tmp;); } else if (n1 == 1) { double tmp = REAL(s1)[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = tmp * dy[i];); } else if (n1 == n2) R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = dx[i] * dy[i];); else MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, da[i] = dx[i1] * dy[i2];); } else if(TYPEOF(s1) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = R_INTEGER(s1, i1) * REAL(s2)[i2];); else if(TYPEOF(s2) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = REAL(s1)[i1] * R_INTEGER(s2, i2);); break; case DIVOP: if(TYPEOF(s1) == REALSXP && TYPEOF(s2) == REALSXP) { double *da = REAL(ans); double *dx = REAL(s1); double *dy = REAL(s2); if (n2 == 1) { double tmp = dy[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = dx[i] / tmp;); } else if (n1 == 1) { double tmp = dx[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = tmp / dy[i];); } else if (n1 == n2) R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = dx[i] / dy[i];); else MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, da[i] = dx[i1] / dy[i2];); } else if(TYPEOF(s1) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = R_INTEGER(s1, i1) / REAL(s2)[i2];); else if(TYPEOF(s2) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = REAL(s1)[i1] / R_INTEGER(s2, i2);); break; case POWOP: if(TYPEOF(s1) == REALSXP && TYPEOF(s2) == REALSXP) { double *da = REAL(ans); double *dx = REAL(s1); double *dy = REAL(s2); if (n2 == 1) { double tmp = dy[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = R_POW(dx[i], tmp);); } else if (n1 == 1) { double tmp = dx[0]; R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = R_POW(tmp, dy[i]);); } else if (n1 == n2) R_ITERATE_CHECK(NINTERRUPT, n, i, da[i] = R_POW(dx[i], dy[i]);); else MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, da[i] = R_POW(dx[i1], dy[i2]);); } else if(TYPEOF(s1) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = R_POW( R_INTEGER(s1, i1), REAL(s2)[i2]);); else if(TYPEOF(s2) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = R_POW(REAL(s1)[i1], R_INTEGER(s2, i2));); break; case MODOP: if(TYPEOF(s1) == REALSXP && TYPEOF(s2) == REALSXP) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = myfmod(REAL(s1)[i1], REAL(s2)[i2]);); else if(TYPEOF(s1) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = myfmod( R_INTEGER(s1, i1), REAL(s2)[i2]);); else if(TYPEOF(s2) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = myfmod(REAL(s1)[i1], R_INTEGER(s2, i2));); break; case IDIVOP: if(TYPEOF(s1) == REALSXP && TYPEOF(s2) == REALSXP) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = myfloor(REAL(s1)[i1], REAL(s2)[i2]);); else if(TYPEOF(s1) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = myfloor(R_INTEGER(s1, i1), REAL(s2)[i2]);); else if(TYPEOF(s2) == INTSXP ) MOD_ITERATE_CHECK(NINTERRUPT, n, n1, n2, i, i1, i2, REAL(ans)[i] = myfloor(REAL(s1)[i1], R_INTEGER(s2,i2));); break; } UNPROTECT(1); /* quick return if there are no attributes */ if (ATTRIB(s1) == R_NilValue && ATTRIB(s2) == R_NilValue) return ans; /* Copy attributes from longer argument. */ if (ans != s2 && n == n2 && ATTRIB(s2) != R_NilValue) copyMostAttrib(s2, ans); if (ans != s1 && n == n1 && ATTRIB(s1) != R_NilValue) copyMostAttrib(s1, ans); /* Done 2nd so s1's attrs overwrite s2's */ return ans; } /* Mathematical Functions of One Argument */ static SEXP math1(SEXP sa, double(*f)(double), SEXP lcall) { SEXP sy; double *y, *a; R_xlen_t i, n; int naflag; if (!isNumeric(sa)) errorcall(lcall, R_MSG_NONNUM_MATH); n = XLENGTH(sa); /* coercion can lose the object bit */ PROTECT(sa = coerceVector(sa, REALSXP)); PROTECT(sy = NO_REFERENCES(sa) ? sa : allocVector(REALSXP, n)); a = REAL(sa); y = REAL(sy); naflag = 0; for (i = 0; i < n; i++) { if (ISNAN(a[i])) y[i] = a[i]; else { y[i] = f(a[i]); if (ISNAN(y[i])) naflag = 1; } } /* These are primitives, so need to use the call */ if(naflag) warningcall(lcall, R_MSG_NA); if (sa != sy && ATTRIB(sa) != R_NilValue) DUPLICATE_ATTRIB(sy, sa); UNPROTECT(2); return sy; } #ifdef HAVE_TANPI // we document that tanpi(0.5) is NaN, but the draft C11 extension // does not require this and the Solaris version gives Inf. double Rtanpi(double); #endif SEXP attribute_hidden do_math1(SEXP call, SEXP op, SEXP args, SEXP env) { SEXP s; checkArity(op, args); check1arg(args, call, "x"); if (DispatchGroup("Math", call, op, args, env, &s)) return s; if (isComplex(CAR(args))) return complex_math1(call, op, args, env); #define MATH1(x) math1(CAR(args), x, call); switch (PRIMVAL(op)) { case 1: return MATH1(floor); case 2: return MATH1(ceil); case 3: return MATH1(sqrt); case 4: return MATH1(sign); /* case 5: return MATH1(trunc); separate from 2.6.0 */ case 10: return MATH1(exp); case 11: return MATH1(expm1); case 12: return MATH1(log1p); case 20: return MATH1(cos); case 21: return MATH1(sin); case 22: return MATH1(tan); case 23: return MATH1(acos); case 24: return MATH1(asin); case 25: return MATH1(atan); case 30: return MATH1(cosh); case 31: return MATH1(sinh); case 32: return MATH1(tanh); case 33: return MATH1(acosh); case 34: return MATH1(asinh); case 35: return MATH1(atanh); case 40: return MATH1(lgammafn); case 41: return MATH1(gammafn); case 42: return MATH1(digamma); case 43: return MATH1(trigamma); /* case 44: return MATH1(tetragamma); case 45: return MATH1(pentagamma); removed in 2.0.0 case 46: return MATH1(Rf_gamma_cody); removed in 2.8.0 */ case 47: return MATH1(cospi); case 48: return MATH1(sinpi); #ifndef HAVE_TANPI case 49: return MATH1(tanpi); #else case 49: return MATH1(Rtanpi); #endif default: errorcall(call, _("unimplemented real function of 1 argument")); } return s; /* never used; to keep -Wall happy */ } /* methods are allowed to have more than one arg */ SEXP attribute_hidden do_trunc(SEXP call, SEXP op, SEXP args, SEXP env) { SEXP s; if (DispatchGroup("Math", call, op, args, env, &s)) return s; checkArity(op, args); /* but is -1 in names.c */ check1arg(args, call, "x"); if (isComplex(CAR(args))) errorcall(call, _("unimplemented complex function")); return math1(CAR(args), trunc, call); } /* Note that this is slightly different from the do_math1 set, both for integer/logical inputs and what it dispatches to for complex ones. */ SEXP attribute_hidden do_abs(SEXP call, SEXP op, SEXP args, SEXP env) { SEXP x, s = R_NilValue /* -Wall */; checkArity(op, args); check1arg(args, call, "x"); x = CAR(args); if (DispatchGroup("Math", call, op, args, env, &s)) return s; if (isInteger(x) || isLogical(x)) { /* integer or logical ==> return integer, factor was covered by Math.factor. */ R_xlen_t i, n = XLENGTH(x); s = (NO_REFERENCES(x) && TYPEOF(x) == INTSXP) ? x : allocVector(INTSXP, n); PROTECT(s); /* Note: relying on INTEGER(.) === LOGICAL(.) : */ for(i = 0 ; i < n ; i++) { int xi = INTEGER(x)[i]; INTEGER(s)[i] = (xi == NA_INTEGER) ? xi : abs(xi); } } else if (TYPEOF(x) == REALSXP) { R_xlen_t i, n = XLENGTH(x); PROTECT(s = NO_REFERENCES(x) ? x : allocVector(REALSXP, n)); for(i = 0 ; i < n ; i++) REAL(s)[i] = fabs(REAL(x)[i]); } else if (isComplex(x)) { return do_cmathfuns(call, op, args, env); } else errorcall(call, R_MSG_NONNUM_MATH); if (x != s && ATTRIB(x) != R_NilValue) DUPLICATE_ATTRIB(s, x); UNPROTECT(1); return s; } /* Mathematical Functions of Two Numeric Arguments (plus 1 int) */ /* math2_1 and math2_2 and related can be removed once the byte compiler knows how to optimize to .External rather than .Internal */ #define if_NA_Math2_set(y,a,b) \ if (ISNA (a) || ISNA (b)) y = NA_REAL; \ else if (ISNAN(a) || ISNAN(b)) y = R_NaN; static SEXP math2(SEXP sa, SEXP sb, double (*f)(double, double), SEXP lcall) { SEXP sy; R_xlen_t i, ia, ib, n, na, nb; double ai, bi, *a, *b, *y; int naflag; if (!isNumeric(sa) || !isNumeric(sb)) errorcall(lcall, R_MSG_NONNUM_MATH); /* for 0-length a we want the attributes of a, not those of b as no recycling will occur */ #define SETUP_Math2 \ na = XLENGTH(sa); \ nb = XLENGTH(sb); \ if ((na == 0) || (nb == 0)) { \ PROTECT(sy = allocVector(REALSXP, 0)); \ if (na == 0) DUPLICATE_ATTRIB(sy, sa); \ UNPROTECT(1); \ return(sy); \ } \ n = (na < nb) ? nb : na; \ PROTECT(sa = coerceVector(sa, REALSXP)); \ PROTECT(sb = coerceVector(sb, REALSXP)); \ PROTECT(sy = allocVector(REALSXP, n)); \ a = REAL(sa); \ b = REAL(sb); \ y = REAL(sy); \ naflag = 0 SETUP_Math2; mod_iterate(na, nb, ia, ib) { // if ((i+1) % NINTERRUPT == 0) R_CheckUserInterrupt(); ai = a[ia]; bi = b[ib]; if_NA_Math2_set(y[i], ai, bi) else { y[i] = f(ai, bi); if (ISNAN(y[i])) naflag = 1; } } #define FINISH_Math2 \ if(naflag) warning(R_MSG_NA); \ if (n == na) DUPLICATE_ATTRIB(sy, sa); \ else if (n == nb) DUPLICATE_ATTRIB(sy, sb); \ UNPROTECT(3) FINISH_Math2; return sy; } /* math2() */ static SEXP math2_1(SEXP sa, SEXP sb, SEXP sI, double (*f)(double, double, int), SEXP lcall) { SEXP sy; R_xlen_t i, ia, ib, n, na, nb; double ai, bi, *a, *b, *y; int m_opt; int naflag; if (!isNumeric(sa) || !isNumeric(sb)) errorcall(lcall, R_MSG_NONNUM_MATH); SETUP_Math2; m_opt = asInteger(sI); mod_iterate(na, nb, ia, ib) { // if ((i+1) % NINTERRUPT == 0) R_CheckUserInterrupt(); ai = a[ia]; bi = b[ib]; if_NA_Math2_set(y[i], ai, bi) else { y[i] = f(ai, bi, m_opt); if (ISNAN(y[i])) naflag = 1; } } FINISH_Math2; return sy; } /* math2_1() */ static SEXP math2_2(SEXP sa, SEXP sb, SEXP sI1, SEXP sI2, double (*f)(double, double, int, int), SEXP lcall) { SEXP sy; R_xlen_t i, ia, ib, n, na, nb; double ai, bi, *a, *b, *y; int i_1, i_2; int naflag; if (!isNumeric(sa) || !isNumeric(sb)) errorcall(lcall, R_MSG_NONNUM_MATH); SETUP_Math2; i_1 = asInteger(sI1); i_2 = asInteger(sI2); mod_iterate(na, nb, ia, ib) { // if ((i+1) % NINTERRUPT == 0) R_CheckUserInterrupt(); ai = a[ia]; bi = b[ib]; if_NA_Math2_set(y[i], ai, bi) else { y[i] = f(ai, bi, i_1, i_2); if (ISNAN(y[i])) naflag = 1; } } FINISH_Math2; return sy; } /* math2_2() */ /* This is only used directly by .Internal for Bessel functions, so managing R_alloc stack is only prudence */ static SEXP math2B(SEXP sa, SEXP sb, double (*f)(double, double, double *), SEXP lcall) { SEXP sy; R_xlen_t i, ia, ib, n, na, nb; double ai, bi, *a, *b, *y; int naflag; double amax, *work; size_t nw; if (!isNumeric(sa) || !isNumeric(sb)) errorcall(lcall, R_MSG_NONNUM_MATH); /* for 0-length a we want the attributes of a, not those of b as no recycling will occur */ SETUP_Math2; /* allocate work array for BesselJ, BesselY large enough for all arguments */ amax = 0.0; for (i = 0; i < nb; i++) { double av = b[i] < 0 ? -b[i] : b[i]; if (av > amax) amax = av; } const void *vmax = vmaxget(); nw = 1 + (size_t)floor(amax); work = (double *) R_alloc(nw, sizeof(double)); mod_iterate(na, nb, ia, ib) { // if ((i+1) % NINTERRUPT == 0) R_CheckUserInterrupt(); ai = a[ia]; bi = b[ib]; if_NA_Math2_set(y[i], ai, bi) else { y[i] = f(ai, bi, work); if (ISNAN(y[i])) naflag = 1; } } vmaxset(vmax); FINISH_Math2; return sy; } /* math2B() */ #define Math2(A, FUN) math2(CAR(A), CADR(A), FUN, call); #define Math2_1(A, FUN) math2_1(CAR(A), CADR(A), CADDR(A), FUN, call); #define Math2_2(A, FUN) math2_2(CAR(A), CADR(A), CADDR(A), CADDDR(A), FUN, call) #define Math2B(A, FUN) math2B(CAR(A), CADR(A), FUN, call); SEXP attribute_hidden do_math2(SEXP call, SEXP op, SEXP args, SEXP env) { checkArity(op, args); if (isComplex(CAR(args)) || (PRIMVAL(op) == 0 && isComplex(CADR(args)))) return complex_math2(call, op, args, env); switch (PRIMVAL(op)) { case 0: return Math2(args, atan2); case 10001: return Math2(args, fround);/* round(), src/nmath/fround.c */ case 10004: return Math2(args, fprec); /* signif(), src/nmath/fprec.c */ case 2: return Math2(args, lbeta); case 3: return Math2(args, beta); case 4: return Math2(args, lchoose); case 5: return Math2(args, choose); case 6: return Math2_1(args, dchisq); case 7: return Math2_2(args, pchisq); case 8: return Math2_2(args, qchisq); case 9: return Math2_1(args, dexp); case 10: return Math2_2(args, pexp); case 11: return Math2_2(args, qexp); case 12: return Math2_1(args, dgeom); case 13: return Math2_2(args, pgeom); case 14: return Math2_2(args, qgeom); case 15: return Math2_1(args, dpois); case 16: return Math2_2(args, ppois); case 17: return Math2_2(args, qpois); case 18: return Math2_1(args, dt); case 19: return Math2_2(args, pt); case 20: return Math2_2(args, qt); case 21: return Math2_1(args, dsignrank); case 22: return Math2_2(args, psignrank); case 23: return Math2_2(args, qsignrank); case 24: return Math2B(args, bessel_j_ex); case 25: return Math2B(args, bessel_y_ex); case 26: return Math2(args, psigamma); default: errorcall(call, _("unimplemented real function of %d numeric arguments"), 2); } return op; /* never used; to keep -Wall happy */ } /* The S4 Math2 group, round and signif */ /* This is a primitive SPECIALSXP with internal argument matching */ SEXP attribute_hidden do_Math2(SEXP call, SEXP op, SEXP args, SEXP env) { SEXP res, ap, call2; int n, nprotect = 2; if (length(args) >= 2 && isSymbol(CADR(args)) && R_isMissing(CADR(args), env)) { double digits = 0; if(PRIMVAL(op) == 10004) digits = 6.0; // for signif() PROTECT(args = list2(CAR(args), ScalarReal(digits))); nprotect++; } PROTECT(args = evalListKeepMissing(args, env)); PROTECT(call2 = lang2(CAR(call), R_NilValue)); SETCDR(call2, args); n = length(args); if (n != 1 && n != 2) error(_("%d arguments passed to '%s' which requires 1 or 2"), n, PRIMNAME(op)); if (! DispatchGroup("Math", call2, op, args, env, &res)) { if(n == 1) { double digits = 0.0; if(PRIMVAL(op) == 10004) digits = 6.0; SETCDR(args, CONS(ScalarReal(digits), R_NilValue)); } else { /* If named, do argument matching by name */ if (TAG(args) != R_NilValue || TAG(CDR(args)) != R_NilValue) { PROTECT(ap = CONS(R_NilValue, list1(R_NilValue))); SET_TAG(ap, install("x")); SET_TAG(CDR(ap), install("digits")); PROTECT(args = matchArgs(ap, args, call)); nprotect +=2; } if (length(CADR(args)) == 0) errorcall(call, _("invalid second argument of length 0")); } res = do_math2(call, op, args, env); } UNPROTECT(nprotect); return res; } /* log{2,10} are builtins */ SEXP attribute_hidden do_log1arg(SEXP call, SEXP op, SEXP args, SEXP env) { SEXP res, call2, args2, tmp = R_NilValue /* -Wall */; checkArity(op, args); check1arg(args, call, "x"); if (DispatchGroup("Math", call, op, args, env, &res)) return res; if(PRIMVAL(op) == 10) tmp = ScalarReal(10.0); if(PRIMVAL(op) == 2) tmp = ScalarReal(2.0); PROTECT(call2 = lang3(install("log"), CAR(args), tmp)); PROTECT(args2 = lang2(CAR(args), tmp)); if (! DispatchGroup("Math", call2, op, args2, env, &res)) { if (isComplex(CAR(args))) res = complex_math2(call2, op, args2, env); else res = math2(CAR(args), tmp, logbase, call); } UNPROTECT(2); return res; } /* This is a primitive SPECIALSXP with internal argument matching */ SEXP attribute_hidden do_log(SEXP call, SEXP op, SEXP args, SEXP env) { SEXP res, ap = args, call2; int n = length(args), nprotect = 2; if (n >= 2 && isSymbol(CADR(args)) && R_isMissing(CADR(args), env)) { #ifdef M_E double e = M_E; #else double e = exp(1.); #endif PROTECT(args = list2(CAR(args), ScalarReal(e))); nprotect++; } PROTECT(args = evalListKeepMissing(args, env)); PROTECT(call2 = lang2(CAR(call), R_NilValue)); SETCDR(call2, args); if (! DispatchGroup("Math", call2, op, args, env, &res)) { switch (n) { case 1: if (isComplex(CAR(args))) res = complex_math1(call, op, args, env); else res = math1(CAR(args), R_log, call); break; case 2: { /* match argument names if supplied */ PROTECT(ap = list2(R_NilValue, R_NilValue)); SET_TAG(ap, install("x")); SET_TAG(CDR(ap), install("base")); PROTECT(args = matchArgs(ap, args, call)); nprotect += 2; if (length(CADR(args)) == 0) errorcall(call, _("invalid argument 'base' of length 0")); if (isComplex(CAR(args)) || isComplex(CADR(args))) res = complex_math2(call, op, args, env); else res = math2(CAR(args), CADR(args), logbase, call); break; } default: error(_("%d arguments passed to 'log' which requires 1 or 2"), n); } } UNPROTECT(nprotect); return res; } /* Mathematical Functions of Three (Real) Arguments */ /* math3_1 and math3_2 and related can be removed once the byte compiler knows how to optimize to .External rather than .Internal */ #define if_NA_Math3_set(y,a,b,c) \ if (ISNA (a) || ISNA (b)|| ISNA (c)) y = NA_REAL; \ else if (ISNAN(a) || ISNAN(b)|| ISNAN(c)) y = R_NaN; #define mod_iterate3(n1,n2,n3,i1,i2,i3) for (i=i1=i2=i3=0; i amax) amax = av; } const void *vmax = vmaxget(); nw = 1 + (size_t)floor(amax); work = (double *) R_alloc(nw, sizeof(double)); mod_iterate3 (na, nb, nc, ia, ib, ic) { // if ((i+1) % NINTERRUPT == 0) R_CheckUserInterrupt(); ai = a[ia]; bi = b[ib]; ci = c[ic]; if_NA_Math3_set(y[i], ai,bi,ci) else { y[i] = f(ai, bi, ci, work); if (ISNAN(y[i])) naflag = 1; } } FINISH_Math3; vmaxset(vmax); return sy; } /* math3B */ #define Math3_1(A, FUN) math3_1(CAR(A), CADR(A), CADDR(A), CADDDR(A), FUN, call); #define Math3_2(A, FUN) math3_2(CAR(A), CADR(A), CADDR(A), CADDDR(A), CAD4R(A), FUN, call) #define Math3B(A, FUN) math3B (CAR(A), CADR(A), CADDR(A), FUN, call); SEXP attribute_hidden do_math3(SEXP call, SEXP op, SEXP args, SEXP env) { checkArity(op, args); switch (PRIMVAL(op)) { case 1: return Math3_1(args, dbeta); case 2: return Math3_2(args, pbeta); case 3: return Math3_2(args, qbeta); case 4: return Math3_1(args, dbinom); case 5: return Math3_2(args, pbinom); case 6: return Math3_2(args, qbinom); case 7: return Math3_1(args, dcauchy); case 8: return Math3_2(args, pcauchy); case 9: return Math3_2(args, qcauchy); case 10: return Math3_1(args, df); case 11: return Math3_2(args, pf); case 12: return Math3_2(args, qf); case 13: return Math3_1(args, dgamma); case 14: return Math3_2(args, pgamma); case 15: return Math3_2(args, qgamma); case 16: return Math3_1(args, dlnorm); case 17: return Math3_2(args, plnorm); case 18: return Math3_2(args, qlnorm); case 19: return Math3_1(args, dlogis); case 20: return Math3_2(args, plogis); case 21: return Math3_2(args, qlogis); case 22: return Math3_1(args, dnbinom); case 23: return Math3_2(args, pnbinom); case 24: return Math3_2(args, qnbinom); case 25: return Math3_1(args, dnorm); case 26: return Math3_2(args, pnorm); case 27: return Math3_2(args, qnorm); case 28: return Math3_1(args, dunif); case 29: return Math3_2(args, punif); case 30: return Math3_2(args, qunif); case 31: return Math3_1(args, dweibull); case 32: return Math3_2(args, pweibull); case 33: return Math3_2(args, qweibull); case 34: return Math3_1(args, dnchisq); case 35: return Math3_2(args, pnchisq); case 36: return Math3_2(args, qnchisq); case 37: return Math3_1(args, dnt); case 38: return Math3_2(args, pnt); case 39: return Math3_2(args, qnt); case 40: return Math3_1(args, dwilcox); case 41: return Math3_2(args, pwilcox); case 42: return Math3_2(args, qwilcox); case 43: return Math3B(args, bessel_i_ex); case 44: return Math3B(args, bessel_k_ex); case 45: return Math3_1(args, dnbinom_mu); case 46: return Math3_2(args, pnbinom_mu); case 47: return Math3_2(args, qnbinom_mu); default: errorcall(call, _("unimplemented real function of %d numeric arguments"), 3); } return op; /* never used; to keep -Wall happy */ } /* do_math3() */ /* Mathematical Functions of Four (Real) Arguments */ /* This can be removed completely once the byte compiler knows how to optimize to .External rather than .Internal */ #define if_NA_Math4_set(y,a,b,c,d) \ if (ISNA (a)|| ISNA (b)|| ISNA (c)|| ISNA (d)) y = NA_REAL;\ else if (ISNAN(a)|| ISNAN(b)|| ISNAN(c)|| ISNAN(d)) y = R_NaN; #define mod_iterate4(n1,n2,n3,n4,i1,i2,i3,i4) for (i=i1=i2=i3=i4=0; i