Nuke custom menu.py

#------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
#satyajeet's custom nuke settings
# version 1.0.0
# Date- May 24 2020
#------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

#Global imports

import nuke
import nukescripts
import os

#set .nuke directory

dir = 'C:\Users\tmi\.nuke'
#nuke.pluginAddPath(dir)

#--------------------------------------------------------------------------------------
import knob_scripter
import stamps
#--------------------------------------------------------------------------------------
#SHORTCUTS
#shortcut for tracker, set merge bbox to B by default

nuke.menu('Nodes').addCommand('Transform/Tracker', 'nuke.createNode("Tracker4").knob("shutteroffset").setValue("centered")', "ctrl+alt+t", icon= 'Tracker.png', shortcutContext = 2)
nuke.menu("Nodes").addCommand("Time/FrameHold", 'nuke.createNode("FrameHold").knob("first_frame").setValue(nuke.frame())', "ctrl+alt+h", icon= "FrameHold.png", shortcutContext = 2)
nuke.menu("Nodes").addCommand("Channel/Shuffle", 'nuke.createNode("Shuffle")', "ctrl+alt+s", icon="Shuffle.png", shortcutContext = 2)
nuke.knobDefault("Merge2.bbox", "B" )
nuke.knobDefault("Merge2.label", "pipe:[value bbox]")
nuke.knobDefault("Copy.label", "pipe:[value bbox]")
nuke.knobDefault("Transform.shutteroffset", "centered")
nuke.knobDefault("ScanlineRender.shutteroffset", "centered")




#switch merge operations
def operationSwitcher():
    ns = nuke.selectedNode()
    if ns.Class() == "Merge2":

        if ns.knob("operation").value() == "mask":
            ns.knob("operation").setValue("stencil")
        elif ns.knob("operation").value() == 'stencil':
            ns.knob("operation").setValue("over")
        elif ns.knob("operation").value() == "over":
            ns.knob("operation").setValue("plus")
        elif ns.knob("operation").value() == "plus":
            ns.knob("operation").setValue("difference")
        elif ns.knob("operation").value() == "difference":
            ns.knob("operation").setValue("mask")

nuke.menu("Nuke").addCommand('Switch operation', operationSwitcher, "ctrl+shift+m")

# switch bbox pipe in merge& copy node
def pipeSwitcher():
    ns = nuke.selectedNode()
    if ns.Class() == "Merge2":
        if ns.knob('bbox').value() == "union":
            ns.knob("bbox").setValue("A")
        elif ns.knob("bbox").value() == "A":
            ns.knob("bbox").setValue("B")
        elif ns.knob("bbox").value() == "B":
            ns.knob("bbox").setValue("union")
        elif ns.Class() == "Copy":
            if ns.knob('bbox').value() == "union":
                ns.knob("bbox").setValue("A")
            elif ns.knob("bbox").value() == "A":
                ns.knob("bbox").setValue("B")
            elif ns.knob("bbox").value() == "B":
                ns.knob("bbox").setValue("union")

nuke.menu("Nuke").addCommand('Pipe switcher', pipeSwitcher, "ctrl+shift+b")

#copy premult shortcut ctrl+alt+k
def  kPre_short():

    nc=nuke.nodes.Copy(from0='rgba.alpha', to0='rgba.alpha', label='pipe: [value bbox]')
    nc.setInput(0, nuke.selectedNode())
    nuke.nodes.Premult().setInput(0, nc).hideControlPanel()

nuke.menu('Nodes').addMenu('Channel').addCommand('Create Copy premult', kPre_short, 'ctrl+alt+k', icon='Copy.png')

#roto blur shortcut o
def rotoBlur():
    nr= nuke.createNode("Roto")
    nuke.nodes.Blur(size=2, channel='rgba', label="[value size]").setInput(0, nr).hideControlPanel()

nuke.menu("Nodes").addCommand("Draw/Roto", rotoBlur, "o", icon= "Roto.png")