@@ -6,6 +6,7 @@ CHANGES in VERSION 2.1.0

 * `draw_legend()`: legends for annotations with the same names are merged

 * `densityHeatmap()`: `ylim` works as it is expected.

 * add `cluster_row_slices` and `cluster_column_slices` to `draw,HeatmapList-method()`

+* `densityHeatmap()`: `col` can be set as a function

 ========================

@@ -3340,6 +3340,7 @@ anno_zoom = function(align_to, panel_fun = function(index, nm = NULL) { grid.rec

 			if(n_boxes == 1) {

 				h = data.frame(bottom = .scale[1] - extend[1], top = .scale[2] + extend[2])

 			} else {

+				size = as.numeric(size)

 				gap = convertHeight(gap, "native", valueOnly = TRUE)

 				box_height = size/sum(size) * (1 + sum(extend) - sum(gap[1:(n_boxes-1)]))

 				h = data.frame(

@@ -3489,6 +3490,7 @@ anno_zoom = function(align_to, panel_fun = function(index, nm = NULL) { grid.rec

 			if(n_boxes == 1) {

 				h = data.frame(left = .scale[1] - extend[1], right = .scale[2] + extend[2])

 			} else {

+				size = as.numeric(size)

 				gap = convertWidth(gap, "native", valueOnly = TRUE)

 				box_width = size/sum(size) * (1 + sum(extend) - sum(gap[1:(n_boxes-1)]))

 				h = data.frame(

@@ -151,7 +151,11 @@ densityHeatmap = function(data,

 		}

 	}

-	col = colorRamp2(seq(0, max(mat, na.rm = TRUE), length = length(col)), col, space = color_space)

+	if(inherits(col, "function")) {

+		col = col(mat)

+	} else {

+		col = colorRamp2(seq(0, max(mat, na.rm = TRUE), length = length(col)), col, space = color_space)

+	}

 	bb = grid.pretty(c(min_x, max_x))

 	ht = Heatmap(mat, col = col, name = "density",