Remove variance partition component and recompute fractions

removeVarPartComponent.R

#' Remove variance partition component and recompute fractions
#' 
#' @param vp result of \code{fitVarPart()}
#' @param exclude array of column names to exclude 
removeVarPartComponent = function(vp, exclude){

	# get column index of exclude
	j = match(exclude, colnames(vp)[-c(1:2)])

	# compute new variance fractions
	df = t(apply(vp[,-c(1:2)], 1, function(x) x[-j] / sum(x[-j])))

	# label with gene and cell type
	df = data.frame(vp[,1:2], df)

	# return
	new("vpDF", S4Vectors::DataFrame(df))
}

# usage
df2 = removeVarPartComponent(vp, exclude = c("Dx", "Sex"))
plotVarPart(df2)