twenty-twenty: Rust中的高效图像处理与视觉增强工具

twenty-twenty是一个Rust编写的图像处理和视觉增强库，专注于提供高性能的图像比较和差异可视化功能。

主要特性

• 高效的图像差异计算
• 视觉增强的差异显示
• 支持多种图像格式
• 可定制的比较参数
• 零成本抽象带来的高性能

安装

在Cargo.toml中添加依赖：

[dependencies]
twenty_twenty = "0.3"
image = "0.24"  # 需要image库来加载图像

完整示例代码

use image::GenericImageView;
use twenty_twenty::{Compare, Visualization};

fn main() {
    // 示例1: 基本图像比较
    basic_comparison();
    
    // 示例2: 带参数的自定义比较
    custom_comparison();
    
    // 示例3: 批量比较目录中的图像
    batch_comparison("images");
}

/// 基本图像比较示例
fn basic_comparison() {
    // 加载两幅测试图像
    let image1 = image::open("test1.png").unwrap().to_rgba8();
    let image2 = image::open("test2.png").unwrap().to_rgba8();
    
    // 使用默认参数比较图像
    let result = twenty_twenty::compare(&image1, &image2, Visualization::Diff);
    
    // 保存差异图像
    result.save("basic_diff.png").unwrap();
    println!("基本比较完成，结果保存为 basic_diff.png");
}

/// 自定义比较参数示例
fn custom_comparison() {
    let image1 = image::open("ui_screenshot.png").unwrap().to_rgba8();
    let image2 = image::open("ui_reference.png").unwrap().to_rgba8();
    
    // 创建自定义比较
    let comparison = Compare::new(&image1, &image2)
        .ignore_alpha()  // 忽略alpha通道
        .threshold(0.05) // 设置差异阈值
        .visualization(Visualization::SideBySide {
            alignment: twenty_twenty::Alignment::Center,
            spacing: 20,
            background: [240, 240, 240, 255], // 浅灰色背景
        })
        .compare();
    
    // 保存比较结果
    comparison.save("custom_comparison.png").unwrap();
    
    // 获取并打印差异指标
    let metrics = Compare::new(&image1, &image2)
        .ignore_alpha()
        .compare_metrics();
    
    println!("自定义比较结果:");
    println!("差异百分比: {:.4}%", metrics.diff_percentage * 100.0);
    println!("平均差异: {:.2}", metrics.mean_diff);
    println!("最大差异: {}", metrics.max_diff);
}

/// 批量比较目录中的图像
fn batch_comparison(dir: &str) {
    use std::path::Path;
    use twenty_twenty::{compare, Visualization};
    
    let dir_path = Path::new(dir);
    if !dir_path.exists() {
        println!("目录 {} 不存在", dir);
        return;
    }
    
    println!("开始在目录 {} 中批量比较图像...", dir);
    
    for entry in dir_path.read_dir().unwrap() {
        let entry = entry.unwrap();
        if entry.file_type().unwrap().is_file() {
            let path = entry.path();
            if let Some(ext) = path.extension() {
                if ext == "png" && !path.to_string_lossy().contains("reference") {
                    let base_name = path.file_stem().unwrap().to_string_lossy();
                    
                    // 查找对应的参考图像
                    let reference_path = path.with_file_name(format!("{}_reference.png", base_name));
                    
                    if reference_path.exists() {
                        println!("比较: {} 和 {}", path.display(), reference_path.display());
                        
                        let image = image::open(&path).unwrap().to_rgba8();
                        let reference = image::open(&reference_path).unwrap().to_rgba8();
                        
                        let result = compare(&image, &reference, Visualization::Diff);
                        let output_path = path.with_file_name(format!("{}_diff.png", base_name));
                        result.save(&output_path).unwrap();
                        
                        println!("差异结果保存为: {}", output_path.display());
                    }
                }
            }
        }
    }
    
    println!("批量比较完成");
}

代码说明

1. 基本比较:

   • 使用twenty_twenty::compare函数进行最简单的图像比较
   • 生成标准差异图像并保存

2. 自定义比较:

   • 使用Compare构建器模式自定义比较参数
   • 设置忽略alpha通道、差异阈值等
   • 生成并排比较视图
   • 获取详细的差异指标数据

3. 批量处理:

   • 扫描目录中的PNG文件
   • 自动查找对应的参考图像(以"_reference.png"结尾)
   • 为每对图像生成差异图
   • 保存差异结果为"原文件名_diff.png"

使用建议

1. 对于UI测试，建议阈值设为0.01-0.05之间
2. 处理大量图像时，考虑使用多线程并行处理
3. 对于持续集成环境，可以将差异百分比作为测试通过/失败的标准

这个示例展示了twenty-twenty库的核心功能，包括基本比较、自定义参数比较和批量处理能力。您可以根据实际需求调整参数和可视化方式。