tbb 0.6.0

Intel Threading Building Blocks
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use std::{
    env,
    fs::{self, File},
    io,
    path::{Path, PathBuf},
};

use cxx_build::CFG;
use flate2::read::GzDecoder;
use glob::glob;
use sha2::{Digest, Sha256};
use tar::Archive;

const URL: &str = "https://github.com/oneapi-src/oneTBB/archive/refs/tags/v2021.5.0.tar.gz";
const SHA256: &str = "e5b57537c741400cf6134b428fc1689a649d7d38d9bb9c1b6d64f092ea28178a";
const PREFIX: &str = "oneTBB-2021.5.0";

fn main() {
    let mut response = reqwest::blocking::get(URL).unwrap();
    let tmp_dir = tempfile::tempdir().unwrap();
    let filename = URL.split('/').last().unwrap();
    let tmp_path = tmp_dir.path().join(filename);
    let mut tmp_file = File::create(&tmp_path).unwrap();

    response.copy_to(&mut tmp_file).unwrap();

    let mut tmp_file = File::open(&tmp_path).unwrap();

    let mut hasher = Sha256::new();
    let _ = io::copy(&mut tmp_file, &mut hasher).unwrap();
    let hash = hasher.finalize();
    let hex_hash = base16ct::lower::encode_string(&hash);
    assert_eq!(hex_hash, SHA256);

    let out_dir = env::var("OUT_DIR").unwrap();
    let out_dir = Path::new(&out_dir);
    let tar_path = out_dir.join(filename);

    fs::rename(&tmp_path, &tar_path).unwrap();
    println!("Downloaded: {}", tar_path.display());

    let tar_gz = File::open(tar_path).unwrap();
    let tar = GzDecoder::new(tar_gz);
    let mut archive = Archive::new(tar);
    archive.unpack(out_dir).unwrap();

    let src_dir = out_dir.join(PREFIX);
    let inc_dir = src_dir.join("include");
    let srcs: Vec<PathBuf> = glob(&format!("{}/src/tbb/*.cpp", src_dir.display()))
        .unwrap()
        .map(|x| x.unwrap())
        .collect();

    CFG.exported_header_dirs = vec![&inc_dir];

    let mut build = cxx_build::bridge("src/lib.rs");
    build
        .files(srcs)
        .include(&inc_dir)
        .define("__TBB_BUILD", None)
        .define("__TBB_NO_IMPLICIT_LINKAGE", None)
        // MSVC flags
        .flag_if_supported("/std:c++17")
        .flag_if_supported("/Zc:preprocessor")
        .flag_if_supported("/EHsc")
        // GCC/Clang flags
        .flag_if_supported("-w")
        .flag_if_supported("-std=c++17")
        .warnings(false);

    if cfg!(target_os = "macos") {
        build.define("_XOPEN_SOURCE", None);
    }

    if cfg!(target_arch = "x86_64") {
        build.define("__TBB_USE_ITT_NOTIFY", None);
    } else {
        build.define("USE_PTHREAD", None);
    }

    if cfg!(all(target_os = "linux", target_arch = "x86_64")) {
        build.flag_if_supported("-mwaitpkg");
    }

    build.compile("tbb");
}