13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117 | class VdbLoader(plugin.HoudiniLoader):
"""Load VDB"""
product_types = {"vdbcache"}
label = "Load VDB"
representations = {"vdb"}
order = -10
icon = "code-fork"
color = "orange"
def load(self, context, name=None, namespace=None, data=None):
# Get the root node
obj = hou.node("/obj")
# Define node name
namespace = namespace if namespace else context["folder"]["name"]
node_name = "{}_{}".format(namespace, name) if namespace else name
# Create a new geo node
container = obj.createNode("geo", node_name=node_name)
# Remove the file node, it only loads static meshes
# Houdini 17 has removed the file node from the geo node
file_node = container.node("file1")
if file_node:
file_node.destroy()
# Explicitly create a file node
file_node = container.createNode("file", node_name=node_name)
path = self.filepath_from_context(context)
file_node.setParms(
{"file": self.format_path(path, context["representation"])})
# Set display on last node
file_node.setDisplayFlag(True)
nodes = [container, file_node]
self[:] = nodes
return pipeline.containerise(
node_name,
namespace,
nodes,
context,
self.__class__.__name__,
suffix="",
)
@staticmethod
def format_path(path, representation):
"""Format file path correctly for single vdb or vdb sequence."""
# The path is either a single file or sequence in a folder.
is_sequence = bool(representation["context"].get("frame"))
if is_sequence:
folder, filename = os.path.split(path)
filename = re.sub(r"(.*)\.(\d+)\.vdb$", "\\1.$F4.vdb", filename)
path = os.path.join(folder, filename)
path = os.path.normpath(path)
path = path.replace("\\", "/")
return path
def update(self, container, context):
repre_entity = context["representation"]
node = container["node"]
try:
file_node = next(
n for n in node.children() if n.type().name() == "file"
)
except StopIteration:
self.log.error("Could not find node of type `alembic`")
return
# Update the file path
file_path = self.filepath_from_context(context)
file_path = self.format_path(file_path, repre_entity)
file_node.setParms({"file": file_path})
# Update attribute
node.setParms({"representation": repre_entity["id"]})
def remove(self, container):
node = container["node"]
node.destroy()
def switch(self, container, context):
self.update(container, context)
def create_load_placeholder_node(
self, node_name: str, placeholder_data: dict
) -> hou.Node:
"""Define how to create a placeholder node for this loader for the
Workfile Template Builder system."""
# Create node
network = lib.find_active_network(
category=hou.sopNodeTypeCategory(),
default="/obj/geo1"
)
if not network:
network = hou.node("/obj").createNode("geo", "geo1")
node = network.createNode("null", node_name=node_name)
node.moveToGoodPosition()
return node
|