Add autoscale, autoscale_None, scaled methods for BivariateNorm

patniharshit · patniharshit · commit 537b4586f326 · 2017-07-13T12:26:21.000+05:30
diff --git a/lib/matplotlib/colors.py b/lib/matplotlib/colors.py
@@ -1414,7 +1414,7 @@ def __call__(self, values, clip=None):
         if clip is None:
             clip = [self.norm1.clip, self.norm2.clip]
 
-        return np.array([self.norm1(values[0], clip=clip[0]),
+        return np.asarray([self.norm1(values[0], clip=clip[0]),
                 self.norm2(values[1], clip=clip[1])])
 
     def inverse(self, values):
@@ -1431,6 +1431,22 @@ def inverse(self, values):
         return np.asarray([self.norm1.inverse(values[0]),
                 self.norm2.inverse(values[1])])
 
+    def autoscale(self, A):
+        """
+        Set *vmin*, *vmax* to min, max of *A*.
+        """
+        self.norm1.autoscale(A[0])
+        self.norm2.autoscale(A[1])
+
+    def autoscale_None(self, A):
+        'autoscale only None-valued vmin or vmax'
+        self.norm1.autoscale_None(A[0])
+        self.norm2.autoscale_None(A[1])
+
+    def scaled(self):
+        'return true if vmin and vmax set for both normalizers'
+        return self.norm1.scaled() and self.norm2.scaled()
+
 
 def rgb_to_hsv(arr):
     """
