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
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(_) => Num(1.),
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)
}
}
#[macro_use]
mod macros {
#[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.);
}