54 lines
1.9 KiB
Rust
54 lines
1.9 KiB
Rust
use serde::{Deserialize, Serialize};
|
|
use std::io::Write;
|
|
|
|
pub fn compile_spirv(network: &graph_craft::document::NodeNetwork, input_type: &str, output_type: &str, compile_dir: Option<&str>, manifest_path: &str) -> anyhow::Result<Vec<u8>> {
|
|
let serialized_graph = serde_json::to_string(&network)?;
|
|
let features = "";
|
|
#[cfg(feature = "profiling")]
|
|
let features = "profiling";
|
|
|
|
println!("calling cargo run!");
|
|
let non_cargo_env_vars = std::env::vars().filter(|(k, _)| k.starts_with("PATH")).collect::<Vec<_>>();
|
|
let mut cargo_command = std::process::Command::new("/usr/bin/cargo")
|
|
.arg("run")
|
|
.arg("--release")
|
|
.arg("--manifest-path")
|
|
.arg(manifest_path)
|
|
.current_dir(manifest_path.replace("Cargo.toml", ""))
|
|
.env_clear()
|
|
.envs(non_cargo_env_vars)
|
|
.arg("--features")
|
|
.arg(features)
|
|
.arg("--")
|
|
.arg(input_type)
|
|
.arg(output_type)
|
|
// TODO: handle None case properly
|
|
.arg(compile_dir.unwrap())
|
|
.stdin(std::process::Stdio::piped())
|
|
.stdout(std::process::Stdio::piped())
|
|
.spawn()?;
|
|
|
|
cargo_command.stdin.as_mut().unwrap().write_all(serialized_graph.as_bytes())?;
|
|
let output = cargo_command.wait_with_output()?;
|
|
if !output.status.success() {
|
|
return Err(anyhow::anyhow!("cargo failed: {}", String::from_utf8_lossy(&output.stderr)));
|
|
}
|
|
Ok(output.stdout)
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug, Clone, PartialEq)]
|
|
pub struct CompileRequest {
|
|
network: graph_craft::document::NodeNetwork,
|
|
input_type: String,
|
|
output_type: String,
|
|
}
|
|
|
|
impl CompileRequest {
|
|
pub fn new(network: graph_craft::document::NodeNetwork, input_type: String, output_type: String) -> Self {
|
|
Self { network, input_type, output_type }
|
|
}
|
|
pub fn compile(&self, compile_dir: &str, manifest_path: &str) -> anyhow::Result<Vec<u8>> {
|
|
compile_spirv(&self.network, &self.input_type, &self.output_type, Some(compile_dir), manifest_path)
|
|
}
|
|
}
|