Fix color graphing to loop through blue, red, green.

Author: ashgillman

nipype/pipeline/engine/workflows.py

@@ -923,8 +923,13 @@ def _get_dot(self, prefix=None, hierarchy=None, colored=False,
             prefix = '  '
         if hierarchy is None:
             hierarchy = []
-        colorset = ['#FFFFC8', '#0000FF', '#B4B4FF', '#E6E6FF', '#FF0000',
-                    '#FFB4B4', '#FFE6E6', '#00A300', '#B4FFB4', '#E6FFE6']
+        colorset = ['#FFFFC8',                       # Y
+                    '#0000FF', '#B4B4FF', '#E6E6FF', # B
+                    '#FF0000', '#FFB4B4', '#FFE6E6', # R
+                    '#00A300', '#B4FFB4', '#E6FFE6', # G
+                    '#0000FF', '#B4B4FF'] # loop B
+        if level > len(colorset) - 2:
+            level = 3 # Loop back to blue
 
         dotlist = ['%slabel="%s";' % (prefix, self.name)]
         for node in nx.topological_sort(self._graph):
@@ -963,8 +968,6 @@ def _get_dot(self, prefix=None, hierarchy=None, colored=False,
                                              colored=colored,
                                              simple_form=simple_form, level=level + 3))
                 dotlist.append('}')
-                if level == 6:
-                    level = 2
             else:
                 for subnode in self._graph.successors_iter(node):
                     if node._hierarchy != subnode._hierarchy: