Modify coolBlueHotRed to be based on class attribute?

Question

I am working on SOM and is generating heatmaps using the palette coolBlueHotRed.

The default coolBlueHotRed is this

coolBlueHotRed <- function(n, alpha