R/draw_irf.R

In common2016/VARrf: What the Package Does (One Line, Title Case)

#' Draw IRF
#'
#' @description draw IRF picture according to results from \code{IRFrf_gen}
#' @param irf a returned objective from \code{IRFrf_gen}
#' @param res_var a string vector composed with names of response variable
#' @param ncol_pic multiple pictures are arranged \code{ncol_pic} coloumns
#' @param ci confidence interval, default is 0.9, and if it's NULL, don't draw
#' confidence interval.
#' @examples
#'
#' @export
#' @importFrom stats median quantile

draw_irf <- function(irf, res_var, ncol_pic = 2, ci = 0.9){
  ans <- lapply(irf, reshape2::melt, id.vars = 's')
  ans <- dplyr::bind_rows(ans)
  dw <- (1 - ci)/2 # quantile
  picdata <- dplyr::group_by(ans, s, variable) %>%
    dplyr::summarise(v50 = median(value),
                     v025 = quantile(value, dw),
                     v975 = quantile(value, 1-dw))
  p <- list()
  pcode <- 'p[[1]]'
  for (i in 1:length(res_var)) {
    if (!is.null(ci)){
      p[[res_var[i]]] <- ggplot2::ggplot(picdata[picdata$variable %in% res_var[i],],
                                         ggplot2::aes_string(x = 's' , y = 'v50')) +
        ggplot2::geom_line(ggplot2::aes_string(y = 'v025'), linetype = 2) +
        ggplot2::geom_line(ggplot2::aes_string(y = 'v975'), linetype = 2)
    }else {
      p[[res_var[i]]] <- ggplot2::ggplot(picdata[picdata$variable %in% res_var[i],],
                                         ggplot2::aes_string(x = 's' , y = 'v50'))
    }
    p[[res_var[i]]] <- p[[res_var[i]]] + ggplot2::geom_hline(yintercept = 0) +
      ggplot2::labs(title = res_var[i]) +
      ggplot2::geom_line() + ggplot2::theme_bw()
    if (i != 1) pcode <- paste(pcode, '+p[[',as.character(i),']]',sep = '')
    if (i == length(res_var)) pcode <- paste('ans <- ',pcode, '+ patchwork::plot_layout(ncol =',
                                             as.character(ncol_pic),')', sep = '')
  }
  eval(parse(text = pcode))
  return(ans)
}