library(gplots)

library(RColorBrewer)

args <- commandArgs(trailingOnly = TRUE)

# print(args[1])

# print(args[2])

#########################################################

### reading in data and transform it to matrix format

#########################################################

data <- read.csv(args[1], comment.char="#")

rnames <- data[,1]                            # assign labels in column 1 to "rnames"

mat_data <- data.matrix(data[,2:ncol(data)])  # transform column 2-5 into a matrix

rownames(mat_data) <- rnames                  # assign row names

#########################################################

### customizing and plotting heatmap

#########################################################

offset <- -13

offset <- -40

offset <- -10

colors=NULL

color_map <- bluered

color_map <- colorRampPalette(c("white","red"))(n = 39)

# colors = c(seq(0,0.5,length=20),seq(0.501,1,length=20)) #TODO change for certain graphs

# color_map <- colorRampPalette(c("white","red"))(n = 39)

# # colors = c(seq(0,0.05,length=20),seq(0.501,1,length=20)) #TODO change for certain graphs

png(file = args[2], units="in", width=11, height=8.5, res=300)

heatmap.2(t(mat_data),

    # cellnote = t(mat_data),

    Rowv=NULL,

    Colv="NA",

    srtCol=0,

    labRow = "",

    col = color_map,

  breaks = colors,

    scale="none",

    margins=c(3,0), # ("margin.Y", "margin.X")

    trace='none',

    symkey=FALSE,

    symbreaks=FALSE,

    dendrogram='none',

    offsetCol = offset,

    cexCol=0.9,

  key=FALSE, #change to turn on

  density.info=NULL,

  denscol="black",

    keysize=1,

  #( "bottom.margin", "left.margin", "top.margin", "left.margin" )

  key.par=list(mar=c(3.5,0,3,0)),

  # lmat -- added 2 lattice sections (5 and 6) for padding

  lmat=rbind(c(5, 4, 2), c(6, 1, 3)), lhei=c(2.5, 5), lwid=c(1, 10, 1)

  )

dev.off()