2017-03-21 03:06:38 +01:00
|
|
|
from arm.logicnode.arm_nodes import *
|
|
|
|
|
2020-09-08 21:16:20 +02:00
|
|
|
class ObjectNode(ArmLogicTreeNode):
|
2020-10-05 19:55:56 +02:00
|
|
|
"""Stores the given object as a variable."""
|
2017-04-04 23:11:31 +02:00
|
|
|
bl_idname = 'LNObjectNode'
|
2017-03-21 03:06:38 +01:00
|
|
|
bl_label = 'Object'
|
2020-09-13 11:27:58 +02:00
|
|
|
arm_version = 1
|
2020-09-08 17:03:24 +02:00
|
|
|
|
2017-03-21 03:06:38 +01:00
|
|
|
def init(self, context):
|
2020-09-13 11:27:58 +02:00
|
|
|
super(ObjectNode, self).init(context)
|
2020-09-20 20:10:01 +02:00
|
|
|
self.add_input('ArmNodeSocketObject', 'Object In')
|
2020-11-17 16:01:15 +01:00
|
|
|
|
2020-09-20 20:10:01 +02:00
|
|
|
self.add_output('ArmNodeSocketObject', 'Object Out', is_var=True)
|
2020-12-15 23:47:38 +01:00
|
|
|
|
|
|
|
def draw_label(self) -> str:
|
|
|
|
inp_object = self.inputs['Object In']
|
|
|
|
if inp_object.is_linked:
|
|
|
|
return self.bl_label
|
|
|
|
|
|
|
|
obj_name = inp_object.get_default_value()
|
|
|
|
if obj_name == '':
|
|
|
|
obj_name = '_self'
|
|
|
|
|
|
|
|
return f'{self.bl_label}: {obj_name}'
|