R/viewnew.R

In dtiezzi/COVIDBR: COVID19 information from Brazil (COVID19 Brazil)

#' Plot COVID19 new cases
#'
#' The function creates line chart displaying the number of new COVID19 cases (7-days mean incidence) in Brazil
#' @return A line chart with covid19 incidence in function of time

#' @param x a data frame with COVID19 information from load.covid()
#' @param ibgeinfo a data frame (default = NA, all Brazil data) from ibge.id()
#' @param top5 boolean (default = FALSE). True if id == NA. If True, displays the top5 states in total number of cases
#'

#' @importFrom ggplot2 ggplot geom_line geom_bar theme_minimal ylab ggtitle aes scale_colour_viridis_d scale_colour_brewer
#' @importFrom utils data
#' @export

#' @examples
#' covid <- load.covidBr()
#' myid <- ibge.id('SP')
#' mychart <- view.covid.newcases(covid, myid)
#'

view.covid.newcases <- function(x, ibgeinfo = NA, top5 = FALSE) {
  if (top5) {
    dt <- subset(x, x$date == max(x$date) & x$id < 76)
    t5 <- dt[order(dt$sumcases, decreasing = TRUE), ][1:5, 1]
    df <- subset(x, x$id %in% t5)
    ibge <- covidBR::ibge
    sts <- ibge[ibge$id %in% t5, c(2,3)]
    rownames(sts) <- sts$id
    df$id <- as.factor(df$id)
    sts <- sts[levels(df$id), ]
    tit = 'COVID19 - New Cases per day - Top 5 States in Brazil'
    p <- ggplot(data = df, aes(x = date, y = newav, color = id)) + geom_line(size = 1.1) + ylab("Total number") + ggtitle(tit) + theme_minimal() + scale_colour_viridis_d('State', labels = sts$name, option = "plasma")
    return(p)
  } else if (is.na(ibgeinfo)) {
    df <- subset(x, x$id == 76)
    tit = 'COVID19 - New Cases per day - Brazil'
  } else {
    df <- subset(x, x$id == ibgeinfo$id[1])
    tit = paste('COVID19 - New Cases per day -', ibgeinfo$name[1], sep = ' ')
  }
  p <- ggplot(data = df) + geom_bar(aes(x = date, y = newcases), stat="identity", fill="#5ab4ac", colour="#0a0a0a") + geom_line(aes(x = date, y = newav), colour = 'red', size = 2) + ylab("Total number") + ggtitle(tit) + theme_minimal()
  return(p)
}