1use std::{
2 env,
3 path::{Path, PathBuf},
4 process::{self, Command},
5};
6
7use cbindgen::Config;
8
9fn main() {
10 generate_dispatch_bindings();
11 let header_path = generate_shader_bindings();
12 compile_metal_shaders(&header_path);
13}
14
15fn generate_dispatch_bindings() {
16 println!("cargo:rustc-link-lib=framework=System");
17 println!("cargo:rerun-if-changed=src/platform/mac/dispatch.h");
18
19 let bindings = bindgen::Builder::default()
20 .header("src/platform/mac/dispatch.h")
21 .allowlist_var("_dispatch_main_q")
22 .allowlist_var("DISPATCH_QUEUE_PRIORITY_DEFAULT")
23 .allowlist_function("dispatch_get_global_queue")
24 .allowlist_function("dispatch_async_f")
25 .parse_callbacks(Box::new(bindgen::CargoCallbacks))
26 .layout_tests(false)
27 .generate()
28 .expect("unable to generate bindings");
29
30 let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
31 bindings
32 .write_to_file(out_path.join("dispatch_sys.rs"))
33 .expect("couldn't write dispatch bindings");
34}
35
36fn generate_shader_bindings() -> PathBuf {
37 let output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("scene.h");
38 let crate_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap());
39 let mut config = Config::default();
40 config.include_guard = Some("SCENE_H".into());
41 config.language = cbindgen::Language::C;
42 config.export.include.extend([
43 "Bounds".into(),
44 "Corners".into(),
45 "Edges".into(),
46 "Size".into(),
47 "Pixels".into(),
48 "PointF".into(),
49 "Hsla".into(),
50 "ContentMask".into(),
51 "Uniforms".into(),
52 "AtlasTile".into(),
53 "PathRasterizationInputIndex".into(),
54 "PathVertex_ScaledPixels".into(),
55 "ShadowInputIndex".into(),
56 "Shadow".into(),
57 "QuadInputIndex".into(),
58 "Underline".into(),
59 "UnderlineInputIndex".into(),
60 "Quad".into(),
61 "SpriteInputIndex".into(),
62 "MonochromeSprite".into(),
63 "PolychromeSprite".into(),
64 "PathSprite".into(),
65 ]);
66 config.no_includes = true;
67 config.enumeration.prefix_with_name = true;
68 cbindgen::Builder::new()
69 .with_src(crate_dir.join("src/scene.rs"))
70 .with_src(crate_dir.join("src/geometry.rs"))
71 .with_src(crate_dir.join("src/color.rs"))
72 .with_src(crate_dir.join("src/window.rs"))
73 .with_src(crate_dir.join("src/platform.rs"))
74 .with_src(crate_dir.join("src/platform/mac/metal_renderer.rs"))
75 .with_config(config)
76 .generate()
77 .expect("Unable to generate bindings")
78 .write_to_file(&output_path);
79
80 output_path
81}
82
83fn compile_metal_shaders(header_path: &Path) {
84 let shader_path = "./src/platform/mac/shaders.metal";
85 let air_output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("shaders.air");
86 let metallib_output_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("shaders.metallib");
87
88 println!("cargo:rerun-if-changed={}", header_path.display());
89 println!("cargo:rerun-if-changed={}", shader_path);
90
91 let output = Command::new("xcrun")
92 .args([
93 "-sdk",
94 "macosx",
95 "metal",
96 "-gline-tables-only",
97 "-mmacosx-version-min=10.15.7",
98 "-MO",
99 "-c",
100 shader_path,
101 "-include",
102 &header_path.to_str().unwrap(),
103 "-o",
104 ])
105 .arg(&air_output_path)
106 .output()
107 .unwrap();
108
109 if !output.status.success() {
110 eprintln!(
111 "metal shader compilation failed:\n{}",
112 String::from_utf8_lossy(&output.stderr)
113 );
114 process::exit(1);
115 }
116
117 let output = Command::new("xcrun")
118 .args(["-sdk", "macosx", "metallib"])
119 .arg(air_output_path)
120 .arg("-o")
121 .arg(metallib_output_path)
122 .output()
123 .unwrap();
124
125 if !output.status.success() {
126 eprintln!(
127 "metallib compilation failed:\n{}",
128 String::from_utf8_lossy(&output.stderr)
129 );
130 process::exit(1);
131 }
132}