`node graph generate` with multiple root nodes?

bilke · March 28, 2024, 2:15pm

Is it somehow possible to generate node graphs with multiple root nodes? I know the command signature allows only one ROOT_NODE.

sphuber · March 28, 2024, 4:25pm

The CLI currently doesn’t allow this indeed, but in principle I see no reason why not to allow specifying multiple. I opened a feature request for it: Allow multiple root nodes for node graph generation · Issue #6335 · aiidateam/aiida-core · GitHub
It should be relatively easy to implement.

In the meantime, if you already want to use it, you can look at the implementation and adapt it yourself:

github.com

aiidateam/aiida-core/blob/6898ff4d8c263cf08707c61411a005f6a7f731dd/src/aiida/cmdline/commands/cmd_node.py#L477


      
              '-c',
              '--highlight-classes',
              help='Only color nodes of specific class label (as displayed in the graph e.g. StructureData, FolderData, etc.).',
              type=click.STRING,
              default=None,
              multiple=True,
          )
          @click.option('-s', '--show', is_flag=True, help='Open the rendered result with the default application.')
          @arguments.OUTPUT_FILE(required=False)
          @decorators.with_dbenv()
          def graph_generate(
              root_node,
              link_types,
              identifier,
              ancestor_depth,
              descendant_depth,
              process_out,
              process_in,
              engine,
              output_format,
              highlight_classes,

Essentially you would just call Graph.recurse_ancestors and Graph.recurse_descendants for each root node.

Topic		Replies	Views
Smart way to export all nodes linked to a workchain General Usage question	3	83	February 23, 2024
Do `with_ancestors` and `with_descendants` work? General Usage	5	71	May 1, 2024
Making nice provenance garphs General Usage	2	37	June 7, 2024
Documentation on `HubbardStructureData` and related General Usage	1	74	March 27, 2024
Possible to store multiple StructureData General Usage question	2	60	September 13, 2023

`node graph generate` with multiple root nodes?

Related Topics