FCP-INDI
diff --git a/‎CPAC/pipeline/cpac_pipeline.py‎
Lines changed: 17 additions & 0 deletions b/‎CPAC/pipeline/cpac_pipeline.py‎
Lines changed: 17 additions & 0 deletions
diff --git a/‎CPAC/pipeline/nipype_pipeline_engine/__init__.py‎
Lines changed: 21 additions & 22 deletions b/‎CPAC/pipeline/nipype_pipeline_engine/__init__.py‎
Lines changed: 21 additions & 22 deletions
@@ -440,6 +440,23 @@ def run_workflow(sub_dict, c, run, pipeline_timing_info=None, p_name=None,
         logger.exception('Building workflow failed')
         raise exception
 
+    wf_graph = c['pipeline_setup', 'log_directory', 'graphviz',
+                 'entire_workflow']
+    if wf_graph.get('generate'):
+        dotfilename = os.path.join(log_dir, f'{p_name}.dot')
+        for graph2use in wf_graph.get('graph2use'):
+            for graph_format in wf_graph.get('format'):
+                try:
+                    workflow.write_graph(dotfilename=dotfilename,
+                                         graph2use=graph2use,
+                                         format=graph_format,
+                                         simple_form=wf_graph.get(
+                                             'simple_form', True))
+                except Exception as exception:
+                    raise RuntimeError(f'Failed to visualize {p_name} ('
+                                       f'{graph2use}, {graph_format})'
+                                       ) from exception
+
     if test_config:
         logger.info('This has been a test of the pipeline configuration '
                     'file, the pipeline was built successfully, but was '
 
@@ -1,35 +1,34 @@
-'''Module to import Nipype Pipeline engine and override some Classes.
-See https://fcp-indi.github.io/docs/developer/nodes
-for C-PAC-specific documentation.
-See https://nipype.readthedocs.io/en/latest/api/generated/nipype.pipeline.engine.html
-for Nipype's documentation.
-
-Copyright (C) 2022  C-PAC Developers
+# Copyright (C) 2022  C-PAC Developers
 
-This file is part of C-PAC.
+# This file is part of C-PAC.
 
-C-PAC is free software: you can redistribute it and/or modify it under
-the terms of the GNU Lesser General Public License as published by the
-Free Software Foundation, either version 3 of the License, or (at your
-option) any later version.
+# C-PAC is free software: you can redistribute it and/or modify it under
+# the terms of the GNU Lesser General Public License as published by the
+# Free Software Foundation, either version 3 of the License, or (at your
+# option) any later version.
 
-C-PAC is distributed in the hope that it will be useful, but WITHOUT
-ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
-FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
-License for more details.
+# C-PAC is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
+# License for more details.
 
-You should have received a copy of the GNU Lesser General Public
-License along with C-PAC. If not, see <https://www.gnu.org/licenses/>.'''  # noqa: E501
+# You should have received a copy of the GNU Lesser General Public
+# License along with C-PAC. If not, see <https://www.gnu.org/licenses/>.
+'''Module to import Nipype Pipeline engine and override some Classes.
+See https://fcp-indi.github.io/docs/developer/nodes
+for C-PAC-specific documentation.
+See https://nipype.readthedocs.io/en/latest/api/generated/nipype.pipeline.engine.html
+for Nipype's documentation.'''  # noqa: E501  # pylint: disable=line-too-long
 from nipype.pipeline import engine as pe
 # import everything in nipype.pipeline.engine.__all__
 from nipype.pipeline.engine import *  # noqa: F401,F403
 # import our DEFAULT_MEM_GB and override Node, MapNode
-from .engine import DEFAULT_MEM_GB, get_data_size, Node, MapNode, \
-                    UNDEFINED_SIZE, Workflow
+from .engine import DEFAULT_MEM_GB, export_graph, get_data_size, Node, \
+                    MapNode, UNDEFINED_SIZE, Workflow
 
 __all__ = [
     interface for interface in dir(pe) if not interface.startswith('_')
-] + ['DEFAULT_MEM_GB', 'get_data_size', 'Node', 'MapNode', 'UNDEFINED_SIZE',
-     'Workflow']
+] + ['DEFAULT_MEM_GB', 'export_graph', 'get_data_size', 'Node', 'MapNode',
+     'UNDEFINED_SIZE', 'Workflow']
 
 del pe