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

make sure only video stream is processed explicitly

Browse source
This commit is contained in:
Jakub Jezek 2024-07-29 16:42:51 +02:00
parent c165eb24e6
commit bd359b01b3
No known key found for this signature in database
GPG key ID: 06DBD609ADF27FD9
1 changed files with 20 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

24
client/ayon_core/lib/transcoding.py
View file

@ -868,6 +868,23 @@ def get_image_info_metadata(
output[key] = val
return output
def _get_video_metadata_from_ffprobe(ffprobe_stream):
"""Extract video metadata from ffprobe stream.
Args:
ffprobe_stream (dict): Stream data obtained from ffprobe.
Returns:
dict: Video metadata extracted from the ffprobe stream.
"""
video_stream = None
for stream in ffprobe_stream["streams"]:
if stream["codec_type"] == "video":
video_stream = stream
break
metadata_stream = _ffprobe_metadata_conversion(video_stream)
return metadata_stream
metadata_stream = None
ext = os.path.splitext(path_to_file)[-1].lower()
if ext not in IMAGE_EXTENSIONS:
@ -877,8 +894,7 @@ def get_image_info_metadata(
).format(ext))
ffprobe_stream = get_ffprobe_data(path_to_file, logger)
if "streams" in ffprobe_stream and len(ffprobe_stream["streams"]) > 0:
metadata_stream = _ffprobe_metadata_conversion(
ffprobe_stream["streams"][-1])
metadata_stream = _get_video_metadata_from_ffprobe(ffprobe_stream)
if not metadata_stream and is_oiio_supported():
oiio_stream = get_oiio_info_for_input(path_to_file, logger=logger)
@ -899,8 +915,7 @@ def get_image_info_metadata(
))
ffprobe_stream = get_ffprobe_data(path_to_file, logger)
if "streams" in ffprobe_stream and len(ffprobe_stream["streams"]) > 0:
metadata_stream = _ffprobe_metadata_conversion(
ffprobe_stream["streams"][0])
metadata_stream = _get_video_metadata_from_ffprobe(ffprobe_stream)
if not metadata_stream:
logger.warning("Failed to get metadata from image file.")
@ -938,6 +953,7 @@ def get_image_info_metadata(
if isinstance(v, dict) and key in v:
output[key] = v[key]
break
return output