mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-24 12:54:40 +01:00
103 lines
3.7 KiB
Python
103 lines
3.7 KiB
Python
import os
|
|
|
|
import pyblish.api
|
|
import clique
|
|
import pype.api
|
|
|
|
|
|
class ExtractJpegEXR(pyblish.api.InstancePlugin):
|
|
"""Resolve any dependency issues
|
|
|
|
This plug-in resolves any paths which, if not updated might break
|
|
the published file.
|
|
|
|
The order of families is important, when working with lookdev you want to
|
|
first publish the texture, update the texture paths in the nodes and then
|
|
publish the shading network. Same goes for file dependent assets.
|
|
"""
|
|
|
|
label = "Extract Jpeg EXR"
|
|
hosts = ["shell"]
|
|
order = pyblish.api.ExtractorOrder
|
|
families = ["imagesequence", "render", "render2d", "source"]
|
|
enabled = False
|
|
|
|
def process(self, instance):
|
|
start = instance.data.get("frameStart")
|
|
stagingdir = os.path.normpath(instance.data.get("stagingDir"))
|
|
|
|
collected_frames = os.listdir(stagingdir)
|
|
collections, remainder = clique.assemble(collected_frames)
|
|
|
|
self.log.info("subset {}".format(instance.data['subset']))
|
|
if 'crypto' in instance.data['subset']:
|
|
return
|
|
|
|
# get representation and loop them
|
|
representations = instance.data["representations"]
|
|
|
|
# filter out mov and img sequences
|
|
representations_new = representations[:]
|
|
|
|
for repre in representations:
|
|
self.log.debug(repre)
|
|
if 'review' in repre['tags'] or "thumb-nuke" in repre['tags']:
|
|
if not isinstance(repre['files'], list):
|
|
return
|
|
|
|
input_file = repre['files'][0]
|
|
|
|
# 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))
|
|
|
|
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))
|
|
|
|
config_data = instance.context.data['output_repre_config']
|
|
|
|
proj_name = os.environ.get('AVALON_PROJECT', '__default__')
|
|
profile = config_data.get(proj_name, config_data['__default__'])
|
|
|
|
jpeg_items = []
|
|
jpeg_items.append(
|
|
os.path.join(os.environ.get("FFMPEG_PATH"), "ffmpeg"))
|
|
# override file if already exists
|
|
jpeg_items.append("-y")
|
|
# use same input args like with mov
|
|
jpeg_items.extend(profile.get('input', []))
|
|
# input file
|
|
jpeg_items.append("-i {}".format(full_input_path))
|
|
# output file
|
|
jpeg_items.append(full_output_path)
|
|
|
|
subprocess_jpeg = " ".join(jpeg_items)
|
|
|
|
# run subprocess
|
|
self.log.debug("{}".format(subprocess_jpeg))
|
|
pype.api.subprocess(subprocess_jpeg)
|
|
|
|
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)
|
|
|
|
instance.data["representations"] = representations_new
|