Ryoji CG Memo - temp(houdini)

<?xml version="1.0" encoding="UTF-8" ?>

<menu>

<![CDATA[

parm = kwargs["parms"][0]

framerange = range(int(hou.expandString('$RFSTART')), int(hou.expandString('$RFEND')) + 1 )

keyVals = {}

for kf in framerange:

keyVals[kf] = parm.evalAtFrame(kf)

parm.deleteAllKeyframes()

for kf in framerange:

setKey = hou.Keyframe()

setKey.setFrame(kf)

setKey.setValue( keyVals[kf] )

parm.setKeyframe(setKey)

]]>

</scriptCode>

</scriptItem>

</menu>

</menuDocument>

import sys

import os

import cStringIO

import tempfile

import traceback

def getComponentStart(node):

return None

def getSpecializedProperties(node):

return ()

def getValueInitializations(node):

return ()

def getNeedsBraces(node):

return False

def setUpCallbacksForNode(node, dialog):

pass

##############################################################################

def viewSource(node):

import hdefereval

source = node.node("common").hdaModule().buildUISource(

node, add_window=False)

# Houdini doesn't like it when button callbacks block waiting for popup

# windows, so defer displaying the message dialog.

hdefereval.executeDeferred(lambda: hou.ui.displayMessage(

"UI File Source Code:", details=source, details_expanded=True))

def showWindow(node):

try:

is_window = node.type().name() == "ui_window"

source = node.node("common").hdaModule().buildUISource(

node, add_window=not is_window)

ui_file_path = os.path.join(tempfile.gettempdir(), "ui_file.ui")

ui_file = open(ui_file_path, "wb")

ui_file.write(source + "\n")

ui_file.close()

dialog = hou.ui.createDialog(ui_file_path)

_setUpCallbacks(node, dialog)

if is_window:

value_name = node.evalParm("value_name")

else:

value_name = "show_test_window"

dialog.setValue(value_name, 1)

except:

hou.ui.displayMessage(traceback.format_exc())

def buildUISource(node, add_window):

output_file = cStringIO.StringIO()

output = _UIFileOutput(output_file)

outputValueInitializations(output, node)

if add_window:

_outputTestWindowStart(output)

outputComponent(output, node)

if add_window:

_outputTestWindowEnd(output)

output_file.reset()

return output_file.read()

def _outputTestWindowStart(output):

output.startLine('test_window = DIALOG "Preview Components"')

output.startLine("{")

output.indent_level += 1

output.startLine("VALUE(show_test_window) LOOK(plain) STRETCH")

def _outputTestWindowEnd(output):

output.indent_level -= 1

output.startLine("}")

def outputValueInitializations(output, node):

for line in nodeModule(node).getValueInitializations(node):

output.startLine(line + "\n")

for output_node in _outputsWithoutSubnets(node):

outputValueInitializations(output, output_node)

def outputComponent(output, node):

outputComponentStart(output, node)

actual_outputs = _outputsWithoutSubnets(node)

needs_braces = (len(actual_outputs) != 0 or

nodeModule(node).getNeedsBraces(node))

if needs_braces:

output.startLine("{\n")

output.indent_level += 1

outputProperties(output, node)

if needs_braces:

output.startLine("\n")

for output_node in actual_outputs:

outputComponent(output, output_node)

output.indent_level -= 1

if needs_braces:

output.startLine("}\n")

else:

output.appendToLine(";\n")

def outputComponentStart(output, node):

# Each type of component must implement outputComponentStart.

output.startLine(nodeModule(node).getComponentStart(node) + " ")

def outputProperties(output, node):

properties = list(getCommonProperties(node))

properties.extend(nodeModule(node).getSpecializedProperties(node))

output.appendToLine(" ".join(properties))

def getCommonProperties(node):

properties = []

if node.evalParm("use_width"):

if node.evalParm("hstretch"):

properties.append("MIN_WIDTH(%g)" % node.evalParm("width"))

else:

properties.append("WIDTH(%g)" % node.evalParm("width"))

if node.evalParm("use_height"):

if node.evalParm("vstretch"):

properties.append("MIN_HEIGHT(%g)" % node.evalParm("height"))

else:

properties.append("HEIGHT(%g)" % node.evalParm("height"))

if node.evalParm("hstretch") and node.evalParm("vstretch"):

properties.append("STRETCH")

else:

if node.evalParm("hstretch"):

properties.append("HSTRETCH")

if node.evalParm("vstretch"):

properties.append("VSTRETCH")

if node.evalParm("margin") != 0.0:

properties.append("MARGIN(%g)" % node.evalParm("margin"))

if node.evalParm("spacing") != 0.0:

properties.append("SPACING(%g)" % node.evalParm("spacing"))

return properties

def _setUpCallbacks(node, dialog):

nodeModule(node).setUpCallbacksForNode(node, dialog)

for output_node in _outputsWithoutSubnets(node):

_setUpCallbacks(output_node, dialog)

def _outputsWithoutSubnets(node):

result = []

for output_node in _sortedNodes(node.outputs()):

# Don't dive into components, since they're subnets with the common

# node inside.

if _isLowLevelComponent(output_node):

result.append(output_node)

else:

result.extend(_followIntoSubnet(output_node))

return result

def _isLowLevelComponent(node):

return (len(node.children()) == 1 and

node.children()[0].type().name() == "ui_common")

def _followIntoSubnet(node):

"""If a node is a subnet, return a sequence containing just the node.

Otherwise, return all the nodes connected to the first indirect input

in the subnet.

"""

try:

indirect_input = node.indirectInputs()[0]

except hou.InvalidNodeType:

return (node,)

# Return the nodes inside the subnet that are outputs of indirect_input.

if hasattr(indirect_input, "outputs"):

return _sortedNodes(indirect_input.outputs())

# Find the node inside the subnet that's connected to this indirect

# input, if any.

subnet = indirect_input.parent()

subnet_first_input = subnet.inputs()[0]

return list(_sortedNodes(child_node for child_node in subnet.children()

if subnet_first_input in child_node.inputs()))

def _sortedNodes(nodes):

# Sort outputs by their x position in the network, since the

# order is otherwise random.

return sorted(nodes,

lambda n0, n1: cmp(n0.position()[0], n1.position()[0]))

def doubleQuotedString(s):

return '"' + "".join(_escape_dict.get(char, char) for char in s) + '"'

_escape_dict = {

"\\": "\\\\",

"\n": "\\n",

"\t": "\\t",

'"': '\\"'

}

class FallbackHDAModule(object):

def __init__(self, overriding_hda_module, base_hda_module):

self.overriding_hda_module = overriding_hda_module

self.base_hda_module = base_hda_module

def __getattr__(self, attrib_name):

try:

return getattr(self.overriding_hda_module, attrib_name)

except:

return getattr(self.base_hda_module, attrib_name)

def nodeModule(node):

return FallbackHDAModule(

node.hdaModule(), node.node("common").hdaModule())

class _UIFileOutput(object):

def __init__(self, output_file):

self.output_file = output_file

self.indent_level = 0

self.ended_line = True

def startLine(self, text):

if not self.ended_line:

self.output_file.write("\n")

self.output_file.write(" " * self.indent_level + text)

self.ended_line = text.endswith("\n")

def appendToLine(self, text):

if self.ended_line:

self.output_file.write(" " * self.indent_level)

self.output_file.write(text)

self.ended_line = text.endswith("\n")

import sys

import os

import cStringIO

import tempfile

import traceback

def getComponentStart(node):

return None

def getSpecializedProperties(node):

return ()

def getValueInitializations(node):

return ()

def getNeedsBraces(node):

return False

def setUpCallbacksForNode(node, dialog):

pass

##############################################################################

def viewSource(node):

import hdefereval

source = node.node("common").hdaModule().buildUISource(

node, add_window=False)

# Houdini doesn't like it when button callbacks block waiting for popup

# windows, so defer displaying the message dialog.

hdefereval.executeDeferred(lambda: hou.ui.displayMessage(

"UI File Source Code:", details=source, details_expanded=True))

def showWindow(node):

try:

is_window = node.type().name() == "ui_window"

source = node.node("common").hdaModule().buildUISource(

node, add_window=not is_window)

ui_file_path = os.path.join(tempfile.gettempdir(), "ui_file.ui")

ui_file = open(ui_file_path, "wb")

ui_file.write(source + "\n")

ui_file.close()

dialog = hou.ui.createDialog(ui_file_path)

_setUpCallbacks(node, dialog)

if is_window:

value_name = node.evalParm("value_name")

else:

value_name = "show_test_window"

dialog.setValue(value_name, 1)

except:

hou.ui.displayMessage(traceback.format_exc())

def buildUISource(node, add_window):

output_file = cStringIO.StringIO()

output = _UIFileOutput(output_file)

outputValueInitializations(output, node)

if add_window:

_outputTestWindowStart(output)

outputComponent(output, node)

if add_window:

_outputTestWindowEnd(output)

output_file.reset()

return output_file.read()

def _outputTestWindowStart(output):

output.startLine('test_window = DIALOG "Preview Components"')

output.startLine("{")

output.indent_level += 1

output.startLine("VALUE(show_test_window) LOOK(plain) STRETCH")

def _outputTestWindowEnd(output):

output.indent_level -= 1

output.startLine("}")

def outputValueInitializations(output, node):

for line in nodeModule(node).getValueInitializations(node):

output.startLine(line + "\n")

for output_node in _outputsWithoutSubnets(node):

outputValueInitializations(output, output_node)

def outputComponent(output, node):

outputComponentStart(output, node)

actual_outputs = _outputsWithoutSubnets(node)

needs_braces = (len(actual_outputs) != 0 or

nodeModule(node).getNeedsBraces(node))

if needs_braces:

output.startLine("{\n")

output.indent_level += 1

outputProperties(output, node)

if needs_braces:

output.startLine("\n")

for output_node in actual_outputs:

outputComponent(output, output_node)

output.indent_level -= 1

if needs_braces:

output.startLine("}\n")

else:

output.appendToLine(";\n")

def outputComponentStart(output, node):

# Each type of component must implement outputComponentStart.

output.startLine(nodeModule(node).getComponentStart(node) + " ")

def outputProperties(output, node):

properties = list(getCommonProperties(node))

properties.extend(nodeModule(node).getSpecializedProperties(node))

output.appendToLine(" ".join(properties))

def getCommonProperties(node):

properties = []

if node.evalParm("use_width"):

if node.evalParm("hstretch"):

properties.append("MIN_WIDTH(%g)" % node.evalParm("width"))

else:

properties.append("WIDTH(%g)" % node.evalParm("width"))

if node.evalParm("use_height"):

if node.evalParm("vstretch"):

properties.append("MIN_HEIGHT(%g)" % node.evalParm("height"))

else:

properties.append("HEIGHT(%g)" % node.evalParm("height"))

if node.evalParm("hstretch") and node.evalParm("vstretch"):

properties.append("STRETCH")

else:

if node.evalParm("hstretch"):

properties.append("HSTRETCH")

if node.evalParm("vstretch"):

properties.append("VSTRETCH")

if node.evalParm("margin") != 0.0:

properties.append("MARGIN(%g)" % node.evalParm("margin"))

if node.evalParm("spacing") != 0.0:

properties.append("SPACING(%g)" % node.evalParm("spacing"))

return properties

def _setUpCallbacks(node, dialog):

nodeModule(node).setUpCallbacksForNode(node, dialog)

for output_node in _outputsWithoutSubnets(node):

_setUpCallbacks(output_node, dialog)

def _outputsWithoutSubnets(node):

result = []

for output_node in _sortedNodes(node.outputs()):

# Don't dive into components, since they're subnets with the common

# node inside.

if _isLowLevelComponent(output_node):

result.append(output_node)

else:

result.extend(_followIntoSubnet(output_node))

return result

def _isLowLevelComponent(node):

return (len(node.children()) == 1 and

node.children()[0].type().name() == "ui_common")

def _followIntoSubnet(node):

"""If a node is a subnet, return a sequence containing just the node.

Otherwise, return all the nodes connected to the first indirect input

in the subnet.

"""

try:

indirect_input = node.indirectInputs()[0]

except hou.InvalidNodeType:

return (node,)

# Return the nodes inside the subnet that are outputs of indirect_input.

if hasattr(indirect_input, "outputs"):

return _sortedNodes(indirect_input.outputs())

# Find the node inside the subnet that's connected to this indirect

# input, if any.

subnet = indirect_input.parent()

subnet_first_input = subnet.inputs()[0]

return list(_sortedNodes(child_node for child_node in subnet.children()

if subnet_first_input in child_node.inputs()))

def _sortedNodes(nodes):

# Sort outputs by their x position in the network, since the

# order is otherwise random.

return sorted(nodes,

lambda n0, n1: cmp(n0.position()[0], n1.position()[0]))

def doubleQuotedString(s):

return '"' + "".join(_escape_dict.get(char, char) for char in s) + '"'

_escape_dict = {

"\\": "\\\\",

"\n": "\\n",

"\t": "\\t",

'"': '\\"'

}

class FallbackHDAModule(object):

def __init__(self, overriding_hda_module, base_hda_module):

self.overriding_hda_module = overriding_hda_module

self.base_hda_module = base_hda_module

def __getattr__(self, attrib_name):

try:

return getattr(self.overriding_hda_module, attrib_name)

except:

return getattr(self.base_hda_module, attrib_name)

def nodeModule(node):

return FallbackHDAModule(

node.hdaModule(), node.node("common").hdaModule())

class _UIFileOutput(object):

def __init__(self, output_file):

self.output_file = output_file

self.indent_level = 0

self.ended_line = True

def startLine(self, text):

if not self.ended_line:

self.output_file.write("\n")

self.output_file.write(" " * self.indent_level + text)

self.ended_line = text.endswith("\n")

def appendToLine(self, text):

if self.ended_line:

self.output_file.write(" " * self.indent_level)

self.output_file.write(text)

self.ended_line = text.endswith("\n")