mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2025-06-28 01:18:30 +00:00
This commit is contained in:
parent
2685654a37
commit
b343713d97
@ -9,6 +9,7 @@
|
||||
import time
|
||||
|
||||
from .common import PostProcessor
|
||||
from .. import webvtt
|
||||
from ..compat import imghdr
|
||||
from ..utils import (
|
||||
MEDIA_EXTENSIONS,
|
||||
@ -626,14 +627,33 @@ def run(self, info):
|
||||
sub_ext = sub_info['ext']
|
||||
if sub_ext == 'json':
|
||||
self.report_warning('JSON subtitles cannot be embedded')
|
||||
elif ext != 'webm' or (ext == 'webm' and sub_ext == 'vtt'):
|
||||
elif ext in ('mkv', 'webm') and sub_ext == 'vtt':
|
||||
# This is a workaround for ffmpeg bug https://trac.ffmpeg.org/ticket/11493
|
||||
# If the WebVTT contains empty text cues, it'll corrupt remuxed Matroska files, dropping frames
|
||||
sub_langs.append(lang)
|
||||
sub_names.append(sub_info.get('name'))
|
||||
sub_filename = sub_info['filepath']
|
||||
|
||||
with open(sub_info['filepath'], 'rb') as f:
|
||||
vtt_bytes = f.read()
|
||||
blocks = list(webvtt.parse_fragment(vtt_bytes))
|
||||
|
||||
nonempty_blocks = [block for block in blocks if not isinstance(block, webvtt.CueBlock) or block.text.strip() != '']
|
||||
if len(blocks) != len(nonempty_blocks):
|
||||
self.report_warning(f'Filtering WebVTT for {lang} subtitle due to ffmpeg bug #11493')
|
||||
sub_filename = f'{sub_filename}_filtered.vtt'
|
||||
with open(sub_filename, 'w') as f:
|
||||
for block in nonempty_blocks:
|
||||
block.write_into(f)
|
||||
sub_filenames.append(sub_filename)
|
||||
elif ext == 'webm':
|
||||
if not webm_vtt_warn:
|
||||
webm_vtt_warn = True
|
||||
self.report_warning('Only WebVTT subtitles can be embedded in webm files')
|
||||
else:
|
||||
sub_langs.append(lang)
|
||||
sub_names.append(sub_info.get('name'))
|
||||
sub_filenames.append(sub_info['filepath'])
|
||||
else:
|
||||
if not webm_vtt_warn and ext == 'webm' and sub_ext != 'vtt':
|
||||
webm_vtt_warn = True
|
||||
self.report_warning('Only WebVTT subtitles can be embedded in webm files')
|
||||
if not mp4_ass_warn and ext == 'mp4' and sub_ext == 'ass':
|
||||
mp4_ass_warn = True
|
||||
self.report_warning('ASS subtitles cannot be properly embedded in mp4 files; expect issues')
|
||||
|
Loading…
Reference in New Issue
Block a user