deformer_containers.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588
  1. from .node_container_common import *
  2. from .xForm_containers import xFormGeometryObject
  3. from .misc_containers import InputExistingGeometryObject
  4. from bpy.types import Node
  5. from .base_definitions import MantisNode
  6. from .utilities import (prRed, prGreen, prPurple, prWhite, prOrange,
  7. wrapRed, wrapGreen, wrapPurple, wrapWhite,
  8. wrapOrange,)
  9. def TellClasses():
  10. return [
  11. DeformerArmature,
  12. DeformerHook,
  13. DeformerMorphTarget,
  14. DeformerMorphTargetDeform,
  15. ]
  16. def trace_xForm_back(nc, socket):
  17. from .xForm_containers import xFormGeometryObject
  18. from .misc_containers import InputExistingGeometryObject
  19. from bpy.types import Object
  20. if (trace := trace_single_line(nc, socket)[0] ) :
  21. for i in range(len(trace)): # have to look in reverse, actually
  22. if ( isinstance(trace[ i ], xFormGeometryObject ) ) or ( isinstance(trace[ i ], InputExistingGeometryObject ) ):
  23. return trace[ i ].bGetObject()
  24. raise GraphError(wrapRed(f"No other object found for {nc}."))
  25. def default_evaluate_input(nc, input_name):
  26. # duped from link_containers... should be common?
  27. # should catch 'Target', 'Pole Target' and ArmatureConstraint targets, too
  28. if ('Target' in input_name) and input_name != "Target Space":
  29. socket = nc.inputs.get(input_name)
  30. if socket.is_linked:
  31. return socket.links[0].from_node
  32. return None
  33. else:
  34. return evaluate_input(nc, input_name)
  35. # semi-duplicated from link_containers
  36. def GetxForm(nc):
  37. trace = trace_single_line_up(nc, "Deformer")
  38. for node in trace[0]:
  39. if (node.__class__ in [xFormGeometryObject, InputExistingGeometryObject]):
  40. return node
  41. raise GraphError("%s is not connected to a downstream xForm" % nc)
  42. class DeformerArmature:
  43. '''A node representing an armature deformer'''
  44. def __init__(self, signature, base_tree):
  45. self.base_tree=base_tree
  46. self.signature = signature
  47. self.inputs = {
  48. "Input Relationship" : NodeSocket(is_input = True, name = "Input Relationship", node = self,),
  49. "Armature Object" : NodeSocket(is_input = True, name = "Armature Object", node = self,),
  50. "Blend Vertex Group" : NodeSocket(is_input = True, name = "Blend Vertex Group", node = self),
  51. "Invert Vertex Group" : NodeSocket(is_input = True, name = "Invert Vertex Group", node = self),
  52. "Preserve Volume" : NodeSocket(is_input = True, name = "Preserve Volume", node = self),
  53. "Use Multi Modifier" : NodeSocket(is_input = True, name = "Use Multi Modifier", node = self),
  54. "Use Envelopes" : NodeSocket(is_input = True, name = "Use Envelopes", node = self),
  55. "Use Vertex Groups" : NodeSocket(is_input = True, name = "Use Vertex Groups", node = self),
  56. "Skinning Method" : NodeSocket(is_input = True, name = "Skinning Method", node = self),
  57. "Deformer" : NodeSocket(is_input = True, name = "Deformer", node = self),
  58. "Copy Skin Weights From" : NodeSocket(is_input = True, name = "Copy Skin Weights From", node = self),
  59. }
  60. self.outputs = {
  61. "Deformer" : NodeSocket(is_input = False, name = "Deformer", node=self), }
  62. self.parameters = {
  63. "Name" : None,
  64. "Armature Object" : None,
  65. "Blend Vertex Group" : None,
  66. "Invert Vertex Group" : None,
  67. "Preserve Volume" : None,
  68. "Use Multi Modifier" : None,
  69. "Use Envelopes" : None,
  70. "Use Vertex Groups" : None,
  71. "Skinning Method" : None,
  72. "Deformer" : None,
  73. "Copy Skin Weights From" : None,
  74. }
  75. # now set up the traverse target...
  76. self.inputs["Deformer"].set_traverse_target(self.outputs["Deformer"])
  77. self.outputs["Deformer"].set_traverse_target(self.inputs["Deformer"])
  78. self.node_type = "LINK"
  79. self.hierarchy_connections, self.connections = [], []
  80. self.hierarchy_dependencies, self.dependencies = [], []
  81. self.prepared = True
  82. self.executed = False
  83. def evaluate_input(self, input_name):
  84. return default_evaluate_input(self, input_name)
  85. def GetxForm(self, socket="Deformer"):
  86. if socket == "Deformer":
  87. return GetxForm(self)
  88. else:
  89. trace_xForm_back(self, socket)
  90. # DUPLICATED FROM xForm_containers::xFormBone
  91. # DEDUP HACK HACK HACK HACK HACK
  92. def bGetParentArmature(self):
  93. from .xForm_containers import xFormArmature
  94. from .misc_containers import InputExistingGeometryObject
  95. from bpy.types import Object
  96. if (trace := trace_single_line(self, "Armature Object")[0] ) :
  97. for i in range(len(trace)):
  98. # have to look in reverse, actually
  99. if ( isinstance(trace[ i ], xFormArmature ) ):
  100. return trace[ i ].bGetObject()
  101. elif ( isinstance(trace[i], InputExistingGeometryObject)):
  102. if (ob := trace[i].bGetObject()).type == "ARMATURE":
  103. return ob
  104. raise RuntimeError(f"Cannot find armature for node {self}")
  105. return None
  106. #should do the trick...
  107. def bExecute(self, bContext = None,):
  108. self.executed = True
  109. def initialize_vgroups(self,):
  110. ob = self.GetxForm().bGetObject()
  111. armOb = self.bGetParentArmature()
  112. for b in armOb.data.bones:
  113. if b.use_deform == False:
  114. continue
  115. vg = ob.vertex_groups.get(b.name)
  116. if not vg:
  117. vg = ob.vertex_groups.new(name=b.name)
  118. num_verts = len(ob.data.vertices)
  119. vg.add(range(num_verts), 0, 'REPLACE')
  120. def copy_weights(self):
  121. # we'll use modifiers for this, maybe use GN for it in the future tho
  122. import bpy
  123. ob = self.GetxForm().bGetObject()
  124. try:
  125. copy_from = self.GetxForm(socket="Copy Skin Weights From")
  126. except GraphError:
  127. copy_from = None
  128. prRed(f"No object found for copying weights in {self}, continuing anyway.")
  129. m = ob.modifiers.new(type="DATA_TRANSFER", name="Mantis_temp_data_transfer")
  130. m.object = None; m.use_vert_data = True
  131. m.data_types_verts = {'VGROUP_WEIGHTS'}
  132. m.vert_mapping = 'POLYINTERP_NEAREST'
  133. m.layers_vgroup_select_src = 'ALL'
  134. m.layers_vgroup_select_dst = 'NAME'
  135. m.object = copy_from
  136. # m.use_object_transform = False # testing reveals that this is undesirable - since the objects may not have their transforms applied.
  137. ob.modifiers.move(len(ob.modifiers)-1, 0)
  138. # ob.data = ob.data.copy()
  139. if False: #MAYBE the mouse needs to be in the 3D viewport, no idea how to set this in an override
  140. # TODO: figure out how to apply this, context is incorrect because armature is still in pose mode
  141. original_active = bpy.context.active_object
  142. original_mode = original_active.mode
  143. bpy.ops.object.mode_set(mode='OBJECT')
  144. with bpy.context.temp_override(**{'active_object':ob, 'selected_objects':[ob, copy_from]}):
  145. # bpy.ops.object.datalayout_transfer(modifier=m.name) # note: this operator is used by the modifier or stand-alone in the UI
  146. # the poll for this operator is defined in blender/source/blender/editors/object/object_data_transfer.cc
  147. # and blender/source/blender/editors/object/object_modifier.cc
  148. # bpy.ops.object.modifier_apply(modifier=m.name, single_user=True)
  149. bpy.ops.object.datalayout_transfer(data_type='VGROUP_WEIGHTS')
  150. bpy.ops.object.data_transfer(data_type='VGROUP_WEIGHTS')
  151. bpy.ops.object.mode_set(mode=original_mode)
  152. def bFinalize(self, bContext=None):
  153. prGreen("Executing Armature Deform Node")
  154. mod_name = self.evaluate_input("Name")
  155. d = self.GetxForm().bGetObject().modifiers.new(mod_name, type='ARMATURE')
  156. if d is None:
  157. raise RuntimeError(f"Modifier was not created in node {self} -- the object is invalid.")
  158. self.bObject = d
  159. d.object = self.bGetParentArmature()
  160. props_sockets = {
  161. 'vertex_group' : ("Blend Vertex Group", ""),
  162. 'invert_vertex_group' : ("Invert Vertex Group", ""),
  163. 'use_deform_preserve_volume' : ("Preserve Volume", False),
  164. 'use_multi_modifier' : ("Use Multi Modifier", False),
  165. 'use_bone_envelopes' : ("Use Envelopes", False),
  166. 'use_vertex_groups' : ("Use Vertex Groups", False),
  167. }
  168. evaluate_sockets(self, d, props_sockets)
  169. #
  170. if (skin_method := self.evaluate_input("Skinning Method")) == "AUTOMATIC_HEAT":
  171. # This is bad and leads to somewhat unpredictable
  172. # behaviour, e.g. what object will be selected? What mode?
  173. # also bpy.ops is ugly and prone to error when used in
  174. # scripts. I don't intend to use bpy.ops when I can avoid it.
  175. import bpy
  176. self.initialize_vgroups()
  177. bContext.view_layer.depsgraph.update()
  178. ob = self.GetxForm().bGetObject()
  179. armOb = self.bGetParentArmature()
  180. deform_bones = []
  181. for pb in armOb.pose.bones:
  182. if pb.bone.use_deform == True:
  183. deform_bones.append(pb)
  184. context_override = {
  185. 'active_object':ob,
  186. 'selected_objects':[ob, armOb],
  187. 'active_pose_bone':deform_bones[0],
  188. 'selected_pose_bones':deform_bones,}
  189. #
  190. with bContext.temp_override(**{'active_object':armOb}):
  191. bpy.ops.object.mode_set(mode='POSE')
  192. bpy.ops.pose.select_all(action='SELECT')
  193. with bContext.temp_override(**context_override):
  194. bpy.ops.paint.weight_paint_toggle()
  195. bpy.ops.paint.weight_from_bones(type='AUTOMATIC')
  196. bpy.ops.paint.weight_paint_toggle()
  197. #
  198. with bContext.temp_override(**{'active_object':armOb}):
  199. bpy.ops.object.mode_set(mode='POSE')
  200. bpy.ops.pose.select_all(action='DESELECT')
  201. bpy.ops.object.mode_set(mode='OBJECT')
  202. # TODO: modify Blender to make this available as a Python API function.
  203. elif skin_method == "EXISTING_GROUPS":
  204. pass
  205. elif skin_method == "COPY_FROM_OBJECT":
  206. self.initialize_vgroups()
  207. self.copy_weights()
  208. class DeformerHook:
  209. '''A node representing a hook deformer'''
  210. def __init__(self, signature, base_tree):
  211. self.base_tree=base_tree
  212. self.signature = signature
  213. self.inputs = {
  214. "Hook Target" : NodeSocket(is_input = True, name = "Hook Target", node = self,),
  215. "Index" : NodeSocket(is_input = True, name = "Index", node = self),
  216. "Deformer" : NodeSocket(is_input = True, name = "Deformer", node = self),
  217. }
  218. self.outputs = {
  219. "Deformer" : NodeSocket(is_input = False, name = "Deformer", node=self), }
  220. self.parameters = {
  221. "Hook Target" : None,
  222. "Index" : None,
  223. "Deformer" : None,
  224. "Name" : None,
  225. }
  226. # now set up the traverse target...
  227. self.inputs["Deformer"].set_traverse_target(self.outputs["Deformer"])
  228. self.outputs["Deformer"].set_traverse_target(self.inputs["Deformer"])
  229. self.node_type = "LINK"
  230. self.hierarchy_connections, self.connections = [], []
  231. self.hierarchy_dependencies, self.dependencies = [], []
  232. self.prepared = True
  233. self.executed = False
  234. def evaluate_input(self, input_name):
  235. return default_evaluate_input(self, input_name)
  236. def GetxForm(self, socket="Deformer"):
  237. if socket == "Deformer":
  238. return GetxForm(self)
  239. else:
  240. trace_xForm_back(self, socket)
  241. def bExecute(self, bContext = None,):
  242. self.executed = True
  243. def bFinalize(self, bContext=None):
  244. from bpy.types import Bone, PoseBone, Object
  245. prGreen(f"Executing Hook Deform Node: {self}")
  246. mod_name = self.evaluate_input("Name")
  247. target_node = self.evaluate_input('Hook Target')
  248. target = target_node.bGetObject(); subtarget = ""
  249. if isinstance(target, Bone) or isinstance(target, PoseBone):
  250. subtarget = target.name; target = target.id_data
  251. ob=self.GetxForm().bGetObject()
  252. prOrange(self.GetxForm().bGetObject())
  253. reuse = False
  254. for m in ob.modifiers:
  255. if False and m.type == 'HOOK' and m.object == target and m.subtarget == subtarget:
  256. d = m; reuse = True; prOrange("REUSE"); break
  257. else:
  258. # don't reuse it
  259. d = ob.modifiers.new(mod_name, type='HOOK')
  260. if d is None:
  261. raise RuntimeError(f"Modifier was not created in node {self} -- the object is invalid.")
  262. get_target_and_subtarget(self, d, input_name="Hook Target")
  263. vertices_used=[]
  264. if reuse: # Get the verts in the list... filter out all the unneeded 0's
  265. vertices_used = list(d.vertex_indices)
  266. include_0 = 0 in vertices_used
  267. vertices_used = list(filter(lambda a : a != 0, vertices_used))
  268. if include_0: vertices_used.append(0)
  269. # now we add the selected vertex to the list, too
  270. vertices_used.append(self.evaluate_input("Index"))
  271. d.vertex_indices_set(vertices_used)
  272. # props_sockets = {
  273. # 'vertex_group' : ("Blend Vertex Group", ""),
  274. # }
  275. # evaluate_sockets(self, d, props_sockets)
  276. class DeformerMorphTarget:
  277. '''A node representing an armature deformer'''
  278. def __init__(self, signature, base_tree):
  279. self.base_tree=base_tree
  280. self.signature = signature
  281. self.inputs = {
  282. "Relative to" : NodeSocket(is_input = True, name = "Relative To", node = self,),
  283. "Object" : NodeSocket(is_input = True, name = "Object", node = self,),
  284. "Deformer" : NodeSocket(is_input = True, name = "Deformer", node = self),
  285. "Vertex Group" : NodeSocket(is_input = True, name = "Vertex Group", node = self),
  286. }
  287. self.outputs = {
  288. "Deformer" : NodeSocket(is_input = False, name = "Deformer", node=self),
  289. "Morph Target" : NodeSocket(is_input = False, name = "Morph Target", node=self), }
  290. self.parameters = {
  291. "Name" : None,
  292. "Relative to" : None,
  293. "Object" : None,
  294. "Morph Target" : None,
  295. "Deformer" : None,
  296. "Vertex Group" : None,
  297. }
  298. # now set up the traverse target...
  299. self.inputs["Deformer"].set_traverse_target(self.outputs["Deformer"])
  300. self.outputs["Deformer"].set_traverse_target(self.inputs["Deformer"])
  301. self.node_type = "LINK"
  302. self.hierarchy_connections, self.connections = [], []
  303. self.hierarchy_dependencies, self.dependencies = [], []
  304. self.prepared = True
  305. self.executed = False
  306. def GetxForm(self, trace_input="Object"):
  307. trace = trace_single_line(self, trace_input)
  308. for node in trace[0]:
  309. if (node.__class__ in [xFormGeometryObject, InputExistingGeometryObject]):
  310. return node
  311. raise GraphError("%s is not connected to an upstream xForm" % self)
  312. def bExecute(self, bContext = None,):
  313. prGreen("Executing Morph Target Node")
  314. name = ''
  315. ob = None; relative = None
  316. try:
  317. ob = self.GetxForm().bGetObject().name
  318. except Exception as e: # this will and should throw an error if it fails
  319. prRed(f"Execution failed at {self}: no object found for morph target.")
  320. raise e
  321. if self.inputs["Relative to"].is_linked:
  322. try:
  323. relative = self.GetxForm("Relative to").bGetObject().name
  324. except Exception as e: # same here
  325. prRed(f"Execution failed at {self}: no relative object found for morph target, despite link existing.")
  326. raise e
  327. vg = self.evaluate_input("Vertex Group") if self.evaluate_input("Vertex Group") else "" # just make sure it is a string
  328. mt={"object":ob, "vertex_group":vg, "relative_shape":relative}
  329. self.parameters["Morph Target"] = mt
  330. self.executed = True
  331. class DeformerMorphTargetDeform:
  332. '''A node representing an armature deformer'''
  333. def __init__(self, signature, base_tree):
  334. self.base_tree=base_tree
  335. self.signature = signature
  336. self.inputs = {
  337. "Deformer" : NodeSocket(is_input = True, name = "Deformer", node = self),
  338. }
  339. self.outputs = {
  340. "Deformer" : NodeSocket(is_input = False, name = "Deformer", node=self), }
  341. self.parameters = {
  342. "Name" : None,
  343. "Deformer" : None,}
  344. # now set up the traverse target...
  345. self.inputs["Deformer"].set_traverse_target(self.outputs["Deformer"])
  346. self.outputs["Deformer"].set_traverse_target(self.inputs["Deformer"])
  347. self.node_type = "LINK"
  348. self.hierarchy_connections, self.connections = [], []
  349. self.hierarchy_dependencies, self.dependencies = [], []
  350. self.prepared = True
  351. self.executed = True
  352. self.bObject = None
  353. setup_custom_props(self)
  354. def GetxForm(self):
  355. return GetxForm(self)
  356. def gen_morph_target_modifier(self):
  357. mod_name = self.evaluate_input("Name")
  358. m = self.GetxForm().bGetObject().modifiers.new(mod_name, type='NODES')
  359. self.bObject = m
  360. # at this point we make the node tree
  361. from bpy import data
  362. ng = data.node_groups.new(mod_name, "GeometryNodeTree")
  363. m.node_group = ng
  364. ng.interface.new_socket("Geometry", in_out="INPUT", socket_type="NodeSocketGeometry")
  365. ng.interface.new_socket("Geometry", in_out="OUTPUT", socket_type="NodeSocketGeometry")
  366. inp = ng.nodes.new("NodeGroupInput")
  367. out = ng.nodes.new("NodeGroupOutput")
  368. # TODO CLEANUP here
  369. if (position := ng.nodes.get("Position")) is None: position = ng.nodes.new("GeometryNodeInputPosition")
  370. if (index := ng.nodes.get("Index")) is None: index = ng.nodes.new("GeometryNodeInputIndex")
  371. rest_position = position
  372. add_these = []
  373. props_sockets={}
  374. object_map = {}
  375. targets = []
  376. for k,v in self.inputs.items():
  377. if "Target" in k:
  378. targets.append(v)
  379. for i, t in enumerate(targets):
  380. mt_node = t.links[0].from_node
  381. mt_name = mt_node.GetxForm().bGetObject().name
  382. vg = mt_node.parameters["Morph Target"]["vertex_group"]
  383. if vg: mt_name = mt_name+"."+vg
  384. try:
  385. ob_relative = t.links[0].from_node.inputs["Relative to"].links[0].from_node.bGetObject()
  386. except IndexError:
  387. ob_relative = None
  388. ng.interface.new_socket(mt_name, in_out = "INPUT", socket_type="NodeSocketObject")
  389. ng.interface.new_socket(mt_name+" Value", in_out = "INPUT", socket_type="NodeSocketFloat")
  390. ob_node = ng.nodes.new("GeometryNodeObjectInfo")
  391. sample_index = ng.nodes.new("GeometryNodeSampleIndex"); sample_index.data_type = 'FLOAT_VECTOR'
  392. subtract = ng.nodes.new("ShaderNodeVectorMath"); subtract.operation="SUBTRACT"
  393. scale1 = ng.nodes.new("ShaderNodeVectorMath"); scale1.operation="SCALE"
  394. ng.links.new(input=inp.outputs[mt_name], output=ob_node.inputs["Object"])
  395. ng.links.new(input=index.outputs["Index"], output=sample_index.inputs["Index"])
  396. ng.links.new(input=position.outputs["Position"], output=sample_index.inputs["Value"])
  397. ng.links.new(input=sample_index.outputs["Value"], output=subtract.inputs[0])
  398. ng.links.new(input=ob_node.outputs["Geometry"], output=sample_index.inputs["Geometry"])
  399. if ob_relative: # TODO: this should also be exposed as an input
  400. ob_node1 = ng.nodes.new("GeometryNodeObjectInfo"); ob_node1.inputs["Object"].default_value = ob_relative
  401. sample_index1 = ng.nodes.new("GeometryNodeSampleIndex"); sample_index1.data_type = 'FLOAT_VECTOR'
  402. ng.links.new(input=index.outputs["Index"], output=sample_index1.inputs["Index"])
  403. ng.links.new(input=position.outputs["Position"], output=sample_index1.inputs["Value"])
  404. ng.links.new(input=ob_node1.outputs["Geometry"], output=sample_index1.inputs["Geometry"])
  405. ng.links.new(input=sample_index1.outputs["Value"], output=subtract.inputs[1])
  406. else:
  407. # ng.links.new(input=rest_position.outputs["Attribute"], output=subtract.inputs[1])
  408. ng.links.new(input=rest_position.outputs["Position"], output=subtract.inputs[1])
  409. ng.links.new(input=subtract.outputs["Vector"], output=scale1.inputs[0])
  410. # TODO: this should be exposed as a node tree input
  411. if vg:= mt_node.evaluate_input("Vertex Group"): # works
  412. vg_att = ng.nodes.new("GeometryNodeInputNamedAttribute"); vg_att.inputs["Name"].default_value=vg
  413. multiply = ng.nodes.new("ShaderNodeMath"); multiply.operation = "MULTIPLY"
  414. ng.links.new(input=vg_att.outputs["Attribute"], output=multiply.inputs[1])
  415. ng.links.new(input=inp.outputs[mt_name+" Value"], output=multiply.inputs[0])
  416. ng.links.new(input=multiply.outputs[0], output=scale1.inputs["Scale"])
  417. else:
  418. ng.links.new(input=inp.outputs[mt_name+" Value"], output=scale1.inputs["Scale"])
  419. add_these.append(scale1)
  420. object_map["Socket_"+str((i+1)*2)]=mt_node.GetxForm().bGetObject()
  421. props_sockets["Socket_"+str((i+1)*2+1)]= ("Value."+str(i).zfill(3), 1.0)
  422. set_position = ng.nodes.new("GeometryNodeSetPosition")
  423. ng.links.new(inp.outputs["Geometry"], output=set_position.inputs["Geometry"])
  424. ng.links.new(set_position.outputs["Geometry"], output=out.inputs["Geometry"])
  425. prev_node = rest_position
  426. for i, node in enumerate(add_these):
  427. add = ng.nodes.new("ShaderNodeVectorMath"); add.operation="ADD"
  428. ng.links.new(prev_node.outputs[0], output=add.inputs[0])
  429. ng.links.new(node.outputs[0], output=add.inputs[1])
  430. prev_node = add
  431. ng.links.new(add.outputs[0], output=set_position.inputs["Position"])
  432. from .utilities import SugiyamaGraph
  433. SugiyamaGraph(ng, 12)
  434. evaluate_sockets(self, m, props_sockets)
  435. for socket, ob in object_map.items():
  436. m[socket]=ob
  437. finish_drivers(self)
  438. def gen_shape_key(self): # TODO: make this a feature of the node definition that appears only when there are no prior deformers - and shows a warning!
  439. # TODO: the below works well, but it is quite slow. It does not seem to have better performence. Its only advantage is export to FBX.
  440. # there are a number of things I need to fix here
  441. # - reuse shape keys if possible
  442. # - figure out how to make this a lot faster
  443. # - edit the xForm stuff to delete drivers from shape key ID's, since they belong to the Key, not the Object.
  444. from time import time
  445. start_time = time()
  446. from bpy import data
  447. ob = self.GetxForm().bGetObject()
  448. m = data.meshes.new_from_object(ob, preserve_all_data_layers=True)
  449. ob.data = m
  450. ob.add_rest_position_attribute = True
  451. ob.shape_key_clear()
  452. targets = []
  453. for k,v in self.inputs.items():
  454. if "Target" in k:
  455. targets.append(v)
  456. for i, t in enumerate(targets):
  457. mt_node = t.links[0].from_node
  458. mt_name = "Morph Target."+str(i).zfill(3)
  459. # using the built-in shapekey feature is actually a lot harder in terms of programming because I need...
  460. # min/max, as it is just not a feature of the GN version
  461. # to carry info from the morph target node regarding relative shapes and vertex groups and all that
  462. # the drivers may be more difficult to apply, too.
  463. # hafta make new geometry for the object and add shape keys and all that
  464. # the benefit to all this being maybe better performence and exporting to game engines via .fbx
  465. # first make a basis shape key
  466. ob.shape_key_add(name='Basis', from_mix=False)
  467. keys={}
  468. props_sockets={}
  469. for i, t in enumerate(targets):
  470. mt_node = t.links[0].from_node
  471. # mt_name = "Morph Target."+str(i).zfill(3)
  472. sk_ob = mt_node.GetxForm().bGetObject()
  473. mt_name = sk_ob.name
  474. vg = mt_node.parameters["Morph Target"]["vertex_group"]
  475. if vg: mt_name = mt_name+"."+vg
  476. sk = ob.shape_key_add(name=mt_name, from_mix=False)
  477. # the shapekey data is absolute point data for each vertex, in order, very simple
  478. for j in range(len(m.vertices)):
  479. sk.data[j].co = sk_ob.data.vertices[j].co # assume they match
  480. sk.vertex_group = vg
  481. sk.slider_min = -10
  482. sk.slider_max = 10
  483. keys[mt_name]=sk
  484. props_sockets[mt_name]= ("Value."+str(i).zfill(3), 1.0)
  485. for i, t in enumerate(targets):
  486. mt_node = t.links[0].from_node
  487. # mt_name = "Morph Target."+str(i).zfill(3)
  488. sk_ob = mt_node.GetxForm().bGetObject()
  489. mt_name = sk_ob.name
  490. vg = mt_node.parameters["Morph Target"]["vertex_group"]
  491. if vg: mt_name = mt_name+"."+vg
  492. if rel := mt_node.parameters["Morph Target"]["relative_shape"]:
  493. sk = keys.get(mt_name)
  494. sk.relative_key = keys.get(rel)
  495. # for k,v in props_sockets.items():
  496. # print(wrapWhite(k), wrapOrange(v), wrapRed(self.evaluate_input(v)))
  497. self.bObject = sk.id_data
  498. evaluate_sockets(self, sk.id_data, props_sockets)
  499. finish_drivers(self)
  500. prWhite(f"Initializing morph target took {time() -start_time} seconds")
  501. # then we need to get all the data from the morph targets, pull all the relative shapes first and add them, vertex groups and properties
  502. # next we add all the shape keys that are left, and their vertex groups
  503. # set the slider ranges to -10 and 10
  504. # then set up the drivers
  505. def bFinalize(self, bContext=None):
  506. # let's find out if there is a prior deformer.
  507. # if not, then there should be an option to use plain 'ol shape keys
  508. # GN is always desirable as an option though because it can be baked.
  509. if self.inputs["Deformer"].is_linked and True:
  510. # for now we won't do Blender Shape Keys
  511. self.gen_morph_target_modifier()
  512. else: # TODO: give the user the option to do this via a node property.
  513. self.gen_shape_key()
  514. for c in TellClasses():
  515. setup_container(c)