R/plot_card.R

In JerBoon/yahtzR: Play variations of the well-known dice rolling game in an R session

plot_card <- function(game, work.mode) {

  # I could dplyr this if I wanted, but heh..
  gg.dat <- game$table
  gg.dat <- gg.dat[gg.dat$half != 99,]
  gg.dat$score <- as.integer(gg.dat$score)
  names(gg.dat)[names(gg.dat) == {if (work.mode) "section" else "name"}] <- "label"
  gg.dat$label <- factor(gg.dat$label, levels=gg.dat$label, ordered=T)
  #score with NAs coorced to 0 - some charts work better with this
  gg.dat$score.z <- gg.dat$score
  gg.dat$score.z[is.na(gg.dat$score.z)] <- 0

  # ------------------------ randomness ----------------

  # Pick a random theme
  base_theme <- sample(list(ggplot2::theme_classic(),
                            ggplot2::theme_grey(),
                            ggplot2::theme_bw(),
                            ggplot2::theme_minimal()),
                       1)

  # colour palette
  base_palette <- sample(c("Greens","Oranges","Accent","Dark2","Paired","Set1","Set2"),1)

  random_colour <- sample(c("red","green","orange","blue","black","grey"),1)

  # Print a random chart
  style <-  sample(1:7,1)

  # ------------------------

  if (style == 1)
  {
    #Horizontal column plot
    gg <- ggplot2::ggplot(gg.dat, ggplot2::aes(x=label,y=score, fill=factor(half), group=factor(half))) +
      base_theme +
      ggplot2::scale_fill_brewer(palette = base_palette) +
      ggplot2::geom_col() +
      ggplot2::scale_x_discrete(name="", limits=rev(gg.dat$label)) +
      ggplot2::ylab("") +
      ggplot2::theme(legend.position = "none") +
      ggplot2::coord_flip()
  }
  else if (style == 2)
  {
    #Vertical column plot
    gg <- ggplot2::ggplot(gg.dat, ggplot2::aes(x=label,y=score, fill=factor(half), group=factor(half))) +
      base_theme +
      ggplot2::scale_fill_brewer(palette = base_palette) +
      ggplot2::geom_col() +
      ggplot2::xlab("") +
      ggplot2::ylab("") +
      ggplot2::theme(legend.position = "none")

    if (!work.mode)
      gg <- gg +
        ggplot2::theme(axis.text.x = ggplot2::element_text(angle=30,vjust=1,hjust=1))
  }
  else if (style == 3)
  {
    #Vertical column plot with facets
    gg <- ggplot2::ggplot(gg.dat, ggplot2::aes(x=label,y=score.z, fill=factor(half))) +
      base_theme +
      ggplot2::scale_fill_brewer(palette = base_palette) +
      ggplot2::geom_col() +
      ggplot2::xlab("") +
      ggplot2::ylab("") +
      ggplot2::theme(legend.position = "none") +
      ggplot2::facet_grid(~ factor(half, labels = c("Upper","Lower")),
                          scales="free_x", space="free_x")

    if (!work.mode)
      gg <- gg +
      ggplot2::theme(axis.text.x = ggplot2::element_text(angle=30,vjust=1,hjust=1))
  }
  else if (style == 4)
  {
    #Horizontal lollipoplot
    gg <- ggplot2::ggplot(gg.dat, ggplot2::aes(x=label,y=score, fill=factor(half), group=factor(half))) +
      base_theme +
      ggplot2::scale_fill_brewer(palette = base_palette) +
      ggplot2::geom_linerange(ggplot2::aes(ymin=0,ymax=score), alpha=sample(c(0.3,0.5,1),1), colour=random_colour, size=sample(c(1,1.5,2),1)) +
      ggplot2::geom_point(color=random_colour, size=sample(2:4,1)) +
      ggplot2::scale_x_discrete(name="", limits=rev(gg.dat$label)) +
      ggplot2::ylab("") +
      ggplot2::theme(legend.position = "none") +
      ggplot2::coord_flip()
  }
  else if (style == 5)
  {
    #Vertical  lollipop
    gg <- ggplot2::ggplot(gg.dat, ggplot2::aes(x=label,y=score, fill=factor(half), group=factor(half))) +
      base_theme +
      ggplot2::scale_fill_brewer(palette = base_palette) +
      ggplot2::geom_linerange(ggplot2::aes(ymin=0,ymax=score), alpha=sample(c(0.3,0.5,1),1), colour=random_colour, size=sample(c(1,1.5,2),1)) +
      ggplot2::geom_point(color=random_colour, size=sample(2:6,1)) +
      ggplot2::ylab("") +
      ggplot2::theme(legend.position = "none")

    # Occasionally reverse the y axis - because why not :)
    if (runif(1)<0.3)
      gg <- gg +
        ggplot2::scale_x_discrete(name="", position="top") +
        ggplot2::scale_y_reverse() +
        ggplot2::theme(axis.text.x = ggplot2::element_text(angle=90, hjust=0, vjust=0.5))
    else
      gg <- gg +
        ggplot2::scale_x_discrete(name="") +
        ggplot2::theme(axis.text.x = ggplot2::element_text(angle=90, hjust=1, vjust=0.5))

  }
  else if (style %in% c(6,7))
  {
    #Progress worm - inspired by watching cricket! :)

    gg <-
      ggplot2::ggplot(gg.dat[!is.na(gg.dat$score.sequence),],
                      ggplot2::aes(x=factor(score.sequence),y=score.running)) +
      base_theme +
      ggplot2::scale_fill_brewer(palette = base_palette) +
      ggplot2::geom_line(ggplot2::aes(group="1"),size=1.5,alpha=0.7,color=random_colour) +
      ggplot2::geom_point(size=3,color=random_colour) +
      ggplot2::geom_text(ggplot2::aes(label=label),hjust=-0.2, angle=-15) +
      ggplot2::scale_y_continuous(limits=c(0,max(gg.dat$score.running))) +
      ggplot2::scale_x_discrete(expand=ggplot2::expand_scale(add=1)) +
      ggplot2::ylab("Cumulative total") +
      ggplot2::xlab("Turn") +
      ggplot2::theme(legend.position = "none")

  }

  # Standard titles
  if (work.mode)
    gg <- gg + ggplot2::ggtitle(label = "Stats for Y",
                                subtitle = paste("sum(val) =",game$table[game$table$section == "gt", "score"]))
  else
    gg <- gg + ggplot2::ggtitle(label = paste(game$title),
                                subtitle = paste("Total score = ",game$table[game$table$section == "gt", "score"]))

  suppressWarnings(print(gg))
}