def _import_graph(self, graph_data, parent=None): """ """ if parent: graph = parent else: graph = NodeGraph() nodes = graph_data["nodes"] imported_nodes = [] for node_data in nodes: import_params = node_data["input_params"] output_params = node_data["output_params"] node = self._import_node(node_data, graph, import_params=False) imported_nodes.append((node, import_params, output_params)) for node, input_params, output_params in imported_nodes: for input_param_data in input_params: self._import_param(node, input_param_data, NodeParam.INPUT) for output_param_data in output_params: self._import_param(node, output_param_data, NodeParam.OUTPUT) output_node = graph_data["output_node"] if output_node: output_node = graph.node(output_node) graph.output_node = output_node return graph
def _read_graph(self, data, parent=None): """ """ if parent: graph = parent else: graph = NodeGraph() nodes = data["nodes"] for node in nodes: self._read_node(node, graph) output_node = data["output_node"] if output_node: output_node = graph.node(output_node) graph.output_node = output_node return graph
#from node import Node import networkx as nx from nodegraph import NodeGraph from nodegraph import NodeGraphExecutor from opnode import OperatorNode, EchoOperator, CodeOperator from paramtype import VariantType ng = NodeGraph() op1 = ng.add_node("op1", "Node") op2 = ng.add_node("op2", "Node") op3 = ng.add_node("op3", "SwitchNode") op4 = ng.add_node("op4", "Node") op5 = ng.add_node("op5", "Node") op6 = ng.add_node("op6", "Node") op4.add_input_param("user_param1", VariantType) op4.input_param("netin").value = op1.output_param("netout") op4.input_param("user_param1").value = op3.output_param("chosen_value") op3.input_param("index_0").value = op2.output_param("netout") op3.add_input_param("index_1", VariantType) op3.input_param("index_1").value = op5.output_param("netout") op3.input_param("switch_index").value = 0 #op6.input_param("netin").value = op1.output_param("netout") #op1.input_param("netin").value = op6.output_param("netout") ng.output_node = op4