mathcompile 0.1.2

High-performance symbolic mathematics with final tagless design, egglog optimization, and Rust hot-loading compilation
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use mathcompile::prelude::*;

fn main() -> mathcompile::Result<()> {
    println!("=== A-Normal Form (ANF) Demonstration ===\n");

    // Create a variable registry
    let mut registry = VariableRegistry::new();
    let x_idx = registry.register_variable("x");
    let y_idx = registry.register_variable("y");

    // Create a complex expression with repeated subexpressions:
    // sin(x + y) + cos(x + y) + exp(x + y)
    // Notice how (x + y) appears three times!
    let x = <ASTEval as ASTMathExpr>::var(x_idx);
    let y = <ASTEval as ASTMathExpr>::var(y_idx);
    let x_plus_y = <ASTEval as ASTMathExpr>::add(x, y);

    let sin_term = <ASTEval as ASTMathExpr>::sin(x_plus_y.clone());
    let cos_term = <ASTEval as ASTMathExpr>::cos(x_plus_y.clone());
    let exp_term = <ASTEval as ASTMathExpr>::exp(x_plus_y);

    let sum1 = <ASTEval as ASTMathExpr>::add(sin_term, cos_term);
    let final_expr = <ASTEval as ASTMathExpr>::add(sum1, exp_term);

    println!("Original expression: sin(x + y) + cos(x + y) + exp(x + y)");
    println!("Notice how (x + y) is computed three times!\n");

    // Convert to ANF - this automatically performs CSE!
    let anf = convert_to_anf(&final_expr)?;

    println!("ANF automatically introduces temporary variables:");
    println!("Let count: {}", anf.let_count());
    println!("Variables used: {:?}\n", anf.used_variables());

    // Generate clean Rust code
    let codegen = ANFCodeGen::new(&registry);
    let function_code = codegen.generate_function("optimized_function", &anf);

    println!("Generated optimized Rust code:");
    println!("{function_code}");

    println!("\n✨ ANF Benefits:");
    println!("• Automatic common subexpression elimination");
    println!("• Clean, readable generated code");
    println!("• Efficient variable management");
    println!("• Ready for further optimization passes");

    Ok(())
}