readtree.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605
  1. from .utilities import prRed, prGreen, prPurple, prWhite, prOrange, \
  2. wrapRed, wrapGreen, wrapPurple, wrapWhite, wrapOrange
  3. from .utilities import gen_nc_input_for_data
  4. def grp_node_reroute_common(nc, nc_to, all_nc):
  5. # we need to do this: go to the to-node
  6. # then reroute the link in the to_node all the way to the beginning
  7. # so that the number of links in "real" nodes is unchanged
  8. # then the links in the dummy nodes need to be deleted
  9. for inp_name, inp in nc.inputs.items():
  10. # assume each input socket only has one input for now
  11. if inp.is_connected:
  12. while (inp.links):
  13. in_link = inp.links.pop()
  14. from_nc = in_link.from_node
  15. from_socket = in_link.from_socket
  16. links = []
  17. from_links = from_nc.outputs[from_socket].links.copy()
  18. while(from_links):
  19. from_link = from_links.pop()
  20. if from_link == in_link:
  21. from_link.die()
  22. continue # DELETE the dummy node link
  23. links.append(from_link)
  24. from_nc.outputs[from_socket].links = links
  25. down = nc_to.outputs[inp_name]
  26. for downlink in down.links:
  27. downlink.from_node = from_nc
  28. downlink.from_socket = from_socket
  29. from_nc.outputs[from_socket].links.append(downlink)
  30. if hasattr(downlink.to_node, "reroute_links"):
  31. downlink.to_node.reroute_links(downlink.to_node, all_nc)
  32. in_link.die()
  33. def reroute_links_grp(nc, all_nc):
  34. if nc.inputs:
  35. if (nc_to := all_nc.get( ( *nc.signature, "NodeGroupInput") )):
  36. grp_node_reroute_common(nc, nc_to, all_nc)
  37. else:
  38. raise RuntimeError("internal error: failed to enter a node group ")
  39. def reroute_links_grpout(nc, all_nc):
  40. if (nc_to := all_nc.get( ( *nc.signature[:-1],) )):
  41. grp_node_reroute_common(nc, nc_to, all_nc)
  42. else:
  43. raise RuntimeError("error leaving a node group (maybe you are running the tree from inside a node group?)")
  44. # FIXME I don't think these signatures are unique.
  45. def insert_lazy_parents(nc):
  46. from .link_containers import LinkInherit
  47. from .base_definitions import NodeLink
  48. inherit_nc = None
  49. if nc.inputs["Relationship"].is_connected:
  50. link = nc.inputs["Relationship"].links[0]
  51. # print(nc)
  52. from_nc = link.from_node
  53. if from_nc.node_type in ["XFORM"] and link.from_socket in ["xForm Out"]:
  54. inherit_nc = LinkInherit(("MANTIS_AUTOGENERATED", *nc.signature[1:], "LAZY_INHERIT"), nc.base_tree)
  55. for from_link in from_nc.outputs["xForm Out"].links:
  56. if from_link.to_node == nc and from_link.to_socket == "Relationship":
  57. break # this is it
  58. from_link.to_node = inherit_nc; from_link.to_socket="Parent"
  59. from_link.to_node.inputs[from_link.to_socket].is_linked=True
  60. links=[]
  61. while (nc.inputs["Relationship"].links):
  62. to_link = nc.inputs["Relationship"].links.pop()
  63. if to_link.from_node == from_nc and to_link.from_socket == "xForm Out":
  64. continue # don't keep this one
  65. links.append(to_link)
  66. to_link.from_node.outputs[from_link.from_socket].is_linked=True
  67. nc.inputs["Relationship"].links=links
  68. link=NodeLink(from_node=inherit_nc, from_socket="Inheritance", to_node=nc, to_socket="Relationship")
  69. inherit_nc.inputs["Parent"].links.append(from_link)
  70. inherit_nc.parameters = {
  71. "Parent":None,
  72. "Inherit Rotation":True,
  73. "Inherit Scale":'FULL',
  74. "Connected":False,
  75. }
  76. # because the from node may have already been done.
  77. init_connections(from_nc)
  78. init_dependencies(from_nc)
  79. init_connections(inherit_nc)
  80. init_dependencies(inherit_nc)
  81. return inherit_nc
  82. # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** #
  83. # DATA FROM NODES #
  84. # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** #
  85. from .base_definitions import replace_types, NodeSocket
  86. # TODO: investigate whether I can set the properties in the downstream nodes directly.
  87. # I am doing this in Schema Solver and it seems to work quite efficiently.
  88. def make_connections_to_ng_dummy(base_tree, tree_path_names, local_nc, all_nc, nc_to):
  89. np = nc_to.prototype
  90. for inp in np.inputs:
  91. nc_from = None
  92. if inp.bl_idname in ['WildcardSocket']:
  93. continue # it isn't a real input so I don't think it is good to check it.
  94. to_s = inp.identifier
  95. if not inp.is_linked: # make an autogenerated NC for the inputs of the group node
  96. if inp.bl_idname in ['xFormSocket']:
  97. continue
  98. from .node_container_common import get_socket_value
  99. nc_cls = gen_nc_input_for_data(inp)
  100. if (nc_cls):
  101. sig = ("MANTIS_AUTOGENERATED", *tree_path_names, np.name, inp.name, inp.identifier)
  102. nc_from = nc_cls(sig, base_tree)
  103. # ugly! maybe even a HACK!
  104. nc_from.inputs = {}
  105. nc_from.outputs = {inp.name:NodeSocket(name = inp.name, node=nc_from)}
  106. nc_from.parameters = {inp.name:get_socket_value(inp)}
  107. #
  108. local_nc[sig] = nc_from; all_nc[sig] = nc_from
  109. from_s = inp.name
  110. nc_from.execution_prepared=True
  111. else:
  112. prRed("No available auto-generated class for input", *tree_path_names, np.name, inp.name)
  113. nc_from.outputs[from_s].connect(node=nc_to, socket=to_s, sort_id=0)
  114. def gen_node_containers(base_tree, current_tree, tree_path_names, all_nc, local_nc, dummy_nodes, group_nodes, schema_nodes ):
  115. from .internal_containers import DummyNode
  116. for ui_node in current_tree.nodes:
  117. # HACK I found that this isn't being set sometimes. I wonder why? It makes the most sense to do this here.
  118. if hasattr(ui_node, 'initialized'): ui_node.initialized=True
  119. # end HACK. TODO: find out why this is not set sometimes. This is only needed for UI socket change updates.
  120. if ui_node.bl_idname in ["NodeFrame", "NodeReroute"]:
  121. continue # not a Mantis Node
  122. if ui_node.bl_idname in ["NodeGroupInput", "NodeGroupOutput"]:
  123. # we only want ONE dummy in/out per tree_path, so use the bl_idname to make a Dummy node
  124. sig = (None, *tree_path_names, ui_node.bl_idname)
  125. if not local_nc.get(sig):
  126. nc = DummyNode( signature=sig , base_tree=base_tree, prototype=ui_node )
  127. local_nc[sig] = nc; all_nc[sig] = nc; dummy_nodes[sig] = nc
  128. if ui_node.bl_idname in ["NodeGroupOutput"]:
  129. nc.reroute_links = reroute_links_grpout
  130. elif ui_node.bl_idname in ["MantisNodeGroup", "MantisSchemaGroup"]:
  131. nc = DummyNode( signature= (sig := (None, *tree_path_names, ui_node.name) ), base_tree=base_tree, prototype=ui_node )
  132. local_nc[sig] = nc; all_nc[sig] = nc; dummy_nodes[sig] = nc
  133. make_connections_to_ng_dummy(base_tree, tree_path_names, local_nc, all_nc, nc)
  134. if ui_node.bl_idname == "MantisNodeGroup":
  135. group_nodes.append(nc)
  136. nc.reroute_links = reroute_links_grp
  137. else:
  138. group_nodes.append(nc)
  139. schema_nodes[sig] = nc
  140. # if it wasn't the types we ignore or the types we make a Dummy for, use this to catch all non-special cases.
  141. elif (nc_cls := ui_node.mantis_class):
  142. sig = (None, *tree_path_names, ui_node.name)
  143. if ui_node.bl_idname in replace_types:
  144. sig = (None, *tree_path_names, ui_node.bl_idname)
  145. if local_nc.get(sig):
  146. continue # already made
  147. nc = nc_cls( sig , base_tree)
  148. local_nc[sig] = nc; all_nc[sig] = nc
  149. else:
  150. nc = None
  151. prRed(f"Can't make nc for.. {ui_node.bl_idname}")
  152. # this should be done at init
  153. if nc.signature[0] not in ['MANTIS_AUTOGENERATED'] and nc.node_type not in ['SCHEMA', 'DUMMY', 'DUMMY_SCHEMA']:
  154. nc.fill_parameters()
  155. def data_from_tree(base_tree, tree_path, dummy_nodes, all_nc, all_schema):#
  156. # TODO: it should be relatively easy to make this use a while loop instead of recursion.
  157. local_nc, group_nodes = {}, []
  158. tree_path_names = [tree.name for tree in tree_path if hasattr(tree, "name")]
  159. if tree_path[-1]:
  160. current_tree = tree_path[-1].node_tree # this may be None.
  161. else:
  162. current_tree = base_tree
  163. #
  164. if current_tree: # the node-group may not have a tree set - if so, ignore it.
  165. from .utilities import clear_reroutes
  166. links = clear_reroutes(list(current_tree.links))
  167. gen_node_containers(base_tree, current_tree, tree_path_names, all_nc, local_nc, dummy_nodes, group_nodes, all_schema)
  168. from .utilities import link_node_containers
  169. for link in links:
  170. link_node_containers((None, *tree_path_names), link, local_nc)
  171. # Now, descend into the Node Groups and recurse
  172. for nc in group_nodes:
  173. data_from_tree(base_tree, tree_path+[nc.prototype], dummy_nodes, all_nc, all_schema)
  174. return dummy_nodes, all_nc, all_schema
  175. from .utilities import check_and_add_root, init_connections, init_dependencies, init_schema_dependencies
  176. def is_signature_in_other_signature(parent_signature, child_signature):
  177. # If the other signature is shorter, it isn't a child node
  178. if len(parent_signature) > len(child_signature):
  179. return False
  180. return parent_signature[0:] == child_signature[:len(parent_signature)]
  181. def solve_schema_to_tree(nc, all_nc, roots=[], error_popups=False):
  182. from .utilities import get_node_prototype
  183. np = get_node_prototype(nc.signature, nc.base_tree)
  184. from .schema_solve import SchemaSolver
  185. tree = np.node_tree
  186. length = nc.evaluate_input("Schema Length")
  187. prOrange(f"Expanding schema {tree.name} in node {nc} with length {length}.")
  188. solver = SchemaSolver(nc, all_nc, np, error_popups=error_popups)
  189. solved_nodes = solver.solve()
  190. prWhite(f"Schema declared {len(solved_nodes)} nodes.")
  191. # maybe this should be done in schema solver. TODO invesitigate a more efficient way
  192. del_me = []
  193. for k, v in all_nc.items():
  194. # delete all the schema's internal nodes. The links have already been deleted by the solver.
  195. if v.signature[0] not in ['MANTIS_AUTOGENERATED'] and is_signature_in_other_signature(nc.signature, k):
  196. del_me.append(k)
  197. for k in del_me:
  198. del all_nc[k]
  199. for k,v in solved_nodes.items():
  200. all_nc[k]=v
  201. init_connections(v)
  202. check_and_add_root(v, roots, include_non_hierarchy=True)
  203. return solved_nodes
  204. # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** #
  205. # PARSE NODE TREE #
  206. # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** # *** #
  207. schema_bl_idnames = [ "SchemaIndex",
  208. "SchemaArrayInput",
  209. "SchemaArrayInputGet",
  210. "SchemaArrayInputAll",
  211. "SchemaArrayOutput",
  212. "SchemaConstInput",
  213. "SchemaConstOutput",
  214. "SchemaOutgoingConnection",
  215. "SchemaIncomingConnection",
  216. ]
  217. from .utilities import get_all_dependencies
  218. def get_schema_length_dependencies(node, all_nodes={}):
  219. """ Get a list of all dependencies for the given node's length or array properties.
  220. This function will also recursively search for dependencies in its sub-trees.
  221. """
  222. deps = []
  223. prepare_links_to = ['Schema Length', 'Array', 'Index']
  224. def extend_dependencies_from_inputs(node):
  225. for inp in node.inputs.values():
  226. for l in inp.links:
  227. if not l.from_node in node.hierarchy_dependencies:
  228. continue
  229. if "MANTIS_AUTOGENERATED" in l.from_node.signature:
  230. deps.extend([l.from_node]) # why we need this lol
  231. if inp.name in prepare_links_to:
  232. deps.append(l.from_node)
  233. deps.extend(get_all_dependencies(l.from_node))
  234. def deps_filter(dep): # remove any nodes inside the schema
  235. if len(dep.signature) > len(node.signature):
  236. for i in range(len(node.signature)):
  237. dep_sig_elem, node_sig_elem = dep.signature[i], node.signature[i]
  238. if dep_sig_elem != node_sig_elem: break # they don't match, it isn't an inner-node
  239. else: # remove this, it didn't break, meaning it shares signature with outer node
  240. return False # this is an inner-node
  241. return True
  242. # this way we can handle Schema and Array Get nodes with one function
  243. extend_dependencies_from_inputs(node)
  244. if node.node_type == 'DUMMY_SCHEMA':
  245. trees = [(node.prototype.node_tree, node.signature)] # this is UI data
  246. while trees:
  247. tree, tree_signature = trees.pop()
  248. for sub_ui_node in tree.nodes:
  249. if sub_ui_node.bl_idname in ['NodeReroute', 'NodeFrame']:
  250. continue
  251. if sub_ui_node.bl_idname in schema_bl_idnames:
  252. sub_node = all_nodes[(*tree_signature, sub_ui_node.bl_idname)]
  253. else:
  254. sub_node = all_nodes[(*tree_signature, sub_ui_node.name)]
  255. if sub_node.node_type == 'DUMMY_SCHEMA':
  256. extend_dependencies_from_inputs(sub_node)
  257. trees.append((sub_node.prototype.node_tree, sub_node.signature))
  258. return list(filter(deps_filter, deps))
  259. def parse_tree(base_tree, error_popups=False):
  260. from uuid import uuid4
  261. base_tree.execution_id = uuid4().__str__() # set the unique id of this execution
  262. from .base_definitions import MantisExecutionContext
  263. mContext = MantisExecutionContext(base_tree=base_tree)
  264. import time
  265. data_start_time = time.time()
  266. # annoyingly I have to pass in values for all of the dicts because if I initialize them in the function call
  267. # then they stick around because the function definition inits them once and keeps a reference
  268. # so instead I have to supply them to avoid ugly code or bugs elsewhere
  269. # it's REALLY confusing when you run into this sort of problem. So it warrants four entire lines of comments!
  270. dummy_nodes, all_mantis_nodes, all_schema = data_from_tree(base_tree, tree_path = [None], dummy_nodes = {}, all_nc = {}, all_schema={})
  271. for dummy in dummy_nodes.values(): # reroute the links in the group nodes
  272. if (hasattr(dummy, "reroute_links")):
  273. dummy.reroute_links(dummy, all_mantis_nodes)
  274. prGreen(f"Pulling data from tree took {time.time() - data_start_time} seconds")
  275. start_time = time.time()
  276. solve_only_these = []; solve_only_these.extend(list(all_schema.values()))
  277. roots, array_nodes = [], []
  278. from collections import deque
  279. unsolved_schema = deque()
  280. from .base_definitions import array_output_types, GraphError
  281. for mantis_node in all_mantis_nodes.values():
  282. # add the Mantis Context here, so that it available during parsing.
  283. mantis_node.mContext = mContext
  284. if mantis_node.node_type in ["DUMMY"]: # clean up the groups
  285. if mantis_node.prototype.bl_idname in ("MantisNodeGroup", "NodeGroupOutput"):
  286. continue
  287. # Initialize the dependencies and connections (from/to links) for each node.
  288. # we record & store it because using a getter is much slower (according to profiling)
  289. init_dependencies(mantis_node); init_connections(mantis_node)
  290. check_and_add_root(mantis_node, roots, include_non_hierarchy=True)
  291. # Array nodes need a little special treatment, they're quasi-schemas
  292. if mantis_node.__class__.__name__ in array_output_types:
  293. solve_only_these.append(mantis_node)
  294. array_nodes.append(mantis_node)
  295. from itertools import chain
  296. for schema in chain(all_schema.values(), array_nodes):
  297. # We must remove the schema/array nodes that are inside a schema tree.
  298. for i in range(len(schema.signature)-1): # -1, we don't want to check this node, obviously
  299. if parent := all_schema.get(schema.signature[:i+1]):
  300. # This will be solved along with its parent schema.
  301. solve_only_these.remove(schema)
  302. break
  303. for schema in all_schema.values():
  304. if schema not in solve_only_these: continue
  305. init_schema_dependencies(schema, all_mantis_nodes)
  306. solve_only_these.extend(get_schema_length_dependencies(schema, all_mantis_nodes))
  307. unsolved_schema.append(schema)
  308. for array in array_nodes:
  309. if array not in solve_only_these: continue
  310. solve_only_these.extend(get_schema_length_dependencies(array))
  311. solve_only_these.extend(array_nodes)
  312. schema_solve_done = set()
  313. solve_only_these = set(solve_only_these)
  314. solve_layer = unsolved_schema.copy(); solve_layer.extend(roots)
  315. while(solve_layer):
  316. n = solve_layer.pop()
  317. if n not in solve_only_these:
  318. continue
  319. if n.signature in all_schema.keys():
  320. for dep in n.hierarchy_dependencies:
  321. if dep not in schema_solve_done and (dep in solve_only_these):
  322. if dep.prepared:
  323. continue
  324. solve_layer.appendleft(n)
  325. break
  326. else:
  327. solved_nodes = solve_schema_to_tree(n, all_mantis_nodes, roots, error_popups=error_popups)
  328. unsolved_schema.remove(n)
  329. schema_solve_done.add(n)
  330. for node in solved_nodes.values():
  331. #
  332. init_dependencies(node)
  333. init_connections(node)
  334. #
  335. solve_layer.appendleft(node)
  336. for conn in n.hierarchy_connections:
  337. if conn not in schema_solve_done and conn not in solve_layer:
  338. solve_layer.appendleft(conn)
  339. continue
  340. else:
  341. for dep in n.hierarchy_dependencies:
  342. if dep not in schema_solve_done:
  343. break
  344. else:
  345. try:
  346. n.bPrepare()
  347. except Exception as e:
  348. e = execution_error_cleanup(n, e, show_error=error_popups)
  349. if error_popups == False:
  350. raise e
  351. schema_solve_done.add(n)
  352. for conn in n.hierarchy_connections:
  353. if conn not in schema_solve_done and conn not in solve_layer:
  354. solve_layer.appendleft(conn)
  355. continue
  356. if unsolved_schema:
  357. raise RuntimeError("Failed to resolve all schema declarations")
  358. # I had a problem with this looping forever. I think it is resolved... but I don't know lol
  359. all_mantis_nodes = list(all_mantis_nodes.values())
  360. kept_nc = {}
  361. while (all_mantis_nodes):
  362. nc = all_mantis_nodes.pop()
  363. if nc in array_nodes:
  364. continue
  365. if nc.node_type in ["DUMMY"]:
  366. continue
  367. # cleanup autogen nodes
  368. if nc.signature[0] == "MANTIS_AUTOGENERATED" and len(nc.inputs) == 0 and len(nc.outputs) == 1:
  369. output=list(nc.outputs.values())[0]
  370. value=list(nc.parameters.values())[0] # IDEA modify the dependecy get function to exclude these nodes completely
  371. for l in output.links:
  372. to_node = l.to_node; to_socket = l.to_socket
  373. l.die()
  374. to_node.parameters[to_socket] = value
  375. del to_node.inputs[to_socket]
  376. init_dependencies(to_node)
  377. continue
  378. if (nc.node_type in ['XFORM']) and ("Relationship" in nc.inputs.keys()):
  379. if (new_nc := insert_lazy_parents(nc)):
  380. kept_nc[new_nc.signature]=new_nc
  381. # be sure to add the Mantis context.
  382. new_nc.mContext =mContext
  383. kept_nc[nc.signature]=nc
  384. prWhite(f"Parsing tree took {time.time()-start_time} seconds.")
  385. prWhite("Number of Nodes: %s" % (len(kept_nc)))
  386. return kept_nc
  387. def switch_mode(mode='OBJECT', objects = []):
  388. active = None
  389. if objects:
  390. from bpy import context, ops
  391. active = objects[-1]
  392. context.view_layer.objects.active = active
  393. if (active):
  394. with context.temp_override(**{'active_object':active, 'selected_objects':objects}):
  395. ops.object.mode_set(mode=mode)
  396. return active
  397. def execution_error_cleanup(node, exception, switch_objects = [], show_error=False ):
  398. from bpy import context
  399. ui_sig = None
  400. if show_error: # show a popup and select the relevant nodes
  401. if node:
  402. ui_sig = node.ui_signature
  403. # TODO: see about zooming-to-node.
  404. base_tree = node.base_tree
  405. tree = base_tree
  406. try:
  407. pass
  408. space = context.space_data
  409. for name in ui_sig[1:]:
  410. for n in tree.nodes: n.select = False
  411. n = tree.nodes[name]
  412. n.select = True
  413. tree.nodes.active = n
  414. if hasattr(n, "node_tree"):
  415. tree = n.node_tree
  416. except AttributeError: # not being run in node graph
  417. pass
  418. finally:
  419. def error_popup_draw(self, context):
  420. self.layout.label(text=f"Error: {exception}")
  421. self.layout.label(text=f"see node: {ui_sig[1:]}.")
  422. context.window_manager.popup_menu(error_popup_draw, title="Error", icon='ERROR')
  423. switch_mode(mode='OBJECT', objects=switch_objects)
  424. for ob in switch_objects:
  425. ob.data.pose_position = 'POSE'
  426. prRed(f"Error: {exception} in node {ui_sig}")
  427. return exception
  428. def execute_tree(nodes, base_tree, context, error_popups = False):
  429. import bpy
  430. from time import time
  431. from .node_container_common import GraphError
  432. original_active = context.view_layer.objects.active
  433. start_execution_time = time()
  434. mContext = None
  435. from collections import deque
  436. xForm_pass = deque()
  437. for nc in nodes.values():
  438. if not mContext: # just grab one of these. this is a silly way to do this.
  439. mContext = nc.mContext
  440. nc.reset_execution()
  441. check_and_add_root(nc, xForm_pass)
  442. executed = []
  443. # check for cycles here by keeping track of the number of times a node has been visited.
  444. visited={}
  445. check_max_len=len(nodes)**2 # seems too high but safe. In a well-ordered graph, I guess this number should be less than the number of nodes.
  446. max_iterations = len(nodes)**2
  447. i = 0
  448. switch_me = [] # switch the mode on these objects
  449. active = None # only need it for switching modes
  450. select_me = []
  451. try:
  452. while(xForm_pass):
  453. if i >= max_iterations:
  454. raise GraphError("There is probably a cycle somewhere in the graph.")
  455. i+=1
  456. n = xForm_pass.pop()
  457. if visited.get(n.signature) is not None:
  458. visited[n.signature]+=1
  459. else:
  460. visited[n.signature]=0
  461. if visited[n.signature] > check_max_len:
  462. raise GraphError("There is a probably a cycle in the graph somewhere. Fix it!")
  463. # we're trying to solve the halting problem at this point.. don't do that.
  464. # TODO find a better way! there are algo's for this but they will require using a different solving algo, too
  465. if n.execution_prepared:
  466. continue
  467. if n.node_type not in ['XFORM', 'UTILITY']:
  468. for dep in n.hierarchy_dependencies:
  469. if not dep.execution_prepared:
  470. xForm_pass.appendleft(n) # hold it
  471. break
  472. else:
  473. n.execution_prepared=True
  474. executed.append(n)
  475. for conn in n.hierarchy_connections:
  476. if not conn.execution_prepared:
  477. xForm_pass.appendleft(conn)
  478. else:
  479. for dep in n.hierarchy_dependencies:
  480. if not dep.execution_prepared:
  481. break
  482. else:
  483. try:
  484. if not n.prepared:
  485. n.bPrepare(context)
  486. if not n.executed:
  487. n.bExecute(context)
  488. if (n.__class__.__name__ == "xFormArmature" ):
  489. ob = n.bGetObject()
  490. switch_me.append(ob)
  491. active = ob
  492. if not (n.__class__.__name__ == "xFormBone" ) and hasattr(n, "bGetObject"):
  493. ob = n.bGetObject()
  494. if isinstance(ob, bpy.types.Object):
  495. select_me.append(ob)
  496. except Exception as e:
  497. e = execution_error_cleanup(n, e, show_error=error_popups)
  498. if error_popups == False:
  499. raise e
  500. n.execution_prepared=True
  501. executed.append(n)
  502. for conn in n.hierarchy_connections:
  503. if not conn.execution_prepared:
  504. xForm_pass.appendleft(conn)
  505. switch_mode(mode='POSE', objects=switch_me)
  506. if (active):
  507. with context.temp_override(**{'active_object':active, 'selected_objects':switch_me}):
  508. bpy.ops.object.mode_set(mode='POSE')
  509. for n in executed:
  510. try:
  511. if not n.prepared:
  512. n.bPrepare(context)
  513. if not n.executed:
  514. n.bExecute(context)
  515. except Exception as e:
  516. e = execution_error_cleanup(n, e, show_error=error_popups)
  517. if error_popups == False:
  518. raise e
  519. switch_mode(mode='OBJECT', objects=switch_me)
  520. for ob in switch_me:
  521. ob.data.pose_position = 'POSE'
  522. # switch to pose mode here so that the nodes can use the final pose data
  523. # this will require them to update the depsgraph.
  524. for n in executed:
  525. try:
  526. n.bFinalize(context)
  527. except Exception as e:
  528. e = execution_error_cleanup(n, e, show_error=error_popups)
  529. if error_popups == False:
  530. raise e
  531. tot_time = (time() - start_execution_time)
  532. prGreen(f"Executed tree of {len(executed)} nodes in {tot_time} seconds")
  533. if (original_active):
  534. context.view_layer.objects.active = original_active
  535. original_active.select_set(True)
  536. except Exception as e:
  537. e = execution_error_cleanup(None, e, switch_me, show_error=error_popups)
  538. if error_popups == False:
  539. raise e
  540. finally:
  541. context.view_layer.objects.active = active
  542. # clear the selection first.
  543. from itertools import chain
  544. for ob in context.selected_objects:
  545. try:
  546. ob.select_set(False)
  547. except RuntimeError: # it isn't in the view layer
  548. pass
  549. for ob in chain(select_me, mContext.b_objects.values()):
  550. try:
  551. ob.select_set(True)
  552. except RuntimeError: # it isn't in the view layer
  553. pass