scialg 0.3.0

A collection of scientific algorithms
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

use crate::ode::stepper::{Stepper, StepperData};
use crate::vector::Vector;

/// Euler method
pub struct Euler<const N: usize> {
    data: StepperData<N>,
}

impl<const N: usize> Euler<N> {
    pub fn new(h: f64, p0: Vector<N>, df: fn(f64, Vector<N>) -> Vector<N>) -> Self {
        Euler {
            data: StepperData::new(h, p0, df),
        }
    }

    fn dy(&self, h: f64) -> Vector<N> {
        self.data.y_cur + (self.data.derive)(self.data.x_cur, self.data.y_cur) * h
    }
}

impl<const N: usize> Stepper<N> for Euler<N> {
    fn step(&mut self) -> Vector<N> {
        self.data.y_cur = self.dy(self.data.h_cur);
        self.data.y_cur
    }
}