feat: 集成内置FFmpeg配置系统和中文注释

主要功能: - 将JSON配置文件内容内置到代码中，移除外部文件依赖 - 将ffmpeg.rs中所有英文注释改为中文注释 - 集成ffmpegOpts参数生成系统内置配置: - 音频编解码器: AAC, AC3, PCM, Vorbis (4个) - 视频编码器: H.264/H.265 NVENC, VideoToolbox, AMF, 软件编码器 (8个) - 模型推荐规则: 基于分辨率的智能推荐新增功能: - 智能编码器选择 (根据操作系统和GPU) - 自动模型推荐 (基于视频分辨率) - 动态参数生成 (支持占位符替换) - 兼容性筛选 (按格式、平台、硬件) - 批处理命令生成新增文件: - examples/: 4个演示程序 - tests/: FFmpeg配置测试套件 - 文档: 使用指南、配置说明、功能总结测试验证: - 21个单元测试全部通过 - 4个演示程序运行成功 - 支持Windows/macOS/Linux多平台 - 支持NVIDIA/AMD/Apple硬件加速优势: - 无外部依赖，部署简单 - 启动速度快，内存占用小 - 向后兼容，不影响现有代码 - 完整中文注释，提升可读性
2025-08-18 13:01:03 +08:00
parent 5345c703f5
commit 704272001f
13 changed files with 2196 additions and 190 deletions
--- a/cargos/tvai-v2/examples/builtin_config_test.rs
+++ b/cargos/tvai-v2/examples/builtin_config_test.rs
@@ -0,0 +1,129 @@
+use tvai_sdk::{TvaiSdk, TemplateBuilder};
+
+fn main() -> Result<(), Box<dyn std::error::Error>> {
+    println!("=== 内置配置测试 ===\n");
+
+    // 创建 SDK 实例
+    let sdk = TvaiSdk::new();
+
+    // 测试内置音频编解码器
+    println!("1. 测试内置音频编解码器:");
+    let audio_codecs = sdk.ffmpeg_generator.get_available_audio_codecs();
+    println!("   加载了 {} 个音频编解码器", audio_codecs.len());
+    
+    for codec in audio_codecs {
+        println!("   - {}: {}", codec.name, codec.ffmpeg_opts);
+    }
+
+    // 测试 AAC 编解码器查找
+    if let Some(aac) = sdk.ffmpeg_generator.get_audio_codec("AAC") {
+        println!("   ✓ 成功找到 AAC 编解码器");
+        if let Some(bitrate) = &aac.bitrate {
+            println!("     比特率范围: {}-{} kbps", bitrate.min, bitrate.max);
+        }
+    } else {
+        println!("   ✗ 未找到 AAC 编解码器");
+    }
+
+    println!();
+
+    // 测试内置视频编码器
+    println!("2. 测试内置视频编码器:");
+    let video_encoders = sdk.ffmpeg_generator.get_available_video_encoders();
+    println!("   加载了 {} 个视频编码器", video_encoders.len());
+    
+    for encoder in video_encoders.iter().take(5) { // 只显示前5个
+        println!("   - {} ({}): {}", encoder.id, encoder.encoder, 
+                 encoder.ffmpeg_opts.chars().take(50).collect::<String>());
+    }
+
+    // 测试 H.264 编码器查找
+    if let Some(h264) = sdk.ffmpeg_generator.get_video_encoder_by_name("H264") {
+        println!("   ✓ 成功找到 H.264 编码器: {}", h264.id);
+    } else {
+        println!("   ✗ 未找到 H.264 编码器");
+    }
+
+    println!();
+
+    // 测试模型推荐
+    println!("3. 测试模型推荐:");
+    let test_resolutions = [480, 720, 1080, 2160];
+    for height in test_resolutions {
+        let recommended = sdk.ffmpeg_generator.recommend_model_for_resolution(height);
+        println!("   {}p 推荐模型: {:?}", height, recommended);
+    }
+
+    println!();
+
+    // 测试音频编解码器参数生成
+    println!("4. 测试音频编解码器参数生成:");
+    let aac_params = sdk.ffmpeg_generator.generate_audio_codec_params("AAC", Some(192));
+    println!("   AAC 192kbps: {}", aac_params);
+    
+    let ac3_params = sdk.ffmpeg_generator.generate_audio_codec_params("AC3", Some(448));
+    println!("   AC3 448kbps: {}", ac3_params);
+
+    println!();
+
+    // 测试视频编码器参数生成
+    println!("5. 测试视频编码器参数生成:");
+    if let Some(encoder) = sdk.ffmpeg_generator.get_video_encoder("libx264-default") {
+        let params = sdk.ffmpeg_generator.generate_video_encoder_params(&encoder.id, "vbr", Some(23));
+        println!("   libx264 CRF 23: {}", params);
+    }
+
+    println!();
+
+    // 测试兼容性筛选
+    println!("6. 测试兼容性筛选:");
+    
+    let windows_encoders = sdk.ffmpeg_generator.get_compatible_video_encoders("windows", None);
+    println!("   Windows 兼容编码器: {} 个", windows_encoders.len());
+    
+    let nvidia_encoders = sdk.ffmpeg_generator.get_compatible_video_encoders("windows", Some("nvidia"));
+    println!("   NVIDIA 编码器: {} 个", nvidia_encoders.len());
+    
+    let mp4_encoders = sdk.ffmpeg_generator.get_encoders_for_extension("mp4");
+    println!("   MP4 兼容编码器: {} 个", mp4_encoders.len());
+
+    println!();
+
+    // 测试完整的 FFmpeg 命令生成
+    println!("7. 测试完整的 FFmpeg 命令生成:");
+    
+    let template = TemplateBuilder::new("内置配置测试")
+        .enable_enhancement("prob-4")
+        .video_codec("H264", Some(23))
+        .audio_settings("AAC", 192, 2)
+        .build()?;
+
+    let command = sdk.generate_ffmpeg_command(&template, "input.mp4", "output.mp4")?;
+    println!("   生成的命令: {}", command);
+
+    println!();
+
+    // 测试自动编码器选择
+    println!("8. 测试自动编码器选择:");
+    
+    match sdk.ffmpeg_generator.generate_with_auto_encoder(&template, "input.mp4", "output.mp4", "windows", Some("nvidia")) {
+        Ok(auto_command) => println!("   自动选择 (Windows+NVIDIA): {}", auto_command),
+        Err(e) => println!("   自动选择失败: {}", e),
+    }
+
+    println!();
+
+    // 生成编码器报告
+    println!("9. 编码器报告:");
+    let report = sdk.ffmpeg_generator.generate_encoder_report();
+    let lines: Vec<&str> = report.lines().take(10).collect(); // 只显示前10行
+    for line in lines {
+        println!("   {}", line);
+    }
+    println!("   ... (报告已截断)");
+
+    println!("\n=== 测试完成 ===");
+    println!("所有内置配置功能正常工作！");
+
+    Ok(())
+}
--- a/cargos/tvai-v2/examples/comprehensive_demo.rs
+++ b/cargos/tvai-v2/examples/comprehensive_demo.rs
@@ -0,0 +1,205 @@
+use tvai_sdk::{TvaiSdk, TemplateBuilder};
+
+fn main() -> Result<(), Box<dyn std::error::Error>> {
+    println!("🎬 Topaz Video AI SDK 综合功能演示");
+    println!("=====================================\n");
+
+    let sdk = TvaiSdk::new();
+
+    // 1. 展示内置配置加载
+    println!("📦 1. 内置配置状态:");
+    println!("   音频编解码器: {} 个", sdk.ffmpeg_generator.get_available_audio_codecs().len());
+    println!("   视频编码器: {} 个", sdk.ffmpeg_generator.get_available_video_encoders().len());
+    
+    // 验证关键编解码器
+    let key_codecs = ["AAC", "AC3", "PCM"];
+    for codec in &key_codecs {
+        if sdk.ffmpeg_generator.get_audio_codec(codec).is_some() {
+            println!("   ✓ {} 编解码器已加载", codec);
+        }
+    }
+    println!();
+
+    // 2. 模型推荐演示
+    println!("🤖 2. AI 模型推荐:");
+    let resolutions = [(480, "SD"), (720, "HD"), (1080, "FHD"), (2160, "4K")];
+    for (height, name) in &resolutions {
+        let models = sdk.ffmpeg_generator.recommend_model_for_resolution(*height);
+        println!("   {} ({}p): {:?}", name, height, models);
+    }
+    println!();
+
+    // 3. 平台兼容性演示
+    println!("💻 3. 平台兼容性:");
+    let platforms = [("windows", "Windows"), ("osx", "macOS"), ("linux", "Linux")];
+    for (os, name) in &platforms {
+        let count = sdk.ffmpeg_generator.get_compatible_video_encoders(os, None).len();
+        println!("   {}: {} 个兼容编码器", name, count);
+    }
+    println!();
+
+    // 4. 创建不同用途的模板
+    println!("📝 4. 模板创建演示:");
+    
+    // 4K 上采样模板
+    let upscale_template = TemplateBuilder::new("4K上采样")
+        .description("将视频上采样到4K分辨率")
+        .enable_enhancement("prob-4")
+        .enhancement_params(20, 35, 15)
+        .resolution(3840, 2160)
+        .video_codec("hevc_nvenc", Some(18))
+        .audio_settings("AAC", 256, 2)
+        .build()?;
+    println!("   ✓ 4K上采样模板创建成功");
+
+    // 60fps 插帧模板
+    let fps_template = TemplateBuilder::new("60fps插帧")
+        .description("将视频插帧到60fps")
+        .enable_frame_interpolation("chr-2", 1.0)
+        .output_settings(0, 60.0)
+        .video_codec("h264_nvenc", Some(23))
+        .audio_settings("AAC", 192, 2)
+        .build()?;
+    println!("   ✓ 60fps插帧模板创建成功");
+
+    // 降噪模板
+    let denoise_template = TemplateBuilder::new("降噪处理")
+        .description("使用Nyx模型进行降噪")
+        .enable_enhancement("nyx-3")
+        .enhancement_params(70, 15, 5)
+        .video_codec("libx264", Some(20))
+        .audio_settings("AAC", 160, 2)
+        .build()?;
+    println!("   ✓ 降噪处理模板创建成功");
+    println!();
+
+    // 5. FFmpeg 命令生成演示
+    println!("⚙️  5. FFmpeg 命令生成:");
+    
+    let test_files = [
+        ("input_sd.mp4", "output_4k.mp4", &upscale_template, "4K上采样"),
+        ("input_30fps.mp4", "output_60fps.mp4", &fps_template, "60fps插帧"),
+        ("noisy_video.mp4", "clean_video.mp4", &denoise_template, "降噪处理"),
+    ];
+
+    for (input, output, template, desc) in &test_files {
+        let command = sdk.generate_ffmpeg_command(template, input, output)?;
+        println!("   {} 命令:", desc);
+        println!("   {}\n", truncate_command(&command, 100));
+    }
+
+    // 6. 自动编码器选择演示
+    println!("🔧 6. 自动编码器选择:");
+    let auto_scenarios = [
+        ("windows", Some("nvidia"), "Windows + NVIDIA"),
+        ("osx", Some("appleIntel"), "macOS + Intel"),
+        ("linux", None, "Linux 软件编码"),
+    ];
+
+    for (os, gpu, desc) in &auto_scenarios {
+        match sdk.ffmpeg_generator.generate_with_auto_encoder(
+            &upscale_template,
+            "input.mp4",
+            "output.mp4",
+            os,
+            *gpu
+        ) {
+            Ok(command) => println!("   ✓ {}: 命令生成成功", desc),
+            Err(e) => println!("   ✗ {}: {}", desc, e),
+        }
+    }
+    println!();
+
+    // 7. 批处理演示
+    println!("📁 7. 批处理功能:");
+    let batch_files = vec![
+        "video1.mp4".to_string(),
+        "video2.mp4".to_string(), 
+        "video3.mp4".to_string(),
+    ];
+    
+    let batch_commands = sdk.generate_batch_commands(&upscale_template, &batch_files, "output")?;
+    println!("   生成了 {} 个批处理命令", batch_commands.len());
+    for (i, _) in batch_commands.iter().enumerate() {
+        println!("   ✓ 批处理命令 {} 生成成功", i + 1);
+    }
+    println!();
+
+    // 8. 编码器参数生成演示
+    println!("🎛️  8. 编码器参数生成:");
+    
+    // 音频参数
+    let audio_tests = [("AAC", 192), ("AC3", 448), ("AAC", 128)];
+    for (codec, bitrate) in &audio_tests {
+        let params = sdk.ffmpeg_generator.generate_audio_codec_params(codec, Some(*bitrate));
+        println!("   {} {}kbps: {}", codec, bitrate, params);
+    }
+    
+    // 视频参数
+    if let Some(encoder) = sdk.ffmpeg_generator.get_video_encoder("h264-high-win-nvidia") {
+        let vbr_params = sdk.ffmpeg_generator.generate_video_encoder_params(&encoder.id, "vbr", Some(23));
+        println!("   H.264 NVENC VBR: {}", truncate_command(&vbr_params, 80));
+    }
+    println!();
+
+    // 9. 性能统计
+    println!("📊 9. 功能统计:");
+    println!("   支持的音频格式: {:?}", get_supported_audio_formats(&sdk));
+    println!("   支持的视频格式: {:?}", get_supported_video_formats(&sdk));
+    println!("   硬件加速编码器: {} 个", count_hardware_encoders(&sdk));
+    println!("   跨平台编码器: {} 个", count_cross_platform_encoders(&sdk));
+    println!();
+
+    println!("🎉 综合演示完成！");
+    println!("所有功能正常工作，内置配置系统运行良好。");
+
+    Ok(())
+}
+
+fn truncate_command(command: &str, max_len: usize) -> String {
+    if command.len() <= max_len {
+        command.to_string()
+    } else {
+        format!("{}...", &command[..max_len])
+    }
+}
+
+fn get_supported_audio_formats(sdk: &TvaiSdk) -> Vec<String> {
+    let mut formats = std::collections::HashSet::new();
+    for codec in sdk.ffmpeg_generator.get_available_audio_codecs() {
+        for ext in &codec.ext {
+            formats.insert(ext.clone());
+        }
+    }
+    let mut result: Vec<_> = formats.into_iter().collect();
+    result.sort();
+    result
+}
+
+fn get_supported_video_formats(sdk: &TvaiSdk) -> Vec<String> {
+    let mut formats = std::collections::HashSet::new();
+    for encoder in sdk.ffmpeg_generator.get_available_video_encoders() {
+        for ext in &encoder.ext {
+            formats.insert(ext.clone());
+        }
+    }
+    let mut result: Vec<_> = formats.into_iter().collect();
+    result.sort();
+    result
+}
+
+fn count_hardware_encoders(sdk: &TvaiSdk) -> usize {
+    sdk.ffmpeg_generator.get_available_video_encoders()
+        .iter()
+        .filter(|encoder| encoder.gpu.is_some())
+        .count()
+}
+
+fn count_cross_platform_encoders(sdk: &TvaiSdk) -> usize {
+    sdk.ffmpeg_generator.get_available_video_encoders()
+        .iter()
+        .filter(|encoder| {
+            encoder.os.as_ref().map_or(false, |os| os.contains('|'))
+        })
+        .count()
+}
--- a/cargos/tvai-v2/examples/ffmpeg_config_demo.rs
+++ b/cargos/tvai-v2/examples/ffmpeg_config_demo.rs
@@ -0,0 +1,123 @@
+use tvai_sdk::{TvaiSdk, TemplateBuilder};
+
+fn main() -> Result<(), Box<dyn std::error::Error>> {
+    // 创建 SDK 实例
+    let sdk = TvaiSdk::new();
+
+    println!("=== Topaz Video AI FFmpeg 内置配置演示 ===\n");
+
+    // 显示编码器报告
+    println!("{}", sdk.ffmpeg_generator.generate_encoder_report());
+
+    // 创建一个示例模板
+    let template = TemplateBuilder::new("内置配置演示")
+        .description("使用内置配置的 FFmpeg 命令生成演示")
+        .enable_enhancement("prob-4")
+        .enhancement_params(25, 30, 20)
+        .resolution(1920, 1080)
+        .video_codec("H264", Some(23))
+        .audio_settings("AAC", 192, 2)
+        .build()?;
+
+    // 生成基本 FFmpeg 命令
+    println!("=== 基本 FFmpeg 命令 ===");
+    let basic_command = sdk.generate_ffmpeg_command(&template, "input.mp4", "output.mp4")?;
+    println!("{}\n", basic_command);
+
+    // 演示自动编码器选择
+    println!("=== 自动编码器选择 ===");
+    
+    // Windows + NVIDIA GPU
+    match sdk.ffmpeg_generator.generate_with_auto_encoder(&template, "input.mp4", "output_nvidia.mp4", "windows", Some("nvidia")) {
+        Ok(command) => println!("Windows + NVIDIA: {}", command),
+        Err(e) => println!("Windows + NVIDIA 错误: {}", e),
+    }
+
+    // macOS + Apple Silicon
+    match sdk.ffmpeg_generator.generate_with_auto_encoder(&template, "input.mp4", "output_apple.mov", "osx", Some("appleSilicon")) {
+        Ok(command) => println!("macOS + Apple Silicon: {}", command),
+        Err(e) => println!("macOS + Apple Silicon 错误: {}", e),
+    }
+
+    // Linux + AMD GPU
+    match sdk.ffmpeg_generator.generate_with_auto_encoder(&template, "input.mp4", "output_amd.mp4", "linux", Some("amd")) {
+        Ok(command) => println!("Linux + AMD: {}", command),
+        Err(e) => println!("Linux + AMD 错误: {}", e),
+    }
+
+    println!();
+
+    // 演示模型推荐
+    println!("=== 模型推荐 ===");
+    let resolutions = [480, 720, 1080, 2160];
+    for height in resolutions {
+        let recommended = sdk.ffmpeg_generator.recommend_model_for_resolution(height);
+        println!("{}p 推荐模型: {:?}", height, recommended);
+    }
+
+    println!();
+
+    // 演示内置音频编解码器配置
+    println!("=== 内置音频编解码器配置 ===");
+    for codec in sdk.ffmpeg_generator.get_available_audio_codecs() {
+        println!("- {}: {}", codec.name, codec.ffmpeg_opts);
+        if let Some(bitrate_config) = &codec.bitrate {
+            println!("  比特率范围: {}-{} kbps (推荐: {} kbps)",
+                     bitrate_config.min, bitrate_config.max, bitrate_config.default);
+            println!("  建议值: {:?}", bitrate_config.suggested);
+        }
+    }
+
+    println!();
+
+    // 演示兼容性筛选
+    println!("=== 兼容性筛选 ===");
+    
+    println!("Windows 兼容编码器:");
+    let windows_encoders = sdk.ffmpeg_generator.get_compatible_video_encoders("windows", None);
+    for encoder in windows_encoders.iter().take(5) { // 只显示前5个
+        println!("  - {} ({})", encoder.id, encoder.encoder);
+    }
+
+    println!("\nmacOS 兼容编码器:");
+    let macos_encoders = sdk.ffmpeg_generator.get_compatible_video_encoders("osx", None);
+    for encoder in macos_encoders.iter().take(5) {
+        println!("  - {} ({})", encoder.id, encoder.encoder);
+    }
+
+    println!("\nMP4 格式兼容编码器:");
+    let mp4_encoders = sdk.ffmpeg_generator.get_encoders_for_extension("mp4");
+    for encoder in mp4_encoders.iter().take(5) {
+        println!("  - {} ({})", encoder.id, encoder.encoder);
+    }
+
+    println!();
+
+    // 演示不同质量设置
+    println!("=== 不同质量设置演示 ===");
+    
+    // 高质量设置
+    let hq_template = TemplateBuilder::new("高质量")
+        .enable_enhancement("prob-4")
+        .video_codec("hevc_nvenc", Some(18))
+        .audio_settings("AAC", 320, 2)
+        .build()?;
+    
+    let hq_command = sdk.generate_ffmpeg_command(&hq_template, "input.mp4", "output_hq.mp4")?;
+    println!("高质量: {}", hq_command);
+
+    // 快速编码设置
+    let fast_template = TemplateBuilder::new("快速编码")
+        .enable_enhancement("prob-4")
+        .video_codec("h264_nvenc", Some(25))
+        .audio_settings("AAC", 128, 2)
+        .preset("fast")
+        .build()?;
+    
+    let fast_command = sdk.generate_ffmpeg_command(&fast_template, "input.mp4", "output_fast.mp4")?;
+    println!("快速编码: {}", fast_command);
+
+    println!("\n=== 演示完成 ===");
+
+    Ok(())
+}
--- a/cargos/tvai-v2/examples/quick_verify.rs
+++ b/cargos/tvai-v2/examples/quick_verify.rs
@@ -0,0 +1,65 @@
+use tvai_sdk::{TvaiSdk, TemplateBuilder};
+
+fn main() -> Result<(), Box<dyn std::error::Error>> {
+    println!("🚀 快速验证内置配置功能...\n");
+
+    // 创建 SDK
+    let sdk = TvaiSdk::new();
+    println!("✓ SDK 创建成功");
+
+    // 验证音频编解码器
+    let audio_count = sdk.ffmpeg_generator.get_available_audio_codecs().len();
+    println!("✓ 加载了 {} 个音频编解码器", audio_count);
+
+    // 验证视频编码器
+    let video_count = sdk.ffmpeg_generator.get_available_video_encoders().len();
+    println!("✓ 加载了 {} 个视频编码器", video_count);
+
+    // 验证 AAC 编解码器
+    if sdk.ffmpeg_generator.get_audio_codec("AAC").is_some() {
+        println!("✓ AAC 编解码器可用");
+    } else {
+        println!("✗ AAC 编解码器不可用");
+    }
+
+    // 验证 H.264 编码器
+    if sdk.ffmpeg_generator.get_video_encoder_by_name("H264").is_some() {
+        println!("✓ H.264 编码器可用");
+    } else {
+        println!("✗ H.264 编码器不可用");
+    }
+
+    // 验证模型推荐
+    let recommended = sdk.ffmpeg_generator.recommend_model_for_resolution(1080);
+    if !recommended.is_empty() {
+        println!("✓ 模型推荐功能正常: {:?}", recommended);
+    } else {
+        println!("✗ 模型推荐功能异常");
+    }
+
+    // 验证 FFmpeg 命令生成
+    let template = TemplateBuilder::new("验证测试")
+        .enable_enhancement("prob-4")
+        .build()?;
+
+    let command = sdk.generate_ffmpeg_command(&template, "test.mp4", "output.mp4")?;
+    if command.contains("ffmpeg") && command.contains("tvai_up") {
+        println!("✓ FFmpeg 命令生成正常");
+    } else {
+        println!("✗ FFmpeg 命令生成异常");
+        println!("   命令: {}", command);
+    }
+
+    // 验证音频参数生成
+    let audio_params = sdk.ffmpeg_generator.generate_audio_codec_params("AAC", Some(192));
+    if audio_params.contains("-c:a") && audio_params.contains("192") {
+        println!("✓ 音频参数生成正常");
+    } else {
+        println!("✗ 音频参数生成异常: {}", audio_params);
+    }
+
+    println!("\n🎉 所有核心功能验证通过！");
+    println!("内置配置系统正常工作。");
+
+    Ok(())
+}