sequential-integration 1.0.2

Lightweight library for sequential integration
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

use core::marker::PhantomData;
use fehler::throws;

use super::utils as integrator_utils;
use crate::{
    engine::{
        helper_equation_traits::{EquationOfOneVariable, EquationOfTwoVariable},
        quadrature::GetQuadratureRange,
        Bounds, CalculationResult, CalculationStep,
    },
    errors::Error,
};

pub struct SecondIntegrator<
    'a,
    G: GetQuadratureRange,
    E: EquationOfTwoVariable,
    F1: Fn(f64) -> f64,
    F2: Fn(f64) -> f64,
> {
    a_equation: F1,
    b_equation: F2,
    h: f64,
    equation: &'a E,
    _p: PhantomData<G>,
}

impl<
        'a,
        G: GetQuadratureRange,
        E: EquationOfTwoVariable,
        F1: Fn(f64) -> f64,
        F2: Fn(f64) -> f64,
    > SecondIntegrator<'a, G, E, F1, F2>
{
    #[throws]
    pub fn new(a_equation: F1, b_equation: F2, h: f64, equation: &'a E) -> Self {
        Self {
            a_equation,
            b_equation,
            h,
            equation,
            _p: PhantomData,
        }
    }
}

impl<
        'a,
        G: GetQuadratureRange,
        E: EquationOfTwoVariable,
        F1: Fn(f64) -> f64,
        F2: Fn(f64) -> f64,
    > EquationOfOneVariable for SecondIntegrator<'a, G, E, F1, F2>
{
    #[throws]
    fn calculate(&self, x: CalculationStep, bounds: Bounds) -> CalculationResult {
        let a = (self.a_equation)(*x);
        let b = (self.b_equation)(*x);
        let borders_config = integrator_utils::BoundsConfigurator::configurate(a, b)?;

        let mut result = CalculationResult::new();
        let mut range = if let Some(range) = G::get_range_generator(borders_config.bounds, self.h)?
        {
            range
        } else {
            return result;
        };

        loop {
            let step = range.next()?;
            result += self
                .equation
                .calculate(x, bounds, step, borders_config.bounds)?
                * borders_config.direction_coeff;

            if step.is_last() {
                break;
            }
        }

        result
    }
}