R/immer_create_design_matrix_A_PCM2.R

In alexanderrobitzsch/immer: Item Response Models for Multiple Ratings

## File Name: immer_create_design_matrix_A_PCM2.R
## File Version: 0.061

immer_create_design_matrix_A_PCM2 <- function(maxK, I, K)
{
    NX <- sum(maxK)
    A <- array(0, dim=c(I,K,NX) )
    dimnames(A)[[1]] <- names(maxK)
    dimnames(A)[[2]] <- paste0('Cat', 1L:K)
    dimnames(A)[[3]] <- paste0('parm', 1L:NX)

    pp <- 1

    #---- item difficulties
    for (ii in 1L:I){
        start_ii <- pp
        A[ii,seq(1,maxK[ii]), pp] <- seq(1, maxK[ii])
        dimnames(A)[[3]][pp] <- names(maxK)[ii]
        pp <- pp + 1
    }

    #---- item-step parameters
    for (ii in 1L:I){
        start_ii <- pp
        if (maxK[ii] > 1){
            for (kk in 1L:(maxK[ii]-1) ){
                A[ii,kk, seq(start_ii,pp)] <- 1
                dimnames(A)[[3]][pp] <- paste0( names(maxK)[ii], '_Step', kk )
                pp <- pp + 1
            }
        }
    }

    pp <- 1
    for (ii in 1L:I){
        start_ii <- pp
        for (kk in 1L:maxK[ii] ){
            A[ii,kk, seq(start_ii,pp)] <- 1
            dimnames(A)[[3]][pp] <- paste0( names(maxK)[ii], '_Cat', kk )
            pp <- pp + 1
        }
    }
    return(A)
}