Skip to content

commands

create_folder_id_hash(nodes)

Create a hash based on cbId attribute value Args: nodes (list): a list of nodes

Returns:

Type Description

dict

Source code in client/ayon_maya/tools/mayalookassigner/commands.py
 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
def create_folder_id_hash(nodes):
    """Create a hash based on cbId attribute value
    Args:
        nodes (list): a list of nodes

    Returns:
        dict
    """
    node_id_hash = defaultdict(list)
    for node in nodes:
        # iterate over content of reference node
        if cmds.nodeType(node) == "reference":
            ref_hashes = create_folder_id_hash(
                list(set(cmds.referenceQuery(node, nodes=True, dp=True))))
            for folder_id, ref_nodes in ref_hashes.items():
                node_id_hash[folder_id] += ref_nodes
        elif cmds.pluginInfo('vrayformaya', query=True,
                             loaded=True) and cmds.nodeType(
                node) == "VRayProxy":
            path = cmds.getAttr("{}.fileName".format(node))
            ids = get_alembic_ids_cache(path)
            for k, _ in ids.items():
                id = k.split(":")[0]
                node_id_hash[id].append(node)
        elif cmds.nodeType(node) in {"aiStandIn", "gpuCache"}:
            for id, _ in arnold_standin.get_nodes_by_id(node).items():
                id = id.split(":")[0]
                node_id_hash[id].append(node)
        else:
            value = lib.get_id(node)
            if value is None:
                continue

            folder_id = value.split(":")[0]
            node_id_hash[folder_id].append(node)

    return dict(node_id_hash)

create_items_from_nodes(nodes)

Create an item for the view based the container and content of it

It fetches the look document based on the folder id found in the content. The item will contain all important information for the tool to work.

If there is an folder id which is not registered in the project's collection it will log a warning message.

Parameters:

Name Type Description Default
nodes list

list of maya nodes

required

Returns:

Type Description

list of dicts

Source code in client/ayon_maya/tools/mayalookassigner/commands.py
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
172
173
174
175
176
def create_items_from_nodes(nodes):
    """Create an item for the view based the container and content of it

    It fetches the look document based on the folder id found in the content.
    The item will contain all important information for the tool to work.

    If there is an folder id which is not registered in the project's collection
    it will log a warning message.

    Args:
        nodes (list): list of maya nodes

    Returns:
        list of dicts

    """

    folder_view_items = []

    id_hashes = create_folder_id_hash(nodes)

    if not id_hashes:
        log.warning("No id hashes")
        return folder_view_items

    project_name = get_current_project_name()
    folder_ids = set(id_hashes.keys())

    # Ignore invalid ids
    folder_ids = {
        folder_id for folder_id in folder_ids
        if ayon_api.utils.convert_entity_id(folder_id)
    }

    folder_entities = ayon_api.get_folders(
        project_name, folder_ids, fields={"id", "path"}
    )
    folder_entities_by_id = {
        folder_entity["id"]: folder_entity
        for folder_entity in folder_entities
    }

    for folder_id, id_nodes in id_hashes.items():
        folder_entity = folder_entities_by_id.get(folder_id)
        # Skip if folder id is not found
        if not folder_entity:
            log.warning(
                "Id found on {num} nodes for which no folder is found database,"
                " skipping '{folder_id}'".format(
                    num=len(nodes),
                    folder_id=folder_id
                )
            )
            continue

        # Collect available look products for this folder
        looks = lib.list_looks(project_name, folder_entity["id"])

        # Collect namespaces the folder is found in
        namespaces = set()
        for node in id_nodes:
            namespace = get_namespace_from_node(node)
            namespaces.add(namespace)

        folder_view_items.append({
            "label": folder_entity["path"],
            "folder_entity": folder_entity,
            "looks": looks,
            "namespaces": namespaces
        })

    return folder_view_items

get_all_asset_nodes()

Get all assets from the scene, container based

Returns:

Name Type Description
list

list of dictionaries

Source code in client/ayon_maya/tools/mayalookassigner/commands.py
57
58
59
60
61
62
63
def get_all_asset_nodes():
    """Get all assets from the scene, container based

    Returns:
        list: list of dictionaries
    """
    return cmds.ls(dag=True, noIntermediate=True, long=True)

get_namespace_from_node(node)

Get the namespace from the given node

Parameters:

Name Type Description Default
node str

name of the node

required

Returns:

Type Description

namespace (str)

Source code in client/ayon_maya/tools/mayalookassigner/commands.py
34
35
36
37
38
39
40
41
42
43
44
45
def get_namespace_from_node(node):
    """Get the namespace from the given node

    Args:
        node (str): name of the node

    Returns:
        namespace (str)

    """
    parts = node.rsplit("|", 1)[-1].rsplit(":", 1)
    return parts[0] if len(parts) > 1 else u":"

get_selected_nodes()

Get information from current selection

Source code in client/ayon_maya/tools/mayalookassigner/commands.py
48
49
50
51
52
53
54
def get_selected_nodes():
    """Get information from current selection"""

    selection = cmds.ls(selection=True, long=True)
    hierarchy = lib.get_all_children(selection,
                                     ignore_intermediate_objects=True)
    return list(hierarchy.union(selection))

remove_unused_looks()

Removes all loaded looks for which none of the shaders are used.

This will cleanup all loaded "LookLoader" containers that are unused in the current scene.

Source code in client/ayon_maya/tools/mayalookassigner/commands.py
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
def remove_unused_looks():
    """Removes all loaded looks for which none of the shaders are used.

    This will cleanup all loaded "LookLoader" containers that are unused in
    the current scene.

    """

    host = registered_host()

    unused = []
    for container in host.ls():
        if container['loader'] == "LookLoader":
            members = lib.get_container_members(container['objectName'])
            look_sets = cmds.ls(members, type="objectSet")
            for look_set in look_sets:
                # If the set is used than we consider this look *in use*
                if cmds.sets(look_set, query=True):
                    break
            else:
                unused.append(container)

    for container in unused:
        log.info("Removing unused look container: %s", container['objectName'])
        remove_container(container)

    log.info("Finished removing unused looks. (see log for details)")