rmath 0.1.5

A rust math 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 crate::rand::{RandError, Source, PrimitiveType};
use std::marker::PhantomData;

pub struct Iter<'a, T, P> 
    where P: PrimitiveType, T: 'a + Source<P> {
    pub(super) inner: &'a mut T,
    pub(super) last_err: Option<RandError>,
    pub(super) ph: PhantomData<P>,
}

impl<'a, T, P> Iter<'a, T, P> 
    where P: PrimitiveType, T: 'a + Source<P> {
    
    pub fn new(rd_src: &'a mut T) -> Iter<'a, T, P> {
        Self {
            inner: rd_src,
            last_err: None,
            ph: PhantomData
        }
    }

    pub fn last_error(&self) -> Option<&RandError> {
        self.last_err.as_ref()
    }
    
    pub fn clear_error(&mut self) {
        self.last_err = None;
    }
}

impl<'a, T, P> Iterator for Iter<'a, T, P> 
    where P: PrimitiveType, T: 'a + Source<P> {
    type Item = P;

    fn next(&mut self) -> Option<Self::Item> {
        if self.last_err.is_none() {
            match self.inner.gen() {
                Ok(x) => Some(x),
                Err(e) => {
                    self.last_err = Some(e);
                    None
                }
            }
        } else {
            None
        }
    }
}

macro_rules! iter_impl {
    ($SourceType0: ident, $Type0: ty) => {
        impl IterSource<$Type0> for $SourceType0<$Type0> {
            fn iter_mut(&mut self) -> crate::rand::iter::Iter<'_, Self, $Type0> {
                crate::rand::iter::Iter {
                    inner: self,
                    last_err: None,
                    ph: std::marker::PhantomData,
                }
            }
        }
    };
    ($SourceType1: ident, $Type1: ty, $($Type2: ty),+) => {
        iter_impl!($SourceType1, $Type1);
        iter_impl!($SourceType1, $($Type2),+);
    };
}