Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2025-12-25 05:14:40 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

improve the validation on invalid channel function

Browse source
This commit is contained in:
Kayla Man 2024-05-24 23:16:32 +08:00
parent bd3be36954
commit 339eea017e
1 changed files with 47 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

69
client/ayon_core/hosts/substancepainter/plugins/publish/validate_ouput_maps.py
View file

@ -25,32 +25,17 @@ class ValidateOutputMaps(pyblish.api.InstancePlugin):
def process(self, instance):
config = instance.data["exportConfig"]
creator_attrs = instance.data["creator_attributes"]
# Substance Painter API does not allow to query the actual output maps
# it will generate without actually exporting the files. So we try to
# generate the smallest size / fastest export as possible
config = copy.deepcopy(config)
export_channel = creator_attrs.get("exportChannel", [])
tmp_export_channel = copy.deepcopy(export_channel)
if export_channel:
for export_preset in config.get("exportPresets", {}):
if not export_preset.get("maps", {}):
raise PublishValidationError(
"No Texture Map Exported with texture set:{}.".format(
instance.name)
)
map_names = [channel_map["fileName"] for channel_map
in export_preset["maps"]]
for channel in tmp_export_channel:
for map_name in map_names:
if channel in map_name:
tmp_export_channel.remove(channel)
if tmp_export_channel:
raise PublishValidationError(
"No Channel(s){} found in the texture set:{}".format(
tmp_export_channel, instance.name
))
invalid_channels = self.get_invalid_channels(instance, config)
if invalid_channels:
raise PublishValidationError(
"No Channel(s){} found in the texture set:{}".format(
invalid_channels, instance.name
))
parameters = config["exportParameters"][0]["parameters"]
parameters["sizeLog2"] = [1, 1] # output 2x2 images (smallest)
parameters["paddingAlgorithm"] = "passthrough" # no dilation (faster)
@ -128,3 +113,43 @@ class ValidateOutputMaps(pyblish.api.InstancePlugin):
message=message,
title="Missing output maps"
)
def get_invalid_channels(self, instance, config):
"""Function to get invalid channel(s) from export channel
filtering
Args:
instance (pyblish.api.Instance): Instance
config (dict): export config
Raises:
PublishValidationError: raise Publish Validation
Error if any invalid channel(s) found
Returns:
list: invalid channel(s)
"""
creator_attrs = instance.data["creator_attributes"]
export_channel = creator_attrs.get("exportChannel", [])
tmp_export_channel = copy.deepcopy(export_channel)
invalid_channel = []
if export_channel:
for export_preset in config.get("exportPresets", {}):
if not export_preset.get("maps", {}):
raise PublishValidationError(
"No Texture Map Exported with texture set:{}.".format(
instance.name)
)
map_names = [channel_map["fileName"] for channel_map
in export_preset["maps"]]
for channel in tmp_export_channel:
found = False
for map_name in map_names:
if channel in map_name:
found = True
break # Exit the inner loop once a match is found
if not found:
invalid_channel.append(channel)
return invalid_channel