ENH Brain.add_label(): also use label.color attribute

Author: christianbrodbeck

surfer/viz.py

@@ -991,18 +991,19 @@ def add_annotation(self, annot, borders=True, alpha=1, hemi=None,
         self.annot_list = al
         self._toggle_render(True, views)
 
-    def add_label(self, label, color="crimson", alpha=1, scalar_thresh=None,
+    def add_label(self, label, color=None, alpha=1, scalar_thresh=None,
                   borders=False, hemi=None, subdir=None):
         """Add an ROI label to the image.
 
         Parameters
         ----------
         label : str | instance of Label
             label filepath or name. Can also be an instance of
-            an object with attributes "hemi", "vertices", "name",
-            and (if scalar_thresh is not None) "values".
+            an object with attributes "hemi", "vertices", "name", and
+            optionally "color" and "values" (if scalar_thresh is not None).
         color : matplotlib-style color
-            anything matplotlib accepts: string, RGB, hex, etc.
+            anything matplotlib accepts: string, RGB, hex, etc. (default
+            "crimson")
         alpha : float in [0, 1]
             alpha level to control opacity
         scalar_thresh : None or number
@@ -1028,6 +1029,9 @@ def add_label(self, label, color="crimson", alpha=1, scalar_thresh=None,
         """
         if isinstance(label, basestring):
             hemi = self._check_hemi(hemi)
+            if color is None:
+                color = "crimson"
+
             if os.path.isfile(label):
                 filepath = label
                 label_name = os.path.basename(filepath).split('.')[1]
@@ -1058,6 +1062,13 @@ def add_label(self, label, color="crimson", alpha=1, scalar_thresh=None,
                     label_name = 'unnamed'
                 else:
                     label_name = str(label.name)
+
+                if color is None:
+                    if hasattr(label, 'color') and label.color is not None:
+                        color = label.color
+                    else:
+                        color = "crimson"
+
                 if scalar_thresh is not None:
                     scalars = label.values
             except Exception: