forked from rust-ndarray/ndarray
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathimpl_internal_constructors.rs
More file actions
66 lines (62 loc) · 1.85 KB
/
impl_internal_constructors.rs
File metadata and controls
66 lines (62 loc) · 1.85 KB
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
// Copyright 2021 bluss and ndarray developers.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
use std::ptr::NonNull;
use crate::imp_prelude::*;
// internal "builder-like" methods
impl<A, S> ArrayBase<S, Ix1>
where S: RawData<Elem = A>
{
/// Create an (initially) empty one-dimensional array from the given data and array head
/// pointer
///
/// ## Safety
///
/// The caller must ensure that the data storage and pointer is valid.
///
/// See ArrayView::from_shape_ptr for general pointer validity documentation.
#[inline]
pub(crate) unsafe fn from_data_ptr(data: S, ptr: NonNull<A>) -> Self
{
let array = ArrayBase {
data,
ptr,
dim: Ix1(0),
strides: Ix1(1),
};
debug_assert!(array.pointer_is_inbounds());
array
}
}
// internal "builder-like" methods
impl<A, S, D> ArrayBase<S, D>
where
S: RawData<Elem = A>,
D: Dimension,
{
/// Set strides and dimension of the array to the new values
///
/// The argument order with strides before dimensions is used because strides are often
/// computed as derived from the dimension.
///
/// ## Safety
///
/// The caller needs to ensure that the new strides and dimensions are correct
/// for the array data.
#[inline]
pub(crate) unsafe fn with_strides_dim<E>(self, strides: E, dim: E) -> ArrayBase<S, E>
where E: Dimension
{
debug_assert_eq!(strides.ndim(), dim.ndim());
ArrayBase {
data: self.data,
ptr: self.ptr,
dim,
strides,
}
}
}