Created
June 11, 2019 12:30
-
-
Save BigRoy/4eafd16d416543aeb8f0b4c8b8911d98 to your computer and use it in GitHub Desktop.
For a specific shape or mesh get the assigned shader in a specific renderlayer without switching to that layer.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import maya.cmds as cmds | |
| def get_shader_in_layer(node, layer): | |
| """Return the assigned shader in a renderlayer without switching layers. | |
| This has been developed and tested for Legacy Renderlayers and *not* for | |
| Render Setup. | |
| Returns: | |
| list: The list of assigned shaders in the given layer. | |
| """ | |
| def _get_shader(shape): | |
| """Return current shader""" | |
| return cmds.listConnections(shape + ".instObjGroups", | |
| source=False, | |
| destination=True, | |
| plugs=False, | |
| connections=False, | |
| type="shadingEngine") or [] | |
| # We check the instObjGroups (shader connection) for layer overrides. | |
| attr = node + ".instObjGroups" | |
| # Ignore complex query if we're in the layer anyway (optimization) | |
| current_layer = cmds.editRenderLayerGlobals(query=True, | |
| currentRenderLayer=True) | |
| if layer == current_layer: | |
| return _get_shader(node) | |
| connections = cmds.listConnections(attr, | |
| plugs=True, | |
| source=False, | |
| destination=True, | |
| type="renderLayer") or [] | |
| connections = filter(lambda x: x.endswith(".outPlug"), connections) | |
| if not connections: | |
| # If no overrides anywhere on the shader, just get the current shader | |
| return _get_shader(node) | |
| def _get_override(connections, layer): | |
| """Return the overridden connection for that layer in connections""" | |
| # If there's an override on that layer, return that. | |
| for connection in connections: | |
| if (connection.startswith(layer + ".outAdjustments") and | |
| connection.endswith(".outPlug")): | |
| # This is a shader override on that layer so get the shader | |
| # connected to .outValue of the .outAdjustment[i] | |
| out_adjustment = connection.rsplit(".", 1)[0] | |
| connection_attr = out_adjustment + ".outValue" | |
| override = cmds.listConnections(connection_attr) or [] | |
| return override | |
| override_shader = _get_override(connections, layer) | |
| if override_shader is not None: | |
| return override_shader | |
| else: | |
| # Get the override for "defaultRenderLayer" (=masterLayer) | |
| return _get_override(connections, layer="defaultRenderLayer") |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment