eom 0.11.0

Configurable ODE/PDE solver
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

#[macro_use]
extern crate criterion;
extern crate eom;

use criterion::Criterion;
use eom::traits::*;
use eom::*;
use ndarray::arr1;

fn lorenz63_heun(c: &mut Criterion) {
    c.bench_function("Lorenz63 Heun 10000 steps", |b| {
        b.iter(|| {
            let dt = 0.01;
            let eom = ode::Lorenz63::default();
            let mut teo = explicit::Heun::new(eom, dt);
            let mut x = arr1(&[1.0, 0.0, 0.0]);
            for _ in 0..10000 {
                teo.iterate(&mut x);
            }
        })
    });
}

fn lorenz63_rk4(c: &mut Criterion) {
    c.bench_function("Lorenz63 RK4 10000 steps", |b| {
        b.iter(|| {
            let dt = 0.01;
            let eom = ode::Lorenz63::default();
            let mut teo = explicit::RK4::new(eom, dt);
            let mut x = arr1(&[1.0, 0.0, 0.0]);
            for _ in 0..10000 {
                teo.iterate(&mut x);
            }
        })
    });
}

criterion_group!(lorenz63, lorenz63_heun, lorenz63_rk4);
criterion_main!(lorenz63);