if draw-children, color leaf-level children grey.

goatools/cli/gosubdag_plot.py

@@ -99,13 +99,14 @@ def __init__(self, go2obj):
 
     def get_go_color(self, **kws):
         """Return source GO IDs ."""
+        # kws: GO go_file draw-children
         ret = {'GOs':set(), 'go2color':{}}
         if 'GO' in kws:
             self._goargs(ret, kws['GO'])
         if 'go_file' in kws:
             self._rdtxt_gos(ret, kws['go_file'])
         if 'draw-children' in kws:
-            ret['GOs'].update(get_leaf_children(ret['GOs'], self.go2obj))
+            self._add_gochildleaf(ret)
         # If there have been no GO IDs explicitly specified by the user
         if not ret['GOs']:
             # If the GO-DAG is sufficiently small, print all GO IDs
@@ -156,6 +157,17 @@ def _rdtxt_gos(self, ret, go_file):
                             print("IGNORING: {L}".format(L=line),)
         self._update_ret(ret, goids, go2color)
 
+    def _add_gochildleaf(self, ret):
+        """Add leaf-level GO children to GO list colored uniquely."""
+        leaf_gos = get_leaf_children(ret['GOs'], self.go2obj)
+        if leaf_gos:
+            ret['GOs'].update(leaf_gos)
+            leaf_go_color = Go2Color.key2col['go_leafchild']
+            go2color = ret['go2color']
+            for goid in leaf_gos:
+                if goid not in go2color:
+                    go2color[goid] = leaf_go_color
+
     @staticmethod
     def _update_ret(ret, goids, go2color):
         """Update 'GOs' and 'go2color' in dict with goids and go2color."""

goatools/gosubdag/plot/go2color.py

@@ -24,6 +24,7 @@ class Go2Color(object):
     key2col = {
         'level_01': '#f1fbfd', # Very light blue
         'go_sources': '#ffffe4', # xkcd off white
+        'go_leafchild': '#eae8e8', # 'Grey palette' http://www.color-hex.com/color-palette/45491
     }