aphreco 0.1.7

A small-scale physiological modeling library
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

use super::dopri45::Dopri45;
use super::rk4::Rk4;

#[derive(Clone)]
pub enum StepOptions {
  Default,

  Rk4 {
    h: f64,
  },

  Dopri45 {
    h0: f64,
    abstol: f64,
    reltol: f64,
    hmin: f64,
    hmax: f64,
  },
}

#[derive(Clone)]
pub enum Stepper {
  Rk4(StepOptions),
  Dopri45(StepOptions),
}

impl Stepper {
  pub fn new<Ode, const LEN_Y: usize>(&self, ode: Ode) -> ConcreteStepper<Ode, LEN_Y>
  where
    Ode: Fn(&f64, &[f64; LEN_Y], &mut [f64; LEN_Y]),
  {
    match self {
      Stepper::Rk4(options) => ConcreteStepper::Rk4 {
        concrete_stepper: Rk4::new(ode, options),
      },

      Stepper::Dopri45(options) => ConcreteStepper::Dopri45 {
        concrete_stepper: Dopri45::new(ode, options),
      },
    }
  }
}

pub enum ConcreteStepper<Ode, const LEN_Y: usize>
where
  Ode: Fn(&f64, &[f64; LEN_Y], &mut [f64; LEN_Y]),
{
  Rk4 {
    concrete_stepper: Rk4<Ode, LEN_Y>,
  },
  Dopri45 {
    concrete_stepper: Dopri45<Ode, LEN_Y>,
  },
}

impl<Ode, const LEN_Y: usize> ConcreteStepper<Ode, LEN_Y>
where
  Ode: Fn(&f64, &[f64; LEN_Y], &mut [f64; LEN_Y]),
{
  pub fn run(&mut self, t: &f64, y: &mut [f64; LEN_Y], dy: &mut [f64; LEN_Y]) -> f64 {
    match self {
      ConcreteStepper::Rk4 { concrete_stepper } => concrete_stepper.run(t, y, dy),
      ConcreteStepper::Dopri45 { concrete_stepper } => concrete_stepper.run(t, y, dy),
    }
  }
}