Merge branch 'dev' of github.com:freesurfer/freesurfer into dev

python/freesurfer/lookups.py

@@ -30,7 +30,7 @@ def __str__(self):
             lines.append(str(idx).ljust(col1) + elt.name.ljust(col2) + colorstr)
         return '\n'.join(lines)
 
-    def add(self, index, name, color):
+    def add(self, index, name, color=None):
         self[index] = LookupTable.Element(name, color)
 
     def search(self, name, exact=False):
@@ -72,6 +72,36 @@ def write(self, filename):
                 colorstr = ' '.join([str(c).ljust(3) for c in color])
                 file.write(str(idx).ljust(col1) + elt.name.ljust(col2) + colorstr + '\n')
 
+    def extract(self, labels):
+        """
+        Extract a new LookupTable from a list of label indices.
+        """
+        lut = LookupTable()
+        for label in labels:
+            elt = self.get(label)
+            if elt is None:
+                raise ValueError(f'Index {label} does not exist in the LookupTable.')
+            lut.add(label, elt.name, elt.color)
+        return lut
+
+    def copy_colors(self, source_lut):
+        """
+        Copies colors of matching label indices from a source LookupTable.
+        """
+        for label in self.keys():
+            elt = source_lut.get(label)
+            if elt is not None and elt.color is not None:
+                self[label].color = elt.color
+
+    def copy_names(self, source_lut):
+        """
+        Copies names of matching label indices from a source LookupTable.
+        """
+        for label in self.keys():
+            elt = source_lut.get(label)
+            if elt is not None:
+                self[label].name = elt.name
+
 
 class RecodingLookupTable(dict):
     """