The R Project SVN R

Rev

Rev 4766 | Rev 5107 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed 

/*
 *  R : A Computer Language for Statistical Data Analysis
 *  Copyright (C) 1995, 1996  Robert Gentleman and Ross Ihaka
 *  Copyright (C) 1997--1998  Robert Gentleman, Ross Ihaka and the
 *                            R Development Core Team
 *
 *  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, write to the Free Software
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include "Defn.h"
#include "Mathlib.h"
#include "Print.h"

#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif

SEXP ScalarLogical(int x)
{
    SEXP ans = allocVector(LGLSXP, 1);
    INTEGER(ans)[0] = x;
    return ans;
}

SEXP ScalarInteger(int x)
{
    SEXP ans = allocVector(INTSXP, 1);
    INTEGER(ans)[0] = x;
    return ans;
}

SEXP ScalarReal(double x)
{
    SEXP ans = allocVector(REALSXP, 1);
    REAL(ans)[0] = x;
    return ans;
}

SEXP ScalarComplex(complex x)
{
    SEXP ans = allocVector(CPLXSXP, 1);
    COMPLEX(ans)[0] = x;
    return ans;
}

SEXP ScalarString(SEXP x)
{
    SEXP ans;
    PROTECT(x);
    ans = allocVector(STRSXP, 1);
    STRING(ans)[0] = x;
    UNPROTECT(1);
    return ans;
}

static char *truenames[] = {
    "T",
    "True",
    "TRUE",
    "true",
    (char *) 0,
};

static char *falsenames[] = {
    "F",
    "False",
    "FALSE",
    "false",
    (char *) 0,
};

int asLogical(SEXP x)
{
    if (isVector(x)) {
    if (LENGTH(x) < 1)
        return NA_INTEGER;
    switch (TYPEOF(x)) {
    case LGLSXP:
        return LOGICAL(x)[0];
    case INTSXP:
        return (INTEGER(x)[0] == NA_INTEGER) ?
        NA_LOGICAL : (INTEGER(x)[0]) != 0;
    case REALSXP:
        return FINITE(REAL(x)[0]) ?
        (REAL(x)[0] != 0.0) : NA_LOGICAL;
    case CPLXSXP:
        return FINITE(COMPLEX(x)[0].r) ?
        (COMPLEX(x)[0].r != 0.0) : NA_LOGICAL;
    default:
        return NA_LOGICAL;
    }
    }
    return NA_LOGICAL;
}

int asInteger(SEXP x)
{
    if (isVectorObject(x) && LENGTH(x) >= 1) {
    switch (TYPEOF(x)) {
    case LGLSXP:
        return (LOGICAL(x)[0] == NA_LOGICAL) ?
        NA_INTEGER : ((LOGICAL(x)[0]) != 0);
    case INTSXP:
        return (INTEGER(x)[0]);
    case REALSXP:
        return FINITE(REAL(x)[0]) ?
        ((int)(REAL(x)[0])) : NA_INTEGER;
    case CPLXSXP:
        return FINITE(COMPLEX(x)[0].r) ?
        ((int)(COMPLEX(x)[0].r)) : NA_INTEGER;
    }
    }
    return NA_INTEGER;
}

double asReal(SEXP x)
{
    if (isVector(x)) {
    if (LENGTH(x) < 1) return NA_REAL;
    switch (TYPEOF(x)) {
    case LGLSXP:
    case INTSXP:
        return (INTEGER(x)[0] == NA_INTEGER) ?
        NA_REAL : (INTEGER(x)[0]);
    case REALSXP:
        return REAL(x)[0];
    case CPLXSXP:
        return COMPLEX(x)[0].r;
    default:
        return NA_REAL;
    }
    }
    return NA_REAL;
}


complex asComplex(SEXP x)
{
    complex z;
    z.r = NA_REAL;
    z.i = NA_REAL;
    if (isVectorObject(x) && LENGTH(x) >= 1) {
    switch (TYPEOF(x)) {
    case LGLSXP:
    case INTSXP:
        if (INTEGER(x)[0] != NA_INTEGER) {
        z.r = INTEGER(x)[0];
        z.i = 0;
        }
        return z;
    case REALSXP:
        if (REAL(x)[0] != NA_REAL) {
        z.r = REAL(x)[0];
        z.i = 0;
        }
        return z;
    case CPLXSXP:
        return COMPLEX(x)[0];
    }
    }
    return z;
}

SEXP asChar(SEXP x)
{
    int w, d, e;
    char buf[MAXELTSIZE];

    if (isVector(x)) {
    if (LENGTH(x) < 1)
        return NA_STRING;
    switch (TYPEOF(x)) {
    case LGLSXP:
        if (LOGICAL(x)[0] == NA_LOGICAL)
        return NA_STRING;
        if (LOGICAL(x)[0])
        sprintf(buf, "T");
        else
        sprintf(buf, "F");
        return mkChar(buf);
    case INTSXP:
        if (INTEGER(x)[0] == NA_INTEGER)
        return NA_STRING;
        sprintf(buf, "%d", INTEGER(x)[0]);
        return mkChar(buf);
    case REALSXP:
        formatReal(REAL(x), 1, &w, &d, &e);
        if (e)
        sprintf(buf, "%*.*e", w, d, REAL(x)[0]);
        else
        sprintf(buf, "%*.*f", w, d, REAL(x)[0]);
        return mkChar(buf);
        /* FIXME: CPLXSXP case here */
    case STRSXP:
        return STRING(x)[0];
    default:
        return NA_STRING;
    }
    }
    return NA_STRING;
}


static char type_msg[] = "invalid type passed to internal function\n";


void internalTypeCheck(SEXP call, SEXP s, SEXPTYPE type)
{
    if (TYPEOF(s) != type) {
    if (call)
        errorcall(call, type_msg);
    else
        error(type_msg);
    }
}


int isSymbol(SEXP s)
{
    return TYPEOF(s) == SYMSXP;
}


int isUserBinop(SEXP s)
{
    if (isSymbol(s)) {
    char *str = CHAR(PRINTNAME(s));
    if (str[0] == '%' && str[strlen(str)-1] == '%')
        return 1;
    }
    return 0;
}


int isNull(SEXP s)
{
    return (s == R_NilValue ||
        (TYPEOF(s) == EXPRSXP && LENGTH(s) == 0));
}


int isFunction(SEXP s)
{
    return (TYPEOF(s) == CLOSXP ||
        TYPEOF(s) == BUILTINSXP ||
        TYPEOF(s) == SPECIALSXP);
}


int isList(SEXP s)
{
    return (s == R_NilValue || TYPEOF(s) == LISTSXP);
}


int isNewList(SEXP s)
{
    return (s == R_NilValue || TYPEOF(s) == VECSXP);
}

int isPairList(SEXP s)
{
    switch (TYPEOF(s)) {
    case NILSXP:
    case LISTSXP:
    case LANGSXP:
    return 1;
    default:
    return 0;
    }
}

int isVectorList(SEXP s)
{
    switch (TYPEOF(s)) {
    case VECSXP:
    case EXPRSXP:
    return 1;
    default:
    return 0;
    }
}


int isVectorObject(SEXP s)
{
    switch (TYPEOF(s)) {
    case LGLSXP:
    case INTSXP:
    case REALSXP:
    case CPLXSXP:
    case STRSXP:
    case VECSXP:
    case EXPRSXP:
    return 1;
    default:
    return 0;
    }
}


int isFrame(SEXP s)
{
    SEXP class;
    int i;
    if (isObject(s)) {
    class = getAttrib(s, R_ClassSymbol);
    for (i = 0; i < length(class); i++)
        if (!strcmp(CHAR(STRING(class)[i]), "data.frame")) return 1;
    }
    return 0;
}


int isEnvironment(SEXP s)
{
    return (TYPEOF(s) == NILSXP || TYPEOF(s) == ENVSXP);
}


int isExpression(SEXP s)
{
    return TYPEOF(s) == EXPRSXP;
}


int isLanguage(SEXP s)
{
    return (s == R_NilValue || TYPEOF(s) == LANGSXP);
}


int isVector(SEXP s)
{
    switch(TYPEOF(s)) {
    case LGLSXP:
    case INTSXP:
    case REALSXP:
    case CPLXSXP:
    case STRSXP:
    case VECSXP:
    case EXPRSXP:
    return 1;
    break;
    default:
    return 0;
    break;
    }
}


int isMatrix(SEXP s)
{
    SEXP t;
    if (isVector(s)) {
    t = getAttrib(s, R_DimSymbol);
    if (TYPEOF(t) == INTSXP && LENGTH(t) == 2)
        return 1;
    }
    return 0;
}


int isArray(SEXP s)
{
    SEXP t;
    if (isVector(s)) {
    t = getAttrib(s, R_DimSymbol);
    if (TYPEOF(t) == INTSXP && LENGTH(t) > 0)
        return 1;
    }
    return 0;
}


int isTs(SEXP s)
{
    return (isVector(s) && getAttrib(s, R_TspSymbol) != R_NilValue);
}


int tsConform(SEXP x, SEXP y)
{
    if ((x = getAttrib(x, R_TspSymbol)) != R_NilValue &&
    (y = getAttrib(y, R_TspSymbol)) != R_NilValue)
    return INTEGER(x)[0] == INTEGER(x)[0] &&
        INTEGER(x)[1] == INTEGER(x)[1] &&
        INTEGER(x)[2] == INTEGER(x)[2];
    return 0;
}


/* Check to see if a list can be made into a vector. */
/* it must have every element being a vector of length 1. */

int isVectorizable(SEXP s)
{
    int mode = 0;
    if (isNull(s)) {
    return 1;
    }
    else if (isNewList(s)) {
    int i, n;
    n = LENGTH(s);
    for (i = 0 ; i < n; i++) {
        if (!isVector(VECTOR(s)[i]) || LENGTH(VECTOR(s)[i]) > 1)
        return 0;
        mode = (mode >= TYPEOF(VECTOR(s)[i])) ?
        mode : TYPEOF(VECTOR(s)[i]);
    }
    return mode;
    }
    else if (isList(s)) {
    for ( ; s != R_NilValue; s = CDR(s)) {
        if (!isVector(CAR(s)) || LENGTH(CAR(s)) > 1)
        return 0;
        mode = (mode >= (int) TYPEOF(CAR(s))) ? mode : TYPEOF(CAR(s));
    }
    return mode;
    }
    else return 0;
}


/* Check to see if the arrays "x" and "y" have the identical extents */

int conformable(SEXP x, SEXP y)
{
    int i, n;
    PROTECT(x = getAttrib(x, R_DimSymbol));
    y = getAttrib(y, R_DimSymbol);
    UNPROTECT(1);
    if ((n = length(x)) != length(y))
    return 0;
    for (i = 0; i < n; i++)
    if (INTEGER(x)[i] != INTEGER(y)[i])
        return 0;
    return 1;
}


int nrows(SEXP s)
{
    SEXP t;
    if (isVector(s) || isList(s)) {
    t = getAttrib(s, R_DimSymbol);
    if (t == R_NilValue) return LENGTH(s);
    return INTEGER(t)[0];
    }
    else if (isFrame(s)) {
    return nrows(CAR(s));
    }
    else error("object is not a matrix\n");
    return -1;
}


int ncols(SEXP s)
{
    SEXP t;
    if (isVector(s) || isList(s)) {
    t = getAttrib(s, R_DimSymbol);
    if (t == R_NilValue) return 1;
    return INTEGER(t)[1];
    }
    else if (isFrame(s)) {
    return length(s);
    }
    else error("object is not a matrix\n");
    return -1;/*NOTREACHED*/
}


int nlevels(SEXP f)
{
    if (!isFactor(f))
    return 0;
    return LENGTH(getAttrib(f, R_LevelsSymbol));
}

/* Is an object of numeric type. */
/* FIXME:  the LGLSXP case should be excluded here. */

int isNumeric(SEXP s)
{
    switch(TYPEOF(s)) {
    case LGLSXP:
    case INTSXP:
    case REALSXP:
    return 1;
    default:
    return 0;
    }
}

int isString(SEXP s)
{
    return (TYPEOF(s) == STRSXP);
}


int isLogical(SEXP s)
{
    return (TYPEOF(s) == LGLSXP);
}


int isInteger(SEXP s)
{
    return (TYPEOF(s) == INTSXP);
}


int isReal(SEXP s)
{
    return (TYPEOF(s) == REALSXP);
}


int isComplex(SEXP s)
{
    return (TYPEOF(s) == CPLXSXP);
}


int isUnordered(SEXP s)
{
    return (isInteger(s)
        && inherits(s, "factor")
        && !inherits(s, "ordered"));
}


int isOrdered(SEXP s)
{
    return (isInteger(s)
        && inherits(s, "factor")
        && inherits(s, "ordered"));
}


int isFactor(SEXP s)
{
    return (isInteger(s) && inherits(s, "factor"));
}


int isObject(SEXP s)
{
    return OBJECT(s);
}


int inherits(SEXP s, char *name)
{
    SEXP class;
    int i, nclass;
    if (isObject(s)) {
    class = getAttrib(s, R_ClassSymbol);
    nclass = length(class);
    for (i = 0; i < nclass; i++) {
        if (!strcmp(CHAR(STRING(class)[i]), name))
        return 1;
    }
    }
    return 0;
}


#ifdef neverUser
double realNA()
{
    return NA_REAL;
}
#endif

static struct {
    char *str;
    int type;
}
TypeTable[] = {
    { "NULL",       NILSXP     },  /* real types */
    { "symbol",     SYMSXP     },
    { "pairlist",   LISTSXP    },
    { "closure",    CLOSXP     },
    { "environment",    ENVSXP     },
    { "promise",    PROMSXP    },
    { "language",   LANGSXP    },
    { "special",    SPECIALSXP },
    { "builtin",    BUILTINSXP },
    { "char",       CHARSXP    },
    { "logical",    LGLSXP     },
    { "integer",    INTSXP     },
    { "double",     REALSXP    }, /*-  "real", for R <= 0.61.x */
    { "complex",    CPLXSXP    },
    { "character",  STRSXP     },
    { "...",        DOTSXP     },
    { "any",        ANYSXP     },
    { "expression", EXPRSXP    },
    { "list",       VECSXP     },
    /* aliases : */
    { "numeric",    REALSXP    },
    { "name",       SYMSXP     },

    { (char *)0,    -1     }
};


SEXPTYPE str2type(char *s)
{
    int i;
    for (i = 0; TypeTable[i].str; i++) {
    if (!strcmp(s, TypeTable[i].str))
        return TypeTable[i].type;
    }
    return -1;
}


SEXP type2str(SEXPTYPE t)
{
    int i;

    for (i = 0; TypeTable[i].str; i++) {
    if (TypeTable[i].type == t)
        return mkChar(TypeTable[i].str);
    }
    UNIMPLEMENTED("type2str");
    return R_NilValue; /* for -Wall */
}

int StringBlank(SEXP x)
{
    if (x == R_NilValue) return 1;
    else return CHAR(x)[0] == '\0';
}

/* Function to test whether a string is a true value */

int StringTrue(char *name)
{
    int i;
    for (i = 0; truenames[i]; i++)
    if (!strcmp(name, truenames[i]))
        return (1);
    return (0);
}

int StringFalse(char *name)
{
    int i;
    for (i = 0; falsenames[i]; i++)
    if (!strcmp(name, falsenames[i]))
        return (1);
    return (0);
}

SEXP EnsureString(SEXP s)
{
    switch(TYPEOF(s)) {
    case SYMSXP:
    s = PRINTNAME(s);
    break;
    case STRSXP:
    s = STRING(s)[0];
    break;
    case CHARSXP:
    break;
    case NILSXP:
    s = R_BlankString;
    break;
    default:
    error("invalid tag in name extraction\n");
    }
    return s;
}


void checkArity(SEXP op, SEXP args)
{
    if (PRIMARITY(op) >= 0 && PRIMARITY(op) != length(args))
    error("%d argument%s passed to \"%s\" which requires %d.\n",
          length(args), (length(args) == 1 ? "" : "s"),
          PRIMNAME(op), PRIMARITY(op));
}


SEXP nthcdr(SEXP s, int n)
{
    if (isList(s) || isLanguage(s) || isFrame(s) || TYPEOF(s) == DOTSXP ) {
    while( n-- > 0 ) {
        if (s == R_NilValue)
        error("\"nthcdr\" list shorter than %d\n", n);
        s = CDR(s);
    }
    return s;
    }
    else error("\"nthcdr\" needs a list to CDR down\n");
    return R_NilValue;/* for -Wall */
}


/* mfindVarInFrame - look up symbol in a single environment frame. */
static SEXP mfindVarInFrame(SEXP frame, SEXP symbol)
{
    while (frame != R_NilValue) {
    if (TAG(frame) == symbol)
        return frame;
    frame = CDR(frame);
    }
    return R_NilValue;
}


static int isMissing(SEXP symbol, SEXP rho)
{
    SEXP vl, s;

    if ( DDVAL(symbol) )
    s = R_DotsSymbol;
    else
    s = symbol;

    vl = mfindVarInFrame(FRAME(rho), s);
    if (vl != R_NilValue) {
    if ( DDVAL(symbol) ) {
        if (length(CAR(vl)) < DDVAL(symbol) || CAR(vl) == R_MissingArg )
            return 1;
        else
            vl = nthcdr(CAR(vl), DDVAL(symbol)-1);
    }
    if (MISSING(vl) == 1 || CAR(vl) == R_MissingArg)
       return 1;
    if (TYPEOF(CAR(vl)) == PROMSXP &&
          TYPEOF(PREXPR(CAR(vl))) == SYMSXP)
          return isMissing(PREXPR(CAR(vl)), PRENV(CAR(vl)));
    else
       return 0;
    }
    return 0;
}

/* in do_missing rho is the environment that missing was called from */

SEXP do_missing(SEXP call, SEXP op, SEXP args, SEXP rho)
{
    SEXP rval, t, sym, s;

    checkArity(op, args);
    s = sym = CAR(args);
    if (!isSymbol(sym))
    error("\"missing\" illegal use of missing\n");

    if ( DDVAL(sym) ) {
    sym = R_DotsSymbol;
    }
    rval=allocVector(LGLSXP,1);

    t = mfindVarInFrame(FRAME(rho), sym);
    if (t != R_NilValue) {
    if (DDVAL(s)) {
        if (length(CAR(t)) < DDVAL(s)  || CAR(t) == R_MissingArg ) {
            LOGICAL(rval)[0] = 1;
            return rval;
        }
        else
            t = nthcdr(CAR(t), DDVAL(s)-1);
    }
    if (MISSING(t) || CAR(t) == R_MissingArg ) {
        LOGICAL(rval)[0] = 1;
        return rval;
    }
    else goto havebinding;
    }
    else  /* it wasn't an argument to the function */
    error("\"missing\" illegal use of missing\n");

  havebinding:

    t = CAR(t);
    if (TYPEOF(t) != PROMSXP) {
    LOGICAL(rval)[0] = 0;
    return rval;
    }

    if (!isSymbol(PREXPR(t))) LOGICAL(rval)[0] = 0;
    else LOGICAL(rval)[0] = isMissing(PREXPR(t), PRENV(t));
    return rval;
}


SEXP do_nargs(SEXP call, SEXP op, SEXP args, SEXP rho)
{
    SEXP t;
    t = allocVector(INTSXP, 1);
    *INTEGER(t) = NARGS(rho);
    return (t);
}


void setIVector(int * vec, int len, int val)
{
    int i;
    for (i = 0; i < len; i++)
    vec[i] = val;
}


void setRVector(double * vec, int len, double val)
{
    int i;
    for (i = 0; i < len; i++)
    vec[i] = val;
}


void setSVector(SEXP * vec, int len, SEXP val)
{
    int i;
    for (i = 0; i < len; i++)
    vec[i] = val;
}


int isFree(SEXP val)
{
    SEXP t;
    for (t = R_FreeSEXP; t != R_NilValue; t = CAR(t))
    if (val == t)
        return 1;
    return (0);
}


/* Debugging functions (hence the d-prefix). */
/* These are intended to be called interactively from */
/* a debugger such as gdb, so you don't have to remember */
/* the names of the data structure components. */

int dtype(SEXP q)
{
    return((int)TYPEOF(q));
}


SEXP dcar(SEXP l)
{
    return(CAR(l));
}


SEXP dcdr(SEXP l)
{
    return(CDR(l));
}

/* Functions for getting and setting the working directory. */

SEXP
do_getwd(SEXP call, SEXP op, SEXP args, SEXP rho) {
    SEXP rval = R_NilValue;
    char buf[2 * PATH_MAX];

    checkArity(op, args);

#ifdef R_GETCWD
    R_GETCWD(buf, PATH_MAX);
    rval = mkString(buf);
#endif
    return(rval);
}

SEXP
do_setwd(SEXP call, SEXP op, SEXP args, SEXP rho) {
    SEXP s;
    const char *path;

    checkArity(op, args);
    
    s = CAR(args);
    if (!isString(s))
    errorcall(call, "character argument expected\n");
    path = R_ExpandFileName(CHAR(STRING(s)[0]));
    if(chdir(path) < 0)
    errorcall(call, "cannot change working directory\n");
    return(R_NilValue);
}