Lightningbeam/nam-ffi/build.rs

46 lines
1.8 KiB
Rust

use std::env;
use std::path::Path;
/// Canonicalize a path without the \\?\ prefix on Windows
fn clean_canonicalize(p: &Path) -> String {
let canon = p.canonicalize().unwrap();
let s = canon.to_str().unwrap();
s.strip_prefix(r"\\?\").unwrap_or(s).to_string()
}
fn main() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
let wrapper_dir = Path::new(&manifest_dir).join("cmake");
let neural_audio_dir = Path::new(&manifest_dir).join("../vendor/NeuralAudio");
let dst = cmake::Config::new(&wrapper_dir)
.define("CMAKE_BUILD_TYPE", "Release")
.define("NEURALAUDIO_SOURCE_DIR", clean_canonicalize(&neural_audio_dir))
.define("BUILD_NAMCORE", "OFF")
.define("BUILD_STATIC_RTNEURAL", "OFF")
.define("BUILD_UTILS", "OFF")
.define("WAVENET_FRAMES", "64")
.define("WAVENET_MATH", "FastMath")
.define("LSTM_MATH", "FastMath")
.build_target("NeuralAudioCAPI")
.build();
let build_dir = dst.join("build");
println!("cargo:rustc-link-search=native={}", build_dir.join("NeuralAudio").join("NeuralAudioCAPI").display());
println!("cargo:rustc-link-search=native={}", build_dir.join("NeuralAudio").join("NeuralAudio").display());
println!("cargo:rustc-link-lib=static=NeuralAudioCAPI");
println!("cargo:rustc-link-lib=static=NeuralAudio");
let target_os = env::var("CARGO_CFG_TARGET_OS").unwrap_or_default();
match target_os.as_str() {
"macos" => println!("cargo:rustc-link-lib=c++"),
"linux" => println!("cargo:rustc-link-lib=stdc++"),
_ => {}
}
println!("cargo:rerun-if-changed=../vendor/NeuralAudio/NeuralAudioCAPI/NeuralAudioCApi.h");
println!("cargo:rerun-if-changed=../vendor/NeuralAudio/NeuralAudioCAPI/NeuralAudioCApi.cpp");
}