R/CytoImageList-class.R

In BodenmillerGroup/cytomapper: Visualization of highly multiplexed imaging data in R

#' @export
#' @rdname CytoImageList
#' @importFrom utils packageVersion
#' @importFrom S4Vectors setValidity2
#' @importClassesFrom S4Vectors SimpleList
setClass(
    Class = "CytoImageList",
    package = "cytomapper",
    slots = c(int_metadata="list"),
    contains = "SimpleList",
    prototype = prototype(
    int_metadata = list(
        version = utils::packageVersion("cytomapper")),
    elementType = "Image_OR_DelayedArray"
    )
)

# Validity checks
#' @importFrom S4Vectors setValidity2
#' @importFrom EBImage colorMode
S4Vectors:::setValidity2(Class = "CytoImageList", .ImageList_validity)

.ImageList_validity <- function(object) {

    msg <- NULL

    # Check if all images have the same number of channels
    dims <- unlist(lapply(object, function(x){
        dim(x)[3]
    }))
    
    if (length(unique(dims)) > 1L) {
        msg <- c(msg, "The images contain different number of channels.\n")
    }

    # Check if all channels have the same names
    if (length(dim(object[[1]])) == 3L) {
        
        cur_names <- dimnames(object[[1]])[[3]]
        errors <- unlist(lapply(object, function(x){
            !identical(cur_names, dimnames(x)[[3]])
        }))
        
        if (sum(errors) > 0) {
            msg <- c(msg, "Not all channels have the same names.")
        }
    }

    # Check if entry names are unique
    if (!is.null(names(object)) &&
        length(unique(names(object))) < length(names(object))) {
        msg <- c(msg, "Only unique entries allowed in a CytoImageList object.")
    }

    # Check if names contain NA or empties
    if (!is.null(names(object)) && (sum(is.na(names(object))) > 0 ||
            sum(names(object) %in% "") > 0)) {
        msg <- c(msg, "Empty or NA names not supported.")
    }

    # Check if channelNames are unique
    if (!is.null(channelNames(object)) &&
        length(unique(channelNames(object))) < length(channelNames(object))) {
        msg <- c(msg, paste("Only unique channels",
                            "allowed in a CytoImageList object."))
    }

    if (length(msg)) { return(msg) }

    return(TRUE)
}