R/cov_clm__fitThetaModel__fitThetaCovariance__fitThetaCovarPML__getCovPhiTheta__getAiWiCalculation.R

In phmpacheco-ufjf/clm: A package to deal with complex longitudinal survey data

getAiWiCalculation <- function(sigmaTheta, individuals, derivSigmaThetaList) {
  individuals <- individuals %>%
    rlist::list.map(rlist::list.append(., zi = unlist(lapply(
      derivSigmaThetaList,
      function(derivSigmaThetaj) {
        -t(resi) %*% solve(sigmaTheta) %*% derivSigmaThetaj %*% solve(sigmaTheta) %*% resi
      }
    )))) %>%
    rlist::list.map(rlist::list.append(., ai = wi * zi)) %>%
    rlist::list.select(ai, wi, hi, ji)

  return(individuals)
}