deformer_containers.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642
  1. from .node_container_common import *
  2. from .xForm_containers import xFormGeometryObject, xFormObjectInstance
  3. from .misc_nodes import InputExistingGeometryObject
  4. from .base_definitions import MantisNode, MantisSocketTemplate
  5. from .utilities import (prRed, prGreen, prPurple, prWhite, prOrange,
  6. wrapRed, wrapGreen, wrapPurple, wrapWhite,
  7. wrapOrange,)
  8. from .deformer_socket_templates import *
  9. from bpy.types import NodeTree
  10. def TellClasses():
  11. return [
  12. DeformerArmature,
  13. DeformerHook,
  14. DeformerMorphTarget,
  15. DeformerMorphTargetDeform,
  16. DeformerSurfaceDeform,
  17. DeformerMeshDeform,
  18. ]
  19. # object instance probably can't use the deformer but it doesn't hurt to try.
  20. deformable_types= (xFormGeometryObject, InputExistingGeometryObject, xFormObjectInstance)
  21. def trace_xForm_back(nc, socket):
  22. if (trace := trace_single_line(nc, socket)[0] ) :
  23. for i in range(len(trace)): # have to look in reverse, actually
  24. if ( isinstance(trace[ i ], deformable_types ) ):
  25. return trace[ i ].bGetObject()
  26. raise GraphError(wrapRed(f"No other object found for {nc}."))
  27. class MantisDeformerNode(MantisNode):
  28. def __init__(self, signature : tuple,
  29. base_tree : NodeTree,
  30. socket_templates : list[MantisSocketTemplate]=[]):
  31. super().__init__(signature, base_tree, socket_templates)
  32. self.node_type = 'LINK'
  33. self.prepared = True
  34. self.bObject=[]
  35. # we need evaluate_input to have the same behaviour as links.
  36. def evaluate_input(self, input_name, index=0):
  37. if (input_name in ['Target', 'Object']):
  38. socket = self.inputs.get(input_name)
  39. if socket.is_linked:
  40. return socket.links[0].from_node
  41. return None
  42. else:
  43. return super().evaluate_input(input_name, index)
  44. def GetxForm(nc, output_name="Deformer"):
  45. break_condition= lambda node : node.__class__ in deformable_types
  46. xforms = trace_line_up_branching(nc, output_name, break_condition)
  47. return_me=[]
  48. for xf in xforms:
  49. if xf.node_type != 'XFORM':
  50. continue
  51. if xf in return_me:
  52. continue
  53. return_me.append(xf)
  54. return return_me
  55. def reset_execution(self):
  56. super().reset_execution()
  57. self.bObject=[]; self.prepared=True
  58. class DeformerArmature(MantisDeformerNode):
  59. '''A node representing an armature deformer'''
  60. def __init__(self, signature, base_tree):
  61. super().__init__(signature, base_tree)
  62. inputs = [
  63. "Input Relationship",
  64. "Armature Object",
  65. "Blend Vertex Group",
  66. "Invert Vertex Group",
  67. "Preserve Volume",
  68. "Use Multi Modifier",
  69. "Use Envelopes",
  70. "Use Vertex Groups",
  71. "Skinning Method",
  72. "Deformer",
  73. "Copy Skin Weights From"
  74. ]
  75. outputs = [
  76. "Deformer"
  77. ]
  78. self.outputs.init_sockets(outputs)
  79. self.inputs.init_sockets(inputs)
  80. self.init_parameters(additional_parameters={"Name":None})
  81. self.set_traverse([("Deformer", "Deformer")])
  82. self.node_type = "LINK"
  83. self.prepared = True
  84. def GetxForm(self, socket="Deformer"):
  85. if socket == "Deformer":
  86. return super().GetxForm()
  87. else:
  88. trace_xForm_back(self, socket)
  89. # DUPLICATED FROM xForm_containers::xFormBone
  90. # DEDUP HACK HACK HACK HACK HACK
  91. def bGetParentArmature(self):
  92. from .xForm_containers import xFormArmature
  93. from .misc_nodes import InputExistingGeometryObject
  94. from bpy.types import Object
  95. if (trace := trace_single_line(self, "Armature Object")[0] ) :
  96. for i in range(len(trace)):
  97. # have to look in reverse, actually
  98. if ( isinstance(trace[ i ], xFormArmature ) ):
  99. return trace[ i ].bGetObject()
  100. elif ( isinstance(trace[i], InputExistingGeometryObject)):
  101. if (ob := trace[i].bGetObject()).type == "ARMATURE":
  102. return ob
  103. raise RuntimeError(f"Cannot find armature for node {self}")
  104. return None
  105. #should do the trick...
  106. def bExecute(self, bContext = None,):
  107. self.executed = True
  108. def initialize_vgroups(self, xf):
  109. ob = xf.bGetObject()
  110. armOb = self.bGetParentArmature()
  111. for b in armOb.data.bones:
  112. if b.use_deform == False:
  113. continue
  114. vg = ob.vertex_groups.get(b.name)
  115. if not vg:
  116. vg = ob.vertex_groups.new(name=b.name)
  117. num_verts = len(ob.data.vertices)
  118. vg.add(range(num_verts), 0, 'REPLACE')
  119. def copy_weights(self, xf):
  120. # we'll use modifiers for this, maybe use GN for it in the future tho
  121. import bpy
  122. ob = xf.bGetObject()
  123. try:
  124. copy_from = self.GetxForm(socket="Copy Skin Weights From")
  125. except GraphError:
  126. copy_from = None
  127. prRed(f"No object found for copying weights in {self}, continuing anyway.")
  128. m = ob.modifiers.new(type="DATA_TRANSFER", name="Mantis_temp_data_transfer")
  129. m.object = None; m.use_vert_data = True
  130. m.data_types_verts = {'VGROUP_WEIGHTS'}
  131. m.vert_mapping = 'POLYINTERP_NEAREST'
  132. m.layers_vgroup_select_src = 'ALL'
  133. m.layers_vgroup_select_dst = 'NAME'
  134. m.object = copy_from
  135. # m.use_object_transform = False # testing reveals that this is undesirable - since the objects may not have their transforms applied.
  136. ob.modifiers.move(len(ob.modifiers)-1, 0)
  137. # ob.data = ob.data.copy()
  138. if False: #MAYBE the mouse needs to be in the 3D viewport, no idea how to set this in an override
  139. # TODO: figure out how to apply this, context is incorrect because armature is still in pose mode
  140. original_active = bpy.context.active_object
  141. original_mode = original_active.mode
  142. bpy.ops.object.mode_set(mode='OBJECT')
  143. with bpy.context.temp_override(**{'active_object':ob, 'selected_objects':[ob, copy_from]}):
  144. # bpy.ops.object.datalayout_transfer(modifier=m.name) # note: this operator is used by the modifier or stand-alone in the UI
  145. # the poll for this operator is defined in blender/source/blender/editors/object/object_data_transfer.cc
  146. # and blender/source/blender/editors/object/object_modifier.cc
  147. # bpy.ops.object.modifier_apply(modifier=m.name, single_user=True)
  148. bpy.ops.object.datalayout_transfer(data_type='VGROUP_WEIGHTS')
  149. bpy.ops.object.data_transfer(data_type='VGROUP_WEIGHTS')
  150. bpy.ops.object.mode_set(mode=original_mode)
  151. def bFinalize(self, bContext=None):
  152. prGreen("Executing Armature Deform Node")
  153. mod_name = self.evaluate_input("Name")
  154. for xf in self.GetxForm():
  155. ob = xf.bGetObject()
  156. d = ob.modifiers.new(mod_name, type='ARMATURE')
  157. if d is None:
  158. raise RuntimeError(f"Modifier was not created in node {self} -- the object is invalid.")
  159. self.bObject.append(d)
  160. d.object = self.bGetParentArmature()
  161. props_sockets = {
  162. 'vertex_group' : ("Blend Vertex Group", ""),
  163. 'invert_vertex_group' : ("Invert Vertex Group", ""),
  164. 'use_deform_preserve_volume' : ("Preserve Volume", False),
  165. 'use_multi_modifier' : ("Use Multi Modifier", False),
  166. 'use_bone_envelopes' : ("Use Envelopes", False),
  167. 'use_vertex_groups' : ("Use Vertex Groups", False),
  168. }
  169. evaluate_sockets(self, d, props_sockets)
  170. #
  171. if (skin_method := self.evaluate_input("Skinning Method")) == "AUTOMATIC_HEAT":
  172. # This is bad and leads to somewhat unpredictable
  173. # behaviour, e.g. what object will be selected? What mode?
  174. # also bpy.ops is ugly and prone to error when used in
  175. # scripts. I don't intend to use bpy.ops when I can avoid it.
  176. import bpy
  177. self.initialize_vgroups(xf)
  178. bContext.view_layer.depsgraph.update()
  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. for b in armOb.data.bones:
  190. b.select = True
  191. with bContext.temp_override(**context_override):
  192. bpy.ops.paint.weight_paint_toggle()
  193. bpy.ops.paint.weight_from_bones(type='AUTOMATIC')
  194. bpy.ops.paint.weight_paint_toggle()
  195. for b in armOb.data.bones:
  196. b.select = False
  197. # TODO: modify Blender to make this available as a Python API function.
  198. elif skin_method == "COPY_FROM_OBJECT":
  199. self.initialize_vgroups(xf)
  200. self.copy_weights(xf)
  201. # elif skin_method == "EXISTING_GROUPS":
  202. # pass
  203. class DeformerHook(MantisDeformerNode):
  204. '''A node representing a hook deformer'''
  205. def __init__(self, signature, base_tree):
  206. super().__init__(signature, base_tree, HookSockets)
  207. # now set up the traverse target...
  208. self.init_parameters(additional_parameters={"Name":None})
  209. self.set_traverse([("Deformer", "Deformer")])
  210. self.prepared = True
  211. def driver_for_radius(self, object, hook, index, influence, bezier=True):
  212. """ Creates a driver to control the radius of a curve point with the hook."""
  213. from bpy.types import Bone, PoseBone
  214. var_template = {"owner":hook,
  215. "name":"a",
  216. "type":"TRANSFORMS",
  217. "space":'WORLD_SPACE',
  218. "channel":'SCALE_X',}
  219. var1_template = {"owner":hook.id_data,
  220. "name":"b",
  221. "type":"TRANSFORMS",
  222. "space":'WORLD_SPACE',
  223. "channel":'SCALE_X',}
  224. keys_template = [{"co":(0,0),
  225. "interpolation": "LINEAR",
  226. "type":"KEYFRAME",},
  227. {"co":(1,influence),
  228. "interpolation": "LINEAR",
  229. "type":"KEYFRAME",},]
  230. if bezier:
  231. owner=object.data.splines[0].bezier_points
  232. else:
  233. owner=object.data.splines[0].points
  234. driver = {
  235. "owner":owner[index],
  236. "prop":"radius",
  237. "ind":-1,
  238. "extrapolation":"LINEAR",
  239. "type":"AVERAGE",
  240. "vars":[],
  241. "keys":keys_template,
  242. }
  243. if isinstance(hook, (Bone, PoseBone)):
  244. driver['type']='SCRIPTED'
  245. driver['expression']="(((1/b)*a)+((1/b_001)*a_001)+((1/b_002)*a_002))/3"
  246. from .drivers import CreateDrivers
  247. axes='XYZ'
  248. for i in range(3):
  249. var = var_template.copy()
  250. var["channel"]="SCALE_"+axes[i]
  251. driver["vars"].append(var)
  252. if isinstance(hook, (Bone, PoseBone)):
  253. var1=var1_template.copy()
  254. var1['channel']="SCALE_"+axes[i]
  255. driver['vars'].append(var1)
  256. CreateDrivers([driver])
  257. def bExecute(self, bContext = None,):
  258. self.executed = True
  259. def bFinalize(self, bContext=None):
  260. from bpy.types import Bone, PoseBone, Object
  261. prGreen(f"Executing Hook Deform Node: {self}")
  262. mod_name = self.evaluate_input("Name")
  263. affect_radius = self.evaluate_input("Affect Curve Radius")
  264. auto_bezier = self.evaluate_input("Auto-Bezier")
  265. target_node = self.evaluate_input('Hook Target')
  266. target = target_node.bGetObject(); subtarget = ""
  267. props_sockets = self.gen_property_socket_map()
  268. if isinstance(target, Bone) or isinstance(target, PoseBone):
  269. subtarget = target.name; target = target.id_data
  270. for xf in self.GetxForm():
  271. ob=xf.bGetObject()
  272. if ob.type == 'CURVE':
  273. spline_index = self.evaluate_input("Spline Index")
  274. from .utilities import get_extracted_spline_object
  275. ob = get_extracted_spline_object(ob, spline_index, self.mContext)
  276. reuse = False
  277. for m in ob.modifiers:
  278. if m.type == 'HOOK' and m.object == target and m.subtarget == subtarget:
  279. if self.evaluate_input("Influence") != m.strength:
  280. continue # make a new modifier so they can have different strengths
  281. if ob.animation_data: # this can be None
  282. drivers = ob.animation_data.drivers
  283. for k in props_sockets.keys():
  284. if driver := drivers.find(k):
  285. # TODO: I should check to see if the drivers are the same...
  286. break # continue searching for an equivalent modifier
  287. else: # There was no driver - use this one.
  288. d = m; reuse = True; break
  289. else: # use this one, there can't be drivers without animation_data.
  290. d = m; reuse = True; break
  291. else:
  292. d = ob.modifiers.new(mod_name, type='HOOK')
  293. if d is None:
  294. raise RuntimeError(f"Modifier was not created in node {self} -- the object is invalid.")
  295. self.bObject.append(d)
  296. self.get_target_and_subtarget(d, input_name="Hook Target")
  297. vertices_used=[]
  298. if reuse: # Get the verts in the list... filter out all the unneeded 0's
  299. vertices_used = list(d.vertex_indices)
  300. include_0 = 0 in vertices_used
  301. vertices_used = list(filter(lambda a : a != 0, vertices_used))
  302. if include_0: vertices_used.append(0)
  303. # now we add the selected vertex to the list, too
  304. vertex = self.evaluate_input("Point Index")
  305. if ob.type == 'CURVE' and ob.data.splines[0].type == 'BEZIER' and auto_bezier:
  306. if affect_radius:
  307. self.driver_for_radius(ob, target_node.bGetObject(), vertex, d.strength)
  308. vertex*=3
  309. vertices_used.extend([vertex, vertex+1, vertex+2])
  310. else:
  311. vertices_used.append(vertex)
  312. # if we have a curve and it is NOT using auto-bezier for the verts..
  313. if ob.type == 'CURVE' and ob.data.splines[0].type == 'BEZIER' and affect_radius and not auto_bezier:
  314. print (f"WARN: {self}: \"Affect Radius\" may not behave as expected"
  315. " when used on Bezier curves without Auto-Bezier")
  316. #bezier point starts at 1, and then every third vert, so 4, 7, 10...
  317. if vertex%3==1:
  318. self.driver_for_radius(ob, target_node.bGetObject(), vertex, d.strength)
  319. if ob.type == 'CURVE' and ob.data.splines[0].type != 'BEZIER' and \
  320. affect_radius:
  321. self.driver_for_radius(ob, target_node.bGetObject(), vertex, d.strength, bezier=False)
  322. d.vertex_indices_set(vertices_used)
  323. evaluate_sockets(self, d, props_sockets)
  324. finish_drivers(self)
  325. # todo: this node should be able to take many indices in the future.
  326. # Also: I have a Geometry Nodes implementation of this I can use... maybe...
  327. class DeformerMorphTarget(MantisDeformerNode):
  328. '''A node representing an armature deformer'''
  329. def __init__(self, signature, base_tree):
  330. super().__init__(signature, base_tree)
  331. inputs = [
  332. "Relative to",
  333. "Object",
  334. "Deformer",
  335. "Vertex Group",
  336. ]
  337. outputs = [
  338. "Deformer",
  339. "Morph Target",
  340. ]
  341. # now set up the traverse target...
  342. self.outputs.init_sockets(outputs)
  343. self.inputs.init_sockets(inputs)
  344. self.init_parameters(additional_parameters={"Name":None})
  345. self.set_traverse([("Deformer", "Deformer")])
  346. self.node_type = "LINK"
  347. self.prepared = True
  348. def GetxForm(self, trace_input="Object"):
  349. trace = trace_single_line(self, trace_input)
  350. for node in trace[0]:
  351. if (isinstance(node, deformable_types)):
  352. return node
  353. raise GraphError("%s is not connected to an upstream xForm" % self)
  354. def bExecute(self, bContext = None,):
  355. prGreen("Executing Morph Target Node")
  356. ob = None; relative = None
  357. # do NOT check if the object exists here. Just let the next node deal with that.
  358. try:
  359. ob = self.GetxForm().bGetObject().name
  360. except Exception as e: # this will and should throw an error if it fails
  361. ob = self.GetxForm().evaluate_input("Name")
  362. if self.inputs["Relative to"].is_linked:
  363. try:
  364. relative = self.GetxForm("Relative to").bGetObject().name
  365. except Exception as e: # same here
  366. prRed(f"Execution failed at {self}: no relative object found for morph target, despite link existing.")
  367. raise e
  368. vg = self.evaluate_input("Vertex Group") if self.evaluate_input("Vertex Group") else "" # just make sure it is a string
  369. mt={"object":ob, "vertex_group":vg, "relative_shape":relative}
  370. self.parameters["Morph Target"] = mt
  371. self.parameters["Name"] = ob # this is redundant but it's OK since accessing the mt is tedious
  372. self.executed = True
  373. class DeformerMorphTargetDeform(MantisDeformerNode):
  374. '''A node representing an armature deformer'''
  375. def __init__(self, signature, base_tree):
  376. super().__init__(signature, base_tree)
  377. inputs = [
  378. "Deformer",
  379. "Use Shape Key",
  380. "Use Offset",
  381. ]
  382. outputs = [
  383. "Deformer",
  384. ]
  385. self.outputs.init_sockets(outputs)
  386. self.inputs.init_sockets(inputs)
  387. self.init_parameters(additional_parameters={"Name":None})
  388. self.set_traverse([("Deformer", "Deformer")])
  389. self.node_type = "LINK"
  390. self.prepared = True
  391. self.executed = True
  392. setup_custom_props(self)
  393. # bpy.data.node_groups["Morph Deform.045"].nodes["Named Attribute.020"].data_type = 'FLOAT_VECTOR'
  394. # bpy.context.object.add_rest_position_attribute = True
  395. def reset_execution(self):
  396. return super().reset_execution()
  397. self.executed=True
  398. def gen_morph_target_modifier(self, xf, context):
  399. # first let's see if this is a no-op
  400. targets = []
  401. for k,v in self.inputs.items():
  402. if "Target" in k:
  403. targets.append(v)
  404. if not targets:
  405. return # nothing to do here.
  406. # at this point we make the node tree
  407. from .geometry_node_graphgen import gen_morph_target_nodes
  408. m, props_sockets = gen_morph_target_nodes(
  409. self.evaluate_input("Name"),
  410. xf.bGetObject(),
  411. targets,
  412. context,
  413. use_offset=self.evaluate_input("Use Offset"))
  414. self.bObject.append(m)
  415. evaluate_sockets(self, m, props_sockets)
  416. finish_drivers(self)
  417. def gen_shape_key(self, xf, context): # TODO: make this a feature of the node definition that appears only when there are no prior deformers - and shows a warning!
  418. # 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.
  419. # there are a number of things I need to fix here
  420. # - reuse shape keys if possible
  421. # - figure out how to make this a lot faster
  422. # - edit the xForm stuff to delete drivers from shape key ID's, since they belong to the Key, not the Object.
  423. # first check if we need to do anythign
  424. targets = []
  425. for k,v in self.inputs.items():
  426. if "Target" in k:
  427. targets.append(v)
  428. if not targets:
  429. return # nothing to do here
  430. from time import time
  431. start_time = time()
  432. from bpy import data
  433. ob = xf.bGetObject()
  434. dg = context.view_layer.depsgraph
  435. dg.update()
  436. if xf.has_shape_keys == False:
  437. m = data.meshes.new_from_object(ob, preserve_all_data_layers=True, depsgraph=dg)
  438. ob.data = m
  439. ob.add_rest_position_attribute = True
  440. ob.shape_key_clear()
  441. ob.shape_key_add(name='Basis', from_mix=False)
  442. else:
  443. m = ob.data
  444. xf.has_shape_keys = True
  445. # using the built-in shapekey feature is actually a lot harder in terms of programming because I need...
  446. # min/max, as it is just not a feature of the GN version
  447. # to carry info from the morph target node regarding relative shapes and vertex groups and all that
  448. # the drivers may be more difficult to apply, too.
  449. # hafta make new geometry for the object and add shape keys and all that
  450. # the benefit to all this being exporting to game engines via .fbx
  451. # first make a basis shape key
  452. keys={}
  453. props_sockets={}
  454. for i, t in enumerate(targets):
  455. mt_node = t.links[0].from_node; sk_ob = mt_node.GetxForm().bGetObject()
  456. if sk_ob is None:
  457. sk_ob = data.objects.new(mt_node.evaluate_input("Name"), data.meshes.new_from_object(ob))
  458. context.collection.objects.link(sk_ob)
  459. prOrange(f"WARN: no object found for f{mt_node}; creating duplicate of current object ")
  460. sk_ob = dg.id_eval_get(sk_ob)
  461. mt_name = sk_ob.name
  462. vg = mt_node.parameters["Morph Target"]["vertex_group"]
  463. if vg: mt_name = mt_name+"."+vg
  464. sk = ob.shape_key_add(name=mt_name, from_mix=False)
  465. # the shapekey data is absolute point data for each vertex, in order, very simple
  466. # SERIOUSLY IMPORTANT:
  467. # use the current position of the vertex AFTER SHAPE KEYS AND DEFORMERS
  468. # easiest way to do it is to eval the depsgraph
  469. # TODO: try and get it without depsgraph update, since that may be (very) slow
  470. sk_m = sk_ob.data#data.meshes.new_from_object(sk_ob, preserve_all_data_layers=True, depsgraph=dg)
  471. for j in range(len(m.vertices)):
  472. sk.data[j].co = sk_m.vertices[j].co # assume they match
  473. # data.meshes.remove(sk_m)
  474. sk.vertex_group = vg
  475. sk.slider_min = -10
  476. sk.slider_max = 10
  477. keys[mt_name]=sk
  478. props_sockets[mt_name]= ("Value."+str(i).zfill(3), 1.0)
  479. for i, t in enumerate(targets):
  480. mt_node = t.links[0].from_node; sk_ob = mt_node.GetxForm().bGetObject()
  481. if sk_ob is None: continue
  482. if rel := mt_node.parameters["Morph Target"]["relative_shape"]:
  483. sk = keys.get(mt_name)
  484. sk.relative_key = keys.get(rel)
  485. self.bObject.append(sk.id_data)
  486. evaluate_sockets(self, sk.id_data, props_sockets)
  487. finish_drivers(self)
  488. prWhite(f"Initializing morph target took {time() -start_time} seconds")
  489. def bFinalize(self, bContext=None):
  490. prGreen(f"Executing Morph Deform node {self}")
  491. use_shape_keys = self.evaluate_input("Use Shape Key")
  492. # if there is a not a prior deformer then there should be an option to use plain 'ol shape keys
  493. # GN is always desirable as an option though because it can be baked & many other reasons
  494. if use_shape_keys: # check and see if we can.
  495. if self.inputs.get("Deformer"): # I guess this isn't available in some node group contexts... bad. FIXME
  496. if (links := self.inputs["Deformer"].links):
  497. if not links[0].from_node.parameters.get("Use Shape Key"):
  498. use_shape_keys = False
  499. elif links[0].from_node.parameters.get("Use Shape Key") == False:
  500. use_shape_keys = False
  501. self.parameters["Use Shape Key"] = use_shape_keys
  502. for xf in self.GetxForm():
  503. if use_shape_keys:
  504. self.gen_shape_key(xf, bContext)
  505. else:
  506. self.gen_morph_target_modifier(xf, bContext)
  507. class DeformerSurfaceDeform(MantisDeformerNode):
  508. '''A node representing an surface deform modifier'''
  509. def __init__(self, signature, base_tree):
  510. super().__init__(signature, base_tree, SurfaceDeformSockets)
  511. # now set up the traverse target...
  512. self.init_parameters(additional_parameters={"Name":None})
  513. self.set_traverse([("Deformer", "Deformer")])
  514. self.prepared = True
  515. def GetxForm(self, socket="Deformer"):
  516. if socket == "Deformer":
  517. return super().GetxForm()
  518. else:
  519. trace_xForm_back(self, socket)
  520. def bExecute(self, bContext = None,):
  521. self.executed = True
  522. def bFinalize(self, bContext=None):
  523. prGreen("Executing Surface Deform Node")
  524. mod_name = self.evaluate_input("Name")
  525. for xf in self.GetxForm():
  526. ob = xf.bGetObject()
  527. d = ob.modifiers.new(mod_name, type='SURFACE_DEFORM')
  528. if d is None:
  529. raise RuntimeError(f"Modifier was not created in node {self} -- the object is invalid.")
  530. self.bObject.append(d)
  531. self.get_target_and_subtarget(d, input_name="Target")
  532. props_sockets = self.gen_property_socket_map()
  533. evaluate_sockets(self, d, props_sockets)
  534. # now we have to bind it
  535. import bpy
  536. target = d.target
  537. with bpy.context.temp_override(**{'active_object':ob, 'selected_objects':[ob, target]}):
  538. bpy.ops.object.surfacedeform_bind(modifier=d.name)
  539. class DeformerMeshDeform(MantisDeformerNode):
  540. '''A node representing a mesh deform modifier'''
  541. def __init__(self, signature, base_tree):
  542. super().__init__(signature, base_tree, MeshDeformSockets)
  543. # now set up the traverse target...
  544. self.init_parameters(additional_parameters={"Name":None})
  545. self.set_traverse([("Deformer", "Deformer")])
  546. self.prepared = True
  547. def GetxForm(self, socket="Deformer"):
  548. if socket == "Deformer":
  549. return super().GetxForm()
  550. else:
  551. trace_xForm_back(self, socket)
  552. def bExecute(self, bContext = None,):
  553. self.executed = True
  554. def bFinalize(self, bContext=None):
  555. prGreen("Executing Mesh Deform Node")
  556. mod_name = self.evaluate_input("Name")
  557. for xf in self.GetxForm():
  558. ob = xf.bGetObject()
  559. d = ob.modifiers.new(mod_name, type='MESH_DEFORM')
  560. if d is None:
  561. raise RuntimeError(f"Modifier was not created in node {self} -- the object is invalid.")
  562. self.bObject.append(d)
  563. self.get_target_and_subtarget(d, input_name="Object")
  564. props_sockets = self.gen_property_socket_map()
  565. evaluate_sockets(self, d, props_sockets)
  566. # now we have to bind it
  567. import bpy
  568. target = d.object
  569. with bpy.context.temp_override(**{'active_object':ob, 'selected_objects':[ob, target]}):
  570. bpy.ops.object.meshdeform_bind(modifier=d.name)
  571. # todo: add influence parameter and set it up with vertex group and geometry nodes
  572. # todo: make cage object display as wireframe if it is not being used for something else
  573. # or add the option in the Geometry Object node