R/distance-functions.R

In harpomaxx/GSgalgoR: An Evolutionary Framework for the Identification and Study of Prognostic Gene Expression Signatures in Cancer

#' Functions to calculate distance matrices using cpu computing
#'
#' @param distancetype a \code{character} that can be either \code{'pearson'}, 
#' \code{'uncentered'}, \code{'spearman'} or \code{'euclidean'}
#' @param x an expression matrix with features as rows and samples as columns
#'
#' @return
#' \code{select_distance(distancetype)} assigns global function 
#' calculate_distance according to the parameters specified
#'
#' \code{calculate_distance_pearson_cpu(x)} returns columnwise pearson 
#' distance calculated using the CPU
#'
#' \code{calculate_distance_uncentered_cpu(x)} returns columnwise 
#' uncentered pearson distance calculated using the CPU
#'
#' \code{calculate_distance_spearman_cpu(x)} returns columnwise 
#' spearman distance calculated using the CPU
#'
#' \code{calculate_distance_euclidean_cpu(x)} returns columnwise 
#' euclidean distance calculated using the CPU
#'
#' @author Martin E Guerrero-Gimenez, \email{mguerrero@mendoza-conicet.gob.ar}
#' @name calculate_distance
#' @examples
#' # load example dataset
#' require(iC10TrainingData)
#' require(pamr)
#'
#' data(train.Exp)
#'
#' calculate_distance <- select_distance(distancetype = "pearson")
#' Dist <- calculate_distance(train.Exp)
#' k <- 4
#' Pam <- cluster_algorithm(Dist, k)
#' table(Pam$cluster)
NULL


#' @rdname calculate_distance
#' @export
calculate_distance_pearson_cpu <- function(x) {
    # print("pearson....")
    mx <- base::colMeans(x)
    x2 <- base::colMeans(x^2)
    mx2 <- mx^2
    sx <- sqrt(x2 - mx2)
    pDist <- ((base::crossprod(x, x) / nrow(x)) - (mx %o% mx)) / sx %o% sx
    stats::as.dist(1 - pDist)
}

#' @rdname calculate_distance
#' @export
calculate_distance_spearman_cpu <- function(x) {
    x <- apply(x, 2, rank)
    mx <- colMeans(x)
    x2 <- colMeans(x^2)
    mx2 <- mx^2
    sx <- sqrt(x2 - mx2)
    pDist <- ((crossprod(x, x) / nrow(x)) - (mx %o% mx)) / sx %o% sx
    stats::as.dist(1 - pDist)
}

#' @rdname calculate_distance
#' @export
calculate_distance_uncentered_cpu <- function(x) {
    mgpu <- t(x)
    d2 <- matrix(1, ncol = 1, nrow = dim(mgpu)[2])
    a1 <- tcrossprod(mgpu, mgpu)
    a2 <- mgpu^2 %*% d2
    pDist <- a1 / sqrt(tcrossprod(a2, a2))
    stats::as.dist(1 - pDist)
}

#' @rdname calculate_distance
#' @export
calculate_distance_euclidean_cpu <- function(x) {
    d <- stats::dist(t(x), method = "euclidean")
}

#' @rdname calculate_distance
#' @export
select_distance <- function(distancetype = "pearson") {
    distancetype <- match.arg(distancetype, c("pearson", "uncentered", 
                                                "euclidean", "spearman"))

    computingtype <- "Using CPU for computing "
    # Centered pearson distance
    if (distancetype == "pearson") {
        calculate_distance <- calculate_distance_pearson_cpu
    }
    # Spearman distance
    if (distancetype == "spearman") {
        calculate_distance <- calculate_distance_spearman_cpu
    }
    if (distancetype == "uncentered") {
        calculate_distance <- calculate_distance_uncentered_cpu
    }
        if (distancetype == "euclidean") {
            calculate_distance <- calculate_distance_euclidean_cpu
    }
    
    message(computingtype, distancetype, " distance")
    # assign("calculate_distance",calculate_distance,envir=.GlobalEnv)
    # return(invisible(calculate_distance))
    return(calculate_distance)
}