mirrors/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-12-23 16:38:57 +00:00
Code Issues Projects Releases Wiki Activity

[ModifyChapters] add --round-cuts-to-keyframes option

Browse Source
This commit is contained in:
Quantum
2024-12-25 19:45:05 -05:00
parent 3905f64920
commit cfc29f62c8
5 changed files with 122 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
yt_dlp/postprocessor/ffmpeg.py
View File

@@ -393,6 +393,32 @@ class FFmpegPostProcessor(PostProcessor):
string = string[1:] if string[0] == "'" else "'" + string
return string[:-1] if string[-1] == "'" else string + "'"
def get_keyframe_timestamps(self, path, opts=[]):
if self.probe_basename != 'ffprobe':
if self.probe_available:
self.report_warning('Only ffprobe is supported for keyframe timestamp extraction')
raise PostProcessingError('ffprobe not found. Please install or provide the path using --ffmpeg-location')
self.check_version()
cmd = [
self.probe_executable,
encodeArgument('-select_streams'),
encodeArgument('v:0'),
encodeArgument('-show_entries'),
encodeArgument('packet=pts_time,flags'),
encodeArgument('-print_format'),
encodeArgument('json'),
]
cmd += opts
cmd.append(self._ffmpeg_filename_argument(path))
self.write_debug(f'ffprobe command line: {shell_quote(cmd)}')
stdout, _, _ = Popen.run(cmd, text=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE)
result = json.loads(stdout)
return [float(packet['pts_time']) for packet in result['packets'] if 'K' in packet['flags']]
def force_keyframes(self, filename, timestamps):
timestamps = orderedSet(timestamps)
if timestamps[0] == 0: