R/toStan.R

In gppm: Gaussian Process Panel Modeling

validate_toStan <- function(parsedModel, myData) {
  stopifnot(is(parsedModel, "ParsedModel"))
}

.pkgglobalenv <- new.env(parent = emptyenv())

toStan <- function(parsedModel, control) {
  validate_toStan(parsedModel)

  templateLocation <- file.path(system.file(package = "gppm"), "stanTemplate.stan")
  theTemplate <- readChar(templateLocation, file.info(templateLocation)$size)

  theCode <- theTemplate
  paramSect <- paste0("real ", parsedModel$params, ";", collapse = "\n ")
  theCode <- gsub("<parameters>", paramSect, theCode)
  theCode <- gsub("<meanfunction>", parsedModel$mFormula, theCode)
  theCode <- gsub("<covfunction>", parsedModel$kFormula, theCode)
  if (control$stanModel) {
    theModel <- rstan::stan_model(model_code = theCode, auto_write = TRUE)
  }else{
    theModel <- NULL
  }
  return(theModel)
}