R/carrega-dados.R

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

#' @title Carrega dados utilizados no boletim
#'
#' @description
#'
#' @param caminho_estados caminho para os dados de covid 19 em formato csv do ministerio da saude ou brasilio
#' @param caminho_ssp  vetor caminho para os dados de covid-19 por da SSP-DF
#' @param brasilio valor logico se os dados de estados virem do site brasil.io
#'
#' @return lista com cinco data.frames e data.tables
#'
#' @import data.table

carrega_dados = function(caminho_estados,
                         caminho_ssp,
                         brasilio = TRUE){
  # Carrega dados do ministerio da saude
  if (brasilio == TRUE){
    dados_ms = carrega_brasilio(caminho_estados)
  } else {
    dados_ms = data.table::fread(caminho_estados, encoding = "UTF-8")
  }

  # Define capitais de acordo com os dois modelos
  if (brasilio == TRUE){
    vetor_capitais = codeplan.boletim2::capitais$ibge
  } else {
    vetor_capitais = codeplan.boletim2::capitais$ibge %/% 10
  }

  # Filtra dados para capitais
  capitais = dados_ms[codmun %in% vetor_capitais,]

  # Carrega dados em lista
  dados = list(
    dados_ssp = carrega_ssp(caminho_ssp = caminho_ssp), # Dados da SSP
    ufs = dados_ms[is.na(codmun), ], # Dados do MS por UF
    df_ms = dados_ms[estado == "DF" & is.na(codmun),], # Dados do MS para o DF
    capitais = capitais # Dados do MS para capitais
    )

  # Retorna Resultados
  return(dados)
}