trend-comparison-shinyapp/plot_functions.R

In COMHIS/ecco-shinyapps: English Short Title Catalogue (ESTC) Metadata Toolkit

library(reshape2)


get_yearly_freqs_molten_df <- function(query_sets_list) {
  year <- (query_sets_list[[1]]$data)$year
  graphdata_df <- data.frame(year)

  for (query_set in query_sets_list) {
    colname <- query_set$term
    freqs <- (query_set$data)$frequency
    graphdata_df[colname] <- freqs
  }

  molten_df <- reshape2::melt(data = graphdata_df, id.vars = "year")
  return(molten_df)
}


plot_titlecount_relative <- function(title = "placeholder",
                                     yearly_title_frequencies_list, style = "smooth", plot_colour = "Dark2") {

  graphdata_df <- get_yearly_freqs_molten_df(yearly_title_frequencies_list)

  if (style == "smooth") {
    geom <- geom_smooth(method = "loess", span = 0.2, se = FALSE, , size = 1.7)
  } else if (style == "very_smooth") {
    geom <- stat_smooth(method = 'lm',  formula = y ~ poly(x,2), se = FALSE, size = 1.7)
  } else if (style == "not_smooth") {
    geom <- geom_line(size = 1.7)
  } else {
    graphdata_df$value[is.na(graphdata_df$value)] <- 0
    geom <- geom_ribbon(aes(ymin=0, ymax=value, fill = variable, colour = NA), position = "stack")
    # geom <- geom_density(aes(ymin=0, ymax=value, fill = variable, colour = NA), position = "stack")
  }

  plot <- ggplot(data = graphdata_df,
                 aes(x = year, y = value, colour = variable)) + 
    geom +
    labs(title = title,
         x = "Year",
         y = "Frequency",
         colour = "Search term") +
    scale_x_continuous(breaks = 
      round(seq(min(graphdata_df$year), max(graphdata_df$year + 4), 5), 1)) +
    scale_y_continuous(limits = c(min(graphdata_df$value, na.rm = TRUE),
                                  max(graphdata_df$value, na.rm = TRUE))) + 
    theme_linedraw() +
    theme(text = element_text(size = 20),
          axis.text.x = element_text(angle = -90),
          axis.title.x = element_blank(),
          legend.title = element_blank(),
          legend.position = "bottom",
          legend.text = element_text(size = 20),
          legend.key.size = unit(4,"line")) +
    scale_colour_brewer(type = "qual", palette = plot_colour) +
    scale_fill_brewer(type = "qual", palette = plot_colour)
  if (style == "very_smooth") {
    plot <- plot + geom_point()
  }
  return(plot)
}