Source code for nukescripts.unrealreader

"""Functions used by the UnrealReader node"""

import nuke
import nukescripts
import os

[docs]def createCamera(unrealReaderNode): """Create a camera node based on the values set in the given unreal reader node""" x = unrealReaderNode.xpos() y = unrealReaderNode.ypos() w = unrealReaderNode.screenWidth() h = unrealReaderNode.screenHeight() m = int(x + w/2) camera = nuke.createNode('Camera', '', False) camera.setInput(0,None) camera.setXYpos(m - int(camera.screenWidth()/2), y + w) link = False linkKnob = unrealReaderNode.knob("linkOutput") if linkKnob: link = bool(linkKnob.getValue()) if link: # Set camera values as links to what's stored in the Unreal Reader node camera.knob("projection_mode").setExpression(unrealReaderNode.name() + ".projectionMode") camera.knob("focal").setExpression(unrealReaderNode.name() + ".focalLength") camera.knob("haperture").setExpression(unrealReaderNode.name() + ".aperture.x") camera.knob("vaperture").setExpression(unrealReaderNode.name() + ".aperture.y") camera.knob("translate").setExpression(unrealReaderNode.name() + ".camTranslate") camera.knob("rotate").setExpression(unrealReaderNode.name() + ".camRotate") else: camera.knob("projection_mode").fromScript(unrealReaderNode.knob("projectionMode").toScript(False)) camera.knob("focal").fromScript(unrealReaderNode.knob("focalLength").toScript(False)) camera.knob("translate").fromScript(unrealReaderNode.knob("camTranslate").toScript(False)) camera.knob("rotate").fromScript(unrealReaderNode.knob("camRotate").toScript(False)) numKeys = unrealReaderNode["aperture"].getNumKeys() camera.knob("haperture").setAnimated() camera.knob("vaperture").setAnimated() for i in range(numKeys): frame = unrealReaderNode.knob("aperture").getKeyTime(i) camera.knob("haperture").setValueAt( unrealReaderNode.knob("aperture").getValueAt(frame, 0), frame ) camera.knob("vaperture").setValueAt( unrealReaderNode.knob("aperture").getValueAt(frame, 1), frame )
[docs]def createReadNode(unrealReaderNode): """Create a read node for the rendered output""" createRead = False createReadKnob = unrealReaderNode.knob("createReadNode") if createReadKnob : createRead = bool(createReadKnob.getValue()) if createRead : # Create node and set file filename = unrealReaderNode['file'].getValue() read = nuke.nodes.Read(file="%s" %(filename)) # Set frame range frameStart = unrealReaderNode['startFrame'].getValue() frameEnd = unrealReaderNode['endFrame'].getValue() read["first"].setValue(int(frameStart)) read["last"].setValue(int(frameEnd)) # Set position in node graph x = unrealReaderNode.xpos() y = unrealReaderNode.ypos() w = unrealReaderNode.screenWidth() m = int(x + w/2) read.setXYpos(m - int(read.screenWidth()/2), y + w)