openinfer-dsl 0.1.0

Rust-embedded DSL for defining OpenInfer graphs with explicit control flow and memory semantics.
Documentation 
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

use crate::types::{Dim, DimAtom};
use quote::quote;

pub(crate) fn dims_expr(dims: &[Dim]) -> proc_macro2::TokenStream {
    let items = dims.iter().map(|dim| match dim {
        Dim::Ident(ident) => {
            let s = ident.to_string();
            quote! { #s.to_string() }
        }
        Dim::Lit(lit) => {
            let s = lit.to_string();
            quote! { #s.to_string() }
        }
        Dim::Mul { left, right } => {
            let left = dim_atom_string(left);
            let right = dim_atom_string(right);
            let s = format!("{}*{}", left, right);
            quote! { #s.to_string() }
        }
    });
    quote! { vec![#(#items),*] }
}

fn dim_atom_string(atom: &DimAtom) -> String {
    match atom {
        DimAtom::Ident(ident) => ident.to_string(),
        DimAtom::Lit(lit) => lit.to_string(),
    }
}