Untitled
unknown
plain_text
2 years ago
12 kB
1
Indexable
import nuke
nuke.menu("Nodes").addCommand('User/Utility/JS_Backdrop/JS_Backdrop', "create_JS_Backdrop()" , 'alt+b')
nuke.menu("Nodes").addCommand('User/Utility/JS_Backdrop/set random nuke colors', "JS_Backdrop_randomColors()")
menubar = nuke.menu('Nuke')
menubar.addCommand("@;scale backdrop right", "JS_Backdrop_Scale('Right')", 'alt+Right')
menubar.addCommand("@;scale backdrop down", "JS_Backdrop_Scale('Down')", 'alt+Down')
menubar.addCommand("@;scale backdrop left", "JS_Backdrop_Scale('Left')", 'alt+Left')
menubar.addCommand("@;scale backdrop up", "JS_Backdrop_Scale('Up')", 'alt+Up')
def JS_Backdrop_Scale(dir):
undo = nuke.Undo()
undo.disable()
grow = round(50/(nuke.zoom()*4))
for node in nuke.selectedNodes():
if node.Class() == 'BackdropNode':
node.setSelected(False)
width = node['bdwidth']
height = node['bdheight']
x = node['xpos']
y = node['ypos']
try:
bdNodes = JS_Backdrop_getNodes()
if dir == 'Left' or dir == 'Right':
Xmin = min([n.xpos() for n in bdNodes])
Xmax = max([n.xpos() + n.screenWidth() for n in bdNodes])
Xanchor = float( (Xmin + Xmax)/2 - node.xpos() ) / float(node.screenWidth())
else:
Ymin = min([n.ypos() for n in bdNodes])
Ymax = max([n.ypos() + n.screenHeight() for n in bdNodes])
Yanchor = float( (Ymin + Ymax)/2 - node.ypos() ) / float(node.screenHeight())
except:
Xanchor = .5
Yanchor = .5
if dir == 'Left':
width.setValue( width.value() - grow )
x.setValue( x.value() + grow*Xanchor )
elif dir == 'Down':
height.setValue( height.value() - grow )
y.setValue( y.value() + grow*Yanchor )
elif dir == 'Right':
width.setValue( width.value() + grow )
x.setValue( x.value() - grow*Xanchor )
elif dir == 'Up':
height.setValue( height.value() + grow )
y.setValue( y.value() - grow*Yanchor )
node.setSelected(True)
undo.enable()
def JS_Backdrop_getNodes():
origSel = nuke.selectedNodes()
node.selectNodes()
bdNodes = nuke.selectedNodes()
[n.setSelected( False ) for n in bdNodes]
[n.setSelected( True ) for n in origSel]
return bdNodes
def create_JS_Backdrop():
#create backdrop
if len(nuke.selectedNodes()) == 0:
node = nuke.createNode('BackdropNode', inpanel = False)
else:
node = nukescripts.autoBackdrop()
###build UI###
node.addKnob(nuke.Tab_Knob('JS_Backdrop'))
#color knobs
node.addKnob( nuke.Enumeration_Knob('color','color', ['custom', 'grey', 'red', 'green', 'blue', 'nuke lime', 'nuke purple', 'nuke pink', 'nuke teal', 'nuke yellow']) )
cc = nuke.nuke.Color_Knob('customColor','custom color')
cc.clearFlag(nuke.STARTLINE)
if node['z_order'].value() == -3:
zValue = 0.15
else:
zValue = min(max(.5 + node['z_order'].value()/10, 0.05), 1)
cc.setValue(zValue)
node.addKnob( cc )
node.addKnob( nuke.nuke.Text_Knob('divider','') )
#icon knob
icons = ['None', 'Blur', 'Camera', 'Card', 'CheckerBoard', 'ColorAdd', 'ColorGamma', 'ColorMult', 'Cube', 'Defocus', 'denoise', 'Exposure', 'Flare', 'FloodFill', 'Glint', 'Glow', 'Grade','Grain', 'Keyer', 'LensDistort', 'Merge', 'Modify', 'Noise', 'Paint', 'Read', 'Render', 'Roto', 'Tracker', 'Write', 'ZBlur']
node.addKnob( nuke.Enumeration_Knob('iconKnob','icon', icons) )
nodeName = node.name()
labelLink = nuke.Link_Knob( 'labelLink', 'label' )
labelLink.makeLink(nodeName, 'label')
node.addKnob( labelLink )
note_fontLink = nuke.Link_Knob( 'note_fontLink', 'font' )
note_fontLink.makeLink(nodeName, 'note_font')
node.addKnob( note_fontLink )
node['note_font_size'].setFlag(0x0000000010000000) # read only flag
note_font_sizeKnob = nuke.Double_Knob( 'note_font_sizeKnob', '' )
note_font_sizeKnob.clearFlag(0x00000002) #remove slider
note_font_sizeKnob.setFlag(0x00000100) #no_animation
note_font_sizeKnob.setRange(0,10)
note_font_sizeKnob.clearFlag(nuke.STARTLINE)
node.addKnob( note_font_sizeKnob )
note_font_colorLink = nuke.Link_Knob( 'note_font_colorLink', 'color' )
note_font_colorLink.makeLink(nodeName, 'note_font_color')
note_font_colorLink.clearFlag(nuke.STARTLINE)
node.addKnob( note_font_colorLink )
#font knobs
fs = nuke.Double_Knob('fontSize','dynamic font size')
fs.setValue(.5)
node.addKnob(fs)
dynamic = nuke.nuke.Boolean_Knob('dynamic', 'enable')
dynamic.clearFlag(nuke.STARTLINE)
dynamic.setValue(True)
node.addKnob( dynamic )
node.addKnob( nuke.nuke.Text_Knob('zorder_divider','') )
#z order buttons
node.addKnob( nuke.PyScript_Knob('up', 'move up', "knob=nuke.thisNode()['z_order'];knob.setValue(knob.value()+1)") )
down = nuke.PyScript_Knob('down', 'move down', "knob=nuke.thisNode()['z_order'];knob.setValue(knob.value()-1)")
down.clearFlag(nuke.STARTLINE)
node.addKnob( down )
front = nuke.PyScript_Knob('front', 'move to front', "node = nuke.thisNode()\nz = node['z_order'].value()\n\nbackdropNodes = [i for i in nuke.allNodes() if i.Class() == 'BackdropNode' and i != node]\n\nax = node.xpos()\nay = node.ypos()\naw = ax + node['bdwidth'].value()\nah = ay + node['bdheight'].value()\n\nfor bdNode in backdropNodes:\n bx = bdNode.xpos()\n by = bdNode.ypos()\n bw = bx + bdNode['bdwidth'].value()\n bh = by + bdNode['bdheight'].value()\n\n intersect = ax > bw and aw > bx and ay > bh and ah > by \n\n if intersect and bdNode['z_order'].value() > z:\n z = bdNode['z_order'].value()\n\nnode['z_order'].setValue(z + 1)")
front.clearFlag(nuke.STARTLINE)
node.addKnob( front )
back = nuke.PyScript_Knob('back', 'move to back', "node = nuke.thisNode()\nz = node['z_order'].value()\n\nbackdropNodes = [i for i in nuke.allNodes() if i.Class() == 'BackdropNode' and i != node]\n\nax = node.xpos()\nay = node.ypos()\naw = ax + node['bdwidth'].value()\nah = ay + node['bdheight'].value()\n\nfor bdNode in backdropNodes:\n bx = bdNode.xpos()\n by = bdNode.ypos()\n bw = bx + bdNode['bdwidth'].value()\n bh = by + bdNode['bdheight'].value()\n\n intersect = ax > bw and aw > bx and ay > bh and ah > by \n\n if intersect and bdNode['z_order'].value() > z:\n z = bdNode['z_order'].value()\n\nnode['z_order'].setValue(z - 1)")
back.clearFlag(nuke.STARTLINE)
node.addKnob( back )
#get KnobChanged python script
try:
from Scripts import JS_Backdrop
except:
import JS_Backdrop
import inspect
lines = inspect.getsource(JS_Backdrop.JS_Backdrop_knobChanged)
lineList = [line[4:].replace(' ', '\t') + '\n' for line in lines.split('\n') if line[:4] == ' ']
KCcode = (''.join(lineList))
#get OnCreate python script
lines = inspect.getsource(JS_Backdrop.JS_Backdrop_onCreate)
lineList = [line[4:].replace(' ', '\t') + '\n' for line in lines.split('\n') if line[:4] == ' ']
OCcode = (''.join(lineList))
#add code to python knobs
node['knobChanged'].setValue(KCcode)
node['onCreate'].setValue(OCcode)
node['autolabel'].setValue("'@' + nuke.thisNode()['iconKnob'].value() + ' ' + nuke.thisNode()['label'].value() if nuke.thisNode()['iconKnob'].value() != 'None' else nuke.thisNode()['label'].value()")
JS_Backdrop_knobChanged(node)
def JS_Backdrop_randomColors():
import random
selnodes = nuke.selectedNodes()
if len(selnodes) == 0:
nodes = nuke.allNodes()
else:
nodes = selnodes
foo=[1, 5, 6, 7, 8, 9]
for node in nodes:
if node.Class() == 'BackdropNode' and node.knob('JS_Backdrop') is not None:
node['color'].setValue(random.choice(foo))
def JS_Backdrop_knobChanged(node, initial = True):
from math import sqrt
undo = nuke.Undo()
undo.disable()
try:
initial
thisKnob = ''
except:
initial = False
if not initial:
node = nuke.thisNode()
thisKnob = nuke.thisKnob()
try:
if thisKnob in [node['fontSize'], node['dynamic'], node['bdwidth'], node['bdheight'], node['note_font_sizeKnob']] or initial:
#dynamic font size
if node['dynamic'].value() == True:
x = node['bdwidth'].value()
y = node['bdheight'].value()
c = sqrt(pow(x,2) + pow(y,2))
val = 6+(c/10)*node['fontSize'].value()
node['note_font_size'].setValue( int(val) )
node['note_font_sizeKnob'].setValue( int(val) )
else:
node['note_font_size'].setValue( node['note_font_sizeKnob'].value() )
if thisKnob == node['dynamic'] or initial:
if node['dynamic'].value() == True:
node['note_font_sizeKnob'].setFlag(0x0000000010000000) #read only
else:
node['note_font_sizeKnob'].clearFlag(0x0000000010000000) #read only
if thisKnob in [node['color'], node['customColor']] or initial:
#tile color
colorDropdown = node['color'].value()
if colorDropdown == 'custom':
customColor = node['customColor'].value()
if type(customColor) == type([]):
customColor = [max(min(i*255, 255),0) for i in customColor]
cc = int('%02x%02x%02x%02x' % (int(customColor[0]),int(customColor[1]),int(customColor[2]),1),16)
else:
customColor = max(min(customColor*255, 255),0)
cc = int('%02x%02x%02x%02x' % (int(customColor),int(customColor),int(customColor),1),16)
if cc == 1:
cc = 255
node['tile_color'].setValue(cc)
elif colorDropdown == 'grey':
node['tile_color'].setValue(2290649088)
elif colorDropdown == 'red':
node['tile_color'].setValue(2131956481)
elif colorDropdown == 'green':
node['tile_color'].setValue(4987137)
elif colorDropdown == 'blue':
node['tile_color'].setValue(2522881)
elif colorDropdown == 'nuke lime':
node['tile_color'].setValue(1908830464)
elif colorDropdown == 'nuke purple':
node['tile_color'].setValue(1903281664)
elif colorDropdown == 'nuke pink':
node['tile_color'].setValue(2386071040)
elif colorDropdown == 'nuke teal':
node['tile_color'].setValue(948866560)
elif colorDropdown == 'nuke yellow':
node['tile_color'].setValue(2391685120)
elif thisKnob == node['name']:
#reset link knobs in case of missing knob errors
lk = [knob for knob in ['labelLink', 'note_fontLink', 'note_font_colorLink'] if node[knob].Class() == 'Link_Knob']
if len(lk) > 0:
for knob in lk:
node[knob].makeLink(node.name(), knob[:-4])
#hacky way to force the UI to update
wk = node['bdwidth']
wkv = wk.value()
wk.setValue(wkv+1)
wk.setValue(wkv)
except:
pass
undo.enable()
def JS_Backdrop_onCreate():
try:
node = nuke.thisNode()
if node.knob('JS_Backdrop') is not None:
node['note_font_size'].setFlag(0x0000000010000000) #read only
node['note_font_sizeKnob'].clearFlag(0x00000002) #remove slider
node['note_font_sizeKnob'].setFlag(0x00000100) #no_animation
if node['dynamic'].value() == True:
node['note_font_sizeKnob'].setFlag(0x0000000010000000) #read only
except:
pass