mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-26 13:52:15 +01:00
143 lines
4.8 KiB
Python
143 lines
4.8 KiB
Python
import os
|
|
|
|
import pyblish.api
|
|
import openpype.api
|
|
import openpype.lib
|
|
from openpype.lib import should_decompress, \
|
|
get_decompress_dir, decompress
|
|
import shutil
|
|
|
|
|
|
class ExtractJpegEXR(pyblish.api.InstancePlugin):
|
|
"""Create jpg thumbnail from sequence using ffmpeg"""
|
|
|
|
label = "Extract Jpeg EXR"
|
|
order = pyblish.api.ExtractorOrder
|
|
families = [
|
|
"imagesequence", "render", "render2d",
|
|
"source", "plate", "take"
|
|
]
|
|
hosts = ["shell", "fusion", "resolve"]
|
|
enabled = False
|
|
|
|
# presetable attribute
|
|
ffmpeg_args = None
|
|
|
|
def process(self, instance):
|
|
self.log.info("subset {}".format(instance.data['subset']))
|
|
if 'crypto' in instance.data['subset']:
|
|
return
|
|
|
|
do_decompress = False
|
|
# ffmpeg doesn't support multipart exrs, use oiiotool if available
|
|
if instance.data.get("multipartExr") is True:
|
|
return
|
|
|
|
# Skip review when requested.
|
|
if not instance.data.get("review", True):
|
|
return
|
|
|
|
# get representation and loop them
|
|
representations = instance.data["representations"]
|
|
|
|
# filter out mov and img sequences
|
|
representations_new = representations[:]
|
|
|
|
for repre in representations:
|
|
tags = repre.get("tags", [])
|
|
self.log.debug(repre)
|
|
valid = 'review' in tags or "thumb-nuke" in tags
|
|
if not valid:
|
|
continue
|
|
|
|
if not isinstance(repre['files'], (list, tuple)):
|
|
input_file = repre['files']
|
|
else:
|
|
file_index = int(float(len(repre['files'])) * 0.5)
|
|
input_file = repre['files'][file_index]
|
|
|
|
stagingdir = os.path.normpath(repre.get("stagingDir"))
|
|
|
|
# input_file = (
|
|
# collections[0].format('{head}{padding}{tail}') % start
|
|
# )
|
|
full_input_path = os.path.join(stagingdir, input_file)
|
|
self.log.info("input {}".format(full_input_path))
|
|
|
|
decompressed_dir = ''
|
|
do_decompress = should_decompress(full_input_path)
|
|
if do_decompress:
|
|
decompressed_dir = get_decompress_dir()
|
|
|
|
decompress(
|
|
decompressed_dir,
|
|
full_input_path)
|
|
# input path changed, 'decompressed' added
|
|
full_input_path = os.path.join(
|
|
decompressed_dir,
|
|
input_file)
|
|
|
|
filename = os.path.splitext(input_file)[0]
|
|
if not filename.endswith('.'):
|
|
filename += "."
|
|
jpeg_file = filename + "jpg"
|
|
full_output_path = os.path.join(stagingdir, jpeg_file)
|
|
|
|
self.log.info("output {}".format(full_output_path))
|
|
|
|
ffmpeg_path = openpype.lib.get_ffmpeg_tool_path("ffmpeg")
|
|
ffmpeg_args = self.ffmpeg_args or {}
|
|
|
|
jpeg_items = []
|
|
jpeg_items.append("\"{}\"".format(ffmpeg_path))
|
|
# override file if already exists
|
|
jpeg_items.append("-y")
|
|
# use same input args like with mov
|
|
jpeg_items.extend(ffmpeg_args.get("input") or [])
|
|
# input file
|
|
jpeg_items.append("-i {}".format(full_input_path))
|
|
# output arguments from presets
|
|
jpeg_items.extend(ffmpeg_args.get("output") or [])
|
|
|
|
# If its a movie file, we just want one frame.
|
|
if repre["ext"] == "mov":
|
|
jpeg_items.append("-vframes 1")
|
|
|
|
# output file
|
|
jpeg_items.append(full_output_path)
|
|
|
|
subprocess_jpeg = " ".join(jpeg_items)
|
|
|
|
# run subprocess
|
|
self.log.debug("{}".format(subprocess_jpeg))
|
|
try: # temporary until oiiotool is supported cross platform
|
|
openpype.api.run_subprocess(
|
|
subprocess_jpeg, shell=True, logger=self.log
|
|
)
|
|
except RuntimeError as exp:
|
|
if "Compression" in str(exp):
|
|
self.log.debug("Unsupported compression on input files. " +
|
|
"Skipping!!!")
|
|
return
|
|
raise
|
|
|
|
if "representations" not in instance.data:
|
|
instance.data["representations"] = []
|
|
|
|
representation = {
|
|
'name': 'thumbnail',
|
|
'ext': 'jpg',
|
|
'files': jpeg_file,
|
|
"stagingDir": stagingdir,
|
|
"thumbnail": True,
|
|
"tags": ['thumbnail']
|
|
}
|
|
|
|
# adding representation
|
|
self.log.debug("Adding: {}".format(representation))
|
|
representations_new.append(representation)
|
|
|
|
if do_decompress and os.path.exists(decompressed_dir):
|
|
shutil.rmtree(decompressed_dir)
|
|
|
|
instance.data["representations"] = representations_new
|