def __init__(self, allcdl):
super(_SelectCCCIDPanel, self).__init__()
self.available = {}
for cur in allcdl:
self.available[cur.getID()] = cur
self.addKnob(nuke.Enumeration_Knob("cccid", "cccid", self.available.keys()))
self.addKnob(nuke.Text_Knob("divider"))
self.addKnob(nuke.Color_Knob("slope"))
self.addKnob(nuke.Color_Knob("offset"))
self.addKnob(nuke.Color_Knob("power"))
self.addKnob(nuke.Double_Knob("saturation"))
def showModalDialog(self, parser):
"""Called to display the dialogue to the user."""
self.parser = parser
self.cdlcorrectionid = nuke.Enumeration_Knob(
"cdlcorrectionid", "ID",
sorted(self.parser.cdlcorrectionvalues.keys()))
self.addKnob(self.cdlcorrectionid)
self.dividerone = nuke.Text_Knob("dividerone", "")
self.addKnob(self.dividerone)
self.cdlcorrectiondesc = nuke.Multiline_Eval_String_Knob(
"cdlcorrectiondesc", "Description", "")
self.addKnob(self.cdlcorrectiondesc)
self.cdlcorrectiondesc.setFlag(nuke.DISABLED)
self.dividertwo = nuke.Text_Knob("dividertwo", "")
self.addKnob(self.dividertwo)
self.extrefsearchpath = nuke.Enumeration_Knob("extrefsearchpath", "Ref Search", \
["Same Name, Same Dir", "Any Name, Same Dir", "Specified"])
self.addKnob(self.extrefsearchpath)
self.extrefpath = nuke.File_Knob("extrefpath", "Ref Path")
self.addKnob(self.extrefpath)
self.dividerthree = nuke.Text_Knob("dividerthree", "")
self.addKnob(self.dividerthree)
self.cdloffset = nuke.Color_Knob("cdloffset", "Offset")
self.addKnob(self.cdloffset)
self.cdloffset.setFlag(nuke.DISABLED)
self.cdlslope = nuke.Color_Knob("cdlslope", "Slope")
self.addKnob(self.cdlslope)
self.cdlslope.setFlag(nuke.DISABLED)
self.cdlpower = nuke.Color_Knob("cdlpower", "Power")
self.addKnob(self.cdlpower)
self.cdlpower.setFlag(nuke.DISABLED)
self.cdlsaturation = nuke.Double_Knob("cdlsaturation", "Saturation")
self.addKnob(self.cdlsaturation)
self.cdlsaturation.setFlag(nuke.DISABLED)
self.dividerfour = nuke.Text_Knob("dividerfour", "")
self.addKnob(self.dividerfour)
self.updateDesc()
return nukescripts.PythonPanel.showModalDialog(
self), self.cdlcorrectionid.value()
def addPoint():
#print pointKnobs()
#print numberOfPointKnobs()
num = numberOfPointKnobs()
node = nuke.thisNode()
knob = nuke.Color_Knob('color{}'.format(num))
cKnobs = colorKnobs()
#avgColorR = sum([k.getValue(0) for k in cKnobs()]) / len(cKnobs())
#avgColorG = sum([k.getValue(1) for k in cKnobs()]) / len(cKnobs())
#avgColorB = sum([k.getValue(2) for k in cKnobs()]) / len(cKnobs())
avgColor = [sum([k.getValue(i) for k in cKnobs]) / len(cKnobs) for i in range(3)] if cKnobs else [0, 0, 0]
knob.setValue(avgColor)
node.addKnob(knob)
knob = nuke.XY_Knob('p{}'.format(num))
knob.clearFlag(nuke.STARTLINE) #creates the knob at the same line
pKnobs = pointKnobs()
avgPos = [sum([k.getValue(i) for k in pKnobs]) / len(pKnobs) for i in range(2)] if pKnobs else [0, 0] #calculates average value of all colors. if knob does not exists sets 0 value
knob.setValue(avgPos) #sets values to the knob
knob.clearFlag(nuke.STARTLINE) #creates the knob at the same line
node.addKnob(knob) #adds the knob
knob = nuke.PyScript_Knob('deletePoint{}'.format(num)) #creates knob 'deletePoints' with the number of the point
knob.clearFlag(nuke.STARTLINE) #creates the knob at the same line
knob.setLabel('delete')
knob.setCommand('nPointGrad.deletePoints()')
node.addKnob(knob) #adds the knob
makeExpression()
def __init__(self):
super(Core_PickTranslate, self).__init__('PickTranslate')
self.k_pick = nuke.Color_Knob('pick', "pick translate")
self.k_node = nuke.Enumeration_Knob('geo', "pick geo node", [])
self.addKnob(self.k_pick)
self.addKnob(self.k_node)
def setup(cls, groupmo):
"""Builds all the interior nodes and knobs of SpillSuppress"""
# =====================================================================
# Nodes
# =====================================================================
# Input ===============================================================
input_node = nuke.nodes.Input()
dot = nuke.nodes.Dot(
inputs=[input_node],
xpos=input_node.xpos() + 40 - 6,
ypos=input_node.ypos() + 100
)
# Shuffles ============================================================
def set_shuffle(shuffle, color):
"""Sets all the color channels of a shuffle to be the same color"""
shuffle['red'].setValue(color)
shuffle['green'].setValue(color)
shuffle['blue'].setValue(color)
shuffle['alpha'].setValue('white')
# Shuffle to extract the green channel
green_shuffle = nuke.nodes.Shuffle(
inputs=[dot],
name='Green',
xpos=input_node.xpos(),
ypos=dot.ypos() + 100
)
set_shuffle(green_shuffle, 'green')
# Shuffle to extract the red channel
red_shuffle = nuke.nodes.Shuffle(
inputs=[dot],
name='Red',
xpos=green_shuffle.xpos() - 160,
ypos=green_shuffle.ypos()
)
set_shuffle(red_shuffle, 'red')
# Shuffle to extract the blue channel
blue_shuffle = nuke.nodes.Shuffle(
inputs=[dot],
name='Blue',
xpos=green_shuffle.xpos() + 160,
ypos=green_shuffle.ypos()
)
set_shuffle(blue_shuffle, 'blue')
dot2 = nuke.nodes.Dot(
inputs=[dot],
xpos=blue_shuffle.xpos() + 160,
ypos=blue_shuffle.ypos() + 4,
)
# Color Switches ======================================================
# This will autoswitch from green primary to blue primary, depending
# on which color has a higher value.
main_switch = nuke.nodes.Switch(
inputs=[blue_shuffle, green_shuffle],
name='BGSwitch',
xpos=blue_shuffle.xpos(),
ypos=blue_shuffle.ypos() + 100
)
# We set this to an expression which looks at the AutoBalance node.
# Expression Reads:
# If AutoBalance's Source Color Green channel minus the Blue channel is
# greater than 0, value should be 1. Else, value should be 0.
main_switch['which'].setExpression(
'AutoBalance.source.g - AutoBalance.source.b > 0 ? 1 : 0'
)
# Inverse Switch (will be the opposite of Main Switch)
inverse_switch = nuke.nodes.Switch(
inputs=[green_shuffle, blue_shuffle],
name='BGSwitchInverse',
xpos=green_shuffle.xpos(),
ypos=green_shuffle.ypos() + 100
)
inverse_switch['which'].setExpression(
'BGSwitch.which'
)
# Channel Mixing ======================================================
# These nodes will determine how the alternate color channels
# (the color channels that are NOT the spill channel) are mixed
# together to help determine how much of the spill channel is spill.
cross = nuke.nodes.Merge2(
inputs=[red_shuffle, inverse_switch],
name='Cross',
xpos=red_shuffle.xpos(),
ypos=red_shuffle.ypos() + 200)
cross.addKnob(nuke.Array_Knob('chanMix', 'channel mix'))
cross['chanMix'].setValue(0.5)
cross['chanMix'].setTooltip(
"Adjust the weighted average of the alternate color channels. 0 "
"is always the full red channel, 1 is either the blue or green "
"channel. If the spill color is blue, 1 is the green channel and "
"vice versa."
)
cross['chanMix'].setFlag(flags.SLIDER)
cross['chanMix'].setFlag(flags.FORCE_RANGE)
cross['mix'].setExpression('chanMix')
max = nuke.nodes.Merge2(
inputs=[red_shuffle, inverse_switch],
name='Max',
xpos=cross.xpos() - 160,
ypos=cross.ypos(),
)
max['operation'].setValue('max')
# Mix Switch will be a user controlled switch. When checked, it will
# use the Max method instead of Crossing between the two.
mix_switch = nuke.nodes.Switch(
inputs=[cross, max],
name='MaxOrCross',
xpos=cross.xpos(),
ypos=cross.ypos() + 100
)
mix_switch['which'].setExpression('parent.useMax')
# Spill Controls ======================================================
# These control the ebb and flow of the spill mask.
thresh_gamma = nuke.nodes.Gamma(
inputs=[mix_switch],
name='ThresholdGamma',
xpos=mix_switch.xpos(),
ypos=mix_switch.ypos() + 26
)
thresh_gamma['value'].setExpression('parent.gamma')
thresh_gain = nuke.nodes.Multiply(
inputs=[thresh_gamma],
name='ThresholdGain',
xpos=thresh_gamma.xpos(),
ypos=thresh_gamma.ypos() + 38)
thresh_gain['value'].setExpression('parent.gain')
# Spill Removal =======================================================
# After the adjustments to the spill matte with gain and gamma, we
# can now subtract the spill.
subtract_spill = nuke.nodes.Merge2(
inputs=[thresh_gain, main_switch],
name='SubtractSpill',
xpos=main_switch.xpos(),
ypos=thresh_gain.ypos() + 6,
)
subtract_spill['operation'].setValue('minus')
remove_negatives = nuke.nodes.Expression(
inputs=[subtract_spill],
name='RemoveNegatives',
xpos=subtract_spill.xpos(),
ypos=subtract_spill.ypos() + 26
)
remove_negatives['expr0'].setValue('r>0?r:0')
remove_negatives['expr1'].setValue('g>0?g:0')
remove_negatives['expr2'].setValue('b>0?b:0')
# Balancing ===========================================================
# These nodes, by way of adding or removing spill channel, determine
# the resultant color balance of the shot.
auto_balance = nuke.nodes.Expression(
inputs=[remove_negatives],
name='AutoBalance',
xpos=remove_negatives.xpos() - 80,
ypos=remove_negatives.ypos() + 100
)
# We'll be using these variables in the channel calculations
#
# Essentially, we're determining the multiplier value needed to get
# from the despilled source color to the destination color.
# Most of these calculations are determining what the despilled
# color will look like.
auto_balance['temp_name0'].setValue('bspill')
auto_balance['temp_expr0'].setValue(
'source.b - pow(source.r * (1 - Cross.mix) '
'+ source.g * Cross.mix, 1 / (ThresholdGamma.value '
'+ 0.000001) ) * ThresholdGain.value'
)
auto_balance['temp_name1'].setValue('gspill')
auto_balance['temp_expr1'].setValue(
'source.g - pow(source.r * (1 - Cross.mix) '
'+ source.b * Cross.mix, 1 / (ThresholdGamma.value '
'+ 0.000001) ) * ThresholdGain.value'
)
# This expression simply compares the two colors to determine which
# spill calculation to use, then we'll that below.
auto_balance['temp_name2'].setValue('spill')
auto_balance['temp_expr2'].setValue(
'source.g - source.b > 0 ? gspill : bspill'
)
auto_balance['expr0'].setValue(
'r * (dest.r - source.r) / spill'
)
auto_balance['expr1'].setValue(
'g * (dest.g - source.g) / spill'
)
auto_balance['expr2'].setValue(
'b * (dest.b - source.b) / spill'
)
# Now we need to add the knobs it's referencing to the AutoBalance
# node.
auto_balance.addKnob(nuke.Tab_Knob('spillcontrols', 'Spill Controls'))
auto_balance.addKnob(nuke.Color_Knob('source', 'Source Color'))
auto_balance.addKnob(nuke.Color_Knob('dest', 'Destination Color'))
auto_balance['source'].setValue([0.1, 0.2, 0.3])
auto_balance['dest'].setValue([0.3, 0.3, 0.3])
# Auto Balancing Tooltips
auto_balance['source'].setTooltip(
"Color of spill to remove. This color will be used to determine if "
"the screen color is blue or green, based on which color channel"
"has a higher value."
)
auto_balance['dest'].setTooltip(
"Color to change the spill color to. This should be a sampled "
"average of the background behind your character. This color is "
"normally not the neutral grey most spill suppression corrects to. "
"Spill suppression should not only suppress the spill color, but "
"transform the spill and reflected spill to reflect the "
"environment. This will lead to better edges and color. "
"Extremely bright or dark values will break. Adjust values until "
"result is no longer 'grainy.'"
)
# Manual Balancing
manual_balance = nuke.nodes.Multiply(
inputs=[remove_negatives],
name='ManualBalance',
channels='rgb',
xpos=remove_negatives.xpos() + 80,
ypos=auto_balance.ypos(),
)
# We need to set the values to trigger the channels to split.
manual_balance['value'].setValue([0.1, 0.1, 0.1, 1])
# But now we can set their expressions
manual_balance['value'].setExpression('AutoBalance.dest.r', 0)
manual_balance['value'].setExpression('AutoBalance.dest.g', 1)
manual_balance['value'].setExpression('AutoBalance.dest.b', 2)
auto_switch = nuke.nodes.Switch(
inputs=[auto_balance, manual_balance],
name='AutoManualSwitch',
xpos=remove_negatives.xpos(),
ypos=remove_negatives.ypos() + 200
)
auto_switch['which'].setExpression('parent.useManual')
blur = nuke.nodes.Blur(
inputs=[auto_switch],
name='BlurSpill',
xpos=auto_switch.xpos(),
ypos=auto_switch.ypos() + 26
)
dot3 = nuke.nodes.Dot(
inputs=[dot2],
xpos=dot2.xpos(),
ypos=blur.ypos() + 42,
)
remove_spill = nuke.nodes.Merge2(
inputs=[dot3, blur],
name='RemoveSpill',
xpos=blur.xpos(),
ypos=blur.ypos() + 38,
)
remove_spill['operation'].setValue('plus')
# Output ==============================================================
copy_channels = nuke.nodes.Copy(
inputs=[dot3, remove_spill],
name='ProtectAll',
xpos=remove_spill.xpos(),
ypos=remove_spill.ypos() + 100
)
copy_channels['from0'].setValue('rgba.red')
copy_channels['from1'].setValue('rgba.green')
copy_channels['from2'].setValue('rgba.blue')
copy_channels['to0'].setValue('rgba.red')
copy_channels['to1'].setValue('rgba.green')
copy_channels['to2'].setValue('rgba.blue')
nuke.nodes.Output(
inputs=[copy_channels],
xpos=copy_channels.xpos(),
ypos=copy_channels.ypos() + 100
)
# =====================================================================
# Knobs
# =====================================================================
set_link('source', groupmo, auto_balance)
groupmo.addKnob(nuke.Text_Knob('source_divider', ''))
set_link('chanMix', groupmo, cross)
use_max = nuke.Boolean_Knob('useMax', 'use maximum for mix')
use_max.clearFlag(nuke.STARTLINE)
use_max.setTooltip(
"Instead of averaging the alternate color channels, use a max "
"operation to combine them."
)
groupmo.addKnob(use_max)
gain = nuke.Array_Knob('gain', 'gain')
gamma = nuke.Array_Knob('gamma', 'gamma')
gamma.clearFlag(nuke.STARTLINE)
gain.setValue(0.95)
gamma.setValue(0.95)
gain.setFlag(flags.SLIDER)
gain.setFlag(flags.LOG_SLIDER)
gamma.setFlag(flags.SLIDER)
gamma.setFlag(flags.LOG_SLIDER)
gain.setTooltip(
"Multiples the spill threshold to increase or decrease what gets "
"considered spill. Has a larger effect on brighter areas."
)
gamma.setTooltip(
"Does a gamma operation on the spill threshold, which has a "
"larger effect on midtones."
)
groupmo.addKnob(gain)
groupmo.addKnob(gamma)
set_link('size', groupmo, blur, label='blur spill')
groupmo.addKnob(nuke.Text_Knob('destination_divider', ''))
set_link('dest', groupmo, auto_balance)
use_manual = nuke.Boolean_Knob('useManual', 'use for manual balancing')
use_manual.clearFlag(nuke.STARTLINE)
use_manual.setTooltip(
"If selected, the Destination Color will instead become a manual "
"gain control for the final result."
)
groupmo.addKnob(use_manual)
# =====================================================================
# Callbacks
# =====================================================================
groupmo['knobChanged'].setValue(
'keying.SpillSuppress.disable_channel_mix_callback()'
)
def __init__( self, node ):
nukescripts.PythonPanel.__init__( self, 'Multiple Nodes' )
#Creating Knobs
self.name = nuke.String_Knob( 'name', 'Knob name' )
self.enum = nuke.Enumeration_Knob('enum', 'Knob Type', ['integer', 'slider', 'double',
'XY', 'XYZ', 'rgb', 'rgba', 'bbox', 'string', 'boolean'])
self.enum.setFlag(nuke.STARTLINE)
self.boolean = nuke.Boolean_Knob('boolean', 'boolean')
self.boolean.setFlag(nuke.STARTLINE)
self.boolean.setFlag(nuke.INVISIBLE)
self.integer = nuke.Int_Knob( 'integer', 'integer' '')
self.integer.setFlag(nuke.STARTLINE)
self.integer.setFlag(nuke.INVISIBLE)
self.slider = nuke.Double_Knob( 'slider', 'slider' )
self.slider.setFlag(nuke.STARTLINE)
self.double = nuke.WH_Knob( 'double', 'double' )
self.double.setFlag(nuke.STARTLINE)
self.double.setFlag(nuke.INVISIBLE)
self.XY = nuke.UV_Knob( 'XY', 'XY' )
self.XY.setFlag(nuke.STARTLINE)
self.XY.setFlag(nuke.INVISIBLE)
self.XYZ = nuke.Scale_Knob( 'XYZ', 'XYZ' )
self.XYZ.setFlag(nuke.STARTLINE)
self.XYZ.setFlag(nuke.INVISIBLE)
self.rgb = nuke.Color_Knob( 'rgb', 'rgb' )
self.rgb.setFlag(nuke.STARTLINE)
self.rgb.setFlag(nuke.INVISIBLE)
self.rgba = nuke.AColor_Knob( 'rgba', 'rgba' )
self.rgba.setFlag(nuke.STARTLINE)
self.rgba.setFlag(nuke.INVISIBLE)
self.bbox = nuke.BBox_Knob( 'bbox', 'bbox' )
self.bbox.setFlag(nuke.STARTLINE)
self.bbox.setFlag(nuke.INVISIBLE)
self.string = nuke.String_Knob( 'string', 'string' )
self.string.setFlag(nuke.STARTLINE)
self.string.setFlag(nuke.INVISIBLE)
#set initial values
self.enum.setValue('slider')
#Adding knobs
for k in ( self.name, self.enum, self.integer, self.slider, self.double, self.XY,
self.XYZ, self.rgb, self.rgba, self.bbox, self.string, self.boolean ):
self.addKnob( k )
date_button = nuke.PyScript_Knob(DATE_BUTTON_NAME, 'SET DATE')
set_date_script = 'import datetime' \
'\nd = datetime.datetime.now().strftime("%Y.%m.%d - %H:%M")' \
'\nnuke.thisNode()["{}"].setValue(d)'.format(DATE_INPUT_NAME)
date_button.setCommand(set_date_script)
g.addKnob(date_button)
# TEXT COLORS TAB
colors_tab = nuke.Tab_Knob(KNOB_BASE_NAME + 'text_colors_tab', 'Text colors')
g.addKnob(colors_tab)
stamp_color_label = nuke.Text_Knob(KNOB_BASE_NAME + 'Stamp colors', '',
'<font color=white size=5>Text color')
g.addKnob(stamp_color_label)
stamp_text_color = nuke.Color_Knob(TEXT_COLOR_NAME, 'Stamp text color:')
stamp_text_color.setValue([1, 1, 1])
g.addKnob(stamp_text_color)
colors_separator = nuke.Text_Knob(KNOB_BASE_NAME + 'second_separator', '', '')
g.addKnob(colors_separator)
# ------- #
use_background_color_checkbox = nuke.Boolean_Knob(BACKGROUND_COLOR_CHECKBOX,
'Use background color?',
True)
use_background_color_checkbox.setFlag(nuke.STARTLINE)
g.addKnob(use_background_color_checkbox)
stamp_background_color = nuke.Color_Knob(BACKGROUND_COLOR_NAME,