R/plotWeekAhead.R

In nulloa/FluViz: R Functions for Visualizing a CDC FluView Submission

#' A Week Ahead Plot Function
#'
#' This function allows you to plot Weak Ahead Predictions
#' @param dat Expects a data csv in the form of a CDC fluview submission see \code{FluSight} package for a minimal submission
#' @param region Specifies the region to be plotted
#' @param wk Numeric. How many weeks ahead to plot. Defaults to 1.
#' @param ilimax Numeric. Max level of ILI percentage to be plotted
#' @keywords Week Ahead Prediction Plots
#' @export
#' @examples
#' plotWeekAhead()

plotWeekAhead <- function(dat, region, wk=1, ilimax){
  require(ggplot2)
  require(cdcfluview)
  
  d <- subset(dat, location==region & as.numeric(as.character(bin_start_incl)) <= ilimax)
  d <- d[grep("wk ahead", d$target),]
  
  NatFluDat <- get_flu_data("national", NA, "ilinet")
  RegFluDat <- get_flu_data("hhs", 1:10, "ilinet")
  NatFluDat$REGION[NatFluDat$REGION=="X"] <- "National"
  CFluDat <- rbind(tail(RegFluDat, n=10), tail(NatFluDat, n=1))[,c("REGION","% WEIGHTED ILI")]
  if(region=="US National"){
    CurrentILIPer <- CFluDat[CFluDat$REGION=="National", "% WEIGHTED ILI"]
  }else{
    CurrentILIPer <- CFluDat[CFluDat$REGION==paste("Region", strsplit(region, " ")[[1]][3]), "% WEIGHTED ILI"]
  }
  
  if(wk==1){
    return(ggplot(data=subset(d, target=="1 wk ahead"), aes(x=as.numeric(as.character(bin_start_incl)), y=value)) + 
            geom_point() + labs(title = "1 Week Ahead", x="ILI%", y="Prob") + geom_vline(aes(xintercept = CurrentILIPer)))
  }
  
  
  if(wk==2){
    return(ggplot(data=subset(d, target=="2 wk ahead"), aes(x=as.numeric(as.character(bin_start_incl)), y=value)) + 
             geom_point() + labs(title = "2 Week Ahead", x="ILI%", y="Prob") + geom_vline(aes(xintercept = CurrentILIPer)))
  }
  
  if(wk==3){
    return(ggplot(data=subset(d, target=="3 wk ahead"), aes(x=as.numeric(as.character(bin_start_incl)), y=value)) + 
             geom_point() + labs(title = "3 Week Ahead", x="ILI%", y="Prob") + geom_vline(aes(xintercept = CurrentILIPer)))
  }
  
  if(wk==4){
    return(ggplot(data=subset(d, target=="4 wk ahead"), aes(x=as.numeric(as.character(bin_start_incl)), y=value)) + 
             geom_point() + labs(title = "4 Week Ahead", x="ILI%", y="Prob") + geom_vline(aes(xintercept = CurrentILIPer)))
  }
  
}