link_socket_templates.py 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502
  1. from .base_definitions import MantisSocketTemplate as SockTemplate
  2. from bpy import app
  3. from dataclasses import replace
  4. from .misc_nodes_socket_templates import SplineIndexTemplate
  5. # Socket Templates we will reuse:
  6. # inputs:
  7. InputRelationshipTemplate : SockTemplate = SockTemplate(
  8. name="Input Relationship", is_input=True, bl_idname='RelationshipSocket',
  9. traverse_target='Output Relationship')
  10. TargetTemplate : SockTemplate = SockTemplate(
  11. name="Target", is_input=True, bl_idname='xFormSocket', )
  12. Head_Tail_Template : SockTemplate = SockTemplate(
  13. name="Head/Tail", is_input=True, bl_idname='FloatFactorSocket',
  14. default_value=0.0, blender_property='head_tail' )
  15. UseBBoneTemplate : SockTemplate = SockTemplate(
  16. name="UseBBone", is_input=True, bl_idname='BooleanSocket',
  17. default_value=False, blender_property='use_bbone_shape' )
  18. AxeSockTemplate : SockTemplate = SockTemplate(
  19. name="Axes", is_input=True, bl_idname='BooleanThreeTupleSocket',
  20. default_value=[True, True, True], blender_property=['use_x', 'use_y', 'use_z'])
  21. AxesInvertTemplate : SockTemplate = SockTemplate(
  22. name="Invert", is_input=True, bl_idname='BooleanThreeTupleSocket',
  23. default_value=[False, False, False], blender_property=['invert_x', 'invert_y', 'invert_z'])
  24. TargetSpaceTemplate : SockTemplate = SockTemplate(
  25. name="Target Space", is_input=True, bl_idname='TransformSpaceSocket',
  26. default_value="WORLD", blender_property='target_space' )
  27. OwnerSpaceTemplate : SockTemplate = SockTemplate(
  28. name="Owner Space", is_input=True, bl_idname='TransformSpaceSocket',
  29. default_value="WORLD", blender_property='owner_space' )
  30. InfluenceTemplate : SockTemplate = SockTemplate(
  31. name="Influence", is_input=True, bl_idname='FloatFactorSocket',
  32. default_value=1.0, blender_property='influence')
  33. EnableTemplate : SockTemplate = SockTemplate(
  34. name="Enable", is_input=True, bl_idname='EnableSocket',
  35. default_value=True, blender_property='mute')
  36. OffsetTemplate : SockTemplate = SockTemplate(
  37. name="Offset", bl_idname='BooleanSocket', is_input=True,
  38. default_value=False, blender_property='use_offset')
  39. # Limit Constraints follow a pattern and can use this generator
  40. LimitTemplateGenerator = lambda name_stub, axis : SockTemplate(
  41. name=name_stub+axis.upper(), is_input=True, bl_idname='BooleanSocket' if "Use" in name_stub else "FloatSocket",
  42. default_value=False, blender_property=name_stub.lower().replace(' ', '_')+axis.lower())
  43. LimitAxesSocketTemplates = [] # could generate these with loops, but this is easier to understand
  44. LimitAxesSocketTemplates.append(UseMaxXTemplate := LimitTemplateGenerator("Use Max ", "X"))
  45. LimitAxesSocketTemplates.append(MaxXTemplate := LimitTemplateGenerator("Max ", "X"))
  46. LimitAxesSocketTemplates.append(UseMaxYTemplate := LimitTemplateGenerator("Use Max ", "Y"))
  47. LimitAxesSocketTemplates.append(MaxYTemplate := LimitTemplateGenerator("Max ", "Y"))
  48. LimitAxesSocketTemplates.append(UseMaxZTemplate := LimitTemplateGenerator("Use Max ", "Z"))
  49. LimitAxesSocketTemplates.append(MaxZTemplate := LimitTemplateGenerator("Max ", "Z"))
  50. LimitAxesSocketTemplates.append(UseMinXTemplate := LimitTemplateGenerator("Use Min ", "X"))
  51. LimitAxesSocketTemplates.append(MinXTemplate := LimitTemplateGenerator("Min ", "X"))
  52. LimitAxesSocketTemplates.append(UseMinYTemplate := LimitTemplateGenerator("Use Min ", "Y"))
  53. LimitAxesSocketTemplates.append(MinYTemplate := LimitTemplateGenerator("Min ", "Y"))
  54. LimitAxesSocketTemplates.append(UseMinZTemplate := LimitTemplateGenerator("Use Min ", "Z"))
  55. LimitAxesSocketTemplates.append(MinZTemplate := LimitTemplateGenerator("Min ", "Z"))
  56. LimitRotationSocketTemplates = [
  57. UseXTemplate := LimitTemplateGenerator("Use ", "X"),
  58. MaxXTemplate,
  59. MinXTemplate,
  60. UseYTemplate := LimitTemplateGenerator("Use ", "Y"),
  61. MaxYTemplate,
  62. MinYTemplate,
  63. UseZTemplate := LimitTemplateGenerator("Use ", "Z"),
  64. MaxZTemplate,
  65. MinZTemplate,
  66. ]
  67. # annoyingly these are a little different than the pattern:
  68. UseXTemplate.blender_property='use_limit_x'
  69. UseYTemplate.blender_property='use_limit_y'
  70. UseZTemplate.blender_property='use_limit_z'
  71. AffectTransformTemplate : SockTemplate = SockTemplate(
  72. name="Affect Transform", bl_idname='BooleanSocket', is_input=True,
  73. default_value=False, blender_property='use_transform_limit')
  74. # Tracking
  75. TrackAxisTemplate= SockTemplate(name="Track Axis", bl_idname="EnumTrackAxis",
  76. is_input=True, default_value='TRACK_Y', blender_property='track_axis')
  77. # outputs:
  78. OutputRelationshipTemplate : SockTemplate = SockTemplate(
  79. name="Output Relationship", is_input=False, bl_idname='RelationshipSocket',
  80. traverse_target='Input Relationship')
  81. LinkInheritSockets = [
  82. SockTemplate(name="Inherit Rotation", is_input=True,
  83. bl_idname='BooleanSocket', default_value=True,),
  84. SockTemplate(name="Inherit Scale", is_input=True,
  85. bl_idname='EnumInheritScale', default_value="FULL",),
  86. SockTemplate(name="Connected", is_input=True,
  87. bl_idname='BooleanSocket', default_value=False,),
  88. SockTemplate(name="Parent", is_input=True, bl_idname='xFormSocket',
  89. traverse_target='Inheritance'),
  90. SockTemplate(name="Inheritance", is_input=False, bl_idname='RelationshipSocket',
  91. traverse_target='Parent'),
  92. ]
  93. LinkCopyLocationSockets = [
  94. InputRelationshipTemplate,
  95. Head_Tail_Template,
  96. UseBBoneTemplate,
  97. AxeSockTemplate,
  98. AxesInvertTemplate,
  99. TargetSpaceTemplate,
  100. OwnerSpaceTemplate,
  101. OffsetTemplate,
  102. InfluenceTemplate,
  103. TargetTemplate,
  104. EnableTemplate,
  105. OutputRelationshipTemplate,
  106. ]
  107. LinkCopyRotationSockets = [
  108. InputRelationshipTemplate,
  109. SockTemplate(name='RotationOrder', bl_idname='RotationOrderSocket', is_input=True,
  110. default_value='AUTO', blender_property='euler_order'),
  111. SockTemplate(name='Rotation Mix', bl_idname='EnumRotationMix', is_input=True,
  112. default_value='REPLACE', blender_property='mix_mode'),
  113. AxeSockTemplate,
  114. AxesInvertTemplate,
  115. TargetSpaceTemplate,
  116. OwnerSpaceTemplate,
  117. InfluenceTemplate,
  118. TargetTemplate,
  119. EnableTemplate,
  120. OutputRelationshipTemplate,
  121. ]
  122. LinkCopyScaleSockets = [
  123. InputRelationshipTemplate,
  124. OffsetTemplate,
  125. SockTemplate(name='Average', bl_idname = 'BooleanSocket', is_input=True,
  126. default_value=False, blender_property='use_make_uniform'),
  127. SockTemplate(name='Additive', bl_idname = 'BooleanSocket', is_input=True,
  128. default_value=False, blender_property='use_add'),
  129. AxeSockTemplate,
  130. SockTemplate(name='Power', bl_idname = 'FloatFactorSocket', is_input=True,
  131. default_value=1.0, blender_property='power'),
  132. TargetSpaceTemplate,
  133. OwnerSpaceTemplate,
  134. InfluenceTemplate,
  135. TargetTemplate,
  136. EnableTemplate,
  137. OutputRelationshipTemplate,
  138. ]
  139. LinkCopyTransformsSockets = [
  140. InputRelationshipTemplate,
  141. Head_Tail_Template,
  142. UseBBoneTemplate,
  143. SockTemplate(name='Mix', bl_idname = 'EnumRotationMixCopyTransforms', is_input=True,
  144. default_value="REPLACE", blender_property='mix_mode'),
  145. TargetSpaceTemplate,
  146. OwnerSpaceTemplate,
  147. InfluenceTemplate,
  148. TargetTemplate,
  149. EnableTemplate,
  150. OutputRelationshipTemplate,
  151. ]
  152. TransformationMinMaxTemplateGenerator = lambda name, bprop : SockTemplate(
  153. name=name, is_input=True,
  154. bl_idname='EnumTransformationAxes' if "Source" in name else "FloatSocket",
  155. default_value='X' if "Source" in name else 1.0,
  156. blender_property=bprop)
  157. LinkTransformationSockets = [
  158. InputRelationshipTemplate,
  159. TargetTemplate,
  160. OwnerSpaceTemplate,
  161. TargetSpaceTemplate,
  162. SockTemplate(name="Extrapolate", is_input=True, bl_idname='BooleanSocket',
  163. default_value=False, blender_property='use_motion_extrapolate'),
  164. SockTemplate(name="Map From", is_input=True, bl_idname='EnumTransformationMap',
  165. default_value="LOCATION", blender_property='map_from'),
  166. SockTemplate(name="Rotation Mode", is_input=True, bl_idname='EnumTransformationRotationMode',
  167. default_value="AUTO", blender_property='from_rotation_mode', hide=True),
  168. TransformationMinMaxTemplateGenerator("X Min From", "from_min_x"),
  169. TransformationMinMaxTemplateGenerator("X Max From", "from_max_x"),
  170. TransformationMinMaxTemplateGenerator("Y Min From", "from_min_y"),
  171. TransformationMinMaxTemplateGenerator("Y Max From", "from_max_y"),
  172. TransformationMinMaxTemplateGenerator("Z Min From", "from_min_z"),
  173. TransformationMinMaxTemplateGenerator("Z Max From", "from_max_z"),
  174. SockTemplate(name="Map To", is_input=True, bl_idname='EnumTransformationMap',
  175. default_value="LOCATION", blender_property='map_to'),
  176. SockTemplate(name="Rotation Order", is_input=True, bl_idname='EnumTransformationRotationOrder',
  177. default_value="AUTO", blender_property='to_euler_order', hide=True),
  178. TransformationMinMaxTemplateGenerator("X Source Axis", "map_to_x_from"),
  179. TransformationMinMaxTemplateGenerator("X Min To", "to_min_x"),
  180. TransformationMinMaxTemplateGenerator("X Max To", "to_max_x"),
  181. TransformationMinMaxTemplateGenerator("Y Source Axis", "map_to_y_from"),
  182. TransformationMinMaxTemplateGenerator("Y Min To", "to_min_y"),
  183. TransformationMinMaxTemplateGenerator("Y Max To", "to_max_y"),
  184. TransformationMinMaxTemplateGenerator("Z Source Axis", "map_to_z_from"),
  185. TransformationMinMaxTemplateGenerator("Z Min To", "to_min_z"),
  186. TransformationMinMaxTemplateGenerator("Z Max To", "to_max_z"),
  187. SockTemplate(name="Mix Mode (Translation)", is_input=True,
  188. bl_idname='EnumTransformationTranslationMixMode',
  189. default_value="REPLACE", blender_property='mix_mode',),
  190. SockTemplate(name="Mix Mode (Rotation)", is_input=True,
  191. bl_idname='EnumTransformationRotationMixMode',
  192. default_value="ADD", blender_property='mix_mode_rot', hide=True),
  193. SockTemplate(name="Mix Mode (Scale)", is_input=True,
  194. bl_idname='EnumTransformationScaleMixMode',
  195. default_value="REPLACE", blender_property='mix_mode_scale', hide=True),
  196. InfluenceTemplate,
  197. EnableTemplate,
  198. OutputRelationshipTemplate,
  199. ]
  200. LinkLimitLocationSockets = [
  201. InputRelationshipTemplate,
  202. *LimitAxesSocketTemplates, # we generated these ahead of time in a list
  203. AffectTransformTemplate,
  204. OwnerSpaceTemplate,
  205. InfluenceTemplate,
  206. EnableTemplate,
  207. OutputRelationshipTemplate,
  208. ]
  209. LinkLimitRotationSockets = [
  210. InputRelationshipTemplate,
  211. *LimitRotationSocketTemplates, # we generated these ahead of time in a list
  212. AffectTransformTemplate,
  213. OwnerSpaceTemplate,
  214. InfluenceTemplate,
  215. EnableTemplate,
  216. OutputRelationshipTemplate,
  217. ]
  218. LinkLimitScaleSockets = [
  219. InputRelationshipTemplate,
  220. *LimitAxesSocketTemplates, # we generated these ahead of time in a list
  221. AffectTransformTemplate,
  222. OwnerSpaceTemplate,
  223. InfluenceTemplate,
  224. EnableTemplate,
  225. OutputRelationshipTemplate,
  226. ]
  227. LinkLimitDistanceSockets = [
  228. InputRelationshipTemplate,
  229. Head_Tail_Template,
  230. UseBBoneTemplate,
  231. SockTemplate(name="Distance", bl_idname='FloatSocket', is_input=True,
  232. default_value=0.0, blender_property='distance'),
  233. SockTemplate(name="Clamp Region", bl_idname="EnumLimitMode", is_input=True,
  234. default_value='LIMITDIST_INSIDE', blender_property='limit_mode'),
  235. AffectTransformTemplate,
  236. OwnerSpaceTemplate,
  237. TargetSpaceTemplate,
  238. InfluenceTemplate,
  239. TargetTemplate,
  240. EnableTemplate,
  241. OutputRelationshipTemplate,
  242. ]
  243. LinkStretchToSockets = [
  244. InputRelationshipTemplate,
  245. Head_Tail_Template,
  246. UseBBoneTemplate,
  247. SockTemplate(name="Original Length", bl_idname='FloatSocket', is_input=True,
  248. default_value=0.0, blender_property='rest_length'),
  249. SockTemplate(name="Volume Variation", bl_idname='FloatSocket', is_input=True,
  250. default_value=1.0, blender_property='bulge'),
  251. SockTemplate(name="Use Volume Min", bl_idname='BoolUpdateParentNode', is_input=True,
  252. default_value=False, blender_property='use_bulge_min'),
  253. SockTemplate(name="Volume Min", bl_idname='FloatSocket', is_input=True,
  254. default_value=1.0, blender_property='bulge_min'),
  255. SockTemplate(name="Use Volume Max", bl_idname='BoolUpdateParentNode', is_input=True,
  256. default_value=False, blender_property='use_bulge_max'),
  257. SockTemplate(name="Volume Max", bl_idname='FloatSocket', is_input=True,
  258. default_value=1.0, blender_property='bulge_max'),
  259. SockTemplate(name="Smooth", bl_idname='FloatFactorSocket', is_input=True,
  260. default_value=0.0, blender_property='bulge_smooth'),
  261. SockTemplate(name="Maintain Volume", bl_idname='EnumMaintainVolumeStretchToSocket', is_input=True,
  262. default_value="VOLUME_XZX", blender_property='volume'),
  263. SockTemplate(name="Rotation", bl_idname='EnumRotationStretchTo', is_input=True,
  264. default_value="SWING_Y", blender_property='keep_axis'),
  265. InfluenceTemplate,
  266. TargetTemplate,
  267. EnableTemplate,
  268. OutputRelationshipTemplate,
  269. ]
  270. LinkDampedTrackSockets =[
  271. InputRelationshipTemplate,
  272. Head_Tail_Template,
  273. UseBBoneTemplate,
  274. TrackAxisTemplate,
  275. InfluenceTemplate,
  276. TargetTemplate,
  277. EnableTemplate,
  278. OutputRelationshipTemplate,
  279. ]
  280. LinkLockedTrackSockets =[
  281. InputRelationshipTemplate,
  282. Head_Tail_Template,
  283. UseBBoneTemplate,
  284. TrackAxisTemplate,
  285. SockTemplate(name="Lock Axis", bl_idname="EnumLockAxis", is_input=True,
  286. default_value="LOCK_Z", blender_property='lock_axis'),
  287. InfluenceTemplate,
  288. TargetTemplate,
  289. EnableTemplate,
  290. OutputRelationshipTemplate,
  291. ]
  292. # NOTE: I am setting different default values here than Blender in order to remain
  293. # consistent with the track constraints tracking the bone to the target.
  294. LinkTrackToSockets = [
  295. InputRelationshipTemplate,
  296. Head_Tail_Template,
  297. UseBBoneTemplate,
  298. TrackAxisTemplate,
  299. SockTemplate(name="Up Axis", bl_idname="EnumUpAxis", is_input=True,
  300. default_value="UP_Z", blender_property='up_axis'),
  301. SockTemplate(name="Use Target Z", bl_idname="BooleanSocket", is_input=True,
  302. default_value=False, blender_property='use_target_z'),
  303. TargetSpaceTemplate,
  304. OwnerSpaceTemplate,
  305. InfluenceTemplate,
  306. TargetTemplate,
  307. EnableTemplate,
  308. OutputRelationshipTemplate,
  309. ]
  310. # relationships & misc.
  311. LinkInheritConstraintSockets = [
  312. InputRelationshipTemplate,
  313. SockTemplate(name="Location", bl_idname='BooleanThreeTupleSocket',
  314. is_input=True, default_value=[True, True, True],
  315. blender_property=['use_location_x', 'use_location_y', 'use_location_z']),
  316. SockTemplate(name="Rotation", bl_idname='BooleanThreeTupleSocket',
  317. is_input=True, default_value=[True, True, True],
  318. blender_property=['use_rotation_x', 'use_rotation_y', 'use_rotation_z']),
  319. SockTemplate(name="Scale", bl_idname='BooleanThreeTupleSocket',
  320. is_input=True, default_value=[True, True, True],
  321. blender_property=['use_scale_x', 'use_scale_y', 'use_scale_z']),
  322. InfluenceTemplate,
  323. TargetTemplate,
  324. EnableTemplate,
  325. OutputRelationshipTemplate,
  326. ]
  327. LinkInverseKinematicsSockets = [
  328. InputRelationshipTemplate,
  329. TargetTemplate,
  330. SockTemplate(name="Pole Target", is_input=True, bl_idname='xFormSocket', ),
  331. ChainLengthTemplate := SockTemplate(name="Chain Length",
  332. bl_idname="IntSocket", is_input=True,
  333. default_value=0, blender_property='chain_count'),
  334. SockTemplate(name="Use Tail", bl_idname="BooleanSocket", is_input=True,
  335. default_value=True, blender_property='use_tail'),
  336. SockTemplate(name="Stretch", bl_idname="BooleanSocket", is_input=True,
  337. default_value=True, blender_property='use_stretch'),
  338. SockTemplate(name="Position", bl_idname="FloatFactorSocket", is_input=True,
  339. default_value=1.0, blender_property='weight'),
  340. SockTemplate(name="Rotation", bl_idname="FloatFactorSocket", is_input=True,
  341. default_value=0.0, blender_property='orient_weight'),
  342. InfluenceTemplate,
  343. EnableTemplate,
  344. OutputRelationshipTemplate,
  345. ]
  346. LinkDrivenParameterSockets = [
  347. InputRelationshipTemplate,
  348. SockTemplate(name="Value", bl_idname="FloatSocket", is_input=True,
  349. default_value=-0.0,),
  350. SockTemplate(name="Parameter", bl_idname="ParameterStringSocket", is_input=True,
  351. default_value="",),
  352. SockTemplate(name="Index", bl_idname="IntSocket", is_input=True,
  353. default_value=0,),
  354. OutputRelationshipTemplate,
  355. ]
  356. LinkArmatureSockets=[
  357. InputRelationshipTemplate,
  358. SockTemplate(name="Preserve Volume", bl_idname='BooleanSocket', is_input=True,
  359. default_value=False, blender_property='use_deform_preserve_volume'),
  360. SockTemplate(name="Use Envelopes", bl_idname='BooleanSocket', is_input=True,
  361. default_value=False, blender_property='use_bone_envelopes'),
  362. SockTemplate(name="Use Current Location", bl_idname='BooleanSocket', is_input=True,
  363. default_value=False, blender_property='use_current_location'),
  364. TargetsArray := replace(TargetTemplate, name='Targets', use_multi_input=True),
  365. WeightsArray := replace(InfluenceTemplate, name='Weights', use_multi_input=True),
  366. InfluenceTemplate,
  367. EnableTemplate,
  368. OutputRelationshipTemplate,
  369. ]
  370. LinkSplineIKSockets = [
  371. InputRelationshipTemplate,
  372. TargetTemplate,
  373. SplineIndexTemplate,
  374. ChainLengthTemplate,
  375. SockTemplate(name="Even Divisions", bl_idname="BooleanSocket", is_input=True,
  376. default_value=False, blender_property='use_even_divisions'),
  377. SockTemplate(name="Chain Offset", bl_idname="BooleanSocket", is_input=True,
  378. default_value=False, blender_property='use_chain_offset'),
  379. SockTemplate(name="Use Curve Radius", bl_idname="BooleanSocket", is_input=True,
  380. default_value=True, blender_property='use_curve_radius'),
  381. SockTemplate(name="Y Scale Mode", bl_idname="EnumYScaleMode", is_input=True,
  382. default_value="FIT_CURVE", blender_property='y_scale_mode'),
  383. SockTemplate(name="XZ Scale Mode", bl_idname="EnumXZScaleMode", is_input=True,
  384. default_value="NONE", blender_property='xz_scale_mode'),
  385. #IMPORTANT NOTE: This socket is removed in 4.5
  386. SockTemplate(name="Use Original Scale", bl_idname="BooleanSocket", is_input=True,
  387. default_value=False, blender_property='use_original_scale'),
  388. InfluenceTemplate,
  389. EnableTemplate,
  390. OutputRelationshipTemplate,
  391. ]
  392. LinkFloorSockets = [
  393. InputRelationshipTemplate,
  394. TargetTemplate,
  395. FloorOffsetTemplate := SockTemplate(name="Offset", bl_idname="FloatSocket",
  396. is_input=True, default_value=False, blender_property='offset'),
  397. FloorAxisTemplate := SockTemplate(name='Min/Max', bl_idname='EnumFloorAxis',
  398. is_input = True, default_value="FLOOR_X", blender_property="floor_location"),
  399. FloorUseRotation := SockTemplate(name="Rotation", bl_idname='BooleanSocket',
  400. is_input=True, default_value=False, blender_property='use_rotation',),
  401. TargetSpaceTemplate,
  402. OwnerSpaceTemplate,
  403. InfluenceTemplate,
  404. EnableTemplate,
  405. OutputRelationshipTemplate,
  406. ]
  407. # DO: arrange these guys in order for goodness' sake
  408. LinkShrinkWrapSockets = [
  409. InputRelationshipTemplate,
  410. TargetTemplate, # IMPORTANT TO DO: targets should be an array
  411. # and the constraints are made ONLY if the target is valid...
  412. # for BONE targets, maybe auto-magically build a mesh for the user.
  413. # this is a little bit too hard to do for now. Users can just make a schema.
  414. SWDistanceTemplate := SockTemplate(name="Distance", bl_idname="FloatSocket",
  415. is_input=True, default_value=0.0, blender_property='distance'),
  416. ShrinkwrapModeTemplate := SockTemplate(name="Mode",
  417. bl_idname="EnumShrinkwrapTypeSocket", is_input=True,
  418. blender_property='shrinkwrap_type'),
  419. ProjectAxisTemplate := SockTemplate(name="Project Axis",
  420. bl_idname="EnumShrinkwrapProjectAxisSocket", is_input=True,
  421. blender_property='project_axis', default_value='NEG_Y'),
  422. ProjectAxisSpaceTemplate := replace(TargetSpaceTemplate, name='Space',
  423. blender_property='project_axis_space', default_value='LOCAL'),
  424. SWDistanceTemplate := SockTemplate(name="Project Distance", bl_idname="FloatSocket",
  425. is_input=True, default_value=0.0, blender_property='project_limit'),
  426. UseProjectOppositeTemplate:= SockTemplate(name="Project Opposite", is_input=True,
  427. bl_idname="BooleanSocket", blender_property='use_project_opposite'),
  428. CullFaceTemplate := SockTemplate(name='Face Cull', is_input = True,
  429. bl_idname="EnumShrinkwrapFaceCullSocket", default_value='OFF',
  430. blender_property = "cull_face"),
  431. UseInvertCullTemplate:= SockTemplate(name="Invert Cull",
  432. bl_idname="BooleanSocket", is_input=True,
  433. blender_property='use_invert_cull'),
  434. SnapModeTemplate := SockTemplate(name="Snap Mode",
  435. bl_idname="EnumShrinkwrapModeSocket", is_input=True,
  436. blender_property='wrap_mode'),
  437. UseTrackNormalTemplate:= SockTemplate(name="Align to Normal", is_input=True,
  438. bl_idname="BooleanSocket", blender_property='use_track_normal'),
  439. TrackAxisTemplate:= SockTemplate(name="Align Normal Axis", bl_idname="EnumTrackAxis",
  440. is_input=True, blender_property='track_axis', default_value='TRACK_Y'),
  441. InfluenceTemplate,
  442. EnableTemplate,
  443. OutputRelationshipTemplate,
  444. ]
  445. LinkGeometryAttributeSockets = [
  446. InputRelationshipTemplate,
  447. TargetTemplate,
  448. OffsetTargetTransformTemplate:= SockTemplate(name="Offset target transform", is_input=True,
  449. bl_idname="BooleanSocket", blender_property='apply_target_transform'),
  450. AttributeTemplate:= SockTemplate(name="Attribute", bl_idname="ParameterStringSocket", is_input=True,
  451. default_value="position", blender_property='attribute_name'),
  452. DataTypeTemplate := SockTemplate(name="Data Type",
  453. bl_idname="EnumGeometryAttributeDataTypeSocket", is_input=True,
  454. blender_property='data_type'),
  455. DomainTemplate := SockTemplate(name="Domain",
  456. bl_idname="EnumGeometryAttributeDomainSocket", is_input=True,
  457. blender_property='domain'),
  458. SampleIndex := SockTemplate(name="Sample Index", bl_idname='UnsignedIntSocket',
  459. is_input=True, default_value=0,blender_property='sample_index'),
  460. EnabledLocationTemplate:= SockTemplate(name="Enabled Location", is_input=True,
  461. bl_idname="BooleanSocket", blender_property='mix_loc'),
  462. EnabledRotationTemplate:= SockTemplate(name="Enabled Rotation", is_input=True,
  463. bl_idname="BooleanSocket", blender_property='mix_rot'),
  464. EnabledScaleTemplate:= SockTemplate(name="Enabled Scale", is_input=True,
  465. bl_idname="BooleanSocket", blender_property='mix_scl'),
  466. SockTemplate(name='Mix Mode', bl_idname='EnumMixModeGeometryAttribute', is_input=True,
  467. default_value='REPLACE', blender_property='mix_mode'),
  468. InfluenceTemplate,
  469. EnableTemplate,
  470. OutputRelationshipTemplate,
  471. ]
  472. # Remove this socket because of Blender changes.
  473. if (app.version >= (4, 5, 0)):
  474. LinkSplineIKSockets.pop(9)