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

extern crate symbolics_core;
use symbolics_core::Expr;
use symbolics_core::{s, apply};

trait Derivative {
    fn deriv1(self, &'static str) -> Self;
}

impl Derivative for Expr {
    fn deriv1(self, wrt: &'static str) -> Expr {
        use Expr::*;
        match self {
            Num(_) => Num(0.),
            Symbol(s) => Num(if s == wrt {1.} else {0.}),
            Add(a, b) => a.clone().deriv1(wrt) + b.clone().deriv1(wrt),
            Mul(a, b) => a.clone().deriv1(wrt)**b.clone() + *a.clone()*b.clone().deriv1(wrt),
            Pow(a, b) => (*a.clone()^*b.clone()) * (a.clone().ln()*b.clone().deriv1(wrt) + (*b.clone()*a.clone().deriv1(wrt)/(*a.clone()))),
            Log(a, b) => a.clone().deriv1(wrt) / (*a.clone() * b.ln()),
            Sin(x) => x.clone().cos() * x.clone().deriv1(wrt),
            Cos(x) => -x.clone().sin() * x.clone().deriv1(wrt),
            Arcsin(x) => x.clone().deriv1(wrt) / (1 - (*x.clone() ^ 2)).sqrt(),
            Arccos(x) => -x.clone().deriv1(wrt) / (1 - (*x.clone() ^ 2)).sqrt(),
            Arctan(x) => x.clone().deriv1(wrt) / (1 + (*x.clone() ^ 2)),
        }.apply1("_", 0) // This is to auto-simplify
    }
}

#[macro_use]
mod macros {
/// Differentiate an expression with respect to any amount of variables.
///
/// # Examples
///
/// To take the derivative of `y` with respect to `x`:
/// diff!(y, x)
///
/// To take the second derivative of `x` with respect to `t`:
/// diff!(x, t, t)
///
/// To take the derivative of `(s!(x) ^ 2) * (s!(y) ^ 3)` with respect to `x`, then `y`:
/// diff!((s!(x) ^ 2) * (s!(y) ^ 3), x, y)
    #[macro_export]
    macro_rules! diff {
        ($expr:expr, $($sym:ident),+) => {
            $expr $(.deriv1(stringify!($sym)))+
        }
    }
}

#[test]
fn linear1() {
    let y = 3 * s!(x);
    assert_eq!(apply!(diff!(y, x), x=0).val().unwrap(), 3.);
}

#[test]
fn polynomial1() {
    let y = (s!(x)^3) + (s!(x)^2) + s!(x) + 2;
    assert_eq!(apply!(diff!(y, x), x=1).val().unwrap(), 6.);
}

#[test]
fn trig1() {
    use symbolics_core::consts::pi;
    let y = s!(x).sin() * s!(x).cos();
    assert_eq!(apply!(diff!(y, x), x=pi()/2).val().unwrap(), -1.);
}

#[test]
fn partial1() {
    let z = 2*s!(x) + s!(y);
    assert_eq!(apply!(diff!(z, x), y=5, x=0).val().unwrap(), 2.);
}