R/calc_Bhat.R

In qtl2pleio: Testing Pleiotropy in Multiparental Populations

#' Calculate estimated allele effects, B matrix
#'
#' @param X dn by df block-diagonal design matrix that incorporates genetic info for d markers. Note that we can use the same marker data twice.
#' @param Sigma_inv dn by dn inverse covariance matrix, often composed as the inverse of \eqn{K \otimes V_g + I_n \otimes V_e}
#' @param Y dn by 1 matrix, ie, a column vector, of d phenotypes' measurements
#' @return a df by 1 matrix of GLS-estimated allele effects
#' @examples
#' X1 <- as.matrix(rbinom(n = 100, size = 1, prob = 1 / 2))
#' X <- gemma2::stagger_mats(X1, X1)
#' Sigma_inv <- diag(200)
#' Y <- runif(200)
#' calc_Bhat(X, Sigma_inv, Y)
#' @export
calc_Bhat <- function(X, Sigma_inv, Y) {
    out <- solve(t(X) %*% Sigma_inv %*% X) %*% t(X) %*% Sigma_inv %*% as.vector(as.matrix(Y))
    return(out)
}