This commit is contained in:
mofeng-git
2025-12-28 18:19:16 +08:00
commit d143d158e4
771 changed files with 220548 additions and 0 deletions

View File

@@ -0,0 +1,12 @@
[package]
name = "libyuv"
version = "0.1.0"
edition = "2021"
description = "libyuv bindings for high-performance pixel format conversion"
license = "BSD-3-Clause"
[dependencies]
[build-dependencies]
cc = "1.0"
bindgen = "0.59"

270
res/vcpkg/libyuv/build.rs Normal file
View 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
}

View File

@@ -0,0 +1,337 @@
#ifndef YUV_FFI_H
#define YUV_FFI_H
#include <stdint.h>
#include <stddef.h>
#ifdef __cplusplus
extern "C" {
#endif
// ============================================================================
// libyuv function declarations
// These are the most commonly used functions for video format conversion
// ============================================================================
// ----------------------------------------------------------------------------
// YUYV (YUY2) conversions - Common for USB capture cards
// ----------------------------------------------------------------------------
// YUYV -> I420 (YUV420P planar)
int YUY2ToI420(const uint8_t* src_yuy2, int src_stride_yuy2,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
// YUYV -> NV12 (optimal for VAAPI)
int YUY2ToNV12(const uint8_t* src_yuy2, int src_stride_yuy2,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_uv, int dst_stride_uv,
int width, int height);
// ----------------------------------------------------------------------------
// UYVY conversions
// ----------------------------------------------------------------------------
int UYVYToI420(const uint8_t* src_uyvy, int src_stride_uyvy,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
int UYVYToNV12(const uint8_t* src_uyvy, int src_stride_uyvy,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_uv, int dst_stride_uv,
int width, int height);
// ----------------------------------------------------------------------------
// I420 (YUV420P) conversions
// ----------------------------------------------------------------------------
// I422 (YUV422P) -> I420 (YUV420P) with vertical chroma downsampling
int I422ToI420(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_u, int src_stride_u,
const uint8_t* src_v, int src_stride_v,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
// I420 -> NV12
int I420ToNV12(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_u, int src_stride_u,
const uint8_t* src_v, int src_stride_v,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_uv, int dst_stride_uv,
int width, int height);
// I420 -> NV21
int I420ToNV21(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_u, int src_stride_u,
const uint8_t* src_v, int src_stride_v,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_vu, int dst_stride_vu,
int width, int height);
// ----------------------------------------------------------------------------
// NV12/NV21 conversions
// ----------------------------------------------------------------------------
// NV12 -> I420
int NV12ToI420(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_uv, int src_stride_uv,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
// NV21 -> I420
int NV21ToI420(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_vu, int src_stride_vu,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
// ----------------------------------------------------------------------------
// ARGB/BGRA conversions (32-bit RGB)
// Note: libyuv uses ARGB to mean BGRA in memory (little-endian)
// ----------------------------------------------------------------------------
// BGRA -> I420
int ARGBToI420(const uint8_t* src_argb, int src_stride_argb,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
// BGRA -> NV12
int ARGBToNV12(const uint8_t* src_argb, int src_stride_argb,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_uv, int dst_stride_uv,
int width, int height);
// RGBA -> I420
int ABGRToI420(const uint8_t* src_abgr, int src_stride_abgr,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
// RGBA -> NV12
int ABGRToNV12(const uint8_t* src_abgr, int src_stride_abgr,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_uv, int dst_stride_uv,
int width, int height);
// ARGB <-> ABGR swap (BGRA <-> RGBA)
int ARGBToABGR(const uint8_t* src_argb, int src_stride_argb,
uint8_t* dst_abgr, int dst_stride_abgr,
int width, int height);
int ABGRToARGB(const uint8_t* src_abgr, int src_stride_abgr,
uint8_t* dst_argb, int dst_stride_argb,
int width, int height);
// ----------------------------------------------------------------------------
// RGB24/BGR24 conversions (24-bit RGB)
// ----------------------------------------------------------------------------
// RGB24 -> I420
int RGB24ToI420(const uint8_t* src_rgb24, int src_stride_rgb24,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
// BGR24 (RAW) -> I420
int RAWToI420(const uint8_t* src_raw, int src_stride_raw,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
// RGB24 -> ARGB
int RGB24ToARGB(const uint8_t* src_rgb24, int src_stride_rgb24,
uint8_t* dst_argb, int dst_stride_argb,
int width, int height);
// BGR24 (RAW) -> ARGB
int RAWToARGB(const uint8_t* src_raw, int src_stride_raw,
uint8_t* dst_argb, int dst_stride_argb,
int width, int height);
// ----------------------------------------------------------------------------
// YUV to RGB conversions (for display/JPEG encoding)
// ----------------------------------------------------------------------------
// I420 -> RGB24
int I420ToRGB24(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_u, int src_stride_u,
const uint8_t* src_v, int src_stride_v,
uint8_t* dst_rgb24, int dst_stride_rgb24,
int width, int height);
// I420 -> ARGB (BGRA)
int I420ToARGB(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_u, int src_stride_u,
const uint8_t* src_v, int src_stride_v,
uint8_t* dst_argb, int dst_stride_argb,
int width, int height);
// NV12 -> RGB24
int NV12ToRGB24(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_uv, int src_stride_uv,
uint8_t* dst_rgb24, int dst_stride_rgb24,
int width, int height);
// NV12 -> ARGB (BGRA)
int NV12ToARGB(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_uv, int src_stride_uv,
uint8_t* dst_argb, int dst_stride_argb,
int width, int height);
// YUYV -> ARGB (BGRA)
int YUY2ToARGB(const uint8_t* src_yuy2, int src_stride_yuy2,
uint8_t* dst_argb, int dst_stride_argb,
int width, int height);
// UYVY -> ARGB (BGRA)
int UYVYToARGB(const uint8_t* src_uyvy, int src_stride_uyvy,
uint8_t* dst_argb, int dst_stride_argb,
int width, int height);
// ARGB -> RGB24
int ARGBToRGB24(const uint8_t* src_argb, int src_stride_argb,
uint8_t* dst_rgb24, int dst_stride_rgb24,
int width, int height);
// ARGB -> RAW (BGR24)
int ARGBToRAW(const uint8_t* src_argb, int src_stride_argb,
uint8_t* dst_raw, int dst_stride_raw,
int width, int height);
// ----------------------------------------------------------------------------
// MJPEG decoding (libyuv built-in, faster than FFmpeg for simple cases)
// ----------------------------------------------------------------------------
// MJPEG -> I420
int MJPGToI420(const uint8_t* sample, size_t sample_size,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int src_width, int src_height,
int dst_width, int dst_height);
// MJPEG -> NV12
int MJPGToNV12(const uint8_t* sample, size_t sample_size,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_uv, int dst_stride_uv,
int src_width, int src_height,
int dst_width, int dst_height);
// MJPEG -> ARGB
int MJPGToARGB(const uint8_t* sample, size_t sample_size,
uint8_t* dst_argb, int dst_stride_argb,
int src_width, int src_height,
int dst_width, int dst_height);
// Get MJPEG dimensions without decoding
int MJPGSize(const uint8_t* sample, size_t sample_size,
int* width, int* height);
// ----------------------------------------------------------------------------
// Scaling
// ----------------------------------------------------------------------------
// Scale filter modes
enum FilterMode {
kFilterNone = 0, // Point sample; Fastest
kFilterLinear = 1, // Filter horizontally only
kFilterBilinear = 2, // Faster than box, but lower quality scaling down
kFilterBox = 3 // Highest quality
};
// I420 scale
int I420Scale(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_u, int src_stride_u,
const uint8_t* src_v, int src_stride_v,
int src_width, int src_height,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int dst_width, int dst_height,
enum FilterMode filtering);
// NV12 scale
int NV12Scale(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_uv, int src_stride_uv,
int src_width, int src_height,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_uv, int dst_stride_uv,
int dst_width, int dst_height,
enum FilterMode filtering);
// ARGB scale
int ARGBScale(const uint8_t* src_argb, int src_stride_argb,
int src_width, int src_height,
uint8_t* dst_argb, int dst_stride_argb,
int dst_width, int dst_height,
enum FilterMode filtering);
// ----------------------------------------------------------------------------
// Rotation
// ----------------------------------------------------------------------------
enum RotationMode {
kRotate0 = 0,
kRotate90 = 90,
kRotate180 = 180,
kRotate270 = 270
};
// I420 rotate
int I420Rotate(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_u, int src_stride_u,
const uint8_t* src_v, int src_stride_v,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height,
enum RotationMode mode);
// NV12 rotate
int NV12ToI420Rotate(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_uv, int src_stride_uv,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height,
enum RotationMode mode);
// ----------------------------------------------------------------------------
// Copy functions
// ----------------------------------------------------------------------------
// Copy I420
int I420Copy(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_u, int src_stride_u,
const uint8_t* src_v, int src_stride_v,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_u, int dst_stride_u,
uint8_t* dst_v, int dst_stride_v,
int width, int height);
// Copy NV12
int NV12Copy(const uint8_t* src_y, int src_stride_y,
const uint8_t* src_uv, int src_stride_uv,
uint8_t* dst_y, int dst_stride_y,
uint8_t* dst_uv, int dst_stride_uv,
int width, int height);
#ifdef __cplusplus
}
#endif
#endif // YUV_FFI_H

1298
res/vcpkg/libyuv/src/lib.rs Normal file

File diff suppressed because it is too large Load Diff