fcm.iterate.R

# FCM iteration function
# returns a table of (iterations x state vectors)
fcm.iterate = function(mat, init, peg, iter, trans, fn, ...) {
  acc = matrix(0, iter, ncol(mat))
  acc[1,] = init
  if(!missing(peg)) {
    acc[1,which(!is.na(peg))] = peg[!is.na(peg)]
  }

  for(i in 2:iter) {
    if(trans) {
      acc[i,] = sapply(t(acc[i-1,]) %*% mat, fn, ...)
    } else {
      acc[i,] = sapply(mat %*% acc[i-1,], fn, ...)
    }
    if(!missing(peg)) {
      acc[i,which(!is.na(peg))] = peg[!is.na(peg)]
    }
  }

  return(acc)
}