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

In nimbleMacros: Macros Generating 'nimble' Code

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.

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

Value

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

Author(s)

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()

nimbleMacros documentation built on April 3, 2025, 11:38 p.m.