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

added function to get thumbnail bg color

Browse source
This commit is contained in:
iLLiCiTiT 2021-06-22 13:13:51 +02:00
parent 4628b7627a
commit 4d484189ed
1 changed files with 28 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

30
openpype/hosts/tvpaint/plugins/publish/extract_sequence.py
View file

@ -300,10 +300,18 @@ class ExtractSequence(pyblish.api.Extractor):
# - just making sure
source_img = Image.open(first_frame_filepath)
if source_img.mode.lower() == "rgba":
bg_image = Image.new("RGBA", source_img.size, (255, 255, 255))
bg_color = self._get_thumbnail_bg_color()
self.log.debug("Adding thumbnail background color {}.".format(
" ".join(bg_color)
))
bg_image = Image.new("RGBA", source_img.size, bg_color)
thumbnail_obj = Image.alpha_composite(bg_image, source_img)
thumbnail_obj.convert("RGB").save(thumbnail_filepath)
else:
self.log.info((
"Source for thumbnail has mode \"{}\" (Expected: RGBA)."
" Can't use thubmanail background color."
).format(source_img.mode))
source_img.save(thumbnail_filepath)
return output_filenames, thumbnail_filepath
@ -403,14 +411,32 @@ class ExtractSequence(pyblish.api.Extractor):
# Composite background only on rgba images
# - just making sure
if source_img.mode.lower() == "rgba":
bg_image = Image.new("RGBA", source_img.size, (255, 255, 255))
bg_color = self._get_thumbnail_bg_color()
self.log.debug("Adding thumbnail background color {}.".format(
" ".join(bg_color)
))
bg_image = Image.new("RGBA", source_img.size, bg_color)
thumbnail_obj = Image.alpha_composite(bg_image, source_img)
thumbnail_obj.convert("RGB").save(thumbnail_filepath)
else:
self.log.info((
"Source for thumbnail has mode \"{}\" (Expected: RGBA)."
" Can't use thubmanail background color."
).format(source_img.mode))
source_img.save(thumbnail_filepath)
return output_filenames, thumbnail_filepath
def _get_thumbnail_bg_color(self):
red = green = blue = 255
if self.thumbnail_bg:
if len(self.thumbnail_bg) == 4:
red, green, blue, _ = self.thumbnail_bg
elif len(self.thumbnail_bg) == 3:
red, green, blue = self.thumbnail_bg
return (red, green, blue)
def _render_layer(
self,
layer,