schema_solve.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526
  1. from .utilities import (prRed, prGreen, prPurple, prWhite,
  2. prOrange,
  3. wrapRed, wrapGreen, wrapPurple, wrapWhite,
  4. wrapOrange,)
  5. from .utilities import init_connections, init_dependencies
  6. from .utilities import class_for_mantis_prototype_node
  7. from .base_definitions import SchemaUINode, GraphError, replace_types, custom_props_types
  8. from .node_container_common import setup_custom_props_from_np
  9. # a class that solves Schema nodes
  10. class SchemaSolver:
  11. def __init__(self, schema_dummy, nodes, prototype, signature=None,):
  12. self.all_nodes = nodes # this is the parsed tree from Mantis
  13. self.node = schema_dummy
  14. self.tree = prototype.node_tree
  15. self.uuid = self.node.uuid
  16. if signature:
  17. self.signature = signature
  18. else:
  19. self.signature = self.node.signature
  20. self.schema_nodes={}
  21. self.solved_nodes = {}
  22. self.incoming_connections = {}
  23. self.outgoing_connections = {}
  24. self.constant_in = {}
  25. self.constant_out = {}
  26. self.array_input_connections = {}
  27. self.array_output_connections = {}
  28. self.nested_schemas = {}
  29. self.autogenerated_nodes = {}
  30. self.held_links = []
  31. self.tree_path_names = [*self.node.signature] # same tree as the schema node
  32. self.autogen_path_names = ['SCHEMA_AUTOGENERATED', *self.node.signature[1:]]
  33. self.index_link = self.node.inputs['Schema Length'].links[0]
  34. self.solve_length = self.node.evaluate_input("Schema Length")
  35. # I'm making this a property of the solver because the solver's data is modified as it solves each iteration
  36. self.index = 0
  37. self.init_schema_links()
  38. self.set_index_strings()
  39. for ui_node in self.tree.nodes:
  40. if isinstance(ui_node, SchemaUINode):
  41. # first we need to fill the parameters of the schema nodes.
  42. # we use the bl_idname because all schema nodes should be single-instance
  43. signature = (*self.tree_path_names, ui_node.bl_idname)
  44. # We use the solver's signature here because it represents the "original" signature of the schema UI group node
  45. # since this schema solver may be in a nested schema, and its node's signature may have uuid/index attached.
  46. get_sig = (*self.signature, ui_node.bl_idname)
  47. if not (mantis_node := self.all_nodes.get(get_sig)): raise RuntimeError(wrapRed(f"Not found: {get_sig}"))
  48. self.schema_nodes[signature] = mantis_node
  49. mantis_node.fill_parameters(ui_node)
  50. def set_index_strings(self):
  51. self.index_str = lambda : '.'+str(self.uuid)+'.'+str(self.index).zfill(4)
  52. self.prev_index_str = lambda : '.'+str(self.uuid)+'.'+str(self.index-1).zfill(4)
  53. def init_schema_links(self,):
  54. """ Sort and store the links to/from the Schema group node."""
  55. for item in self.tree.interface.items_tree:
  56. if item.item_type == 'PANEL': continue
  57. if not item.parent:
  58. raise GraphError("ERROR: Schema tree has inputs that are not in categories. This is not supported")
  59. match item.parent.name:
  60. case 'Connection':
  61. if item.in_out == 'INPUT':
  62. if incoming_links := self.node.inputs[item.identifier].links:
  63. self.incoming_connections[item.name] = incoming_links[0]
  64. else:
  65. self.incoming_connections[item.name] = None
  66. else: # OUTPUT
  67. if outgoing_links := self.node.outputs[item.identifier].links:
  68. self.outgoing_connections[item.name] = outgoing_links.copy()
  69. else:
  70. self.outgoing_connections[item.name] = []
  71. case 'Constant':
  72. if item.in_out == 'INPUT':
  73. if constant_in_links := self.node.inputs[item.identifier].links:
  74. self.constant_in[item.name] = constant_in_links[0]
  75. else:
  76. self.constant_in[item.name] = None
  77. else: # OUTPUT
  78. if constant_out_links := self.node.outputs[item.identifier].links:
  79. self.constant_out[item.name] = constant_out_links.copy()
  80. else:
  81. self.constant_out[item.name] = []
  82. case 'Array':
  83. if item.in_out == 'INPUT':
  84. if item.identifier not in self.array_input_connections.keys():
  85. self.array_input_connections[item.identifier]=[]
  86. if in_links := self.node.inputs[item.identifier].links:
  87. self.array_input_connections[item.identifier]=in_links.copy()
  88. else: # OUTPUT
  89. if item.identifier not in self.array_output_connections.keys():
  90. self.array_output_connections[item.identifier]=[]
  91. if out_links := self.node.outputs[item.identifier].links:
  92. self.array_output_connections[item.identifier] = out_links.copy()
  93. def gen_solve_iteration_mantis_nodes(self, frame_mantis_nodes, unprepared):
  94. for prototype_ui_node in self.tree.nodes:
  95. if isinstance(prototype_ui_node, SchemaUINode):
  96. continue # IGNORE the schema interface nodes, we already made them in __init__()
  97. # they are reused for each iteration.
  98. elif prototype_ui_node.bl_idname in ['NodeFrame', 'NodeReroute']:
  99. continue # IGNORE stuff that is purely UI - frames, reroutes.
  100. signature = (*self.autogen_path_names, prototype_ui_node.name+self.index_str())
  101. prototype_mantis_node = self.all_nodes[(*self.signature, prototype_ui_node.name)]
  102. # the prototype_mantis_node was generated inside the schema when we parsed the tree.
  103. # it is the prototype of the mantis node which we make for this iteration
  104. # for Schema sub-nodes ... they need a prototype to init.
  105. if prototype_mantis_node.node_type in ['DUMMY', 'DUMMY_SCHEMA']:
  106. from .utilities import get_node_prototype
  107. ui_node = get_node_prototype(prototype_mantis_node.signature, prototype_mantis_node.base_tree)
  108. if prototype_mantis_node.node_type == 'DUMMY_SCHEMA':
  109. mantis_node = prototype_mantis_node.__class__(
  110. signature, prototype_mantis_node.base_tree, prototype=ui_node,
  111. natural_signature = (*self.node.signature, ui_node.name) )
  112. else:
  113. mantis_node = prototype_mantis_node.__class__(signature, prototype_mantis_node.base_tree, prototype=ui_node)
  114. else:
  115. mantis_node = prototype_mantis_node.__class__(signature, prototype_mantis_node.base_tree)
  116. frame_mantis_nodes[mantis_node.signature] = mantis_node
  117. if mantis_node.prepared == False:
  118. unprepared.append(mantis_node)
  119. if mantis_node.__class__.__name__ in custom_props_types:
  120. setup_custom_props_from_np(mantis_node, prototype_ui_node)
  121. mantis_node.fill_parameters(prototype_ui_node)
  122. def handle_link_from_index_input(self, index, frame_mantis_nodes, ui_link):
  123. from .utilities import get_link_in_out
  124. _from_name, to_name = get_link_in_out(ui_link)
  125. to_node = frame_mantis_nodes[ (*self.autogen_path_names, to_name+self.index_str()) ]
  126. if to_node.node_type in ['DUMMY', 'DUMMY_SCHEMA']:
  127. from .utilities import gen_nc_input_for_data
  128. nc_cls = gen_nc_input_for_data(ui_link.from_socket)
  129. if (nc_cls): #HACK
  130. sig = ("MANTIS_AUTOGENERATED", *self.tree_path_names[1:-1], self.index_str(), ui_link.from_socket.name, ui_link.from_socket.identifier)
  131. nc_from = nc_cls(sig, self.node.base_tree)
  132. # ugly! maybe even a HACK!
  133. nc_from.inputs = {}
  134. from .base_definitions import NodeSocket
  135. nc_from.outputs = {ui_link.from_socket.name:NodeSocket(name = ui_link.from_socket.name, node=nc_from)}
  136. from .base_definitions import get_socket_value
  137. nc_from.parameters = {ui_link.from_socket.name:index}
  138. frame_mantis_nodes[sig]=nc_from
  139. from_node = nc_from
  140. self.solved_nodes[sig]=from_node
  141. _connection = from_node.outputs[ui_link.from_socket.name].connect(node=to_node, socket=ui_link.to_socket.identifier)
  142. return
  143. # Since the index is already determined, it is safe to remove the socket and just keep the value.
  144. to_node.parameters[ui_link.to_socket.name] = index
  145. del to_node.inputs[ui_link.to_socket.name]
  146. def handle_link_from_schema_length_input(self, frame_mantis_nodes, ui_link):
  147. from .utilities import get_link_in_out
  148. # see, here I can just use the schema node
  149. _from_name, to_name = get_link_in_out(ui_link)
  150. to_node = frame_mantis_nodes[(*self.autogen_path_names, to_name+self.index_str())]
  151. # this self.index_link is only used here?
  152. if (self.index_link.from_node):
  153. connection = self.index_link.from_node.outputs[self.index_link.from_socket].connect(node=to_node, socket=ui_link.to_socket.name)
  154. # otherwise we can autogen an int input I guess...?
  155. else:
  156. raise RuntimeError("I was expecting there to be an incoming connection here for Schema Length")
  157. def handle_link_from_incoming_connection_input(self, frame_mantis_nodes, ui_link):
  158. from .utilities import get_link_in_out
  159. incoming = self.incoming_connections[ui_link.from_socket.name]
  160. from_node = incoming.from_node
  161. _from_name, to_name = get_link_in_out(ui_link)
  162. to_node = frame_mantis_nodes[ (*self.autogen_path_names, to_name+self.index_str()) ]
  163. connection = from_node.outputs[incoming.from_socket].connect(node=to_node, socket=ui_link.to_socket.name)
  164. init_connections(from_node)
  165. def handle_link_from_constant_input(self, frame_mantis_nodes, ui_link, to_ui_node):
  166. from .utilities import get_link_in_out
  167. incoming = self.constant_in[ui_link.from_socket.name]
  168. from_node = incoming.from_node
  169. to_name = get_link_in_out(ui_link)[1]
  170. to_node = frame_mantis_nodes[(*self.autogen_path_names, to_name+self.index_str())]
  171. to_socket=ui_link.to_socket.name
  172. from .base_definitions import SchemaGroup
  173. if isinstance(to_ui_node, SchemaGroup):
  174. to_socket=ui_link.to_socket.identifier
  175. connection = from_node.outputs[incoming.from_socket].connect(node=to_node, socket=to_socket)
  176. init_connections(from_node)
  177. def handle_link_to_array_input_get(self, frame_mantis_nodes, ui_link, index):
  178. from .utilities import get_link_in_out
  179. from_name, to_name = get_link_in_out(ui_link)
  180. from_nc = frame_mantis_nodes[(*self.autogen_path_names, from_name+self.index_str())]
  181. to_nc = self.schema_nodes[(*self.tree_path_names, to_name)]
  182. # this only needs to be done once:
  183. if index == 0: # BUG? HACK? TODO find out what is going on here.
  184. # Kill the link between the schema node group and the node connecting to it
  185. old_nc = self.all_nodes[(*self.tree_path_names, from_name)]
  186. # I am not sure about this!
  187. existing_link = old_nc.outputs[ui_link.from_socket.name].links[0]
  188. existing_link.die()
  189. #
  190. connection = from_nc.outputs[ui_link.from_socket.name].connect(node=to_nc, socket=ui_link.to_socket.name)
  191. def handle_link_from_array_input(self, frame_mantis_nodes, ui_link, index):
  192. from .utilities import get_link_in_out
  193. get_index = index
  194. try:
  195. incoming = self.array_input_connections[ui_link.from_socket.identifier][get_index]
  196. except IndexError:
  197. if len(self.array_input_connections[ui_link.from_socket.identifier]) > 0:
  198. incoming = self.array_input_connections[ui_link.from_socket.identifier][0]
  199. # prOrange(incoming.from_node.node_type)
  200. if incoming.from_node.node_type not in ['DUMMY_SCHEMA']:
  201. raise NotImplementedError(wrapRed("dev: make it so Mantis checks if there are enough Array inputs."))
  202. else: # do nothing
  203. return
  204. else:
  205. raise RuntimeError(wrapRed("make it so Mantis checks if there are enough Array inputs!"))
  206. to_name = get_link_in_out(ui_link)[1]
  207. to_node = frame_mantis_nodes[(*self.autogen_path_names, to_name+self.index_str())]
  208. connection = incoming.from_node.outputs[incoming.from_socket].connect(node=to_node, socket=ui_link.to_socket.name)
  209. init_connections(incoming.from_node)
  210. def handle_link_to_constant_output(self, frame_mantis_nodes, index, ui_link, to_ui_node):
  211. from .utilities import get_link_in_out
  212. to_node = self.schema_nodes[(*self.tree_path_names, to_ui_node.bl_idname)]
  213. expose_when = to_node.evaluate_input('Expose when N==')
  214. if index == expose_when:
  215. for outgoing in self.constant_out[ui_link.to_socket.name]:
  216. to_node = outgoing.to_node
  217. from_name = get_link_in_out(ui_link)[0]
  218. from_node = frame_mantis_nodes[(*self.autogen_path_names, from_name+self.index_str()) ]
  219. connection = from_node.outputs[ui_link.from_socket.name].connect(node=to_node, socket=outgoing.to_socket)
  220. # WTF is even happening here?? TODO BUG HACK
  221. def handle_link_to_array_output(self, frame_mantis_nodes, index, ui_link, to_ui_node, from_ui_node):# if this duplicated code works, dedupe!
  222. from .utilities import get_link_in_out
  223. to_node = self.schema_nodes[(*self.tree_path_names, to_ui_node.bl_idname)] # get it by [], we want a KeyError if this fails
  224. for outgoing in self.array_output_connections[ui_link.to_socket.identifier]:
  225. # print (type(outgoing))
  226. from .schema_containers import SchemaIndex
  227. from_name = get_link_in_out(ui_link)[0]
  228. from_node = frame_mantis_nodes[ (*self.autogen_path_names, from_name+self.index_str()) ]
  229. if not from_node:
  230. from_node = self.schema_nodes[(*self.tree_path_names, from_ui_node.bl_idname)]
  231. to_node = outgoing.to_node
  232. if isinstance(from_node, SchemaIndex): # I think I need to dedup this stuff
  233. # print("INDEX")
  234. from .utilities import gen_nc_input_for_data
  235. nc_cls = gen_nc_input_for_data(ui_link.from_socket)
  236. if (nc_cls): #HACK
  237. sig = ("MANTIS_AUTOGENERATED", *self.tree_path_names[1:-1], self.index_str(), ui_link.from_socket.name, ui_link.from_socket.identifier)
  238. nc_from = nc_cls(sig, self.node.base_tree)
  239. # ugly! maybe even a HACK!
  240. nc_from.inputs = {}
  241. from .node_container_common import NodeSocket
  242. nc_from.outputs = {ui_link.from_socket.name:NodeSocket(name = ui_link.from_socket.name, node=nc_from)}
  243. from .node_container_common import get_socket_value
  244. if ui_link.from_socket.name in ['Index']:
  245. nc_from.parameters = {ui_link.from_socket.name:index}
  246. else:
  247. nc_from.parameters = {ui_link.from_socket.name:self.solve_length}
  248. frame_mantis_nodes[sig]=nc_from
  249. from_node = nc_from
  250. self.solved_nodes[sig]=from_node
  251. # I have a feeling that something bad will happen if both of these conditions (above and below) are true
  252. if to_node.node_type == 'DUMMY_SCHEMA' and to_node.prepared:
  253. other_stem = ('SCHEMA_AUTOGENERATED', *to_node.signature[1:])
  254. from .utilities import get_node_prototype
  255. other_schema_np = get_node_prototype(to_node.signature, to_node.base_tree)
  256. other_schema_tree = other_schema_np.node_tree
  257. for n in other_schema_tree.nodes:
  258. if n.bl_idname not in ["SchemaArrayInput", "SchemaArrayInputGet"]:
  259. continue
  260. out = n.outputs[outgoing.to_socket]
  261. for l in out.links:
  262. other_index_str = lambda : '.'+str(to_node.uuid)+'.'+str(index).zfill(4)
  263. # get it by [], we want a KeyError if this fails
  264. try:
  265. out_node = self.all_nodes[(*other_stem, l.to_node.name+other_index_str())]
  266. except KeyError as e:
  267. for n in self.all_nodes:
  268. if len(n) > len(other_stem)+1: break
  269. for elem in other_stem:
  270. if elem not in n: break
  271. else:
  272. print(n)
  273. raise e
  274. connection = from_node.outputs[ui_link.from_socket.name].connect(node=out_node, socket=l.to_socket.name)
  275. else:
  276. connection = from_node.outputs[ui_link.from_socket.name].connect(node=to_node, socket=outgoing.to_socket)
  277. def handle_link_from_array_input_get(self, frame_mantis_nodes, index, ui_link, from_ui_node ):
  278. from .utilities import get_link_in_out
  279. get_index = index
  280. from_node = self.schema_nodes[(*self.tree_path_names, from_ui_node.bl_idname)]
  281. from .utilities import cap, wrap
  282. get_index = from_node.evaluate_input("Index", index)
  283. oob = from_node.evaluate_input("OoB Behaviour")
  284. # we must assume that the array has sent the correct number of links
  285. if oob == 'WRAP':
  286. get_index = wrap(get_index, len(self.array_input_connections[ui_link.from_socket.identifier])-1, 0)
  287. if oob == 'HOLD':
  288. get_index = cap(get_index, len(self.array_input_connections[ui_link.from_socket.identifier])-1)
  289. try:
  290. incoming = self.array_input_connections[ui_link.from_socket.identifier][get_index]
  291. except IndexError:
  292. raise RuntimeError(wrapRed("Dummy! You need to make it so Mantis checks if there are enough Array inputs! It should probably have a Get Index!"))
  293. to_name = get_link_in_out(ui_link)[1]
  294. to_node = frame_mantis_nodes[(*self.autogen_path_names, to_name+self.index_str())]
  295. connection = incoming.from_node.outputs[incoming.from_socket].connect(node=to_node, socket=ui_link.to_socket.name)
  296. init_connections(incoming.from_node)
  297. def prepare_nodes(self, unprepared):
  298. # At this point, we've already run a pretty exhaustive preperation phase to prep the schema's dependencies
  299. # So we should not need to add any new dependencies unless there is a bug elsewhere.
  300. # and in fact, I could skip this in some cases, and should investigate if profiling reveals a slowdown here.
  301. while unprepared:
  302. nc = unprepared.pop()
  303. if sum([dep.prepared for dep in nc.hierarchy_dependencies]) == len(nc.hierarchy_dependencies):
  304. nc.bPrepare()
  305. if nc.node_type == 'DUMMY_SCHEMA':
  306. self.solve_nested_schema(nc)
  307. else: # Keeping this for-loop as a fallback, it should never add dependencies though
  308. for dep in nc.hierarchy_dependencies:
  309. if not dep.prepared and dep not in unprepared:
  310. prOrange(f"Adding dependency... {dep}")
  311. unprepared.appendleft(dep)
  312. unprepared.appendleft(nc) # just rotate them until they are ready.
  313. def solve_iteration(self):
  314. """ Solve an iteration of the schema.
  315. - 1 Create the Mantis Node instances that represent this iteration of the schema
  316. - 2 Connect the links from the entrypoint or previous iteration.
  317. - 3 Connect the constant and array links, and any link between nodes entirely within the tree
  318. - 4 Prepare the nodes that modify data (in case of e.g. array get index or nested schema length input)
  319. - 5 Connect the final prepared nodes
  320. and return the nodes that were created in this schema iteration (frame).
  321. This function also adds to held_links to pass data between iterations.
  322. """
  323. from .schema_definitions import (SchemaIndex,
  324. SchemaArrayInput,
  325. SchemaArrayInputGet,
  326. SchemaArrayOutput,
  327. SchemaConstInput,
  328. SchemaConstOutput,
  329. SchemaOutgoingConnection,
  330. SchemaIncomingConnection,)
  331. from .utilities import clear_reroutes
  332. from .utilities import get_link_in_out, link_node_containers
  333. self.set_index_strings()
  334. frame_mantis_nodes = {}
  335. # Later we have to run bPrepare() on these guys, so we make the deque and fill it now.
  336. from collections import deque
  337. unprepared= deque()
  338. self.gen_solve_iteration_mantis_nodes(frame_mantis_nodes, unprepared)
  339. # This is where we handle node connections BETWEEN frames
  340. while(self.held_links):
  341. ui_link = self.held_links.pop()
  342. to_ui_node = ui_link.to_socket.node; from_ui_node = ui_link.from_socket.node
  343. if isinstance(to_ui_node, SchemaOutgoingConnection):
  344. mantis_incoming_node = self.schema_nodes[*self.tree_path_names, 'SchemaIncomingConnection']
  345. for mantis_link in mantis_incoming_node.outputs[ui_link.to_socket.name].links:
  346. to_mantis_node, to_mantis_socket = mantis_link.to_node, mantis_link.to_socket
  347. from_name = get_link_in_out(ui_link)[0]
  348. from_mantis_node = self.solved_nodes[ (*self.autogen_path_names, from_name+self.prev_index_str()) ]
  349. to_mantis_node = frame_mantis_nodes[ (*self.autogen_path_names, to_mantis_node.signature[-1]+self.index_str()) ]
  350. connection = from_mantis_node.outputs[ui_link.from_socket.name].connect(node=to_mantis_node, socket=to_mantis_socket)
  351. # We want to delete the links from the tree into the schema node.
  352. # TODO: this is not robust enough and I do not feel sure this is doing the right thing.
  353. if existing_link := self.incoming_connections[ui_link.to_socket.name]:
  354. if existing_link.to_node == self.node:
  355. print ("Deleting...", existing_link)
  356. if self.node.signature[-1] in existing_link.to_node.signature:
  357. existing_link.die()
  358. # BUG may exist here.
  359. self.incoming_connections[ui_link.to_socket.name] = connection
  360. # Get the rerouted links from the graph. We don't really need to do this every iteration.
  361. # TODO: use profiling to determine if this is slow; if so: copy & reuse the data, refactor the pop()'s out.
  362. ui_links = clear_reroutes(list(self.tree.links))
  363. # Now we handle ui_links in the current frame, including those ui_links between Schema nodes and "real" nodes
  364. awaiting_prep_stage = []
  365. for ui_link in ui_links:
  366. to_ui_node = ui_link.to_socket.node; from_ui_node = ui_link.from_socket.node
  367. if isinstance(from_ui_node, SchemaIndex):
  368. if ui_link.from_socket.name == "Index":
  369. self.handle_link_from_index_input(self.index, frame_mantis_nodes, ui_link)
  370. elif ui_link.from_socket.name == "Schema Length":
  371. self.handle_link_from_schema_length_input(frame_mantis_nodes, ui_link)
  372. continue
  373. if isinstance(from_ui_node, SchemaIncomingConnection):
  374. if ui_link.from_socket.name in self.incoming_connections.keys():
  375. self.handle_link_from_incoming_connection_input(frame_mantis_nodes, ui_link)
  376. continue
  377. if isinstance(from_ui_node, SchemaConstInput):
  378. if ui_link.from_socket.name in self.constant_in.keys():
  379. self.handle_link_from_constant_input( frame_mantis_nodes, ui_link, to_ui_node)
  380. continue
  381. if isinstance(to_ui_node, SchemaArrayInputGet):
  382. self.handle_link_to_array_input_get( frame_mantis_nodes, ui_link, self.index)
  383. continue
  384. if isinstance(from_ui_node, SchemaArrayInput):
  385. self.handle_link_from_array_input(frame_mantis_nodes, ui_link, self.index)
  386. continue
  387. # HOLD these links to the next iteration:
  388. if isinstance(to_ui_node, SchemaOutgoingConnection):
  389. self.held_links.append(ui_link)
  390. continue
  391. # HOLD these links until prep is done a little later
  392. if isinstance(to_ui_node, SchemaConstOutput) or isinstance(to_ui_node, SchemaArrayOutput) or \
  393. isinstance(from_ui_node, SchemaArrayInputGet):
  394. awaiting_prep_stage.append(ui_link)
  395. continue
  396. # for any of the special cases, we hit a 'continue' block. So this connection is not special, and is made here.
  397. connection = link_node_containers(self.autogen_path_names, ui_link, frame_mantis_nodes, from_suffix=self.index_str(), to_suffix=self.index_str())
  398. for k,v in frame_mantis_nodes.items():
  399. self.solved_nodes[k]=v
  400. init_dependencies(v) # it is hard to overstate how important this single line of code is
  401. self.prepare_nodes(unprepared)
  402. while(awaiting_prep_stage):
  403. ui_link = awaiting_prep_stage.pop()
  404. to_ui_node = ui_link.to_socket.node; from_ui_node = ui_link.from_socket.node
  405. if isinstance(to_ui_node, SchemaConstOutput):
  406. self.handle_link_to_constant_output(frame_mantis_nodes, self.index, ui_link, to_ui_node)
  407. if isinstance(to_ui_node, SchemaArrayOutput):
  408. self.handle_link_to_array_output(frame_mantis_nodes, self.index, ui_link, to_ui_node, from_ui_node)
  409. if isinstance(from_ui_node, SchemaArrayInputGet):
  410. self.handle_link_from_array_input_get(frame_mantis_nodes, self.index, ui_link, from_ui_node )
  411. # end seciton
  412. return frame_mantis_nodes
  413. def solve_nested_schema(self, schema_nc):
  414. """ Solves all schema node groups found in this Schema. This is a recursive function, which will
  415. solve all levels of nested schema - since this function is called by solver.solve().
  416. """
  417. if schema_nc.prepared == False:
  418. all_nodes = self.all_nodes.copy()
  419. ui_node = schema_nc.prototype
  420. length = schema_nc.evaluate_input("Schema Length")
  421. tree = ui_node.node_tree
  422. prOrange(f"Expanding schema {tree.name} in node {schema_nc} with length {length}.")
  423. solver = SchemaSolver(schema_nc, all_nodes, ui_node, schema_nc.natural_signature)
  424. solved_nodes = solver.solve()
  425. schema_nc.prepared = True
  426. for k,v in solved_nodes.items():
  427. self.solved_nodes[k]=v
  428. def finalize(self, frame_nc):
  429. from .schema_definitions import (SchemaOutgoingConnection,)
  430. for i in range(len(self.held_links)):
  431. link = self.held_links.pop()
  432. to_np = link.to_socket.node; from_np = link.from_socket.node
  433. if isinstance(to_np, SchemaOutgoingConnection):
  434. if link.to_socket.name in self.outgoing_connections.keys():
  435. if (outgoing_links := self.outgoing_connections[link.to_socket.name]) is None: continue
  436. for outgoing in outgoing_links:
  437. if outgoing:
  438. to_node = outgoing.to_node
  439. from_node =frame_nc[(*self.autogen_path_names, from_np.name+self.index_str()) ]
  440. connection = from_node.outputs[link.from_socket.name].connect(node=to_node, socket=outgoing.to_socket)
  441. # we need to kill the link between the Schema itself and the next node and update the deps. Otherwise:confusing bugs.
  442. outgoing.die(); init_dependencies(to_node)
  443. # else: # the node just isn't connected out this socket.
  444. # # solve all unsolved nested schemas...
  445. for schema_sig, schema_nc in self.nested_schemas.items():
  446. self.solve_nested_schema(schema_nc)
  447. for n in self.autogenerated_nodes.values():
  448. init_connections(n)
  449. for c in n.connections:
  450. init_dependencies(c)
  451. all_outgoing_links = []
  452. for conn in self.outgoing_connections.values():
  453. for outgoing in conn:
  454. all_outgoing_links.append(outgoing)
  455. for conn in self.constant_out.values():
  456. for outgoing in conn:
  457. all_outgoing_links.append(outgoing)
  458. for conn in self.array_output_connections.values():
  459. for outgoing in conn:
  460. all_outgoing_links.append(outgoing)
  461. for outgoing in all_outgoing_links:
  462. to_node = outgoing.to_node
  463. for l in to_node.inputs[outgoing.to_socket].links:
  464. if self.node == l.from_node:
  465. l.die()
  466. for inp in self.node.inputs.values():
  467. for l in inp.links:
  468. init_connections(l.from_node) # to force it to have hierarchy connections with the new nodes.
  469. def solve(self):
  470. for index in range(self.solve_length):
  471. self.index = index
  472. frame_mantis_nodes = self.solve_iteration()
  473. for sig, nc in frame_mantis_nodes.items():
  474. if nc.node_type == 'DUMMY_SCHEMA':
  475. self.nested_schemas[sig] = nc
  476. self.finalize(frame_mantis_nodes)
  477. self.node.solver = self
  478. return self.solved_nodes