//! Indexing #![allow(clippy::reversed_empty_ranges)] use crate::base::storage::{RawStorage, RawStorageMut}; use crate::base::{ Const, Dim, DimDiff, DimName, DimSub, Dynamic, Matrix, MatrixSlice, MatrixSliceMut, Scalar, U1, }; use std::ops; // N.B.: Not a public trait! trait DimRange { /// The number of elements indexed by this range. type Length: Dim; /// The lower bound of the range, inclusive. fn lower(&self, dimension: D) -> usize; /// The number of elements included in the range. fn length(&self, dimension: D) -> Self::Length; /// Produces true if `Self` is contained within `dimension`. fn contained_by(&self, dimension: D) -> bool; } impl DimRange for usize { type Length = U1; #[inline(always)] fn lower(&self, _: D) -> usize { *self } #[inline(always)] fn length(&self, _: D) -> Self::Length { Const::<1> } #[inline(always)] fn contained_by(&self, dimension: D) -> bool { *self < dimension.value() } } #[test] fn dimrange_usize() { assert!(!DimRange::contained_by(&0, Const::<0>)); assert!(DimRange::contained_by(&0, Const::<1>)); } impl DimRange for ops::Range { type Length = Dynamic; #[inline(always)] fn lower(&self, _: D) -> usize { self.start } #[inline(always)] fn length(&self, _: D) -> Self::Length { Dynamic::new(self.end.saturating_sub(self.start)) } #[inline(always)] fn contained_by(&self, dimension: D) -> bool { (self.start < dimension.value()) && (self.end <= dimension.value()) } } #[test] fn dimrange_range_usize() { assert!(!DimRange::contained_by(&(0..0), Const::<0>)); assert!(!DimRange::contained_by(&(0..1), Const::<0>)); assert!(DimRange::contained_by(&(0..1), Const::<1>)); assert!(DimRange::contained_by( &((usize::MAX - 1)..usize::MAX), Dynamic::new(usize::MAX) )); assert_eq!( DimRange::length(&((usize::MAX - 1)..usize::MAX), Dynamic::new(usize::MAX)), Dynamic::new(1) ); assert_eq!( DimRange::length(&(usize::MAX..(usize::MAX - 1)), Dynamic::new(usize::MAX)), Dynamic::new(0) ); assert_eq!( DimRange::length(&(usize::MAX..usize::MAX), Dynamic::new(usize::MAX)), Dynamic::new(0) ); } impl DimRange for ops::RangeFrom { type Length = Dynamic; #[inline(always)] fn lower(&self, _: D) -> usize { self.start } #[inline(always)] fn length(&self, dimension: D) -> Self::Length { (self.start..dimension.value()).length(dimension) } #[inline(always)] fn contained_by(&self, dimension: D) -> bool { self.start < dimension.value() } } #[test] fn dimrange_rangefrom_usize() { assert!(!DimRange::contained_by(&(0..), Const::<0>)); assert!(!DimRange::contained_by(&(0..), Const::<0>)); assert!(DimRange::contained_by(&(0..), Const::<1>)); assert!(DimRange::contained_by( &((usize::MAX - 1)..), Dynamic::new(usize::MAX) )); assert_eq!( DimRange::length(&((usize::MAX - 1)..), Dynamic::new(usize::MAX)), Dynamic::new(1) ); assert_eq!( DimRange::length(&(usize::MAX..), Dynamic::new(usize::MAX)), Dynamic::new(0) ); } impl DimRange for ops::RangeFrom where D: DimSub, { type Length = DimDiff; #[inline(always)] fn lower(&self, _: D) -> usize { self.start.value() } #[inline(always)] fn length(&self, dimension: D) -> Self::Length { dimension.sub(self.start) } #[inline(always)] fn contained_by(&self, _: D) -> bool { true } } #[test] fn dimrange_rangefrom_dimname() { assert_eq!(DimRange::length(&(Const::<1>..), Const::<5>), Const::<4>); } impl DimRange for ops::RangeFull { type Length = D; #[inline(always)] fn lower(&self, _: D) -> usize { 0 } #[inline(always)] fn length(&self, dimension: D) -> Self::Length { dimension } #[inline(always)] fn contained_by(&self, _: D) -> bool { true } } #[test] fn dimrange_rangefull() { assert!(DimRange::contained_by(&(..), Const::<0>)); assert_eq!(DimRange::length(&(..), Const::<1>), Const::<1>); } impl DimRange for ops::RangeInclusive { type Length = Dynamic; #[inline(always)] fn lower(&self, _: D) -> usize { *self.start() } #[inline(always)] fn length(&self, _: D) -> Self::Length { Dynamic::new(if self.end() < self.start() { 0 } else { self.end().wrapping_sub(self.start().wrapping_sub(1)) }) } #[inline(always)] fn contained_by(&self, dimension: D) -> bool { (*self.start() < dimension.value()) && (*self.end() < dimension.value()) } } #[test] fn dimrange_rangeinclusive_usize() { assert!(!DimRange::contained_by(&(0..=0), Const::<0>)); assert!(DimRange::contained_by(&(0..=0), Const::<1>)); assert!(!DimRange::contained_by( &(usize::MAX..=usize::MAX), Dynamic::new(usize::MAX) )); assert!(!DimRange::contained_by( &((usize::MAX - 1)..=usize::MAX), Dynamic::new(usize::MAX) )); assert!(DimRange::contained_by( &((usize::MAX - 1)..=(usize::MAX - 1)), Dynamic::new(usize::MAX) )); assert_eq!(DimRange::length(&(0..=0), Const::<1>), Dynamic::new(1)); assert_eq!( DimRange::length(&((usize::MAX - 1)..=usize::MAX), Dynamic::new(usize::MAX)), Dynamic::new(2) ); assert_eq!( DimRange::length(&(usize::MAX..=(usize::MAX - 1)), Dynamic::new(usize::MAX)), Dynamic::new(0) ); assert_eq!( DimRange::length(&(usize::MAX..=usize::MAX), Dynamic::new(usize::MAX)), Dynamic::new(1) ); } impl DimRange for ops::RangeTo { type Length = Dynamic; #[inline(always)] fn lower(&self, _: D) -> usize { 0 } #[inline(always)] fn length(&self, _: D) -> Self::Length { Dynamic::new(self.end) } #[inline(always)] fn contained_by(&self, dimension: D) -> bool { self.end <= dimension.value() } } #[test] fn dimrange_rangeto_usize() { assert!(DimRange::contained_by(&(..0), Const::<0>)); assert!(!DimRange::contained_by(&(..1), Const::<0>)); assert!(DimRange::contained_by(&(..0), Const::<1>)); assert!(DimRange::contained_by( &(..(usize::MAX - 1)), Dynamic::new(usize::MAX) )); assert_eq!( DimRange::length(&(..(usize::MAX - 1)), Dynamic::new(usize::MAX)), Dynamic::new(usize::MAX - 1) ); assert_eq!( DimRange::length(&(..usize::MAX), Dynamic::new(usize::MAX)), Dynamic::new(usize::MAX) ); } impl DimRange for ops::RangeToInclusive { type Length = Dynamic; #[inline(always)] fn lower(&self, _: D) -> usize { 0 } #[inline(always)] fn length(&self, _: D) -> Self::Length { Dynamic::new(self.end + 1) } #[inline(always)] fn contained_by(&self, dimension: D) -> bool { self.end < dimension.value() } } #[test] fn dimrange_rangetoinclusive_usize() { assert!(!DimRange::contained_by(&(..=0), Const::<0>)); assert!(!DimRange::contained_by(&(..=1), Const::<0>)); assert!(DimRange::contained_by(&(..=0), Const::<1>)); assert!(!DimRange::contained_by( &(..=(usize::MAX)), Dynamic::new(usize::MAX) )); assert!(DimRange::contained_by( &(..=(usize::MAX - 1)), Dynamic::new(usize::MAX) )); assert_eq!( DimRange::length(&(..=(usize::MAX - 1)), Dynamic::new(usize::MAX)), Dynamic::new(usize::MAX) ); } /// A helper trait used for indexing operations. pub trait MatrixIndex<'a, T, R: Dim, C: Dim, S: RawStorage>: Sized { /// The output type returned by methods. type Output: 'a; /// Produces true if the given matrix is contained by this index. #[doc(hidden)] fn contained_by(&self, matrix: &Matrix) -> bool; /// Produces a shared view of the data at this location if in bounds, /// or `None`, otherwise. #[doc(hidden)] #[inline(always)] fn get(self, matrix: &'a Matrix) -> Option { if self.contained_by(matrix) { Some(unsafe { self.get_unchecked(matrix) }) } else { None } } /// Produces a shared view of the data at this location if in bounds /// without any bounds checking. #[doc(hidden)] unsafe fn get_unchecked(self, matrix: &'a Matrix) -> Self::Output; /// Produces a shared view to the data at this location, or panics /// if out of bounds. #[doc(hidden)] #[inline(always)] fn index(self, matrix: &'a Matrix) -> Self::Output { self.get(matrix).expect("Index out of bounds.") } } /// A helper trait used for indexing operations. pub trait MatrixIndexMut<'a, T, R: Dim, C: Dim, S: RawStorageMut>: MatrixIndex<'a, T, R, C, S> { /// The output type returned by methods. type OutputMut: 'a; /// Produces a mutable view of the data at this location, without /// performing any bounds checking. #[doc(hidden)] unsafe fn get_unchecked_mut(self, matrix: &'a mut Matrix) -> Self::OutputMut; /// Produces a mutable view of the data at this location, if in /// bounds. #[doc(hidden)] #[inline(always)] fn get_mut(self, matrix: &'a mut Matrix) -> Option { if self.contained_by(matrix) { Some(unsafe { self.get_unchecked_mut(matrix) }) } else { None } } /// Produces a mutable view of the data at this location, or panics /// if out of bounds. #[doc(hidden)] #[inline(always)] fn index_mut(self, matrix: &'a mut Matrix) -> Self::OutputMut { self.get_mut(matrix).expect("Index out of bounds.") } } /// # Slicing based on ranges /// ## Indices to Individual Elements /// ### Two-Dimensional Indices /// ``` /// # use nalgebra::*; /// let matrix = Matrix2::new(0, 2, /// 1, 3); /// /// assert_eq!(matrix.index((0, 0)), &0); /// assert_eq!(matrix.index((1, 0)), &1); /// assert_eq!(matrix.index((0, 1)), &2); /// assert_eq!(matrix.index((1, 1)), &3); /// ``` /// /// ### Linear Address Indexing /// ``` /// # use nalgebra::*; /// let matrix = Matrix2::new(0, 2, /// 1, 3); /// /// assert_eq!(matrix.get(0), Some(&0)); /// assert_eq!(matrix.get(1), Some(&1)); /// assert_eq!(matrix.get(2), Some(&2)); /// assert_eq!(matrix.get(3), Some(&3)); /// ``` /// /// ## Indices to Individual Rows and Columns /// ### Index to a Row /// ``` /// # use nalgebra::*; /// let matrix = Matrix2::new(0, 2, /// 1, 3); /// /// assert!(matrix.index((0, ..)) /// .eq(&Matrix1x2::new(0, 2))); /// ``` /// /// ### Index to a Column /// ``` /// # use nalgebra::*; /// let matrix = Matrix2::new(0, 2, /// 1, 3); /// /// assert!(matrix.index((.., 0)) /// .eq(&Matrix2x1::new(0, /// 1))); /// ``` /// /// ## Indices to Parts of Individual Rows and Columns /// ### Index to a Partial Row /// ``` /// # use nalgebra::*; /// let matrix = Matrix3::new(0, 3, 6, /// 1, 4, 7, /// 2, 5, 8); /// /// assert!(matrix.index((0, ..2)) /// .eq(&Matrix1x2::new(0, 3))); /// ``` /// /// ### Index to a Partial Column /// ``` /// # use nalgebra::*; /// let matrix = Matrix3::new(0, 3, 6, /// 1, 4, 7, /// 2, 5, 8); /// /// assert!(matrix.index((..2, 0)) /// .eq(&Matrix2x1::new(0, /// 1))); /// /// assert!(matrix.index((Const::<1>.., 0)) /// .eq(&Matrix2x1::new(1, /// 2))); /// ``` /// ## Indices to Ranges of Rows and Columns /// ### Index to a Range of Rows /// ``` /// # use nalgebra::*; /// let matrix = Matrix3::new(0, 3, 6, /// 1, 4, 7, /// 2, 5, 8); /// /// assert!(matrix.index((1..3, ..)) /// .eq(&Matrix2x3::new(1, 4, 7, /// 2, 5, 8))); /// ``` /// ### Index to a Range of Columns /// ``` /// # use nalgebra::*; /// let matrix = Matrix3::new(0, 3, 6, /// 1, 4, 7, /// 2, 5, 8); /// /// assert!(matrix.index((.., 1..3)) /// .eq(&Matrix3x2::new(3, 6, /// 4, 7, /// 5, 8))); /// ``` impl> Matrix { /// Produces a view of the data at the given index, or /// `None` if the index is out of bounds. #[inline] #[must_use] pub fn get<'a, I>(&'a self, index: I) -> Option where I: MatrixIndex<'a, T, R, C, S>, { index.get(self) } /// Produces a mutable view of the data at the given index, or /// `None` if the index is out of bounds. #[inline] #[must_use] pub fn get_mut<'a, I>(&'a mut self, index: I) -> Option where S: RawStorageMut, I: MatrixIndexMut<'a, T, R, C, S>, { index.get_mut(self) } /// Produces a view of the data at the given index, or /// panics if the index is out of bounds. #[inline] #[must_use] pub fn index<'a, I>(&'a self, index: I) -> I::Output where I: MatrixIndex<'a, T, R, C, S>, { index.index(self) } /// Produces a mutable view of the data at the given index, or /// panics if the index is out of bounds. #[inline] pub fn index_mut<'a, I>(&'a mut self, index: I) -> I::OutputMut where S: RawStorageMut, I: MatrixIndexMut<'a, T, R, C, S>, { index.index_mut(self) } /// Produces a view of the data at the given index, without doing /// any bounds checking. #[inline] #[must_use] pub unsafe fn get_unchecked<'a, I>(&'a self, index: I) -> I::Output where I: MatrixIndex<'a, T, R, C, S>, { index.get_unchecked(self) } /// Returns a mutable view of the data at the given index, without doing /// any bounds checking. #[inline] #[must_use] pub unsafe fn get_unchecked_mut<'a, I>(&'a mut self, index: I) -> I::OutputMut where S: RawStorageMut, I: MatrixIndexMut<'a, T, R, C, S>, { index.get_unchecked_mut(self) } } // EXTRACT A SINGLE ELEMENT BY 1D LINEAR ADDRESS impl<'a, T, R, C, S> MatrixIndex<'a, T, R, C, S> for usize where T: Scalar, R: Dim, C: Dim, S: RawStorage, { type Output = &'a T; #[doc(hidden)] #[inline(always)] fn contained_by(&self, matrix: &Matrix) -> bool { *self < matrix.len() } #[doc(hidden)] #[inline(always)] unsafe fn get_unchecked(self, matrix: &'a Matrix) -> Self::Output { let nrows = matrix.shape().0; let row = self % nrows; let col = self / nrows; matrix.data.get_unchecked(row, col) } } impl<'a, T, R, C, S> MatrixIndexMut<'a, T, R, C, S> for usize where T: Scalar, R: Dim, C: Dim, S: RawStorageMut, { type OutputMut = &'a mut T; #[doc(hidden)] #[inline(always)] unsafe fn get_unchecked_mut(self, matrix: &'a mut Matrix) -> Self::OutputMut where S: RawStorageMut, { let nrows = matrix.shape().0; let row = self % nrows; let col = self / nrows; matrix.data.get_unchecked_mut(row, col) } } // EXTRACT A SINGLE ELEMENT BY 2D COORDINATES impl<'a, T: 'a, R, C, S> MatrixIndex<'a, T, R, C, S> for (usize, usize) where R: Dim, C: Dim, S: RawStorage, { type Output = &'a T; #[doc(hidden)] #[inline(always)] fn contained_by(&self, matrix: &Matrix) -> bool { let (rows, cols) = self; let (nrows, ncols) = matrix.shape_generic(); DimRange::contained_by(rows, nrows) && DimRange::contained_by(cols, ncols) } #[doc(hidden)] #[inline(always)] unsafe fn get_unchecked(self, matrix: &'a Matrix) -> Self::Output { let (row, col) = self; matrix.data.get_unchecked(row, col) } } impl<'a, T: 'a, R, C, S> MatrixIndexMut<'a, T, R, C, S> for (usize, usize) where R: Dim, C: Dim, S: RawStorageMut, { type OutputMut = &'a mut T; #[doc(hidden)] #[inline(always)] unsafe fn get_unchecked_mut(self, matrix: &'a mut Matrix) -> Self::OutputMut where S: RawStorageMut, { let (row, col) = self; matrix.data.get_unchecked_mut(row, col) } } macro_rules! impl_index_pair { ( $R: ident, $C: ident, [<$($RTyP: ident : $RTyPB: ty,)*> usize => $ROut: ty $(where $RConstraintType: ty: $RConstraintBound: ident<$($RConstraintBoundParams: ty $( = $REqBound: ty )*),*>)*], [<$($CTyP: ident : $CTyPB: ty,)*> usize => $COut: ty $(where $CConstraintType: ty: $CConstraintBound: ident<$($CConstraintBoundParams: ty $( = $CEqBound: ty )*),*>)*] ) => {}; ( $R: ident, $C: ident, [<$($RTyP: ident: $RTyPB: tt),*> $RIdx: ty => $ROut: ty $(where $RConstraintType: ty: $RConstraintBound: ident $(<$($RConstraintBoundParams: ty $( = $REqBound: ty )*),*>)* )*], [<$($CTyP: ident: $CTyPB: tt),*> $CIdx: ty => $COut: ty $(where $CConstraintType: ty: $CConstraintBound: ident $(<$($CConstraintBoundParams: ty $( = $CEqBound: ty )*),*>)* )*] ) => { impl<'a, T, $R, $C, S, $($RTyP : $RTyPB,)* $($CTyP : $CTyPB),*> MatrixIndex<'a, T, $R, $C, S> for ($RIdx, $CIdx) where T: Scalar, $R: Dim, $C: Dim, S: RawStorage, $( $RConstraintType: $RConstraintBound $(<$( $RConstraintBoundParams $( = $REqBound )*),*>)* ,)* $( $CConstraintType: $CConstraintBound $(<$( $CConstraintBoundParams $( = $CEqBound )*),*>)* ),* { type Output = MatrixSlice<'a, T, $ROut, $COut, S::RStride, S::CStride>; #[doc(hidden)] #[inline(always)] fn contained_by(&self, matrix: &Matrix) -> bool { let (rows, cols) = self; let (nrows, ncols) = matrix.shape_generic(); DimRange::contained_by(rows, nrows) && DimRange::contained_by(cols, ncols) } #[doc(hidden)] #[inline(always)] unsafe fn get_unchecked(self, matrix: &'a Matrix) -> Self::Output { use crate::base::SliceStorage; let (rows, cols) = self; let (nrows, ncols) = matrix.shape_generic(); let data = SliceStorage::new_unchecked(&matrix.data, (rows.lower(nrows), cols.lower(ncols)), (rows.length(nrows), cols.length(ncols))); Matrix::from_data_statically_unchecked(data) } } impl<'a, T, $R, $C, S, $($RTyP : $RTyPB,)* $($CTyP : $CTyPB),*> MatrixIndexMut<'a, T, $R, $C, S> for ($RIdx, $CIdx) where T: Scalar, $R: Dim, $C: Dim, S: RawStorageMut, $( $RConstraintType: $RConstraintBound $(<$( $RConstraintBoundParams $( = $REqBound )*),*>)* ,)* $( $CConstraintType: $CConstraintBound $(<$( $CConstraintBoundParams $( = $CEqBound )*),*>)* ),* { type OutputMut = MatrixSliceMut<'a, T, $ROut, $COut, S::RStride, S::CStride>; #[doc(hidden)] #[inline(always)] unsafe fn get_unchecked_mut(self, matrix: &'a mut Matrix) -> Self::OutputMut { use crate::base::SliceStorageMut; let (rows, cols) = self; let (nrows, ncols) = matrix.shape_generic(); let data = SliceStorageMut::new_unchecked(&mut matrix.data, (rows.lower(nrows), cols.lower(ncols)), (rows.length(nrows), cols.length(ncols))); Matrix::from_data_statically_unchecked(data) } } } } macro_rules! impl_index_pairs { (index $R: ident with {} index $C: ident with {$($r: tt,)* }) => {}; (index $R: ident with {$lh : tt, $($lt : tt,)*} index $C: ident with { $($r: tt,)* }) => { $( impl_index_pair!{$R, $C, $lh, $r} )* impl_index_pairs!{index $R with {$($lt,)*} index $C with {$($r,)*}} } } impl_index_pairs! { index R with { [<> usize => U1], [<> ops::Range => Dynamic], [<> ops::RangeFrom => Dynamic], [<> ops::RangeFull => R], [<> ops::RangeInclusive => Dynamic], [<> ops::RangeTo => Dynamic], [<> ops::RangeToInclusive => Dynamic], [ ops::RangeFrom => DimDiff where R: DimSub], } index C with { [<> usize => U1], [<> ops::Range => Dynamic], [<> ops::RangeFrom => Dynamic], [<> ops::RangeFull => C], [<> ops::RangeInclusive => Dynamic], [<> ops::RangeTo => Dynamic], [<> ops::RangeToInclusive => Dynamic], [ ops::RangeFrom => DimDiff where C: DimSub], } }