add compiler library

Author: SiebenCorgie

Cargo.toml

@@ -8,9 +8,10 @@ members = [
 "crates/minisdf-common",
 "crates/minisdf-optimizer",
 "crates/minisdf-backend-spirv",
+"crates/msdfc",
 
 "examples/simple"
-]
+, "crates/msdfc"]
 
 
 [workspace.dependencies]

README.md

@@ -11,6 +11,8 @@ Proof of concept RVSDG based compiler for a simple SignedDistanceField language.
 
 `minisdf-backend-spirv`: simple SPIR-V backend based on [rspirv](https://crates.io/crates/rspirv).
 
+`msdfc`: **M**ini**Sdf** **C**ompiler: Simple library that accepts a miniSDF compiler source file, or AST, and generates, if possible, a SPIR-V Binary.
+
 ## Example
 
 The following example defines a field that subtracts smoothed-box from a unit box that repeats in the domain 4³.

crates/minisdf-ast/src/lib.rs

@@ -3,6 +3,7 @@
 mod ts_parser;
 use minisdf_common::Span;
 pub use ts_parser::parse_file;
+pub use ts_parser::TSParseError;
 
 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
 pub enum Ty {

crates/minisdf-ast/src/ts_parser.rs

@@ -11,7 +11,7 @@ use crate::{
     Field, Ident, TypedArg,
 };
 
-use self::err::TSParseError;
+pub use self::err::TSParseError;
 use minisdf_common::ErrorPrintBundle;
 mod err;
 mod tree;

crates/minisdf-backend-spirv/src/lib.rs

@@ -1,5 +1,4 @@
 use ahash::AHashMap;
-use err::LoweringError;
 use minisdf_ast::Ty;
 use minisdf_optimizer::{
     rvsdg::{
@@ -15,6 +14,7 @@ use rspirv::{
     spirv::{self, Decoration, FunctionControl, LinkageType, Word},
 };
 mod err;
+pub use err::LoweringError;
 mod lltraversal;
 
 pub trait SpirvBackend {

crates/minisdf-optimizer/src/lib.rs

@@ -4,6 +4,7 @@ mod highlevel;
 mod lowlevel;
 mod type_check;
 
+pub use err::OptError;
 pub use highlevel::{hlgraph_from_ast, HLGraph};
 pub use lowlevel::{LLGraph, LLOp, LLOpTy};
 pub use rvsdg;

crates/msdfc/Cargo.toml

@@ -0,0 +1,14 @@
+[package]
+name = "msdfc"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+minisdf-ast = {path = "../../crates/minisdf-ast"}
+minisdf-optimizer = {path = "../../crates/minisdf-optimizer"}
+minisdf-backend-spirv = {path = "../../crates/minisdf-backend-spirv"}
+rvsdg-viewer = {path = "../../../vola/crates/rvsdg-viewer"}
+thiserror.workspace = true
+bytemuck = "1.14"

crates/msdfc/src/lib.rs

@@ -0,0 +1,64 @@
+use std::path::Path;
+
+use minisdf_ast::{Field, TSParseError};
+use minisdf_backend_spirv::{rspirv::binary::Assemble, LoweringError, SpirvBackend};
+use minisdf_optimizer::OptError;
+use thiserror::Error;
+
+#[derive(Error, Debug)]
+pub enum CompileError {
+    #[error("Failed to parse file into AST(s): {0}")]
+    AstParsingFailed(#[from] TSParseError),
+    #[error("High level type check failed")]
+    HLTypeCheckFailed,
+    #[error("Optimizer error: {0}")]
+    OptErr(#[from] OptError),
+    #[error("Failed to lower LLGraph to SPIR-V: {0}")]
+    LoweringErr(#[from] LoweringError),
+}
+
+///Compiles fields in `file`. If successful, returns all SPIR-V modules and their field names.
+pub fn compile_file(file: impl AsRef<Path>) -> Result<Vec<(String, Vec<u32>)>, CompileError> {
+    let p = file.as_ref();
+
+    let fields = minisdf_ast::parse_file(p)?;
+
+    let mut asts = Vec::new();
+    for ast in fields {
+        let name = ast.name.0.clone();
+        let spv = compile_ast(ast, file.as_ref())?;
+
+        asts.push((name, spv));
+    }
+
+    Ok(asts)
+}
+
+///Compiles the `ast` based on `file` into a SPIR-V buffer
+pub fn compile_ast(ast: Field, file: impl AsRef<Path>) -> Result<Vec<u32>, CompileError> {
+    let hltree = minisdf_optimizer::hlgraph_from_ast(ast, file).unwrap();
+
+    //rvsdg_viewer::into_svg(&hltree.graph, &format!("hl_{}.svg", name));
+
+    if !hltree.type_check() {
+        println!("Type check failed 😭");
+        return Err(CompileError::HLTypeCheckFailed);
+    }
+    let mut ll_graph = hltree.into_ll_graph();
+
+    //rvsdg_viewer::into_svg(&ll_graph.graph, &format!("ll_{}.svg", name));
+
+    ll_graph.verify()?;
+
+    ll_graph.inline();
+    //rvsdg_viewer::into_svg(&ll_graph.graph, &format!("ll_{}_post_inline.svg", name));
+
+    ll_graph.cne();
+    //rvsdg_viewer::into_svg(&ll_graph.graph, &format!("ll_{}_post_cne.svg", name));
+
+    ll_graph.type_resolve()?;
+
+    let spvmod = ll_graph.lower()?;
+    let dta = spvmod.assemble();
+    Ok(dta)
+}