""" FFmpeg工具模块 - 提供FFmpeg命令构建和处理的公共函数 """ import logging from typing import List, Tuple, Optional from config.settings import get_ffmpeg_config logger = logging.getLogger(__name__) def build_base_ffmpeg_args() -> List[str]: """ 构建基础FFmpeg参数 Returns: 基础参数列表 """ config = get_ffmpeg_config() args = ["ffmpeg", "-y", "-hide_banner"] args.extend(config.progress_args) args.extend(config.loglevel_args) return args def build_null_audio_input() -> List[str]: """ 构建空音频输入参数 Returns: 空音频输入参数列表 """ config = get_ffmpeg_config() return config.null_audio_args def build_amix_filter(input1: str, input2: str, output: str) -> str: """ 构建音频混合滤镜 Args: input1: 第一个音频输入 input2: 第二个音频输入 output: 输出流名称 Returns: 混合滤镜字符串 """ config = get_ffmpeg_config() return f"{input1}[{input2}]{config.amix_args[0]}[{output}]" def build_overlay_scale_filter(video_input: str, overlay_input: str, output: str) -> str: """ 构建覆盖层缩放滤镜 Args: video_input: 视频输入流 overlay_input: 覆盖层输入流 output: 输出流名称 Returns: 缩放滤镜字符串 """ config = get_ffmpeg_config() if config.overlay_scale_mode == "scale": return f"{video_input}[{overlay_input}]scale=iw:ih[{output}]" else: return f"{video_input}[{overlay_input}]{config.overlay_scale_mode}=iw:ih[{output}]" def get_annexb_filter() -> str: """ 获取annexb转换滤镜 Returns: annexb滤镜名称 """ config = get_ffmpeg_config() encoder_args_str = " ".join(config.encoder_args).lower() if "hevc" in encoder_args_str: return "hevc_mp4toannexb" return "h264_mp4toannexb" def build_standard_output_args() -> List[str]: """ 构建标准输出参数 Returns: 输出参数列表 """ config = get_ffmpeg_config() return [ *config.video_args, *config.audio_args, *config.encoder_args, *config.default_args ] def validate_ffmpeg_file_extensions(file_path: str) -> bool: """ 验证文件扩展名是否为FFmpeg支持的格式 Args: file_path: 文件路径 Returns: 是否为支持的格式 """ supported_extensions = { '.mp4', '.avi', '.mov', '.mkv', '.flv', '.wmv', '.webm', '.ts', '.m2ts', '.mts', '.m4v', '.3gp', '.asf', '.rm', '.mp3', '.wav', '.aac', '.flac', '.ogg', '.m4a', '.wma' } import os _, ext = os.path.splitext(file_path.lower()) return ext in supported_extensions def estimate_processing_time(input_duration: float, complexity_factor: float = 1.0) -> float: """ 估算处理时间 Args: input_duration: 输入文件时长(秒) complexity_factor: 复杂度因子(1.0为普通处理) Returns: 预估处理时间(秒) """ # 基础处理速度假设为实时的0.5倍(即处理1秒视频需要2秒) base_processing_ratio = 2.0 return input_duration * base_processing_ratio * complexity_factor