53 lines
1.0 KiB
Python
53 lines
1.0 KiB
Python
import typing as typ
|
|
|
|
from .... import contracts as ct
|
|
from .... import sockets
|
|
from ... import base
|
|
|
|
class BlenderConstantNode(base.MaxwellSimNode):
|
|
node_type = ct.NodeType.BlenderConstant
|
|
bl_label = "Blender Constant"
|
|
|
|
input_socket_sets = {
|
|
"Object": {
|
|
"Value": sockets.BlenderObjectSocketDef(),
|
|
},
|
|
"Collection": {
|
|
"Value": sockets.BlenderCollectionSocketDef(),
|
|
},
|
|
"Text": {
|
|
"Value": sockets.BlenderTextSocketDef(),
|
|
},
|
|
"Image": {
|
|
"Value": sockets.BlenderImageSocketDef(),
|
|
},
|
|
"GeoNode Tree": {
|
|
"Value": sockets.BlenderGeoNodesSocketDef(),
|
|
},
|
|
}
|
|
output_socket_sets = input_socket_sets
|
|
|
|
####################
|
|
# - Callbacks
|
|
####################
|
|
@base.computes_output_socket(
|
|
"Value",
|
|
input_sockets={"Value"}
|
|
)
|
|
def compute_value(self, input_sockets) -> typ.Any:
|
|
return input_sockets["Value"]
|
|
|
|
|
|
|
|
####################
|
|
# - Blender Registration
|
|
####################
|
|
BL_REGISTER = [
|
|
BlenderConstantNode,
|
|
]
|
|
BL_NODES = {
|
|
ct.NodeType.BlenderConstant: (
|
|
ct.NodeCategory.MAXWELLSIM_INPUTS_CONSTANTS
|
|
)
|
|
}
|