R/predict.bbase.interaction.factor.by.curve.bs.R

In DDPstar: Density Regression via Dirichlet Process Mixtures of Normal Structured Additive Regression Models

predict_bbase.interaction.factor.by.curve.bs <-
function(object, newx, newfactor) {
	newfactor <- droplevels(newfactor)
	factor.levels <- levels(newfactor)

	if(length(factor.levels) > 1) {
		ord <- NULL
		for(i in 1:length(factor.levels)) {
			ord <- c(ord, which(newfactor == factor.levels[i]))
		}
	} else {
		ord <- 1:length(newx)
	}
	interaction.smooth.part <- list()
	for(i in 1:length(factor.levels)) {
		interaction.smooth.part[[i]] <- predict_bbase.bs(object[[factor.levels[i]]], newx[newfactor == factor.levels[i]])
	}
	aux <- as.matrix(bdiag(interaction.smooth.part))
	B <- aux[order(ord),]
	B
}