从TMB tutorial中,定义.cpp
文件中的目标函数,以便在C ++函数和从R调用的函数之间共享模型数据结构的参数名称和名称。例如, tutorial.cpp
文件:
#include <TMB.hpp> // Links in the TMB libraries
template<class Type>
Type objective_function<Type>::operator() ()
{
DATA_VECTOR(x); // Data vector transmitted from R
PARAMETER(mu); // Parameter value transmitted from R
PARAMETER(sigma); //
Type f; // Declare the "objective function" (neg. log. likelihood)
f = -sum(dnorm(x,mu,sigma,true)); // Use R-style call to normal density
return f;
}
编译后,dyn.load
可以从R调用此函数,但是,您需要知道数据向量的名称为x
,并且有两个参数值mu
和sigma
。是否可以从R?
答案 0 :(得分:3)
我不知道包中有这样做的功能,但下面的功能可能会帮助你;
TMBsearch = function(path,what='parameter',class=FALSE){
if(!missing(what) | length(what)>1) stop("What should be of length one")
if(!(what %in% c('parameter','data','report','sdreport'))) stop("What should be parameter, data, report or sdreport")
text = paste0(paste0(readLines(path), collapse = "\n"), "\n") # read the text from the cpp file
start = unlist(gregexpr(pattern =toupper(what),text)) # starting position
end.poss = unlist(gregexpr(pattern =')',text)) # possible end positions
end = rep(NA,length(start))
for(i in 1:length(start)){end[i] = end.poss[(end.poss-start[i]) > 0][1]} # actual end position
textsub = substring(text,first=start,last=end) # extract the full PARAMETER/DATA_x(...)
found = gsub("[\\(\\)]", "", regmatches(textsub, gregexpr("\\(.*?\\)", textsub))) # get rid of the brackets
if(class & what %in% c('parameter','data')){
dataclass=tolower(gsub("_", "",gsub(".*PARAMETER\\s*|\\(.*", "", textsub)))
dataclass[dataclass=='']="single value"
names(found)=datatype
}
return(found)
}
TMBsearch(path=paste0(filename,'.cpp'), what='parameter')
&#34;什么&#34;可以是&#39;参数&#39;,&#39;数据&#39;&#39;报告&#39;或者&#39; sdreport&#39;但默认情况下我把它作为参数。
附加:如果class == TRUE而不是参数和数据,则将类(矩阵,数组等)作为每个对象的名称。
答案 1 :(得分:1)
感谢@Wave的有用功能。如果what
有多个参数,我只是对它进行了一些改进以检索列表中的所有类型。我的名字中还有一些剩余的空格,所以我还添加了gsub
。
TMBsearch <- function(path, what = c('parameter', 'data', 'report', 'sdreport')) {
res <- lapply(what, function(what) {
# what <- match.arg(what)
text <- paste0(paste0(readLines(path), collapse = "\n"), "\n") # read the text from the cpp file
start <- unlist(gregexpr(pattern = toupper(what), text)) # starting position
end.poss <- unlist(gregexpr(pattern = ')', text)) # possible end positions
end <- rep(NA,length(start))
for (i in 1:length(start)) {end[i] <- end.poss[(end.poss - start[i]) > 0][1]} # actual end position
textsub <- substring(text, first = start, last = end) # extract the full PARAMETER/DATA_x(...) -> might be handy to now whether array or vector or...
found <- gsub("[\\(\\)]", "", regmatches(textsub, gregexpr("\\(.*?\\)", textsub))) # get rid of the brackets
found_nospace <- gsub(" ", "", found) # get rid of the spaces if some left
return(found_nospace)
})
names(res) <- what
res
}