as_view

rust-math · Jun 20, 2020 · c9e46a9 · c9e46a9
1 parent 8dc079d
commit c9e46a9
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/fftw/src/array.rs b/fftw/src/array.rs
@@ -2,6 +2,7 @@
 
 use crate::types::*;
 use ffi;
+use ndarray::*;
 
 use num_traits::Zero;
 use std::ops::{Deref, DerefMut};
@@ -55,6 +56,14 @@ impl<T> AlignedVec<T> {
  pub fn as_slice_mut(&mut self) -> &mut [T] {
  unsafe { from_raw_parts_mut(self.data, self.n) }
  }
+
+ pub fn as_view<D, Shape>(&self, shape: Shape) -> Result<ArrayView<T, D>, ShapeError>
+ where
+ D: Dimension,
+ Shape: Into<StrideShape<D>>,
+ {
+ ArrayView::from_shape(shape, self.as_slice())
+ }
 }
 
 impl<T> Deref for AlignedVec<T> {