mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2026-01-28 16:41:52 +08:00
init
This commit is contained in:
mofeng-git
270
res/vcpkg/libyuv/build.rs
Normal file
270
res/vcpkg/libyuv/build.rs
Normal file
@@ -0,0 +1,270 @@
|
||||
use std::{
|
||||
env,
|
||||
path::{Path, PathBuf},
|
||||
process::Command,
|
||||
};
|
||||
|
||||
fn main() {
|
||||
let manifest_dir = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
|
||||
let cpp_dir = manifest_dir.join("cpp");
|
||||
|
||||
println!("cargo:rerun-if-changed=cpp/yuv_ffi.h");
|
||||
|
||||
// Generate FFI bindings
|
||||
generate_bindings(&cpp_dir);
|
||||
|
||||
// Link libyuv library
|
||||
link_libyuv();
|
||||
}
|
||||
|
||||
fn generate_bindings(cpp_dir: &Path) {
|
||||
let ffi_header = cpp_dir.join("yuv_ffi.h");
|
||||
|
||||
bindgen::builder()
|
||||
.header(ffi_header.to_string_lossy().to_string())
|
||||
// YUYV conversions
|
||||
.allowlist_function("YUY2ToI420")
|
||||
.allowlist_function("YUY2ToNV12")
|
||||
// UYVY conversions
|
||||
.allowlist_function("UYVYToI420")
|
||||
.allowlist_function("UYVYToNV12")
|
||||
// I420 conversions
|
||||
.allowlist_function("I420ToNV12")
|
||||
.allowlist_function("I420ToNV21")
|
||||
.allowlist_function("I420Copy")
|
||||
// I422 conversions
|
||||
.allowlist_function("I422ToI420")
|
||||
// NV12/NV21 conversions
|
||||
.allowlist_function("NV12ToI420")
|
||||
.allowlist_function("NV21ToI420")
|
||||
.allowlist_function("NV12Copy")
|
||||
// ARGB/BGRA conversions
|
||||
.allowlist_function("ARGBToI420")
|
||||
.allowlist_function("ARGBToNV12")
|
||||
.allowlist_function("ABGRToI420")
|
||||
.allowlist_function("ABGRToNV12")
|
||||
.allowlist_function("ARGBToABGR")
|
||||
.allowlist_function("ABGRToARGB")
|
||||
// RGB24/BGR24 conversions
|
||||
.allowlist_function("RGB24ToI420")
|
||||
.allowlist_function("RAWToI420")
|
||||
.allowlist_function("RGB24ToARGB")
|
||||
.allowlist_function("RAWToARGB")
|
||||
// YUV to RGB conversions
|
||||
.allowlist_function("I420ToRGB24")
|
||||
.allowlist_function("I420ToARGB")
|
||||
.allowlist_function("NV12ToRGB24")
|
||||
.allowlist_function("NV12ToARGB")
|
||||
.allowlist_function("YUY2ToARGB")
|
||||
.allowlist_function("UYVYToARGB")
|
||||
.allowlist_function("ARGBToRGB24")
|
||||
.allowlist_function("ARGBToRAW")
|
||||
// MJPEG decoding
|
||||
.allowlist_function("MJPGToI420")
|
||||
.allowlist_function("MJPGToNV12")
|
||||
.allowlist_function("MJPGToARGB")
|
||||
.allowlist_function("MJPGSize")
|
||||
// Scaling
|
||||
.allowlist_function("I420Scale")
|
||||
.allowlist_function("NV12Scale")
|
||||
.allowlist_function("ARGBScale")
|
||||
// Rotation
|
||||
.allowlist_function("I420Rotate")
|
||||
.allowlist_function("NV12ToI420Rotate")
|
||||
// Enums
|
||||
.allowlist_type("FilterMode")
|
||||
.allowlist_type("RotationMode")
|
||||
.rustified_enum("FilterMode")
|
||||
.rustified_enum("RotationMode")
|
||||
.generate()
|
||||
.expect("Failed to generate libyuv bindings")
|
||||
.write_to_file(Path::new(&env::var_os("OUT_DIR").unwrap()).join("yuv_ffi.rs"))
|
||||
.expect("Failed to write yuv_ffi.rs");
|
||||
}
|
||||
|
||||
fn link_libyuv() {
|
||||
// Try vcpkg first
|
||||
if let Ok(vcpkg_root) = env::var("VCPKG_ROOT") {
|
||||
if link_vcpkg(vcpkg_root.into()) {
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
// Try pkg-config
|
||||
if link_pkg_config() {
|
||||
return;
|
||||
}
|
||||
|
||||
// Try system library directly
|
||||
if link_system() {
|
||||
return;
|
||||
}
|
||||
|
||||
panic!(
|
||||
"libyuv not found!\n\
|
||||
\n\
|
||||
Install via one of:\n\
|
||||
- vcpkg: vcpkg install libyuv && export VCPKG_ROOT=/path/to/vcpkg\n\
|
||||
- apt (Debian/Ubuntu): sudo apt install libyuv-dev\n\
|
||||
- dnf (Fedora): sudo dnf install libyuv-devel\n\
|
||||
- pacman (Arch): sudo pacman -S libyuv\n"
|
||||
);
|
||||
}
|
||||
|
||||
fn link_vcpkg(mut path: PathBuf) -> bool {
|
||||
let target_arch = env::var("CARGO_CFG_TARGET_ARCH").unwrap_or_default();
|
||||
let target_os = env::var("CARGO_CFG_TARGET_OS").unwrap_or_default();
|
||||
|
||||
let triplet = match (target_os.as_str(), target_arch.as_str()) {
|
||||
("linux", "x86_64") => "x64-linux",
|
||||
("linux", "aarch64") => "arm64-linux",
|
||||
("linux", "arm") => "arm-linux",
|
||||
("windows", "x86_64") => "x64-windows-static",
|
||||
("windows", "x86") => "x86-windows-static",
|
||||
("macos", "x86_64") => "x64-osx",
|
||||
("macos", "aarch64") => "arm64-osx",
|
||||
_ => {
|
||||
println!(
|
||||
"cargo:warning=Unsupported vcpkg target: {}-{}",
|
||||
target_os, target_arch
|
||||
);
|
||||
return false;
|
||||
}
|
||||
};
|
||||
|
||||
path.push("installed");
|
||||
path.push(triplet);
|
||||
|
||||
let include_path = path.join("include");
|
||||
let lib_path = path.join("lib");
|
||||
|
||||
if !lib_path.exists() {
|
||||
println!(
|
||||
"cargo:warning=vcpkg libyuv not found at: {}",
|
||||
lib_path.display()
|
||||
);
|
||||
return false;
|
||||
}
|
||||
|
||||
println!("cargo:rustc-link-search=native={}", lib_path.display());
|
||||
|
||||
// For Linux: use dynamic linking
|
||||
#[cfg(target_os = "linux")]
|
||||
{
|
||||
println!("cargo:rustc-link-lib=yuv");
|
||||
println!("cargo:rustc-link-lib=jpeg");
|
||||
println!("cargo:rustc-link-lib=stdc++");
|
||||
}
|
||||
|
||||
// For Windows/macOS: keep static linking
|
||||
#[cfg(not(target_os = "linux"))]
|
||||
{
|
||||
let static_lib = lib_path.join("libyuv.a");
|
||||
let static_lib_win = lib_path.join("yuv.lib");
|
||||
|
||||
if static_lib.exists() || static_lib_win.exists() {
|
||||
println!("cargo:rustc-link-lib=static=yuv");
|
||||
} else {
|
||||
println!("cargo:rustc-link-lib=yuv");
|
||||
}
|
||||
|
||||
let jpeg_static = lib_path.join("libjpeg.a");
|
||||
let jpeg_static_win = lib_path.join("jpeg.lib");
|
||||
let turbojpeg_static = lib_path.join("libturbojpeg.a");
|
||||
|
||||
if turbojpeg_static.exists() {
|
||||
println!("cargo:rustc-link-lib=static=turbojpeg");
|
||||
} else if jpeg_static.exists() || jpeg_static_win.exists() {
|
||||
println!("cargo:rustc-link-lib=static=jpeg");
|
||||
} else {
|
||||
println!("cargo:rustc-link-lib=jpeg");
|
||||
}
|
||||
|
||||
#[cfg(target_os = "windows")]
|
||||
println!("cargo:rustc-link-lib=stdc++");
|
||||
}
|
||||
|
||||
println!(
|
||||
"cargo:info=Using libyuv from vcpkg: {}",
|
||||
include_path.display()
|
||||
);
|
||||
true
|
||||
}
|
||||
|
||||
fn link_pkg_config() -> bool {
|
||||
let output = match Command::new("pkg-config")
|
||||
.args(["--libs", "--cflags", "libyuv"])
|
||||
.output()
|
||||
{
|
||||
Ok(o) => o,
|
||||
Err(_) => return false,
|
||||
};
|
||||
|
||||
if !output.status.success() {
|
||||
return false;
|
||||
}
|
||||
|
||||
let flags = String::from_utf8_lossy(&output.stdout);
|
||||
for flag in flags.split_whitespace() {
|
||||
if flag.starts_with("-L") {
|
||||
println!("cargo:rustc-link-search=native={}", &flag[2..]);
|
||||
} else if flag.starts_with("-l") {
|
||||
println!("cargo:rustc-link-lib={}", &flag[2..]);
|
||||
}
|
||||
}
|
||||
|
||||
// Also need libjpeg
|
||||
if let Ok(jpeg_output) = Command::new("pkg-config")
|
||||
.args(["--libs", "libjpeg"])
|
||||
.output()
|
||||
{
|
||||
if jpeg_output.status.success() {
|
||||
let jpeg_flags = String::from_utf8_lossy(&jpeg_output.stdout);
|
||||
for flag in jpeg_flags.split_whitespace() {
|
||||
if flag.starts_with("-L") {
|
||||
println!("cargo:rustc-link-search=native={}", &flag[2..]);
|
||||
} else if flag.starts_with("-l") {
|
||||
println!("cargo:rustc-link-lib={}", &flag[2..]);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(target_os = "linux")]
|
||||
println!("cargo:rustc-link-lib=stdc++");
|
||||
|
||||
println!("cargo:info=Using libyuv from pkg-config (dynamic linking)");
|
||||
true
|
||||
}
|
||||
|
||||
fn link_system() -> bool {
|
||||
// Try common system library paths (dynamic linking only)
|
||||
let lib_paths = [
|
||||
"/usr/lib",
|
||||
"/usr/lib64",
|
||||
"/usr/local/lib",
|
||||
"/usr/local/lib64",
|
||||
"/usr/lib/x86_64-linux-gnu", // Debian/Ubuntu x86_64
|
||||
"/usr/lib/aarch64-linux-gnu", // Debian/Ubuntu ARM64
|
||||
"/usr/lib/arm-linux-gnueabihf", // Debian/Ubuntu ARMv7
|
||||
];
|
||||
|
||||
for path in &lib_paths {
|
||||
let lib_path = Path::new(path);
|
||||
let libyuv_so = lib_path.join("libyuv.so");
|
||||
|
||||
if libyuv_so.exists() {
|
||||
println!("cargo:rustc-link-search=native={}", path);
|
||||
println!("cargo:rustc-link-lib=yuv");
|
||||
println!("cargo:rustc-link-lib=jpeg");
|
||||
|
||||
#[cfg(target_os = "linux")]
|
||||
println!("cargo:rustc-link-lib=stdc++");
|
||||
|
||||
println!("cargo:info=Using system libyuv from {} (dynamic linking)", path);
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
false
|
||||
}
|
||||
Reference in New Issue
Block a user