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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171 | class MovLoader(load.LoaderPlugin):
"""Load mov into OpenRV"""
label = "Load MOV"
product_types: ClassVar[set] = {"*"}
representations: ClassVar[set] = {"*"}
extensions: ClassVar[set] = {"mov", "mp4"}
order = 0
icon = "code-fork"
color = "orange"
def load(
self,
context: dict,
name: str | None = None,
namespace: str | None = None,
options: dict | None = None,
) -> None:
filepath = self.filepath_from_context(context)
namespace = namespace if namespace else context["folder"]["name"]
rep_name = os.path.basename(filepath)
loaded_node = rv.commands.addSourceVerbose([filepath])
node = self._finalize_loaded_node(loaded_node, rep_name, filepath)
# update colorspace
self.set_representation_colorspace(node, context["representation"])
imprint_container(
node,
name=name,
namespace=namespace,
context=context,
loader=self.__class__.__name__,
)
def _finalize_loaded_node(self, loaded_node, rep_name, filepath):
"""Finalize the loaded node in OpenRV.
We are organizing all loaded sources under a switch group so we can
let user switch between versions later on. Every new updated verion is
added as new media representation under the switch group.
We are removing firstly added source since it does not have a name.
Args:
loaded_node (str): The node that was loaded.
rep_name (str): The name of the representation.
filepath (str): The path of the representation.
Returns:
str: The node that was loaded.
"""
node = loaded_node
rv.commands.addSourceMediaRep(loaded_node, rep_name, [filepath])
rv.commands.setActiveSourceMediaRep(
loaded_node,
rep_name,
)
switch_node = rv.commands.sourceMediaRepSwitchNode(loaded_node)
node_type = rv.commands.nodeType(switch_node)
for node in rv.commands.sourceMediaRepsAndNodes(switch_node):
source_node_name = node[0]
source_node = node[1]
node_type = rv.commands.nodeType(source_node)
node_gorup = rv.commands.nodeGroup(source_node)
# we are removing the firstly added wource since it does not have
# a name and we don't want to confuse the user with multiple
# versions of the same source but one of them without a name
if node_type == "RVFileSource" and source_node_name == "":
rv.commands.deleteNode(node_gorup)
else:
node = source_node
break
rv.commands.setStringProperty(f"{node}.media.name", [rep_name], True)
rv.commands.reload()
return node
def update(self, container, context):
node = container["node"]
filepath = rv.commands.sequenceOfFile(
self.filepath_from_context(context)
)[0]
repre_entity = context["representation"]
new_rep_name = os.path.basename(filepath)
source_reps = rv.commands.sourceMediaReps(node)
self.log.warning(f">> source_reps: {source_reps}")
if new_rep_name not in source_reps:
# change path
rv.commands.addSourceMediaRep(node, new_rep_name, [filepath])
else:
self.log.warning(">> new_rep_name already in source_reps")
rv.commands.setActiveSourceMediaRep(
node,
new_rep_name,
)
source_rep_name = rv.commands.sourceMediaRep(node)
self.log.info(f"New source_rep_name: {source_rep_name}")
# update colorspace
representation = context["representation"]
self.set_representation_colorspace(node, representation)
# add data for inventory manager
rv.commands.setStringProperty(
f"{node}.ayon.representation",
[repre_entity["id"]],
True,
)
rv.commands.reload()
def remove(self, container):
node = container["node"]
# since we are organizing all loaded sources under a switch group
# we need to remove all the source nodes organized under it
switch_node = rv.commands.sourceMediaRepSwitchNode(node)
for node_data in rv.commands.sourceMediaRepsAndNodes(switch_node):
source_node_name = node_data[0]
source_node = node_data[1]
node_type = rv.commands.nodeType(source_node)
node_group = rv.commands.nodeGroup(source_node)
if node_type == "RVFileSource":
self.log.warning(f">> node_type: {node_type}")
self.log.warning(f">> source_node_name: {source_node_name}")
rv.commands.deleteNode(node_group)
rv.commands.reload()
def set_representation_colorspace(self, node, representation):
colorspace_data = representation.get("data", {}).get("colorspaceData")
if colorspace_data:
colorspace = colorspace_data["colorspace"]
# TODO: Confirm colorspace is valid in current OCIO config
# otherwise errors will be spammed from OpenRV for invalid space
self.log.info(f"Setting colorspace: {colorspace}")
group = rv.commands.nodeGroup(node)
# Enable OCIO for the node and set the colorspace
set_group_ocio_active_state(group, state=True)
set_group_ocio_colorspace(group, colorspace)
def switch(self, container, context):
self.update(container, context)
|