CitFuns/R/ggheatmap.R

ggheatmap<-function(df, x=NULL, y=NULL, value=NULL, grouping="mean", exclude_group=NULL, scale="none"){
  if (is.null(x)){x=colnames(df)[1]}
  if (is.null(y)){y=colnames(df)[2]}
  if (is.null(value)){value=colnames(df)[3]}

  df<-rename(df, "VarX"=all_of(x), "VarY"=all_of(y), "Value"=all_of(value))
  
  if (is.null(exclude_group)){
    df<-df %>% group_by(VarX,VarY)
  }else{
    df<-df %>% group_by_("VarX","VarY",exclude_group) #%>% rename(exclude_group="all_of(exclude_group)")
  }
  if (grouping == "mean"){
    df<-df %>% summarise(Value=mean(Value)) %>% ungroup
  }
  if (grouping == "median"){
    df<-df %>% summarise(Value=median(Value)) %>% ungroup
  }
  order<-clustsort(df %>% spread(VarY,Value) %>% select(!all_of(exclude_group)) %>% as.data.frame)

  df$VarX<-factor(df$VarX, levels=order$x)
  df$VarY<-factor(df$VarY, levels=order$y)
  
  if (scale != "none"){
    if (scale == "rows"){
      cols<-unique(df$VarY)
      sca.df<-spread(df, VarY, Value)
      for (i in cols){sca.df[,i]<-scale(sca.df[,i])}
      df<-gather(sca.df, VarY, Value, all_of(cols))
    }
    if (scale == "cols"){
      cols<-unique(df$VarX)
      sca.df<-spread(df, VarX, Value)
      for (i in cols){sca.df[,i]<-scale(sca.df[,i])}
      df<-gather(sca.df, VarX, Value, all_of(cols))
    }
  }
  
  df %>%
    ggplot(aes(VarX, VarY, fill=Value))+
    labs(x=x, y=y)+
    theme_heatmap()
}