R/plot.R

In psc: Personalised Synthetic Controls

#' Function for Plotting PSC objects
#'
#' A function which illsutrates the predicted response under the Counter Factual
#' Model (CFM) and the observed response under the experimental treatment(s).
#' Form of the output will depend on the form of the CFM used
#'
#' @param x an object of class 'psc'
#' @param ... not used
#' @return a survival plot corresponding to the psc fit
#' @details This function plots the expected response of the control treatment
#'    along with the observe response rates of the experimental arms
#' @import ggplot2
#' @examples
#' bin.mod <- psc::bin.mod
#' data <- psc::data
#' bin.psc <- pscfit(bin.mod,data,nsim=3000)
#' plot(bin.psc)
#' @export
plot.psc <- function (x,...){

  model.type <- x$'model.type';model.type
  fam <- x$DC_clean$model_extract$family;fam

  if ("glm" %in% model.type) {
    ### gaussian
    if(fam$family%in%c("gaussian")){
      p <- plot.psc.cont(x)
    }
    ## poisson
    if(fam$family%in%c("poisson")){
      p <- plot.psc.count(x)
    }
    ## binomial
    if(fam$family%in%c("binomial")){
      p <- plot.psc.binary(x)
    }
  }

  if ("flexsurvreg" %in% model.type) {
    p <- plot.psc.flexsurvreg(x)
  }
  p
}