meteor_detect/build.rs

use std::process::Command;
use std::env;

fn main() {
    // 通过包含的 OpenCV 库检测版本
    let opencv_version = detect_opencv_version();
    
    println!("cargo:rustc-env=OPENCV_VERSION={}", opencv_version);
    
    // 解析版本
    let version_parts: Vec<&str> = opencv_version.split('.').collect();
    let major = version_parts.get(0).and_then(|s| s.parse::<u32>().ok()).unwrap_or(0);
    let minor = version_parts.get(1).and_then(|s| s.parse::<u32>().ok()).unwrap_or(0);
    
    // 如果版本 >= 4.11.0，启用特性
    if major > 4 || (major == 4 && minor >= 11) {
        println!("cargo:rustc-cfg=feature=\"opencv-4-11-plus\"");
        println!("cargo:warning=Detected OpenCV {} >= 4.11.0, enabling opencv-4-11-plus feature", opencv_version);
    } else {
        println!("cargo:warning=Detected OpenCV {}, opencv-4-11-plus feature NOT enabled", opencv_version);
    }
}

fn detect_opencv_version() -> String {
    // 方法 1: 尝试使用 pkg-config
    if let Some(version) = try_pkg_config() {
        return version;
    }
    
    // 方法 2: 尝试使用 opencv_version 命令
    if let Some(version) = try_opencv_version_command() {
        return version;
    }
    
    // 方法 3: 尝试编译并运行一个小程序来检测版本
    if let Some(version) = try_compile_check() {
        return version;
    }
    
    // 方法 4: 检查 OpenCV 头文件
    if let Some(version) = try_header_check() {
        return version;
    }
    
    // 回退到默认版本
    let default_version = "4.0.0".to_string();
    println!("cargo:warning=Could not detect OpenCV version, assuming {}", default_version);
    default_version
}

fn try_pkg_config() -> Option<String> {
    // 尝试使用 pkg-config 来获取 OpenCV 版本
    let output = Command::new("pkg-config")
        .args(["--modversion", "opencv4"])
        .output()
        .ok()?;
    
    if output.status.success() {
        let version = String::from_utf8(output.stdout).ok()?;
        let version = version.trim().to_string();
        if !version.is_empty() {
            return Some(version);
        }
    }
    
    // 尝试 opencv 包（旧版本）
    let output = Command::new("pkg-config")
        .args(["--modversion", "opencv"])
        .output()
        .ok()?;
    
    if output.status.success() {
        let version = String::from_utf8(output.stdout).ok()?;
        let version = version.trim().to_string();
        if !version.is_empty() {
            return Some(version);
        }
    }
    
    None
}

fn try_opencv_version_command() -> Option<String> {
    // 尝试运行 opencv_version 命令
    let output = Command::new("opencv_version")
        .output()
        .ok()?;
    
    if output.status.success() {
        let version = String::from_utf8(output.stdout).ok()?;
        let version = version.trim().to_string();
        if !version.is_empty() {
            return Some(version);
        }
    }
    
    None
}

fn try_compile_check() -> Option<String> {
    // 创建一个临时目录
    let out_dir = env::var("OUT_DIR").ok()?;
    let temp_file = format!("{}/version_check.cpp", out_dir);
    let temp_exe = format!("{}/version_check", out_dir);
    
    // 写入一个简单的 C++ 程序来打印 OpenCV 版本
    std::fs::write(&temp_file, r#"
        #include <opencv2/core.hpp>
        #include <iostream>
        
        int main() {
            std::cout << CV_VERSION << std::endl;
            return 0;
        }
    "#).ok()?;
    
    // 尝试编译
    let status = Command::new("c++")
        .args([&temp_file, "-o", &temp_exe, "-lopencv_core"])
        .status()
        .ok()?;
    
    if !status.success() {
        return None;
    }
    
    // 运行程序获取版本
    let output = Command::new(&temp_exe)
        .output()
        .ok()?;
    
    if output.status.success() {
        let version = String::from_utf8(output.stdout).ok()?;
        let version = version.trim().to_string();
        if !version.is_empty() {
            return Some(version);
        }
    }
    
    None
}

fn try_header_check() -> Option<String> {
    // 尝试在常见位置查找 OpenCV 头文件
    let paths = [
        "/usr/include/opencv4/opencv2/core/version.hpp",
        "/usr/local/include/opencv4/opencv2/core/version.hpp",
        "/usr/include/opencv2/core/version.hpp",
        "/usr/local/include/opencv2/core/version.hpp",
        "/opt/homebrew/include/opencv4/opencv2/core/version.hpp",
        "/opt/homebrew/include/opencv2/core/version.hpp",
    ];
    
    for path in paths.iter() {
        if let Ok(content) = std::fs::read_to_string(path) {
            // 查找版本定义 #define CV_VERSION_MAJOR 4
            let mut major = None;
            let mut minor = None;
            let mut patch = None;
            
            for line in content.lines() {
                if let Some(value) = extract_define(line, "CV_VERSION_MAJOR") {
                    major = Some(value);
                } else if let Some(value) = extract_define(line, "CV_VERSION_MINOR") {
                    minor = Some(value);
                } else if let Some(value) = extract_define(line, "CV_VERSION_REVISION") {
                    patch = Some(value);
                }
                
                if major.is_some() && minor.is_some() && patch.is_some() {
                    return Some(format!("{}.{}.{}", major.unwrap(), minor.unwrap(), patch.unwrap()));
                }
            }
        }
    }
    
    None
}

fn extract_define(line: &str, define_name: &str) -> Option<String> {
    let line = line.trim();
    if line.starts_with("#define") && line.contains(define_name) {
        let parts: Vec<&str> = line.split_whitespace().collect();
        if parts.len() >= 3 && parts[1] == define_name {
            return Some(parts[2].to_string());
        }
    }
    None
}