R/gera-crescimento.R

In mellohenrique/codeplan.boletim2: Gerador dos dados do Boletim da Codeplan

#' @title Gera dados de crescimento de casos pro semana
#'
#' @description Avalia os dados de casos e obitos do MS e gera um data.table com as informacoes de crescimento de casos e obitos semanais por UF
#'
#' @param dados um data.frame com dados de casos e obitos do MS
#'
#' @return um data.table com os dados de crescimento de casos e obitos semanais por UF
#'
#' @import data.table


gera_crescimento = function(dados){
  # Conformando ao tipo data.table
  dados = transforma_dt(dados)

  # Selecionando apenas o dia da semana desejado
  dados = dados[weekdays(data) == weekdays(max(data)),]

  # Calculca-se casos e obitos por semana
  dados = dados[, `:=`(casos_semana = casosAcumulado - shift(casosAcumulado),
                       obitos_semana = obitosAcumulado - shift(obitosAcumulado)), by = estado]

  # Selecionando colunas de interesse
  dados = dados[, .(data, estado, casos_semana, obitos_semana)]

  # Calcula-se crescimento
  dados = dados[, `:=`(crescimento_casos_semana = casos_semana/shift(casos_semana) - 1,
                       crescimento_obitos_semana = obitos_semana/shift(obitos_semana) - 1),
                by = estado]

  # Selecionando data de interesse
  dados = dados[data == max(data),]
  dados = dados[estado != "",]

  # Retorna Resultados
  return(dados)
}