Fix kippenhan colormaps when all data is positive

rjfarmer · May 25, 2023 · 66640a3 · 66640a3
1 parent 89e3d79
commit 66640a3
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/mesaPlot/plot.py b/mesaPlot/plot.py
@@ -5549,14 +5549,6 @@ def plotKip3(
         extent = (xmin, xmax, ymin, ymax)
         extent = np.double(np.array(extent))
 
-        if cmap_merge:
-            newCm = self.mergeCmaps(cmap, [[0.0, 0.5], [0.5, 1.0]])
-        else:
-            if not isinstance(cmap, str):
-                newCm = cmap[-1]
-            else:
-                newCm = cmap
-
         if hasdataz:
             if zlog:
                 # Get rid of warnigns about > nan's
@@ -5572,6 +5564,19 @@ def plotKip3(
                 vmax = np.maximum(np.abs(vmax), np.abs(vmin))
                 vmin = -vmax
 
+                if cmap_merge:
+                    newCm = self.mergeCmaps(cmap, [[0.0, 0.5], [0.5, 1.0]])
+                else:
+                    if not isinstance(cmap, str):
+                        newCm = cmap[-1]
+                    else:
+                        newCm = cmap
+            else:
+                if not isinstance(cmap, str):
+                    newCm = cmap[-1]
+                else:
+                    newCm = cmap
+
             if cmin is not None:
                 vmin = cmin