R/plot_AAratio_comparison.R

In matteodelucchi/TRAL-Result-Analysis: Analysis of the output of TRAL

#' Compare the AA frequency in two groups.
#'
#' @param df data.frame. AA frequency count obtained through \code{\link{AAfreq_in_TR}} with an additional column "aa_ratio_overall".
#' @param plot_title str. Plot title.
#' @param ylim int. Scale the y-axis maximum value.
#'
#' @return ggplot object. The values from 'aa_ratio' are displayed in green, the values from 'aa_ratio_overall' in red.
#'
#' @export
#'
plot_AAratio_comparison <- function(df = AAfreq, plot_title = NULL, ylim =1){
  p1 <- ggplot(df, aes(x = as.factor(df$aa), y = df$aa_ratio))+
    geom_segment( aes(x=as.factor(df$aa) ,xend=as.factor(df$aa), y=df$aa_ratio_overall, yend=df$aa_ratio), color="grey")+
    geom_point(color="#2D882D")+
    geom_point(aes(y=df$aa_ratio_overall), color= "#AA3939")+
    scale_x_discrete(labels = as.character(df$aa))+
    theme_minimal()

  if(!is.null(plot_title)){
    p1 <- p1 +
      labs(x ="Amino Acid",
           y = "AA ratio",
           title = plot_title)
  } else {
    p1 <- p1 +
      labs(x ="Amino Acid",
           y = "AA ratio")
  }

  p1 <- beautifier(p1)+
    theme(axis.text.x = element_text(angle = 0, hjust = 0.5))+
    ylim(0,ylim)
  return(p1)
}