faer 0.23.2

linear algebra 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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

use super::*;
use crate::into_range::IntoRange;
use crate::{Idx, IdxInc, assert, debug_assert};

impl<'a, R: Shape, T, Rs: Stride, RowRange: IntoRange<IdxInc<R>, Len<R>: 'a>> ColIndex<RowRange> for ColRef<'a, T, R, Rs> {
	type Target = ColRef<'a, T, RowRange::Len<R>, Rs>;

	#[track_caller]
	#[inline]
	fn get(this: Self, row: RowRange) -> Self::Target {
		let row = row.into_range(R::start(), this.nrows().end());
		assert!(all(row.start <= row.end, row.end <= this.nrows()));
		let nrows = unsafe { RowRange::Len::<R>::new_unbound(row.end.unbound() - row.start.unbound()) };

		this.subrows(row.start, nrows)
	}

	#[track_caller]
	#[inline]
	unsafe fn get_unchecked(this: Self, row: RowRange) -> Self::Target {
		let row = row.into_range(R::start(), this.nrows().end());

		debug_assert!(all(row.start <= row.end, row.end <= this.nrows(),));
		let nrows = unsafe { RowRange::Len::<R>::new_unbound(row.end.unbound() - row.start.unbound()) };

		this.subrows(row.start, nrows)
	}
}

impl<'a, R: Shape, T, Rs: Stride, RowRange: IntoRange<IdxInc<R>, Len<R>: 'a>> ColIndex<RowRange> for ColMut<'a, T, R, Rs> {
	type Target = ColMut<'a, T, RowRange::Len<R>, Rs>;

	#[track_caller]
	#[inline]
	fn get(this: Self, row: RowRange) -> Self::Target {
		let row = row.into_range(R::start(), this.nrows().end());
		assert!(all(row.start <= row.end, row.end <= this.nrows()));
		let nrows = unsafe { RowRange::Len::<R>::new_unbound(row.end.unbound() - row.start.unbound()) };

		this.subrows_mut(row.start, nrows)
	}

	#[track_caller]
	#[inline]
	unsafe fn get_unchecked(this: Self, row: RowRange) -> Self::Target {
		let row = row.into_range(R::start(), this.nrows().end());

		debug_assert!(all(row.start <= row.end, row.end <= this.nrows(),));
		let nrows = unsafe { RowRange::Len::<R>::new_unbound(row.end.unbound() - row.start.unbound()) };

		this.subrows_mut(row.start, nrows)
	}
}

macro_rules! idx_impl {
    ($R: ty $(, $tt: tt)?) => {
        impl<'a $(, $tt)?, T, Rs: Stride> ColIndex<Idx<$R>> for ColRef<'a, T, $R, Rs> {
            type Target = &'a T;

            #[track_caller]
            #[inline]
            fn get(this: Self, row: Idx<$R>) -> Self::Target {
                this.at(row)
            }

            #[track_caller]
            #[inline]
            unsafe fn get_unchecked(this: Self, row: Idx<$R>) -> Self::Target {
                this.at_unchecked(row)
            }
        }

        impl<'a $(, $tt)?, T, Rs: Stride> ColIndex<Idx<$R>> for ColMut<'a, T, $R, Rs> {
            type Target = &'a mut T;

            #[track_caller]
            #[inline]
            fn get(this: Self, row: Idx<$R>) -> Self::Target {
                this.at_mut(row)
            }

            #[track_caller]
            #[inline]
            unsafe fn get_unchecked(this: Self, row: Idx<$R>) -> Self::Target {
                this.at_mut_unchecked(row)
            }
        }
    };
}

idx_impl!(usize);
idx_impl!(Dim<'N>, 'N);