mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-24 12:54:40 +01:00
130 lines
3.9 KiB
Python
130 lines
3.9 KiB
Python
import os
|
|
import shutil
|
|
|
|
from avalon import harmony
|
|
|
|
import pype.api
|
|
import pype.hosts.harmony
|
|
|
|
|
|
class ExtractTemplate(pype.api.Extractor):
|
|
"""Extract the connected nodes to the composite instance."""
|
|
|
|
label = "Extract Template"
|
|
hosts = ["harmony"]
|
|
families = ["harmony.template"]
|
|
|
|
def process(self, instance):
|
|
staging_dir = self.staging_dir(instance)
|
|
filepath = os.path.join(staging_dir, f"{instance.name}.tpl")
|
|
|
|
self.log.info(f"Outputting template to {staging_dir}")
|
|
|
|
dependencies = []
|
|
self.get_dependencies(instance[0], dependencies)
|
|
|
|
# Get backdrops.
|
|
backdrops = {}
|
|
for dependency in dependencies:
|
|
for backdrop in self.get_backdrops(dependency):
|
|
backdrops[backdrop["title"]["text"]] = backdrop
|
|
unique_backdrops = [backdrops[x] for x in set(backdrops.keys())]
|
|
|
|
# Get non-connected nodes within backdrops.
|
|
all_nodes = instance.context.data.get("allNodes")
|
|
for node in [x for x in all_nodes if x not in dependencies]:
|
|
within_unique_backdrops = bool(
|
|
[x for x in self.get_backdrops(node) if x in unique_backdrops]
|
|
)
|
|
if within_unique_backdrops:
|
|
dependencies.append(node)
|
|
|
|
# Make sure we dont export the instance node.
|
|
if instance[0] in dependencies:
|
|
dependencies.remove(instance[0])
|
|
|
|
# Export template.
|
|
pype.hosts.harmony.export_template(
|
|
unique_backdrops, dependencies, filepath
|
|
)
|
|
|
|
# Prep representation.
|
|
os.chdir(staging_dir)
|
|
shutil.make_archive(
|
|
f"{instance.name}",
|
|
"zip",
|
|
os.path.join(staging_dir, f"{instance.name}.tpl")
|
|
)
|
|
|
|
representation = {
|
|
"name": "tpl",
|
|
"ext": "zip",
|
|
"files": f"{instance.name}.zip",
|
|
"stagingDir": staging_dir
|
|
}
|
|
instance.data["representations"] = [representation]
|
|
|
|
def get_backdrops(self, node):
|
|
sig = harmony.signature()
|
|
func = """function %s(probe_node)
|
|
{
|
|
var backdrops = Backdrop.backdrops("Top");
|
|
var valid_backdrops = [];
|
|
for(var i=0; i<backdrops.length; i++)
|
|
{
|
|
var position = backdrops[i].position;
|
|
|
|
var x_valid = false;
|
|
var node_x = node.coordX(probe_node);
|
|
if (position.x < node_x && node_x < (position.x + position.w)){
|
|
x_valid = true
|
|
};
|
|
|
|
var y_valid = false;
|
|
var node_y = node.coordY(probe_node);
|
|
if (position.y < node_y && node_y < (position.y + position.h)){
|
|
y_valid = true
|
|
};
|
|
|
|
if (x_valid && y_valid){
|
|
valid_backdrops.push(backdrops[i])
|
|
};
|
|
}
|
|
return valid_backdrops;
|
|
}
|
|
%s
|
|
""" % (sig, sig)
|
|
return harmony.send(
|
|
{"function": func, "args": [node]}
|
|
)["result"]
|
|
|
|
def get_dependencies(self, node, dependencies):
|
|
sig = harmony.signature()
|
|
func = """function %s(args)
|
|
{
|
|
var target_node = args[0];
|
|
var numInput = node.numberOfInputPorts(target_node);
|
|
var dependencies = [];
|
|
for (var i = 0 ; i < numInput; i++)
|
|
{
|
|
dependencies.push(node.srcNode(target_node, i));
|
|
}
|
|
return dependencies;
|
|
}
|
|
%s
|
|
""" % (sig, sig)
|
|
|
|
current_dependencies = harmony.send(
|
|
{"function": func, "args": [node]}
|
|
)["result"]
|
|
|
|
for dependency in current_dependencies:
|
|
if not dependency:
|
|
continue
|
|
|
|
if dependency in dependencies:
|
|
continue
|
|
|
|
dependencies.append(dependency)
|
|
|
|
self.get_dependencies(dependency, dependencies)
|