hecate-symrs 0.2.4

A pure rust symbolic computation library! Inspired by sympy.
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use serde::{Deserialize, Serialize};

use super::*;
#[derive(Clone, Debug, Hash, PartialEq, Eq, JsonSchema, Deserialize, Serialize)]
#[serde(from = "String", into = "String")]
pub struct Symbol {
    pub name: String,
}

impl Symbol {
    pub fn new(name: &str) -> Symbol {
        Self::new_v2(name.to_string())
    }

    pub fn new_v2(name: String) -> Symbol {
        Symbol {
            name: (match name.as_str() {
                "nabla" => "∇".into(),
                "laplacian" => "Δ".into(),
                "theta" => "θ".into(),
                _ => name,
            }),
        }
    }

    pub fn new_box(name: &str) -> Box<dyn Expr> {
        Box::new(Symbol::new(name))
    }
}

impl std::fmt::Display for Symbol {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.name)
    }
}

impl Expr for Symbol {
    fn known_expr(&self) -> KnownExpr {
        KnownExpr::Symbol(self)
    }
    fn get_ref<'a>(&'a self) -> &'a dyn Expr {
        self as &dyn Expr
    }
    fn for_each_arg(&self, f: &mut dyn FnMut(&dyn Arg) -> ()) {
        f(&self.name);
    }

    fn from_args(&self, args: Vec<Box<dyn Arg>>) -> Box<dyn Expr> {
        let name = (&*args[0]) as &dyn Any;

        let name = name.downcast_ref::<String>().unwrap();
        Box::new(Symbol { name: name.clone() })
    }

    fn clone_box(&self) -> Box<dyn Expr> {
        Box::new(self.clone())
    }

    fn str(&self) -> String {
        self.name.clone()
    }
    fn to_cpp(&self) -> String {
        match self.name.as_str() {
            "M^n" => "mass_mat".to_string(),
            "A^n" => "laplace_mat".to_string(),
            "k" => "time_step".to_string(),
            _ => self
                .name
                .replace("^n-1", "_prev")
                .replace("^n", "")
                .to_lowercase(),
        }
    }
}

impl<E: Expr> std::ops::Add<&E> for &Symbol {
    type Output = Box<dyn Expr>;

    fn add(self, rhs: &E) -> Self::Output {
        self.get_ref() + rhs.get_ref()
    }
}

impl<E: Expr> std::ops::Mul<&E> for &Symbol {
    type Output = Box<dyn Expr>;

    fn mul(self, rhs: &E) -> Self::Output {
        self.get_ref() * rhs.get_ref()
    }
}

impl<E: Expr> std::ops::Div<&E> for &Symbol {
    type Output = Box<dyn Expr>;

    fn div(self, rhs: &E) -> Self::Output {
        self.get_ref() / rhs.get_ref()
    }
}

impl From<String> for Symbol {
    fn from(value: String) -> Self {
        Self::new_v2(value)
    }
}

impl Into<String> for Symbol {
    fn into(self) -> String {
        format!("{self}")
    }
}