mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-25 05:14:40 +01:00
171 lines
5.2 KiB
Python
171 lines
5.2 KiB
Python
from collections import OrderedDict
|
|
import avalon.api
|
|
import avalon.nuke
|
|
from pype import api as pype
|
|
from pype.nuke import plugin
|
|
from pypeapp import config
|
|
|
|
import nuke
|
|
|
|
class CreateWriteRender(plugin.PypeCreator):
|
|
# change this to template preset
|
|
name = "WriteRender"
|
|
label = "Create Write Render"
|
|
hosts = ["nuke"]
|
|
nClass = "write"
|
|
family = "render"
|
|
icon = "sign-out"
|
|
defaults = ["Main", "Mask"]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(CreateWriteRender, self).__init__(*args, **kwargs)
|
|
|
|
data = OrderedDict()
|
|
|
|
data["family"] = self.family
|
|
data["families"] = self.nClass
|
|
|
|
for k, v in self.data.items():
|
|
if k not in data.keys():
|
|
data.update({k: v})
|
|
|
|
self.data = data
|
|
self.nodes = nuke.selectedNodes()
|
|
self.log.info("self.data: '{}'".format(self.data))
|
|
|
|
def process(self):
|
|
from pype.nuke import lib as pnlib
|
|
|
|
inputs = []
|
|
outputs = []
|
|
instance = nuke.toNode(self.data["subset"])
|
|
selected_node = None
|
|
|
|
# use selection
|
|
if (self.options or {}).get("useSelection"):
|
|
nodes = self.nodes
|
|
|
|
assert len(nodes) < 2, self.log.error("Select only one node. The node you want to connect to, or tick off `Use selection`")
|
|
|
|
selected_node = nodes[0]
|
|
inputs = [selected_node]
|
|
outputs = selected_node.dependent()
|
|
|
|
if instance:
|
|
if (instance.name() in selected_node.name()):
|
|
selected_node = instance.dependencies()[0]
|
|
|
|
# if node already exist
|
|
if instance:
|
|
# collect input / outputs
|
|
inputs = instance.dependencies()
|
|
outputs = instance.dependent()
|
|
selected_node = inputs[0]
|
|
# remove old one
|
|
nuke.delete(instance)
|
|
|
|
# recreate new
|
|
write_data = {
|
|
"class": self.nClass,
|
|
"families": [self.family],
|
|
"avalon": self.data
|
|
}
|
|
|
|
if self.presets.get('fpath_template'):
|
|
self.log.info("Adding template path from preset")
|
|
write_data.update(
|
|
{"fpath_template": self.presets["fpath_template"]}
|
|
)
|
|
else:
|
|
self.log.info("Adding template path from plugin")
|
|
write_data.update({
|
|
"fpath_template": "{work}/renders/nuke/{subset}/{subset}.{frame}.{ext}"})
|
|
|
|
write_node = pnlib.create_write_node(
|
|
self.data["subset"],
|
|
write_data,
|
|
input=selected_node)
|
|
|
|
# relinking to collected connections
|
|
for i, input in enumerate(inputs):
|
|
write_node.setInput(i, input)
|
|
|
|
write_node.autoplace()
|
|
|
|
for output in outputs:
|
|
output.setInput(0, write_node)
|
|
|
|
return write_node
|
|
|
|
#
|
|
# class CreateWritePrerender(avalon.nuke.Creator):
|
|
# # change this to template preset
|
|
# preset = "prerender"
|
|
#
|
|
# name = "WritePrerender"
|
|
# label = "Create Write Prerender"
|
|
# hosts = ["nuke"]
|
|
# family = "{}_write".format(preset)
|
|
# families = preset
|
|
# icon = "sign-out"
|
|
# defaults = ["Main", "Mask"]
|
|
#
|
|
# def __init__(self, *args, **kwargs):
|
|
# super(CreateWritePrerender, self).__init__(*args, **kwargs)
|
|
# self.presets = config.get_presets()['plugins']["nuke"]["create"].get(
|
|
# self.__class__.__name__, {}
|
|
# )
|
|
#
|
|
# data = OrderedDict()
|
|
#
|
|
# data["family"] = self.family.split("_")[1]
|
|
# data["families"] = self.families
|
|
#
|
|
# {data.update({k: v}) for k, v in self.data.items()
|
|
# if k not in data.keys()}
|
|
# self.data = data
|
|
#
|
|
# def process(self):
|
|
# self.name = self.data["subset"]
|
|
#
|
|
# instance = nuke.toNode(self.data["subset"])
|
|
# node = 'write'
|
|
#
|
|
# if not instance:
|
|
# write_data = {
|
|
# "class": node,
|
|
# "preset": self.preset,
|
|
# "avalon": self.data
|
|
# }
|
|
#
|
|
# if self.presets.get('fpath_template'):
|
|
# self.log.info("Adding template path from preset")
|
|
# write_data.update(
|
|
# {"fpath_template": self.presets["fpath_template"]}
|
|
# )
|
|
# else:
|
|
# self.log.info("Adding template path from plugin")
|
|
# write_data.update({
|
|
# "fpath_template": "{work}/prerenders/{subset}/{subset}.{frame}.{ext}"})
|
|
#
|
|
# # get group node
|
|
# group_node = create_write_node(self.data["subset"], write_data)
|
|
#
|
|
# # open group node
|
|
# group_node.begin()
|
|
# for n in nuke.allNodes():
|
|
# # get write node
|
|
# if n.Class() in "Write":
|
|
# write_node = n
|
|
# group_node.end()
|
|
#
|
|
# # linking knobs to group property panel
|
|
# linking_knobs = ["first", "last", "use_limit"]
|
|
# for k in linking_knobs:
|
|
# lnk = nuke.Link_Knob(k)
|
|
# lnk.makeLink(write_node.name(), k)
|
|
# lnk.setName(k.replace('_', ' ').capitalize())
|
|
# lnk.clearFlag(nuke.STARTLINE)
|
|
# group_node.addKnob(lnk)
|
|
#
|
|
# return
|