80 lines
2.1 KiB
Rust
80 lines
2.1 KiB
Rust
use anyhow::*;
|
|
use fs_extra::{copy_items, dir::CopyOptions};
|
|
use glob::glob;
|
|
use rayon::prelude::*;
|
|
use std::path::PathBuf;
|
|
use std::{
|
|
env,
|
|
fs::{read_to_string, write},
|
|
};
|
|
|
|
struct ShaderData {
|
|
src: String,
|
|
src_path: PathBuf,
|
|
spv_path: PathBuf,
|
|
kind: shaderc::ShaderKind,
|
|
}
|
|
|
|
impl ShaderData {
|
|
pub fn load(src_path: PathBuf) -> Result<Self> {
|
|
let ext = src_path
|
|
.extension()
|
|
.context("File has no extension")?
|
|
.to_str()
|
|
.context("Extension cannot be converted to &str")?;
|
|
let kind = match ext {
|
|
"vert" => shaderc::ShaderKind::Vertex,
|
|
"frag" => shaderc::ShaderKind::Fragment,
|
|
"comp" => shaderc::ShaderKind::Compute,
|
|
_ => bail!("Unsupported shader: {}", src_path.display()),
|
|
};
|
|
|
|
let src = read_to_string(&src_path)?;
|
|
let spv_path = src_path.with_extension(format!("{}.spv", ext));
|
|
|
|
Ok(Self {
|
|
src,
|
|
src_path,
|
|
spv_path,
|
|
kind,
|
|
})
|
|
}
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
let mut shader_paths = Vec::new();
|
|
shader_paths.extend(glob("./src/**/*.vert")?);
|
|
shader_paths.extend(glob("./src/**/*.frag")?);
|
|
shader_paths.extend(glob("./src/**/*.comp")?);
|
|
|
|
let shaders: Result<Vec<_>> = shader_paths
|
|
.into_par_iter()
|
|
.map(|glob_result| ShaderData::load(glob_result?))
|
|
.collect();
|
|
|
|
let mut compiler = shaderc::Compiler::new().context("Unable to create shader compiler")?;
|
|
|
|
for shader in shaders? {
|
|
println!(
|
|
"cargo:rerun-if-changed={}",
|
|
&shader.src_path.as_os_str().to_str().unwrap()
|
|
);
|
|
let compiled = compiler.compile_into_spirv(
|
|
&shader.src,
|
|
shader.kind,
|
|
&shader.src_path.to_str().unwrap(),
|
|
"main",
|
|
None,
|
|
)?;
|
|
write(shader.spv_path, compiled.as_binary_u8())?;
|
|
}
|
|
|
|
println!("cargo:rerun-if-changed=res/*");
|
|
let out_dir = env::var("OUT_DIR")?;
|
|
let mut copy_options = CopyOptions::new();
|
|
copy_options.overwrite = true;
|
|
copy_items(&vec!["res/"], out_dir, ©_options)?;
|
|
|
|
Ok(())
|
|
}
|