trace_calls.R

trace_calls.R

tracked <- new.env()

makeCounted <- function(fun, loud=FALSE) {
  assign(fun, TRUE, envir=tracked)
  calls <- 0
  countedFun <- function(...) {
    calls <<- .Primitive('+')(1, calls)
    if (loud)
      cat('call made to:', fun, '\n')
    .Primitive(fun)(...)
  }
  return(countedFun)
}

getCounts <- function() {
  tmp <- ls(envir=tracked)
  counts <- cbind(sapply(tmp, function(x)
                         get('calls', envir=environment(get(x)))))
  print(counts)
  invisible(counts)
}


`[` <- makeCounted("[")
`[<-` <- makeCounted("[<-")
`+` <- makeCounted("+")

n <- 100
x <- rnorm(n)
y <- rnorm(n)
r <- numeric(n)

for (i in seq_along(x)) {
  r[i] <- x[i] + y[i]
}

getCounts()

## Explicit calls
## for (i in seq_along(x)) {
##   `[<-`(r, i, `+`(`[`(x, i), `[`(y, i)))
## }