Merge pull request #10 from vaaaaanquish/make_osx

Support osx

Author: vaaaaanquish

Cargo.toml

@@ -9,7 +9,7 @@ readme = "README.md"
 exclude = [".gitignore", ".gitmodules", "examples", "lightgbm-sys"]
 
 [dependencies]
-lightgbm-sys = "0.1.0"
+lightgbm-sys = "0.2.0"
 libc = "0.2.81"
 derive_builder = "0.5.1"
 serde_json = "1.0.59"

README.md

@@ -9,6 +9,9 @@ You need an environment that can build LightGBM.
 ```
 # linux
 apt install -y cmake libclang-dev libc++-dev gcc-multilib
+
+# OS X
+brew install cmake libomp
 ```
 
 Please see below for details.

lightgbm-sys/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "lightgbm-sys"
-version = "0.1.0"
+version = "0.2.0"
 authors = ["vaaaaanquish <[email protected]>"]
 build = "build.rs"
 license = "MIT"

lightgbm-sys/build.rs

@@ -8,6 +8,7 @@ use std::path::{Path, PathBuf};
 
 
 fn main() {
+    let target = env::var("TARGET").unwrap();
     let out_dir = env::var("OUT_DIR").unwrap();
     let lgbm_root = Path::new(&out_dir).join("lightgbm");
 
@@ -23,10 +24,10 @@ fn main() {
 
     // CMake
     let dst = Config::new(&lgbm_root)
-        .profile("Release")
-        .uses_cxx11()
-        .define("BUILD_STATIC_LIB", "ON")
-        .build();
+            .profile("Release")
+            .uses_cxx11()
+            .define("BUILD_STATIC_LIB", "ON")
+            .build();
 
     // bindgen build
     let bindings = bindgen::Builder::default()
@@ -40,9 +41,16 @@ fn main() {
         .write_to_file(out_path.join("bindings.rs"))
         .expect("Couldn't write bindings.");
 
+    // link to appropriate C++ lib
+    if target.contains("apple") {
+        println!("cargo:rustc-link-lib=c++");
+        println!("cargo:rustc-link-lib=dylib=omp");
+    } else {
+        println!("cargo:rustc-link-lib=stdc++");
+        println!("cargo:rustc-link-lib=dylib=gomp");
+    }
+
     println!("cargo:rustc-link-search={}", out_path.join("lib").display());
-    println!("cargo:rustc-link-lib=dylib=gomp");
-    println!("cargo:rustc-link-lib=stdc++");
     println!("cargo:rustc-link-search=native={}", dst.display());
     println!("cargo:rustc-link-lib=static=_lightgbm");
 }