Not  exactly sure what you need. Are you referring to the altitudes (=magnitude) of the displacement? In that case you need to use the same function used for the displacement to drive the colormap. Is that what you want?Like the attached example?

TD