plots.R

options(warn=-1)

yxBox <- function(y, x, title = NA)
{
    t <- title
    p <- wsj_pal(palette = 'colors6')(6)

    b <- qplot(x = x, y = y) +
        geom_boxplot(fill = p[4]) +
        theme_wsj() +
        scale_color_wsj() +
        theme(legend.position = 'none',
              text = element_text(size = 16, family = family),
              title = element_text(size = 24, family = family),
              plot.title = element_text(hjust = 0.5),
              axis.text.x = element_text(angle = 90, hjust = 1)) +
        ggtitle(t)

    b
}

yxScatter <- function(y, x = NA, title = NA)
{
    if(is.na(title))
    {
        if(is.na(x))
        {
            t <- tail(strsplit(deparse(substitute(y)), '\\$')[[1]], 1)
        }
        else
        {
            t <- paste(tail(strsplit(deparse(substitute(x)), '\\$')[[1]], 1), 'vs.',
                       tail(strsplit(deparse(substitute(y)), '\\$')[[1]], 1))
        }
    }
    else
    {
        t <- title
    }

    if(is.na(x))
    {
        n <- 1:length(y)
    }
    else
    {
        n <- x
    }

    p <- qplot(n, y) +
        geom_point(size = 4, alpha = 0.5, color = Cancer$color) +
        theme_wsj() +
        theme(legend.position = 'none',
              text = element_text(size = 12, family = family),
              title = element_text(size = 16, family = family),
              plot.title = element_text(hjust = 0.5)) +
        ggtitle(t)

    if(is.na(x))
    {
        p
    }
    else
    {
        p + geom_smooth(method = 'lm', lwd = 3, alpha = 0.5, color = 'red', fullrange = T)
    }
}

boxHist <- function(v, title = NA, subtitle = NA)
{
    if(is.na(title))
    {
        t <- tail(strsplit(deparse(substitute(v)), '\\$')[[1]], 1)
    }
    else
    {
        t <- title
    }

    vLen    <- length(v)
    vMin    <- min(v, na.rm = T)
    vMax    <- max(v, na.rm = T)
    vBins   <- sqrt(vLen)
    vMean   <- mean(v, na.rm = T)
    vMedian <- median(v, na.rm = T)
    vSD     <- sd(v)

    p       <- wsj_pal(palette = 'colors6')(6)

    b <- qplot(x = 1, y = v) +
        geom_boxplot(fill = p[4]) +
        coord_flip() +
        theme_wsj() +
        scale_color_wsj() +
        theme(legend.position = 'none',
              text = element_text(size = 16, family = family),
              title = element_text(size = 24, family = family),
              plot.title = element_text(hjust = 0.5),
              axis.text.y = element_blank()) +
        ggtitle(t)

    h <- qplot(x = v, geom = 'histogram', bins = vBins) +
        geom_histogram(fill= p[4], bins = vBins) +
       #geom_density(aes(y = 2500 * ..count..), fill = p[2], alpha = 0.1) +
        geom_vline(aes(xintercept = vMean),   size = 1, color = p[1], linetype = 'dashed') +
        geom_vline(aes(xintercept = vMedian), size = 1, color = p[1], linetype = 'solid') +
        theme_wsj() +
        scale_color_wsj() +
        theme(legend.position = 'none',
              title = element_blank(),
              plot.title = element_blank())

    ggarrange(b, h, heights = c(2,3), align = 'hv', ncol = 1, nrow = 2)
}