DataArrayModel : create a new LinearGradient object only when max and min values are changed (i.e. in reset_minmax() method)

Author: alixdamman

larray_editor/arraymodel.py

@@ -284,11 +284,14 @@ def reset_minmax(self):
             if self.vmax == self.vmin:
                 self.vmin -= 1
             self.bgcolor_enabled = True
+            self.bg_gradient = LinearGradient([(self.vmin, self.hsv_min), (self.vmax, self.hsv_max)])
+
         # ValueError for empty arrays
         except (TypeError, ValueError):
             self.vmin = None
             self.vmax = None
             self.bgcolor_enabled = False
+            self.bg_gradient = None
 
     def set_format(self, format):
         """Change display format"""
@@ -343,10 +346,8 @@ def data(self, index, role=Qt.DisplayRole):
                 return to_qvariant(self._format % value)
         elif role == Qt.BackgroundColorRole:
             if self.bgcolor_enabled and value is not np.ma.masked:
-                if self.bg_gradient is None:
-                    color_val = float(self.color_func(value))
-                    bg_gradient = LinearGradient([(self.vmin, self.hsv_min), (self.vmax, self.hsv_max)])
-                    return bg_gradient[color_val]
+                if self.bg_value is None:
+                    return self.bg_gradient[float(self.color_func(value))]
                 else:
                     bg_value = self.bg_value
                     x, y = index.row(), index.column()