mdarray 0.2.0

Multidimensional array for Rust
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

use crate::index::ViewIndex;
use crate::layout::StridedLayout;
use crate::order::Order;
use crate::view::StridedView;
use std::marker::PhantomData;
use std::ops::{Deref, DerefMut, Index, IndexMut};
use std::ptr::NonNull;

pub struct SubGrid<'a, T, const N: usize, const M: usize, O: Order> {
    ptr: NonNull<T>,
    layout: StridedLayout<N, M, O>,
    _marker: PhantomData<&'a T>,
}

pub struct SubGridMut<'a, T, const N: usize, const M: usize, O: Order> {
    ptr: NonNull<T>,
    layout: StridedLayout<N, M, O>,
    _marker: PhantomData<&'a mut T>,
}

impl<'a, T, const N: usize, const M: usize, O: Order> SubGrid<'a, T, N, M, O> {
    pub fn new(ptr: NonNull<T>, layout: StridedLayout<N, M, O>) -> Self {
        Self {
            ptr,
            layout,
            _marker: PhantomData,
        }
    }
}

impl<'a, T, const N: usize, const M: usize, O: Order> SubGridMut<'a, T, N, M, O> {
    pub fn new(ptr: NonNull<T>, layout: StridedLayout<N, M, O>) -> Self {
        Self {
            ptr,
            layout,
            _marker: PhantomData,
        }
    }
}

impl<'a, T, const N: usize, const M: usize, O: Order> Deref for SubGrid<'a, T, N, M, O> {
    type Target = StridedView<T, N, M, O>;

    fn deref(&self) -> &Self::Target {
        unsafe { StridedView::from_raw_parts(self.ptr.as_ptr(), &self.layout) }
    }
}

impl<'a, T, const N: usize, const M: usize, O: Order> Deref for SubGridMut<'a, T, N, M, O> {
    type Target = StridedView<T, N, M, O>;

    fn deref(&self) -> &Self::Target {
        unsafe { StridedView::from_raw_parts(self.ptr.as_ptr(), &self.layout) }
    }
}

impl<'a, T, const N: usize, const M: usize, O: Order> DerefMut for SubGridMut<'a, T, N, M, O> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        unsafe { StridedView::from_raw_parts_mut(self.ptr.as_ptr(), &self.layout) }
    }
}

impl<'a, I: ViewIndex<T, N, M, O>, T, const N: usize, const M: usize, O: Order> Index<I>
    for SubGrid<'a, T, N, M, O>
{
    type Output = I::Output;

    fn index(&self, index: I) -> &I::Output {
        index.index(&**self)
    }
}

impl<'a, I: ViewIndex<T, N, M, O>, T, const N: usize, const M: usize, O: Order> Index<I>
    for SubGridMut<'a, T, N, M, O>
{
    type Output = I::Output;

    fn index(&self, index: I) -> &I::Output {
        index.index(&**self)
    }
}

impl<'a, I: ViewIndex<T, N, M, O>, T, const N: usize, const M: usize, O: Order> IndexMut<I>
    for SubGridMut<'a, T, N, M, O>
{
    fn index_mut(&mut self, index: I) -> &mut I::Output {
        index.index_mut(&mut **self)
    }
}