mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-02 00:44:52 +01:00
add sequence image in photoshop
This commit is contained in:
clement.hector
parent
6f792511a4
commit
0724f70eee
3 changed files with 103 additions and 46 deletions
|
|
@ -1,4 +1,5 @@
|
|||
import os
|
||||
import shutil
|
||||
|
||||
import openpype.api
|
||||
import openpype.lib
|
||||
|
|
@ -7,7 +8,7 @@ from openpype.hosts.photoshop import api as photoshop
|
|||
|
||||
class ExtractReview(openpype.api.Extractor):
|
||||
"""
|
||||
Produce a flattened image file from all 'image' instances.
|
||||
Produce a flattened or sequence image file from all 'image' instances.
|
||||
|
||||
If no 'image' instance is created, it produces flattened image from
|
||||
all visible layers.
|
||||
|
|
@ -20,54 +21,43 @@ class ExtractReview(openpype.api.Extractor):
|
|||
# Extract Options
|
||||
jpg_options = None
|
||||
mov_options = None
|
||||
make_image_sequence = None
|
||||
|
||||
def process(self, instance):
|
||||
staging_dir = self.staging_dir(instance)
|
||||
self.log.info("Outputting image to {}".format(staging_dir))
|
||||
self.staging_dir = self.staging_dir(instance)
|
||||
self.log.info("Outputting image to {}".format(self.staging_dir))
|
||||
|
||||
stub = photoshop.stub()
|
||||
self.stub = photoshop.stub()
|
||||
self.output_seq_filename = os.path.splitext(
|
||||
self.stub.get_active_document_name())[0] + ".%04d.jpg"
|
||||
|
||||
layers = []
|
||||
for image_instance in instance.context:
|
||||
if image_instance.data["family"] != "image":
|
||||
continue
|
||||
layers.append(image_instance[0])
|
||||
|
||||
# Perform extraction
|
||||
output_image = "{}.jpg".format(
|
||||
os.path.splitext(stub.get_active_document_name())[0]
|
||||
)
|
||||
output_image_path = os.path.join(staging_dir, output_image)
|
||||
with photoshop.maintained_visibility():
|
||||
if layers:
|
||||
# Hide all other layers.
|
||||
extract_ids = set([ll.id for ll in stub.
|
||||
get_layers_in_layers(layers)])
|
||||
self.log.debug("extract_ids {}".format(extract_ids))
|
||||
for layer in stub.get_layers():
|
||||
# limit unnecessary calls to client
|
||||
if layer.visible and layer.id not in extract_ids:
|
||||
stub.set_visible(layer.id, False)
|
||||
|
||||
stub.saveAs(output_image_path, 'jpg', True)
|
||||
new_img_list = src_img_list = []
|
||||
if self.make_image_sequence:
|
||||
src_img_list = self._get_image_path_from_instances(instance)
|
||||
if self.make_image_sequence and src_img_list:
|
||||
new_img_list = self._copy_image_to_staging_dir(src_img_list)
|
||||
else:
|
||||
layers = self._get_layers_from_instance(instance)
|
||||
new_img_list = self._saves_flattened_layers(layers)
|
||||
instance.data["representations"].append({
|
||||
"name": "jpg",
|
||||
"ext": "jpg",
|
||||
"files": new_img_list,
|
||||
"stagingDir": self.staging_dir,
|
||||
"tags": self.jpg_options['tags']
|
||||
})
|
||||
|
||||
ffmpeg_path = openpype.lib.get_ffmpeg_tool_path("ffmpeg")
|
||||
|
||||
instance.data["representations"].append({
|
||||
"name": "jpg",
|
||||
"ext": "jpg",
|
||||
"files": output_image,
|
||||
"stagingDir": staging_dir,
|
||||
"tags": self.jpg_options['tags']
|
||||
})
|
||||
instance.data["stagingDir"] = staging_dir
|
||||
instance.data["stagingDir"] = self.staging_dir
|
||||
|
||||
# Generate thumbnail.
|
||||
thumbnail_path = os.path.join(staging_dir, "thumbnail.jpg")
|
||||
thumbnail_path = os.path.join(self.staging_dir, "thumbnail.jpg")
|
||||
self.log.info(f"Generate thumbnail {thumbnail_path}")
|
||||
args = [
|
||||
ffmpeg_path,
|
||||
"-y",
|
||||
"-i", output_image_path,
|
||||
"-i", os.path.join(self.staging_dir, self.output_seq_filename),
|
||||
"-vf", "scale=300:-1",
|
||||
"-vframes", "1",
|
||||
thumbnail_path
|
||||
|
|
@ -78,17 +68,20 @@ class ExtractReview(openpype.api.Extractor):
|
|||
"name": "thumbnail",
|
||||
"ext": "jpg",
|
||||
"files": os.path.basename(thumbnail_path),
|
||||
"stagingDir": staging_dir,
|
||||
"stagingDir": self.staging_dir,
|
||||
"tags": ["thumbnail"]
|
||||
})
|
||||
|
||||
# Generate mov.
|
||||
mov_path = os.path.join(staging_dir, "review.mov")
|
||||
mov_path = os.path.join(self.staging_dir, "review.mov")
|
||||
self.log.info(f"Generate mov review: {mov_path}")
|
||||
img_number = len(new_img_list)
|
||||
args = [
|
||||
ffmpeg_path,
|
||||
"-y",
|
||||
"-i", output_image_path,
|
||||
"-i", os.path.join(self.staging_dir, self.output_seq_filename),
|
||||
"-vf", "pad=ceil(iw/2)*2:ceil(ih/2)*2",
|
||||
"-vframes", "1",
|
||||
"-vframes", str(img_number),
|
||||
mov_path
|
||||
]
|
||||
output = openpype.lib.run_subprocess(args)
|
||||
|
|
@ -97,9 +90,9 @@ class ExtractReview(openpype.api.Extractor):
|
|||
"name": "mov",
|
||||
"ext": "mov",
|
||||
"files": os.path.basename(mov_path),
|
||||
"stagingDir": staging_dir,
|
||||
"stagingDir":self.staging_dir,
|
||||
"frameStart": 1,
|
||||
"frameEnd": 1,
|
||||
"frameEnd": img_number,
|
||||
"fps": 25,
|
||||
"preview": True,
|
||||
"tags": self.mov_options['tags']
|
||||
|
|
@ -107,7 +100,65 @@ class ExtractReview(openpype.api.Extractor):
|
|||
|
||||
# Required for extract_review plugin (L222 onwards).
|
||||
instance.data["frameStart"] = 1
|
||||
instance.data["frameEnd"] = 1
|
||||
instance.data["frameEnd"] = img_number
|
||||
instance.data["fps"] = 25
|
||||
|
||||
self.log.info(f"Extracted {instance} to {staging_dir}")
|
||||
self.log.info(f"Extracted {instance} to {self.staging_dir}")
|
||||
|
||||
def _get_image_path_from_instances(self, instance):
|
||||
img_list = []
|
||||
|
||||
for instance in instance.context:
|
||||
if instance.data["family"] != "image":
|
||||
continue
|
||||
|
||||
for rep in instance.data["representations"]:
|
||||
img_path = os.path.join(
|
||||
rep["stagingDir"],
|
||||
rep["files"]
|
||||
)
|
||||
img_list.append(img_path)
|
||||
|
||||
return img_list
|
||||
|
||||
def _copy_image_to_staging_dir(self, img_list):
|
||||
copy_files = []
|
||||
for i, img_src in enumerate(img_list):
|
||||
img_filename = self.output_seq_filename %i
|
||||
img_dst = os.path.join(self.staging_dir, img_filename)
|
||||
|
||||
self.log.debug(
|
||||
"Copying file .. {} -> {}".format(img_src, img_dst)
|
||||
)
|
||||
shutil.copy(img_src, img_dst)
|
||||
copy_files.append(img_filename)
|
||||
|
||||
return copy_files
|
||||
|
||||
def _get_layers_from_instance(self, instance):
|
||||
layers = []
|
||||
for image_instance in instance.context:
|
||||
if image_instance.data["family"] != "image":
|
||||
continue
|
||||
layers.append(image_instance[0])
|
||||
|
||||
return layers
|
||||
|
||||
def _saves_flattened_layers(self, layers):
|
||||
img_filename = self.output_seq_filename %0
|
||||
output_image_path = os.path.join(self.staging_dir, img_filename)
|
||||
|
||||
with photoshop.maintained_visibility():
|
||||
if layers:
|
||||
# Hide all other layers.
|
||||
extract_ids = set([ll.id for ll in self.stub.
|
||||
get_layers_in_layers(layers)])
|
||||
self.log.debug("extract_ids {}".format(extract_ids))
|
||||
for layer in self.stub.get_layers():
|
||||
# limit unnecessary calls to client
|
||||
if layer.visible and layer.id not in extract_ids:
|
||||
self.stub.set_visible(layer.id, False)
|
||||
|
||||
self.stub.saveAs(output_image_path, 'jpg', True)
|
||||
|
||||
return img_filename
|
||||
|
|
|
|||
|
|
@ -33,6 +33,7 @@
|
|||
]
|
||||
},
|
||||
"ExtractReview": {
|
||||
"make_image_sequence": false,
|
||||
"jpg_options": {
|
||||
"tags": []
|
||||
},
|
||||
|
|
@ -48,4 +49,4 @@
|
|||
"create_first_version": false,
|
||||
"custom_templates": []
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -154,6 +154,11 @@
|
|||
"key": "ExtractReview",
|
||||
"label": "Extract Review",
|
||||
"children": [
|
||||
{
|
||||
"type": "boolean",
|
||||
"key": "make_image_sequence",
|
||||
"label": "Makes an image sequence instead of a flatten image"
|
||||
},
|
||||
{
|
||||
"type": "dict",
|
||||
"collapsible": false,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue