R/abundanceAnalysis.R

In malhamdoosh/abseqR: Reporting and data analysis functionalities for Rep-Seq datasets of antibody libraries

#' Abundance distribution
#'
#' @import ggplot2 utils
#' @include util.R
#' @include distributions.R
#'
#' @param files list type. list of files in abundance directory
#' @param sampleNames vector type. 1-1 correspondance to files
#' @param outputDir string type.
#' @param skipDgene logical type. Skip D germline abundance plot if TRUE.
#' @param .save logical type. Save Rdata ggplot item
#'
#' @return None
.abundancePlot <- function(files,
                           sampleNames,
                           outputDir,
                           skipDgene = FALSE,
                           .save = TRUE) {
    vdj <- if (skipDgene)
        c("v", "j")
    else
        c("v", "d", "j")

    for (expression in c("family", "gene")) {
        for (gene in vdj) {
            # correction, J has no "gene" but rather variant
            if (gene == 'j' && expression == 'gene') {
                expression <- 'variant'
            }
            reg <- paste0(".*_ig",
                          gene,
                          "_dist_",
                          expression,
                          "_level\\.csv(\\.gz)?$")
            selectedFiles <- files[grepl(reg, files)]
            if (length(selectedFiles)) {
                vert <- .checkVert(selectedFiles[1])
                if (vert) {
                    width <- V_WIDTH
                    height <- V_HEIGHT
                } else {
                    width <- H_WIDTH
                    height <- H_HEIGHT
                }
                mashedName <- paste(sampleNames, collapse = ", ")
                dataframes <- lapply(
                    selectedFiles,
                    read.csv,
                    stringsAsFactors = FALSE,
                    skip = 1
                )
                subtitle <- paste("Total is",
                                  paste(lapply(selectedFiles, function(x) {
                                      as.integer(.getTotal(x))
                                  }), collapse = ", "))
                p <- .plotDist(
                    dataframes,
                    sampleNames,
                    paste0(
                        "IG",
                        toupper(gene),
                        " abundance in ",
                        mashedName
                    ),
                    vert,
                    subs = subtitle
                )
                filename <-
                    file.path(
                        outputDir,
                        paste0(
                            paste(sampleNames, collapse = "_"),
                            "_ig",
                            gene,
                            "_dist_",
                            expression,
                            "_level.png"
                        )
                    )
                ggsave(
                    filename,
                    plot = p,
                    width = width,
                    height = height
                )
                .saveAs(.save, filename, plot = p)
            }
        }
    }
}

#' V-J association plot
#'
#' @import circlize
#' @import RColorBrewer
#' @import stats
#' @import grDevices
#' @importFrom graphics title
#'
#' @param sampleName string type
#' @param path string type. Path to _vjassoc.csv
#' @param outputdir string type
#'
#' @return None
.plotCirclize <- function(sampleName, path, outputdir) {
    filename <- file.path(path, paste0(sampleName, "_vjassoc.csv"))

    message("Plotting V-J association for ", sampleName)

    if (file.exists(filename)) {
        df <- read.csv(filename)

        # output file
        outputFileName <- file.path(outputdir, gsub(".csv",
                                                    ".png",
                                                    basename(filename),
                                                    fixed = TRUE))

        png(
            outputFileName,
            width = 3.25,
            height = 3.25,
            units = "in",
            res = 800,
            pointsize = 4
        )

        # circos theme setup
        #if (length(unique(df[[1]]))-1 < 8 && length(unique(df[[2]]))-1 < 8)  {
        #    circos.par(gap.after = c(rep(5, length(unique(df[[1]]))-1), 15,
        #                             rep(5, length(unique(df[[2]]))-1), 15))
        #}

        row = rep(brewer.pal(12, "Paired"), nrow(df))[seq_along(unique(df[[1]]))]
        col = rep(rev(brewer.pal(12, "Paired")),
                  nrow(df))[seq_along(unique(df[[2]]))]

        chordDiagram(
            df,
            annotationTrack = "grid",
            preAllocateTracks = list(track.height = 0.2),
            grid.col = c(row, col)
        )
        title(sampleName, cex = 8)
        circos.trackPlotRegion(
            track.index = 1,
            bg.border = NA,
            panel.fun = function(x, y) {
                sector.name =
                    get.cell.meta.data("sector.index")
                xlim = get.cell.meta.data("xlim")
                ylim = get.cell.meta.data("ylim")
                circos.text(
                    mean(xlim),
                    ylim[1],
                    sector.name,
                    facing = "clockwise",
                    adj = c(0, 1.5)
                )
            }
        )
        circos.clear()
        dev.off()
    } else {
        warning("Could not find file ", filename,
                ". Skipping V-J association plot")
    }
}


#' Plots a plotly heatmap from provided matrix
#'
#' @import reshape2
#' @import ggplot2
#' @importFrom plotly subplot plot_ly plotly_empty layout
#'
#' @param m matrix type
#' @param title character type
#' @param xlabel character type
#' @param ylabel character type
#'
#' @return list with keys: static and interactive (ggplot2 object and plotly
#' object respectivelyb)
.hmFromMatrix <- function(m,
                          title,
                          xlabel = "",
                          ylabel = "") {
    x <- colSums(m)
    y <- rowSums(m)

    xax <- list(title = xlabel)
    yax <- list(title = ylabel)

    interactive <-
        suppressMessages(
            subplot(
                plot_ly(
                    x = as.numeric(colnames(m)),
                    y = x,
                    type = "bar",
                    color = I("DarkBlue")
                ),
                plotly_empty(type = "scatter"),
                plot_ly(
                    x = as.numeric(colnames(m)),
                    y = as.numeric(rownames(m)),
                    z = m,
                    type = "heatmap"
                ),
                plot_ly(
                    y = as.numeric(rownames(m)),
                    x = y,
                    type = "bar",
                    orientation = "h",
                    color = I("DarkBlue")
                ),
                nrows = 2,
                heights = c(0.2, 0.8),
                widths = c(0.8, 0.2),
                margin = 0,
                shareX = TRUE,
                shareY = TRUE,
                titleX = FALSE,
                titleY = FALSE
            )
        )

    static <- ggplot(melt(m), aes(x = Var2, y = Var1)) +
        geom_tile(aes(fill = value)) +
        labs(
            title = title,
            x = xlabel,
            y = ylabel,
            fill = "percent"
        ) +
        theme_bw()

    return(list(
        interactive =  plotly::layout(
            interactive,
            title = title,
            showlegend = FALSE,
            xaxis = xax,
            yaxis = yax
        ),
        static = static
    ))
}


#' Plots all 5 alignment quality heatmaps
#'
#' @description Plots alignment quality vs:
#' \itemize{
#'   \item{mismatches}
#'   \item{gaps}
#'   \item{bitscore}
#'   \item{percent identity}
#'   \item{subject start}
#' }
#'
#' @param abundanceDirectory character type.
#' fully qualified path to abundance directory
#' @param sampleName character type. sample name
#'
#' @include util.R
#' @import utils
#'
#' @return list of ggplotly heatmaps
.alignQualityHeatMaps <- function(abundanceDirectory, sampleName) {
    qualityMeasure <-
        c("mismatches", "gaps", "bitscore", "identity", "start")
    lapply(qualityMeasure, function(qual) {
        heatmapFile <-
            file.path(
                abundanceDirectory,
                paste0(sampleName, "_igv_align_quality_", qual, "_hm.tsv")
            )
        if (file.exists(heatmapFile)) {
            mat <-
                as.matrix(read.table(
                    heatmapFile,
                    skip = 1,
                    check.names = FALSE
                ))
            xlabel <- "Alignment Length"
            if (qual == "identity") {
                qual <- "%Identity"
            } else if (qual == "start") {
                qual <- "Subject start"
                xlabel <- "Query start"
            }
            totalCount <- .getTotal(heatmapFile)
            p <- .hmFromMatrix(
                mat,
                title = paste(
                    "Alignment Quality of",
                    sampleName,
                    "\nTotal is",
                    totalCount
                ),
                xlabel = xlabel,
                ylabel = qual
            )
            .saveAs(TRUE, heatmapFile, plot = p[["interactive"]])
            .saveAs(TRUE,
                    sub(".tsv", "_static.tsv", heatmapFile, fixed = TRUE),
                    plot = p[["static"]])
            ggsave(
                sub(".tsv", "_static.png", heatmapFile, fixed = TRUE),
                plot = p[["static"]],
                width = V_WIDTH,
                height = V_HEIGHT
            )
            return(p)
        } else {
            warning("Could not find ", heatmapFile, " for sample ", sampleName)
        }
    })
}


#' Conducts abundance analysis
#'
#' @import ggplot2 utils
#' @include util.R
#' @include distributions.R
#'
#' @param abundanceDirectories list type. List of sample directories
#' @param abunOut string type. Output directory
#' @param sampleNames vector type. 1-1 correspondence with abundanceDirectories
#' @param combinedNames string type. Title "combined" sample names
#' @param mashedNames string type. File "mashed" names - avoid special chars
#' @param skipDgene logical type. Skip D gene plots?
#' @param .save logical type. Save ggplot as Rdata
#'
#' @return None
.abundanceAnalysis <- function(abundanceDirectories,
                               abunOut,
                               sampleNames,
                               combinedNames,
                               mashedNames,
                               skipDgene = FALSE,
                               .save = TRUE) {
    # where to find the files
    # 3 each from V and D, then 2 from J (no gene) or 5 (exclude the 3 from D)
    searchFiles <-
        .listFilesInOrder(path = abundanceDirectories,
                          pattern = ".*ig[vdj]_dist_[family|gene|variant].*\\.csv(\\.gz)?$",
                          expectedRet = c(8, 5))
    if (length(searchFiles) > 0) {
        message("Plotting V(D)J abundance distribution for ", combinedNames)
        .abundancePlot(searchFiles,
                       sampleNames,
                       abunOut,
                       skipDgene = skipDgene)
    } else {
        warning("Could not find V(D)J abundance CSV files in ",
                paste(abundanceDirectories, collapse = ","))
    }

    # plot igv mismatches distribution
    abunIgvMismatchFiles <-
        .listFilesInOrder(path = abundanceDirectories,
                          pattern = ".*_igv_mismatches_dist\\.csv(\\.gz)?$")

    if (length(abunIgvMismatchFiles) > 0) {
        message("Plotting IGV mismatches distribution for ", combinedNames)
        subtitle <- paste("Total is",
                          paste(lapply(abunIgvMismatchFiles, function(x) {
                              as.integer(.getTotal(x))
                          }), collapse = ", "))

        abunIgvMismatches <- .plotDist(
            lapply(abunIgvMismatchFiles, read.csv, skip = 1),
            sampleNames,
            paste("Number of mismatches in V gene in", combinedNames),
            .checkVert(abunIgvMismatchFiles[[1]]),
            subs = subtitle
        )
        saveName <-
            file.path(abunOut,
                      paste0(mashedNames, "_igv_mismatches_dist.png"))
        ggsave(saveName,
               plot = abunIgvMismatches,
               width = V_WIDTH,
               height = V_HEIGHT)
        .saveAs(.save, saveName, plot = abunIgvMismatches)
    } else {
        warning("Could not find IGV mismatches distribution CSV files in ",
                paste(abundanceDirectories, collapse = ","))
    }

    # plot igv gaps distribution
    abunIgvGapsFiles <-
        .listFilesInOrder(path = abundanceDirectories,
                          pattern = ".*_igv_gaps_dist\\.csv(\\.gz)?$")
    if (length(abunIgvGapsFiles) > 0) {
        message("Plotting IGV indels distribution for ", combinedNames)
        subtitle <- paste("Total is",
                          paste(lapply(abunIgvGapsFiles, function(x) {
                              as.integer(.getTotal(x))
                          }), collapse = ", "))
        abunIgvGaps <- .plotDist(
            lapply(abunIgvGapsFiles, read.csv, skip = 1),
            sampleNames,
            paste("Number of gaps in V gene in ", combinedNames),
            .checkVert(abunIgvGapsFiles[[1]]),
            subs = subtitle
        )
        saveName <-
            file.path(abunOut, paste0(mashedNames, "_igv_gaps_dist.png"))
        ggsave(saveName,
               plot = abunIgvGaps,
               width = V_WIDTH,
               height = V_HEIGHT)
        .saveAs(.save, saveName, plot = abunIgvGaps)
    } else {
        warning("Could not find IGV indels distribution CSV files in ",
                paste(abundanceDirectories, collapse = ","))
    }

    if (length(sampleNames) == 1) {
        # we can plot circlize if there's only one sample
        .plotCirclize(sampleNames[1], abundanceDirectories[[1]], abunOut)
        # alignment quality heatmaps
        .alignQualityHeatMaps(abundanceDirectories, sampleNames[1])
    }
}