nimbleMacros (version 0.1.1)

FORLOOP: Macro to build for loop(s) from code with index ranges in brackets

Description

This macro takes a line of NIMBLE model code with index ranges inside brackets on either the left-hand side of a declaration or both the left- and right-hand sides of a declaration and constructs a corresponding for loop or series of nested for loops.

Value

NIMBLE code for a for loop or series of nested for loops.

Arguments

code

The right-hand side of a parameter declaration

modelInfo

Used internally by nimbleMacros; a list of model information such as constants and dimensions

.env

Used internally by nimbleMacros; the environment where the model was created

Author

Ken Kellner and Perry de Valpine

Examples

code <- nimbleCode({
   y[1:n, 1:2, 1] ~ FORLOOP(dnorm(mu[1:n], sigma))
   mu[1:n] <- FORLOOP(beta[1] + beta[2]*x[1:n])
})

mod <- nimbleModel(code, constants = list(n=10))
mod$getCode()