R/crea_theme.R

In energyandcleanair/rcrea: Query And Plot Air Quality And Weather Data from CREA Database

theme_crea <- function(base_size=11,
                       family='OpenSans',
                       ...) {
  (ggthemes::theme_calc(base_size=base_size) +
     ggplot2::theme(#title = element_text(family='SourceSansPro'),
       plot.title = element_text(size=rel(1.5), face='bold', color=unname(pal_crea['Dark.blue'])),
       plot.subtitle = element_text(color='#060606'),
       plot.caption = element_text(#face='italic',
                                   color=pal_crea['Dark.blue'],
                                   hjust = 0,
                                   lineheight = 1.1),
       strip.background = element_rect(fill=pal_crea['Blue'],
                                       linetype=0),
       plot.background = element_rect(color='white', fill='white'),
       # text = element_text(family=family),
       panel.grid.major.y = element_line(size=0.1),
       panel.border = element_rect(colour="white"),
       axis.line = element_line(size=0.1),
       ...))
}


theme_crea_old <- function(base_size=11, ...) {
  (ggthemes::theme_calc(base_size=base_size) +
     ggplot2::theme(#title = element_text(family='SourceSansPro'),
       plot.title = element_text(size=rel(1.5), face='bold', color=unname(pal_crea['Dark.blue'])),
       plot.subtitle = element_text(face='italic', color='black'),
       plot.caption = element_text(face='italic', color=pal_crea['Dark.blue']),
       strip.background = element_rect(fill=pal_crea['Blue'],
                                       linetype=0),
       plot.background = element_rect(color='white', fill='white'),
       ...))
}


getpal <- function(x) {
  x %<>% textConnection %>% readLines
  x %<>% trimws()
  x %>% gsub('.*#', '#', .) -> pal
  x %>% gsub(' #.*', '', .) %>% make.names -> names(pal)
  return(pal)
}

pal_crea <-
  "Dark blue #35416C
  Blue #8cc9D0
  Light blue #cce7eb
  Turquoise #27a59c
  Green #75b44c
  Dark gray #333333
  Light gray #cacaca
  Yellow #fff2cc
  Orange #f6b26b
  Red #cc0000
  Dark red #990000
  Black #000000
  Dark violet #351c75
  Dark purple #741b47" %>% getpal

pal_crea.change <-
  "Dark blue #35416C
Blue #8cc9D0
Light blue #cce7eb
Yellow #fff2cc
Orange #f6b26b
Red #cc0000
Dark red #990000" %>% getpal

 pal_crea.heatmap <-
  "Black #000000
Dark violet #351c75
Dark purple #741b47
Dark red #990000
Red #cc0000
Orange #f6b26b
Yellow #fff2cc" %>% getpal

pal_crea.dramatic <- pal_crea.heatmap[c(4,2,3,1,5,6,7)]

pal_crea.electricity <- c(
  "Solar"= pal_crea["Yellow"],
  "Wind"= pal_crea["Blue"],
  "Other Renewables"= pal_crea["Light.blue"],
  "Hydro"= pal_crea["Dark.blue"],
  "Biomass"= pal_crea["Greem"],
  "Nuclear"= pal_crea["Dark.red"],
  "Fossil Gas"= pal_crea["Light.gray"],
  "Coal"=pal_crea["Dark.gray"]
)

crea_palettes <- list(CREA = pal_crea,
                      change = pal_crea.change,
                      heatmap = pal_crea.heatmap,
                      dramatic = pal_crea.dramatic,
                      electricity = pal_crea.electricity)

getcols <- function(pal, alpha=1, col.index=T, darken=0) {
  cols <- paste0(crea_palettes[[pal]][col.index],
         format(as.hexmode(round(alpha*255, 0)), width=2))

  if(darken > 0) cols <- colorspace::darken(cols, amount=darken)
  if(darken < 0) cols <- colorspace::lighten(cols, amount=-darken)
  return(cols)
}

makepal <- function(pal, alpha=1, col.index=T, darken=0) {
  cols <- getcols(pal=pal, alpha=alpha, col.index=col.index, darken=darken)
  scales::manual_pal(unname(cols))
}

makegrad <- function(pal, alpha=1, bias=1, reverse.order=F, col.index=T, darken=0) {
  cols <- getcols(pal=pal, alpha=alpha, col.index=col.index, darken=darken)
  if(reverse.order) cols <- rev(cols)
  function(x) { scales::col_numeric(unname(cols), c(0,1))(x^bias) }
}


scale_color_crea_d <- function(palette = "CREA", alpha = 1, col.index=T, darken=0, ...) {
  ggplot2::discrete_scale("colour", scale_name=palette, palette = makepal(palette, alpha, col.index, darken), ...)
}

scale_fill_crea_d <- function(palette = "CREA", alpha = 1, col.index=T, darken=0, ...) {
  ggplot2::discrete_scale("fill", scale_name=palette, palette = makepal(palette, alpha, col.index, darken), ...)
}

scale_color_crea_c <- function(palette = "CREA", alpha = 1, reverse.order=F, bias=1, col.index = T, darken = 0, ...) {
  ggplot2::continuous_scale("colour", scale_name=palette, palette = makegrad(palette, alpha=alpha, reverse.order=reverse.order, bias=bias, col.index = col.index, darken = darken), ...)
}

scale_fill_crea_c <- function(palette = "CREA", alpha = 1, reverse.order=F, bias=1, col.index = T, darken = 0, ...) {
  ggplot2::continuous_scale("fill", scale_name=palette, palette = makegrad(palette, alpha=alpha, reverse.order=reverse.order, bias=bias, col.index = col.index, darken = darken), ...)
}

# Terrible name, but already use in many places
scale_y_crea_zero <- function(mult_high=0.1){
  ggplot2::scale_y_continuous(limits=c(0,NA), expand=expansion(mult=c(0, mult_high)))
}

scale_y_zero <- function(mult_high=0.1){
  ggplot2::scale_y_continuous(limits=c(0,NA), expand=expansion(mult=c(0, mult_high)))
}

scale_percent <- function(x, accuracy = 1, scale = 100, suffix = "%", with_sign=F, ...) {
  formatted <- scales::number(x * scale, accuracy = accuracy, suffix = suffix, ...)
  ifelse(x > 0 & with_sign, paste0("+", formatted), formatted)
}

scale_percent_format <- function(accuracy = 1, scale = 100, suffix = "%", with_sign=F, ...) {
  function(x) ggplot2::scale_percent(x, accuracy = accuracy, scale = scale, suffix = suffix, with_sign = with_sign, ...)
}


# Specific scales ---------------------------------------------------------
scale_fill_electricity <- ggplot2::scale_fill_manual(values=pal_crea.electricity)
scale_color_electricity <- ggplot2::scale_color_manual(values=pal_crea.electricity)

crea.theme <- function(colors='CREA', reverse.order=F) {
  pars = standard.theme(color = FALSE)
  pars$strip.background$col <- pal_crea['Blue']
  pars$par.main.text$col <- pal_crea['Dark.blue']
  pars$par.main.text$cex <- 1.5
  cols <- colorRampPalette(crea_palettes[[colors]])(100)
  if(reverse.order) cols %<>% rev
  pars$regions$col <- cols
  return(pars)
}



# attach(CREAtheme)

c('ribbon', 'col', 'area', 'bar') %>% lapply(ggplot2::update_geom_defaults, list(fill = pal_crea[2])) -> t1
c('line', 'point') %>% lapply(ggplot2::update_geom_defaults, list(color = pal_crea[1])) -> t1
rm(t1)
rm(getpal)

# print('CREA theme loaded')