mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-26 22:02:15 +01:00
79 lines
2.9 KiB
Python
79 lines
2.9 KiB
Python
from avalon.nuke.pipeline import Creator
|
|
from avalon.nuke import lib as anlib
|
|
import nuke
|
|
import nukescripts
|
|
|
|
class CreateGizmo(Creator):
|
|
"""Add Publishable "gizmo" group
|
|
|
|
The name is symbolically gizmo as presumably
|
|
it is something familiar to nuke users as group of nodes
|
|
distributed downstream in workflow
|
|
"""
|
|
|
|
name = "gizmo"
|
|
label = "Gizmo"
|
|
family = "gizmo"
|
|
icon = "file-archive-o"
|
|
defaults = ["ViewerInput", "Lut", "Effect"]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(CreateGizmo, self).__init__(*args, **kwargs)
|
|
self.nodes = nuke.selectedNodes()
|
|
self.node_color = "0x7533c1ff"
|
|
return
|
|
|
|
def process(self):
|
|
if (self.options or {}).get("useSelection"):
|
|
nodes = self.nodes
|
|
self.log.info(len(nodes))
|
|
if len(nodes) == 1:
|
|
anlib.select_nodes(nodes)
|
|
node = nodes[-1]
|
|
# check if Group node
|
|
if node.Class() in "Group":
|
|
node["name"].setValue("{}_GZM".format(self.name))
|
|
node["tile_color"].setValue(int(self.node_color, 16))
|
|
return anlib.imprint(node, self.data)
|
|
else:
|
|
nuke.message("Please select a group node "
|
|
"you wish to publish as the gizmo")
|
|
|
|
if len(nodes) >= 2:
|
|
anlib.select_nodes(nodes)
|
|
nuke.makeGroup()
|
|
gizmo_node = nuke.selectedNode()
|
|
gizmo_node["name"].setValue("{}_GZM".format(self.name))
|
|
gizmo_node["tile_color"].setValue(int(self.node_color, 16))
|
|
|
|
# add sticky node wit guide
|
|
with gizmo_node:
|
|
sticky = nuke.createNode("StickyNote")
|
|
sticky["label"].setValue(
|
|
"Add following:\n- set Input"
|
|
" nodes\n- set one Output1\n"
|
|
"- create User knobs on the group")
|
|
|
|
# add avalon knobs
|
|
return anlib.imprint(gizmo_node, self.data)
|
|
|
|
else:
|
|
nuke.message("Please select nodes you "
|
|
"wish to add to the gizmo")
|
|
return
|
|
else:
|
|
with anlib.maintained_selection():
|
|
gizmo_node = nuke.createNode("Group")
|
|
gizmo_node["name"].setValue("{}_GZM".format(self.name))
|
|
gizmo_node["tile_color"].setValue(int(self.node_color, 16))
|
|
|
|
# add sticky node wit guide
|
|
with gizmo_node:
|
|
sticky = nuke.createNode("StickyNote")
|
|
sticky["label"].setValue(
|
|
"Add following:\n- add Input"
|
|
" nodes\n- add one Output1\n"
|
|
"- create User knobs on the group")
|
|
|
|
# add avalon knobs
|
|
return anlib.imprint(gizmo_node, self.data)
|