From a3f3bb4751cdc7db143e22376119d9346897cf88 Mon Sep 17 00:00:00 2001
From: 21pages <pages21@163.com>
Date: Mon, 8 May 2023 19:57:32 +0800
Subject: [PATCH] aom vcpkg binding

Signed-off-by: 21pages <pages21@163.com>
---
 libs/scrap/build.rs                     | 26 +++++++++++++------------
 libs/scrap/src/bindings/aom_ffi.h       |  8 ++++++++
 libs/scrap/{ => src/bindings}/vpx_ffi.h | 16 +++++++--------
 3 files changed, 30 insertions(+), 20 deletions(-)
 create mode 100644 libs/scrap/src/bindings/aom_ffi.h
 rename libs/scrap/{ => src/bindings}/vpx_ffi.h (96%)

diff --git a/libs/scrap/build.rs b/libs/scrap/build.rs
index 3e08469b0..72fe0707a 100644
--- a/libs/scrap/build.rs
+++ b/libs/scrap/build.rs
@@ -144,16 +144,17 @@ fn generate_bindings(
     include_paths: &[PathBuf],
     ffi_rs: &Path,
     exact_file: &Path,
+    regex: &str,
 ) {
     let mut b = bindgen::builder()
         .header(ffi_header.to_str().unwrap())
-        .allowlist_type("^[vV].*")
-        .allowlist_var("^[vV].*")
-        .allowlist_function("^[vV].*")
-        .rustified_enum("^v.*")
+        .allowlist_type(regex)
+        .allowlist_var(regex)
+        .allowlist_function(regex)
+        .rustified_enum(regex)
         .trust_clang_mangling(false)
         .layout_tests(false) // breaks 32/64-bit compat
-        .generate_comments(false); // vpx comments have prefix /*!\
+        .generate_comments(false); // comments have prefix /*!\
 
     for dir in include_paths {
         b = b.clang_arg(format!("-I{}", dir.display()));
@@ -163,22 +164,22 @@ fn generate_bindings(
     fs::copy(ffi_rs, exact_file).ok(); // ignore failure
 }
 
-fn gen_vpx() {
-    let includes = find_package("libvpx");
+fn gen_vcpkg_package(package: &str, ffi_header: &str, generated: &str, regex: &str) {
+    let includes = find_package(package);
     let src_dir = env::var_os("CARGO_MANIFEST_DIR").unwrap();
     let src_dir = Path::new(&src_dir);
     let out_dir = env::var_os("OUT_DIR").unwrap();
     let out_dir = Path::new(&out_dir);
 
-    let ffi_header = src_dir.join("vpx_ffi.h");
+    let ffi_header = src_dir.join("src").join("bindings").join(ffi_header);
     println!("rerun-if-changed={}", ffi_header.display());
     for dir in &includes {
         println!("rerun-if-changed={}", dir.display());
     }
 
-    let ffi_rs = out_dir.join("vpx_ffi.rs");
-    let exact_file = src_dir.join("generated").join("vpx_ffi.rs");
-    generate_bindings(&ffi_header, &includes, &ffi_rs, &exact_file);
+    let ffi_rs = out_dir.join(generated);
+    let exact_file = src_dir.join("generated").join(generated);
+    generate_bindings(&ffi_header, &includes, &ffi_rs, &exact_file, regex);
 }
 
 fn main() {
@@ -194,7 +195,8 @@ fn main() {
     env::set_var("CARGO_CFG_TARGET_FEATURE", "crt-static");
 
     find_package("libyuv");
-    gen_vpx();
+    gen_vcpkg_package("libvpx", "vpx_ffi.h", "vpx_ffi.rs", "^[vV].*");
+    gen_vcpkg_package("aom", "aom_ffi.h", "aom_ffi.rs", "^(aom|AOM|OBU|AV1).*");
 
     // there is problem with cfg(target_os) in build.rs, so use our workaround
     let target_os = std::env::var("CARGO_CFG_TARGET_OS").unwrap();
diff --git a/libs/scrap/src/bindings/aom_ffi.h b/libs/scrap/src/bindings/aom_ffi.h
new file mode 100644
index 000000000..3b7babf2b
--- /dev/null
+++ b/libs/scrap/src/bindings/aom_ffi.h
@@ -0,0 +1,8 @@
+#include <aom/aom.h>
+#include <aom/aom_image.h>
+#include <aom/aom_integer.h> 
+#include <aom/aom_codec.h>
+#include <aom/aom_external_partition.h>
+#include <aom/aom_frame_buffer.h> 
+#include <aom/aom_encoder.h>
+#include <aom/aomcx.h> 
diff --git a/libs/scrap/vpx_ffi.h b/libs/scrap/src/bindings/vpx_ffi.h
similarity index 96%
rename from libs/scrap/vpx_ffi.h
rename to libs/scrap/src/bindings/vpx_ffi.h
index 3db471e8f..cd44f9865 100644
--- a/libs/scrap/vpx_ffi.h
+++ b/libs/scrap/src/bindings/vpx_ffi.h
@@ -1,9 +1,9 @@
-#include <vpx/vp8.h>
-#include <vpx/vp8cx.h>
-#include <vpx/vp8dx.h>
-#include <vpx/vpx_codec.h>
-#include <vpx/vpx_decoder.h>
-#include <vpx/vpx_encoder.h>
-#include <vpx/vpx_frame_buffer.h>
-#include <vpx/vpx_image.h>
+#include <vpx/vp8.h>
+#include <vpx/vp8cx.h>
+#include <vpx/vp8dx.h>
+#include <vpx/vpx_codec.h>
+#include <vpx/vpx_decoder.h>
+#include <vpx/vpx_encoder.h>
+#include <vpx/vpx_frame_buffer.h>
+#include <vpx/vpx_image.h>
 #include <vpx/vpx_integer.h>
\ No newline at end of file