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
