R/print.etm.R

In etm: Empirical Transition Matrix

print.etm <- function(x, covariance = FALSE, whole = TRUE, ...) {

    if (!inherits(x, "etm"))
        stop("'x' must be of class 'etm'")

    absorb <- setdiff(as.character(x$trans$to), as.character(x$trans$from))
    transient <- unique(x$state.names[!(x$state.names %in% absorb)])

    cat(paste("Multistate model with", length(transient), "transient state(s)\n",
              "and", length(absorb), "absorbing state(s)\n\n", sep = " "))

    cat("Possible transitions:\n")
    print(x$trans, row.names = FALSE)
    cat("\n")

   if (is.null(x$strata)) {

        cat(paste("Estimate of P(", x$s, ", ", x$t, ")\n", sep = ""))
        print(x$est[, , dim(x$est)[3]]); cat("\n")

        if (!is.null(x$cov) & covariance == TRUE) {
            if (whole) {
                cat(paste("Estimate of cov(P(", x$s, ", ", x$t, "))\n", sep = ""))
                print(x$cov[, , dim(x$cov)[3]])
            }
            else {
                cov <- x$cov[, , dim(x$cov)[3]][rowSums(x$cov[, , dim(x$cov)[3]]) != 0, ]
                cova <- cov[, colSums(cov) != 0]
                cat(paste("Estimate of cov(P(", x$s, ", ", x$t, "))\n", sep = ""))
                print(cova)
            }
        }

    } else {

        for (i in seq_along(x$strata)) {

            cat("\t", x$strata[i], ":\n\n")

            cat(paste("Estimate of P(", x[i]$s, ", ", x[i]$t, ")\n", sep = ""))
            print(x[i]$est[, , dim(x[i]$est)[3]]); cat("\n")

            if (!is.null(x$cov) & covariance == TRUE) {
                if (whole) {
                    cat(paste("Estimate of cov(P(", x[i]$s, ", ", x[i]$t, "))\n", sep = ""))
                    print(x[i]$cov[, , dim(x[i]$cov)[3]])
                }
                else {
                    cov <- x[i]$cov[, , dim(x[i]$cov)[3]][rowSums(x[i]$cov[, , dim(x[i]$cov)[3]]) != 0, ]
                    cova <- cov[, colSums(cov) != 0]
                    cat(paste("Estimate of cov(P(", x[i]$s, ", ", x[i]$t, "))\n", sep = ""))
                    print(cova)
                }
            }

        }
    }

    invisible()

}

### etmStratified
## print.etm.stratified <- function(x, covariance = FALSE, whole = TRUE, ...) {

##     if (!inherits(x, "etm.stratified"))
##         stop("'x' must be of class 'etm.stratified'")

##     absorb <- setdiff(levels(x$trans$to), levels(x$trans$from))
##     transient <- unique(x$state.names[!(x$state.names %in% absorb)])

##     cat(paste("Multistate model with", length(transient), "transient state(s)\n",
##               "and", length(absorb), "absorbing state(s)\n\n", sep = " "))

##     cat("Possible transitions:\n")
##     print(x$trans, row.names = FALSE)
##     cat("\n")

##     for (i in seq_along(x$strata)) {

##         cat(x$strata[i], ":\n\n")

##         cat(paste("Estimate of P(", x[i]$s, ", ", x[i]$t, ")\n", sep = ""))
##         print(x[i]$est[, , dim(x[i]$est)[3]]); cat("\n")

##         if (!is.null(x$cov) & covariance == TRUE) {
##             if (whole) {
##                 cat(paste("Estimate of cov(P(", x[i]$s, ", ", x[i]$t, "))\n", sep = ""))
##                 print(x[i]$cov[, , dim(x[i]$cov)[3]])
##             }
##             else {
##                 cov <- x[i]$cov[, , dim(x[i]$cov)[3]][rowSums(x[i]$cov[, , dim(x[i]$cov)[3]]) != 0, ]
##                 cova <- cov[, colSums(cov) != 0]
##                 cat(paste("Estimate of cov(P(", x[i]$s, ", ", x[i]$t, "))\n", sep = ""))
##                 print(cova)
##             }
##         }

##     }

##     invisible()

## }