R/dPlotCoFunction.R

In Theorell/DepecheR: Determination of essential phenotypic elements of clusters in high-dimensional entities

# This function is used for all major plots in the package. Hence the functions
# dColorPlot, dDensityPlot, dResidualPlot, dSplsda, and dWilcox uses it.
# "Unique" parameters
# colorVariable: the color associated with each point for plotting.
# For information on the other parameters, see any of the plotting functions.
dPlotCoFunction <- function(colorVariable, plotName, xYData, title = FALSE,
                            densContour, bandColor, dotSize, plotDir,
                            createOutput = TRUE) {
    colnames(xYData) <- c("V1", "V2")

    # Create the density matrix for xYData.
    if (is.logical(densContour)) {
        if (densContour) {
            densContour <- dContours(xYData)
        }
    }

    if (length(densContour) > 1) {
        xlim <- c(min(densContour[[1]]), max(densContour[[1]]))
        ylim <- c(min(densContour[[2]]), max(densContour[[2]]))
    } else {
        range1 <- range(xYData[, 1])
        sideDist1 <- 0.05 * (range1[2] - range1[1])
        range2 <- range(xYData[, 2])
        sideDist2 <- 0.05 * (range2[2] - range2[1])

        xlim <- c(
            range1[1] - sideDist1,
            range1[2] + sideDist1
        )
        ylim <- c(
            range2[1] - sideDist2,
            range2[2] + sideDist2
        )
    }

    # Plot it
    main <- ifelse(title, plotName, "")



    if (createOutput) {
        png(file.path(plotDir, paste0(plotName, ".png")),
            width = 2500,
            height = 2500, units = "px", bg = "transparent"
        )

        plot(V2 ~ V1,
            data = xYData, main = main, pch = 20, cex = dotSize,
            cex.main = 5, col = colorVariable, xlim = xlim, ylim = ylim,
            axes = FALSE, xaxs = "i", yaxs = "i"
        )

        if (length(densContour) > 1) {
            par(fig = c(0, 1, 0, 1), mar = c(6, 4.5, 4.5, 2.5), new = TRUE)
            contour(
                x = densContour$x, y = densContour$y, z = densContour$z,
                xlim = xlim, ylim = ylim, nlevels = 10, col = bandColor,
                lwd = 8, drawlabels = FALSE, axes = FALSE, xaxs = "i",
                yaxs = "i"
            )
        }
        dev.off()
    }
}