Fix contour colorscale domain missing zmin/zmax

lvlte · Jun 2, 2023 · 66e7b3c · 66e7b3c
1 parent d31839e
commit 66e7b3c
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/src/traces/contour/make_color_map.js b/src/traces/contour/make_color_map.js
@@ -29,10 +29,10 @@ module.exports = function makeColorMap(trace) {
 
     var si, i;
 
-    if(contours.coloring === 'heatmap') {
-        var zmin0 = cOpts.min;
-        var zmax0 = cOpts.max;
+    var zmin0 = cOpts.min;
+    var zmax0 = cOpts.max;
 
+    if(contours.coloring === 'heatmap') {
         for(i = 0; i < len; i++) {
             si = scl[i];
             domain[i] = si[0] * (zmax0 - zmin0) + zmin0;
@@ -65,6 +65,18 @@ module.exports = function makeColorMap(trace) {
             domain[i] = (si[0] * (nc + extra - 1) - (extra / 2)) * cs + start;
             range[i] = si[1];
         }
+
+        // Ensure zmin and zmax are included in the colorscale
+
+        if(domain[0] > zmin0) {
+            domain.unshift(zmin0);
+            range.unshift(range[0]);
+        }
+
+        if(domain[domain.length - 1] < zmax0) {
+            domain.push(zmax0);
+            range.push(range[range.length - 1]);
+        }
     }
 
     return Colorscale.makeColorScaleFunc(