Source Code for Module nukescripts.misc

  1  # Copyright (c) 2009 The Foundry Visionmongers Ltd.  All Rights Reserved. 
  2   
  3  import nuke, nukescripts 
  4  import random 
  5  import os 
  6  import textwrap 
  7   
  8   
  9 -def copy_knobs(args): 
 10    g1 = nuke.thisGroup() 
 11    selNodes = g1.selectedNodes() 
 12   
 13    g2 = nuke.nodes.Group(name = "____tempcopyknobgroup__") 
 14    with g2: 
 15      nuke.nodePaste(nukescripts.cut_paste_file()) 
 16   
 17    excludedKnobs = ["name", "xpos", "ypos"] 
 18   
 19    nodes = g2.nodes() 
 20    for i in g2.nodes(): 
 21      for j in selNodes: 
 22        k1 = i.knobs() 
 23        k2 = j.knobs() 
 24        intersection = dict([(item, k1[item]) for item in k1.keys() if item not in excludedKnobs and k2.has_key(item)]) 
 25        for k in intersection.keys(): 
 26          x1 = i[k] 
 27          x2 = j[k] 
 28          x2.fromScript(x1.toScript()) 
 29   
 30    nuke.delete(g2) 
 31   
 32   
 33 -def connect_selected_to_viewer(inputIndex): 
 34    """Connects the selected node to the given viewer input index, ignoring errors if no node is selected.""" 
 35   
 36    selection = None 
 37    try: 
 38      selection = nuke.selectedNode() 
 39    except ValueError:  # no node selected 
 40      pass 
 41   
 42    if selection is not None and selection.Class() == 'Viewer': 
 43      selection = None 
 44     
 45    nuke.connectViewer(inputIndex, selection) 
 46   
 47   
 48 -def toggle_monitor_output(): 
 49    """Toggles monitor output (switches it on if it's off, or vice versa) for the currently active viewer.""" 
 50    viewer = nuke.activeViewer() 
 51    if viewer is not None: 
 52      enableKnob = viewer.node().knob('MonitorOutEnable') 
 53      enableKnob.setValue(not enableKnob.value()) 
 54   
 55   
 56 -def clear_selection_recursive(group = nuke.root()): 
 57    """Sets all nodes to unselected, including in child groups.""" 
 58    for n in group.selectedNodes(): 
 59      n.setSelected(False) 
 60    groups = [i for i in group.nodes() if i.Class() == 'Group'] 
 61    for i in groups: 
 62      clear_selection_recursive(i) 
 63   
 64   
 65 -def goofy_title(): 
 66    """Returns a random message for use as an untitled script name. 
 67    Can be assigned to nuke.untitled as a callable. 
 68    Put a goofy_title.txt somewhere in your NUKE_PATH to customise.""" 
 69     
 70    goofyFile = None 
 71    for dir in nuke.pluginPath(): 
 72        fileName = os.path.join(dir, "goofy_title.txt") 
 73        if os.path.exists(fileName): 
 74            goofyFile = fileName 
 75            break 
 76     
 77    if goofyFile is None: 
 78        return "Missing goofy_title.txt" 
 79     
 80    file = open(goofyFile) 
 81    lines = file.readlines() 
 82    file.close() 
 83     
 84    lines = [line.strip() for line in lines] 
 85    lines = [line for line in lines if len(line) > 0 and line[0] != '#'] 
 86     
 87    if len(lines) < 1: 
 88      return "Empty goofy_title.txt" 
 89     
 90    return random.choice(lines) 
 91   
 92   
 93 -def declone(node): 
 94    if node.clones() == 0: 
 95      return 
 96    args = node.writeKnobs(nuke.WRITE_ALL | nuke.WRITE_USER_KNOB_DEFS | nuke.WRITE_NON_DEFAULT_ONLY | nuke.TO_SCRIPT) 
 97    newnode = nuke.createNode(node.Class(), knobs = args) 
 98    nuke.inputs(newnode, nuke.inputs(node)) 
 99    num_inputs = nuke.inputs(node) 
100    for i in range(num_inputs): 
101       newnode.setInput(i, node.input(i)) 
102    node.setInput(0, newnode) 
103    nuke.delete(node) 
104   
105   
106 -def showname(): 
107    '''Shows the current script path and, if the selected node is a Read or Write node, the filename from it.''' 
108      
109    # get the nuke script path 
110    # we always need this 
111    nukescript = nuke.value("root.name") 
112     
113    # look if there is a selected node 
114    # if not, output the script only 
115    p = nuke.Panel("Current Info", 500) 
116    try: 
117      n = nuke.selectedNode() 
118      if n.Class() == "Read" or n.Class() == "Write": 
119        a = nuke.value(n.name()+".first", nuke.value("root.first_frame")) 
120        b = nuke.value(n.name()+".last", nuke.value("root.last_frame")) 
121        curfile = n.knob("file").value()+" "+str(a)+"-"+str(b) 
122        p.addSingleLineInput("Filename", curfile) 
123        p.addSingleLineInput("Script", nukescript) 
124        p.show() 
125      else: 
126        p.addSingleLineInput("Script", nukescript) 
127        p.show() 
128    except: 
129      p.addSingleLineInput("Script", nukescript) 
130      p.show() 
131   
132   
133 -def swapAB(n): 
134    """Swaps the first two inputs of a node.""" 
135     
136    if max(n.inputs(), n.minimumInputs()) > 1: 
137      a = n.input(0) 
138      n.setInput(0, n.input(1)) 
139      n.setInput(1, a) 
140   
141 -def print_callback_info(verbose=False, callbackTypes=None): 
142   
143    """Returns a list of all currently active callbacks, with the following optional arguments:""" 
144    """verbose=False : prints the documentation as well as the callback""" 
145    """callbackTypes=None : limit the callback info to a particular callback type (e.g. ['OnCreates'])""" 
146   
147    # list of all callback types 
148    all_Callback_Types = ['onUserCreates', 'onCreates', 'onScriptLoads', 'onScriptSaves', 'onScriptCloses', 'onDestroys', 'knobChangeds', 'updateUIs', 'autolabels', 'beforeRenders', 'beforeFrameRenders', 'afterRenders', 'afterFrameRenders', 'filenameFilters', 'validateFilenames', 'autoSaveFilters', 'autoSaveRestoreFilters', 'autoSaveDeleteFilters' ] 
149   
150    #if no callbackTypes defined or is an invalid type, then default search all callback types 
151    is_valid_type = (type(callbackTypes) is dict) or (type(callbackTypes) is list) 
152    if ( not callbackTypes or (not is_valid_type ) ): 
153      callbackTypes = all_Callback_Types 
154   
155    callback_defs = {} 
156   
157    for callbackType in callbackTypes: 
158      callback_defs[callbackType] = eval('nuke.%s' %(callbackType)) 
159   
160    # find the max target name length 
161    maxTargetNameLen = max( map( len,' '.join( [ (k+'').join( callback_defs[k].keys() ) for k in callback_defs.keys() ] ).split(' ') ) ) 
162   
163    indent = (maxTargetNameLen+8)*' ' 
164   
165    sortedCallbackTypes = sorted( callback_defs.keys() ) 
166    for callbackType in sortedCallbackTypes: 
167      for callbackTarget in callback_defs[callbackType].keys(): 
168        for func, a, b, c in callback_defs[callbackType][callbackTarget]: 
169          id = '%s%s%s : '%(callbackType[:-1],'_'*(maxTargetNameLen-len(callbackType)-len(callbackTarget)+1),callbackTarget) 
170          print '%s%s' %(id, func.__name__) 
171          if verbose: 
172            doc = func.__doc__ 
173            if not doc: 
174              doc='NO DOCUMENTATION' 
175            docNoReturns = str(doc).lstrip().replace('\n','') 
176            docNoConsecutiveSpaces = " ".join(docNoReturns.split()) 
177            docWrappedText = textwrap.wrap(docNoConsecutiveSpaces, 60) 
178            for line in docWrappedText: 
179              print indent + line.replace('\n', '\n' + indent) 
180