2018-10-27 21:16:03 +08:00
use num ::{ One , Zero } ;
2016-12-05 05:44:42 +08:00
2018-05-19 21:41:58 +08:00
use approx ::{ AbsDiffEq , RelativeEq , UlpsEq } ;
use std ::any ::TypeId ;
2016-12-05 05:44:42 +08:00
use std ::cmp ::Ordering ;
2021-07-18 09:19:20 +08:00
use std ::fmt ;
2018-12-28 01:05:25 +08:00
use std ::hash ::{ Hash , Hasher } ;
2018-05-19 21:41:58 +08:00
use std ::marker ::PhantomData ;
2021-08-03 00:41:46 +08:00
use std ::mem ;
2016-12-05 05:44:42 +08:00
2021-04-12 18:14:16 +08:00
#[ cfg(feature = " serde-serialize-no-std " ) ]
2018-02-02 19:26:35 +08:00
use serde ::{ Deserialize , Deserializer , Serialize , Serializer } ;
2017-05-04 10:02:30 +08:00
2021-03-06 00:08:46 +08:00
use simba ::scalar ::{ ClosedAdd , ClosedMul , ClosedSub , Field , SupersetOf } ;
2020-03-21 19:16:46 +08:00
use simba ::simd ::SimdPartialOrd ;
2016-12-05 05:44:42 +08:00
2019-03-23 21:29:07 +08:00
use crate ::base ::allocator ::{ Allocator , SameShapeAllocator , SameShapeC , SameShapeR } ;
use crate ::base ::constraint ::{ DimEq , SameNumberOfColumns , SameNumberOfRows , ShapeConstraint } ;
use crate ::base ::dimension ::{ Dim , DimAdd , DimSum , IsNotStaticOne , U1 , U2 , U3 } ;
2020-03-18 00:58:36 +08:00
use crate ::base ::iter ::{
ColumnIter , ColumnIterMut , MatrixIter , MatrixIterMut , RowIter , RowIterMut ,
} ;
2021-08-03 00:41:46 +08:00
use crate ::base ::storage ::{ Owned , RawStorage , RawStorageMut , SameShapeStorage } ;
2021-04-11 17:00:38 +08:00
use crate ::base ::{ Const , DefaultAllocator , OMatrix , OVector , Scalar , Unit } ;
2021-08-03 00:41:46 +08:00
use crate ::{ ArrayStorage , SMatrix , SimdComplexField , Storage , UninitMatrix } ;
2021-05-07 15:09:10 +08:00
2021-08-03 00:41:46 +08:00
use crate ::storage ::IsContiguous ;
use crate ::uninit ::{ Init , InitStatus , Uninit } ;
2021-05-07 15:09:10 +08:00
#[ cfg(any(feature = " std " , feature = " alloc " )) ]
2021-08-26 17:07:00 +08:00
use crate ::{ DMatrix , DVector , Dynamic , RowDVector , VecStorage } ;
2021-08-03 00:41:46 +08:00
use std ::mem ::MaybeUninit ;
2016-12-05 05:44:42 +08:00
/// A square matrix.
2021-04-11 17:00:38 +08:00
pub type SquareMatrix < T , D , S > = Matrix < T , D , D , S > ;
2016-12-05 05:44:42 +08:00
/// A matrix with one column and `D` rows.
2021-04-11 17:00:38 +08:00
pub type Vector < T , D , S > = Matrix < T , D , U1 , S > ;
2016-12-05 05:44:42 +08:00
2017-08-03 01:37:44 +08:00
/// A matrix with one row and `D` columns .
2021-04-11 17:00:38 +08:00
pub type RowVector < T , D , S > = Matrix < T , U1 , D , S > ;
2017-05-18 04:37:18 +08:00
2016-12-05 05:44:42 +08:00
/// The type of the result of a matrix sum.
2021-04-11 17:00:38 +08:00
pub type MatrixSum < T , R1 , C1 , R2 , C2 > =
Matrix < T , SameShapeR < R1 , R2 > , SameShapeC < C1 , C2 > , SameShapeStorage < T , R1 , C1 , R2 , C2 > > ;
2016-12-05 05:44:42 +08:00
/// The type of the result of a matrix sum.
2021-04-11 17:00:38 +08:00
pub type VectorSum < T , R1 , R2 > =
Matrix < T , SameShapeR < R1 , R2 > , U1 , SameShapeStorage < T , R1 , U1 , R2 , U1 > > ;
2016-12-05 05:44:42 +08:00
/// The type of the result of a matrix cross product.
2021-04-11 17:00:38 +08:00
pub type MatrixCross < T , R1 , C1 , R2 , C2 > =
Matrix < T , SameShapeR < R1 , R2 > , SameShapeC < C1 , C2 > , SameShapeStorage < T , R1 , C1 , R2 , C2 > > ;
2016-12-05 05:44:42 +08:00
2017-02-13 01:17:09 +08:00
/// The most generic column-major matrix (and vector) type.
///
2020-11-14 01:34:47 +08:00
/// # Methods summary
2020-11-15 23:57:49 +08:00
/// Because `Matrix` is the most generic types used as a common representation of all matrices and
/// vectors of **nalgebra** this documentation page contains every single matrix/vector-related
/// method. In order to make browsing this page simpler, the next subsections contain direct links
/// to groups of methods related to a specific topic.
2020-11-14 01:34:47 +08:00
///
/// #### Vector and matrix construction
/// - [Constructors of statically-sized vectors or statically-sized matrices](#constructors-of-statically-sized-vectors-or-statically-sized-matrices)
2020-11-15 23:57:49 +08:00
/// (`Vector3`, `Matrix3x6`…)
2020-11-14 01:34:47 +08:00
/// - [Constructors of fully dynamic matrices](#constructors-of-fully-dynamic-matrices) (`DMatrix`)
/// - [Constructors of dynamic vectors and matrices with a dynamic number of rows](#constructors-of-dynamic-vectors-and-matrices-with-a-dynamic-number-of-rows)
2020-11-15 23:57:49 +08:00
/// (`DVector`, `MatrixXx3`…)
2020-11-14 01:34:47 +08:00
/// - [Constructors of matrices with a dynamic number of columns](#constructors-of-matrices-with-a-dynamic-number-of-columns)
2020-11-15 23:57:49 +08:00
/// (`Matrix2xX`…)
2020-11-14 01:34:47 +08:00
/// - [Generic constructors](#generic-constructors)
/// (For code generic wrt. the vectors or matrices dimensions.)
2020-11-15 23:57:49 +08:00
///
/// #### Computer graphics utilities for transformations
/// - [2D transformations as a Matrix3 <span style="float:right;">`new_rotation`…</span>](#2d-transformations-as-a-matrix3)
/// - [3D transformations as a Matrix4 <span style="float:right;">`new_rotation`, `new_perspective`, `look_at_rh`…</span>](#3d-transformations-as-a-matrix4)
/// - [Translation and scaling in any dimension <span style="float:right;">`new_scaling`, `new_translation`…</span>](#translation-and-scaling-in-any-dimension)
/// - [Append/prepend translation and scaling <span style="float:right;">`append_scaling`, `prepend_translation_mut`…</span>](#appendprepend-translation-and-scaling)
/// - [Transformation of vectors and points <span style="float:right;">`transform_vector`, `transform_point`…</span>](#transformation-of-vectors-and-points)
///
/// #### Common math operations
/// - [Componentwise operations <span style="float:right;">`component_mul`, `component_div`, `inf`…</span>](#componentwise-operations)
/// - [Special multiplications <span style="float:right;">`tr_mul`, `ad_mul`, `kronecker`…</span>](#special-multiplications)
/// - [Dot/scalar product <span style="float:right;">`dot`, `dotc`, `tr_dot`…</span>](#dotscalar-product)
/// - [Cross product <span style="float:right;">`cross`, `perp`…</span>](#cross-product)
/// - [Magnitude and norms <span style="float:right;">`norm`, `normalize`, `metric_distance`…</span>](#magnitude-and-norms)
/// - [In-place normalization <span style="float:right;">`normalize_mut`, `try_normalize_mut`…</span>](#in-place-normalization)
/// - [Interpolation <span style="float:right;">`lerp`, `slerp`…</span>](#interpolation)
/// - [BLAS functions <span style="float:right;">`gemv`, `gemm`, `syger`…</span>](#blas-functions)
/// - [Swizzling <span style="float:right;">`xx`, `yxz`…</span>](#swizzling)
2021-12-31 04:55:29 +08:00
/// - [Triangular matrix extraction <span style="float:right;">`upper_triangle`, `lower_triangle`</span>](#triangular-matrix-extraction)
2020-11-15 23:57:49 +08:00
///
/// #### Statistics
/// - [Common operations <span style="float:right;">`row_sum`, `column_mean`, `variance`…</span>](#common-statistics-operations)
/// - [Find the min and max components <span style="float:right;">`min`, `max`, `amin`, `amax`, `camin`, `cmax`…</span>](#find-the-min-and-max-components)
/// - [Find the min and max components (vector-specific methods) <span style="float:right;">`argmin`, `argmax`, `icamin`, `icamax`…</span>](#find-the-min-and-max-components-vector-specific-methods)
///
/// #### Iteration, map, and fold
/// - [Iteration on components, rows, and columns <span style="float:right;">`iter`, `column_iter`…</span>](#iteration-on-components-rows-and-columns)
/// - [Elementwise mapping and folding <span style="float:right;">`map`, `fold`, `zip_map`…</span>](#elementwise-mapping-and-folding)
/// - [Folding or columns and rows <span style="float:right;">`compress_rows`, `compress_columns`…</span>](#folding-on-columns-and-rows)
2020-11-14 01:34:47 +08:00
///
/// #### Vector and matrix slicing
2020-11-15 23:57:49 +08:00
/// - [Creating matrix slices from `&[T]` <span style="float:right;">`from_slice`, `from_slice_with_strides`…</span>](#creating-matrix-slices-from-t)
/// - [Creating mutable matrix slices from `&mut [T]` <span style="float:right;">`from_slice_mut`, `from_slice_with_strides_mut`…</span>](#creating-mutable-matrix-slices-from-mut-t)
/// - [Slicing based on index and length <span style="float:right;">`row`, `columns`, `slice`…</span>](#slicing-based-on-index-and-length)
/// - [Mutable slicing based on index and length <span style="float:right;">`row_mut`, `columns_mut`, `slice_mut`…</span>](#mutable-slicing-based-on-index-and-length)
/// - [Slicing based on ranges <span style="float:right;">`rows_range`, `columns_range`…</span>](#slicing-based-on-ranges)
/// - [Mutable slicing based on ranges <span style="float:right;">`rows_range_mut`, `columns_range_mut`…</span>](#mutable-slicing-based-on-ranges)
///
/// #### In-place modification of a single matrix or vector
/// - [In-place filling <span style="float:right;">`fill`, `fill_diagonal`, `fill_with_identity`…</span>](#in-place-filling)
/// - [In-place swapping <span style="float:right;">`swap`, `swap_columns`…</span>](#in-place-swapping)
/// - [Set rows, columns, and diagonal <span style="float:right;">`set_column`, `set_diagonal`…</span>](#set-rows-columns-and-diagonal)
///
/// #### Vector and matrix size modification
/// - [Rows and columns insertion <span style="float:right;">`insert_row`, `insert_column`…</span>](#rows-and-columns-insertion)
/// - [Rows and columns removal <span style="float:right;">`remove_row`, `remove column`…</span>](#rows-and-columns-removal)
/// - [Rows and columns extraction <span style="float:right;">`select_rows`, `select_columns`…</span>](#rows-and-columns-extraction)
/// - [Resizing and reshaping <span style="float:right;">`resize`, `reshape_generic`…</span>](#resizing-and-reshaping)
/// - [In-place resizing <span style="float:right;">`resize_mut`, `resize_vertically_mut`…</span>](#in-place-resizing)
///
/// #### Matrix decomposition
/// - [Rectangular matrix decomposition <span style="float:right;">`qr`, `lu`, `svd`…</span>](#rectangular-matrix-decomposition)
/// - [Square matrix decomposition <span style="float:right;">`cholesky`, `symmetric_eigen`…</span>](#square-matrix-decomposition)
///
/// #### Vector basis computation
/// - [Basis and orthogonalization <span style="float:right;">`orthonormal_subspace_basis`, `orthonormalize`…</span>](#basis-and-orthogonalization)
2020-11-14 01:34:47 +08:00
///
/// # Type parameters
/// The generic `Matrix` type has four type parameters:
2021-04-11 17:00:38 +08:00
/// - `T`: for the matrix components scalar type.
2017-02-13 01:17:09 +08:00
/// - `R`: for the matrix number of rows.
/// - `C`: for the matrix number of columns.
/// - `S`: for the matrix data storage, i.e., the buffer that actually contains the matrix
/// components.
///
/// The matrix dimensions parameters `R` and `C` can either be:
2018-09-24 12:48:42 +08:00
/// - type-level unsigned integer constants (e.g. `U1`, `U124`) from the `nalgebra::` root module.
2017-02-13 01:17:09 +08:00
/// All numbers from 0 to 127 are defined that way.
/// - type-level unsigned integer constants (e.g. `U1024`, `U10000`) from the `typenum::` crate.
/// Using those, you will not get error messages as nice as for numbers smaller than 128 defined on
/// the `nalgebra::` module.
/// - the special value `Dynamic` from the `nalgebra::` root module. This indicates that the
/// specified dimension is not known at compile-time. Note that this will generally imply that the
/// matrix data storage `S` performs a dynamic allocation and contains extra metadata for the
/// matrix shape.
///
/// Note that mixing `Dynamic` with type-level unsigned integers is allowed. Actually, a
2021-04-11 17:00:38 +08:00
/// dynamically-sized column vector should be represented as a `Matrix<T, Dynamic, U1, S>` (given
/// some concrete types for `T` and a compatible data storage type `S`).
2021-08-03 00:41:46 +08:00
#[ repr(C) ]
#[ derive(Clone, Copy) ]
2022-03-17 01:07:29 +08:00
#[ cfg_attr(feature = " cuda " , derive(cust_core::DeviceCopy)) ]
2021-04-12 15:18:17 +08:00
pub struct Matrix < T , R , C , S > {
2020-11-14 01:34:47 +08:00
/// The data storage that contains all the matrix components. Disappointed?
///
/// Well, if you came here to see how you can access the matrix components,
/// you may be in luck: you can access the individual components of all vectors with compile-time
/// dimensions <= 6 using field notation like this:
/// `vec.x`, `vec.y`, `vec.z`, `vec.w`, `vec.a`, `vec.b`. Reference and assignation work too:
2020-11-15 23:57:49 +08:00
/// ```
/// # use nalgebra::Vector3;
/// let mut vec = Vector3::new(1.0, 2.0, 3.0);
2020-11-14 01:34:47 +08:00
/// vec.x = 10.0;
2020-11-15 23:57:49 +08:00
/// vec.y += 30.0;
/// assert_eq!(vec.x, 10.0);
/// assert_eq!(vec.y + 100.0, 132.0);
2020-11-14 01:34:47 +08:00
/// ```
/// Similarly, for matrices with compile-time dimensions <= 6, you can use field notation
/// like this: `mat.m11`, `mat.m42`, etc. The first digit identifies the row to address
/// and the second digit identifies the column to address. So `mat.m13` identifies the component
/// at the first row and third column (note that the count of rows and columns start at 1 instead
/// of 0 here. This is so we match the mathematical notation).
///
/// For all matrices and vectors, independently from their size, individual components can
/// be accessed and modified using indexing: `vec[20]`, `mat[(20, 19)]`. Here the indexing
/// starts at 0 as you would expect.
2017-08-03 01:37:44 +08:00
pub data : S ,
2017-02-13 01:17:09 +08:00
2021-04-12 17:27:21 +08:00
// NOTE: the fact that this field is private is important because
// this prevents the user from constructing a matrix with
// dimensions R, C that don't match the dimension of the
// storage S. Instead they have to use the unsafe function
// from_data_statically_unchecked.
// Note that it would probably make sense to just have
// the type `Matrix<S>`, and have `T, R, C` be associated-types
2021-08-03 00:41:46 +08:00
// of the `RawStorage` trait. However, because we don't have
2021-12-30 01:46:01 +08:00
// specialization, this is not possible because these `T, R, C`
2021-08-03 00:41:46 +08:00
// allows us to desambiguate a lot of configurations.
2021-04-11 17:00:38 +08:00
_phantoms : PhantomData < ( T , R , C ) > ,
2016-12-05 05:44:42 +08:00
}
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S : fmt ::Debug > fmt ::Debug for Matrix < T , R , C , S > {
fn fmt ( & self , formatter : & mut fmt ::Formatter < '_ > ) -> Result < ( ) , fmt ::Error > {
2021-09-13 01:52:46 +08:00
self . data . fmt ( formatter )
2021-08-03 00:41:46 +08:00
}
}
impl < T , R , C , S > Default for Matrix < T , R , C , S >
2021-07-18 09:19:20 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
R : Dim ,
C : Dim ,
S : Default ,
2021-07-18 09:19:20 +08:00
{
2019-04-16 16:11:27 +08:00
fn default ( ) -> Self {
2021-08-03 00:41:46 +08:00
Matrix {
data : Default ::default ( ) ,
_phantoms : PhantomData ,
}
2019-04-16 16:11:27 +08:00
}
}
2021-04-12 18:14:16 +08:00
#[ cfg(feature = " serde-serialize-no-std " ) ]
2021-08-03 00:41:46 +08:00
impl < T , R , C , S > Serialize for Matrix < T , R , C , S >
where
T : Scalar ,
R : Dim ,
C : Dim ,
S : Serialize ,
{
2021-04-11 17:00:38 +08:00
fn serialize < Ser > ( & self , serializer : Ser ) -> Result < Ser ::Ok , Ser ::Error >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
Ser : Serializer ,
2020-04-06 00:49:48 +08:00
{
2017-05-04 10:02:30 +08:00
self . data . serialize ( serializer )
}
}
2021-04-12 18:14:16 +08:00
#[ cfg(feature = " serde-serialize-no-std " ) ]
2021-08-03 00:41:46 +08:00
impl < ' de , T , R , C , S > Deserialize < ' de > for Matrix < T , R , C , S >
where
T : Scalar ,
R : Dim ,
C : Dim ,
S : Deserialize < ' de > ,
{
2017-05-04 10:02:30 +08:00
fn deserialize < D > ( deserializer : D ) -> Result < Self , D ::Error >
2020-04-06 00:49:48 +08:00
where
D : Deserializer < ' de > ,
{
2018-02-02 19:26:35 +08:00
S ::deserialize ( deserializer ) . map ( | x | Matrix {
data : x ,
_phantoms : PhantomData ,
} )
2017-05-04 10:02:30 +08:00
}
}
2019-08-12 00:26:12 +08:00
#[ cfg(feature = " compare " ) ]
2021-08-03 00:41:46 +08:00
impl < T : Scalar , R : Dim , C : Dim , S : RawStorage < T , R , C > > matrixcompare_core ::Matrix < T >
2021-04-11 17:00:38 +08:00
for Matrix < T , R , C , S >
2020-06-30 01:03:20 +08:00
{
2019-08-12 00:26:12 +08:00
fn rows ( & self ) -> usize {
self . nrows ( )
}
fn cols ( & self ) -> usize {
self . ncols ( )
}
2021-07-26 01:06:14 +08:00
fn access ( & self ) -> matrixcompare_core ::Access < '_ , T > {
2019-08-12 00:26:12 +08:00
matrixcompare_core ::Access ::Dense ( self )
}
}
#[ cfg(feature = " compare " ) ]
2021-08-03 00:41:46 +08:00
impl < T : Scalar , R : Dim , C : Dim , S : RawStorage < T , R , C > > matrixcompare_core ::DenseAccess < T >
2021-04-11 17:00:38 +08:00
for Matrix < T , R , C , S >
2020-06-30 01:03:20 +08:00
{
2021-04-11 17:00:38 +08:00
fn fetch_single ( & self , row : usize , col : usize ) -> T {
2019-08-12 00:26:12 +08:00
self . index ( ( row , col ) ) . clone ( )
}
}
2021-02-11 12:12:09 +08:00
#[ cfg(feature = " bytemuck " ) ]
2021-08-03 00:41:46 +08:00
unsafe impl < T : Scalar , R : Dim , C : Dim , S : RawStorage < T , R , C > > bytemuck ::Zeroable
for Matrix < T , R , C , S >
where
S : bytemuck ::Zeroable ,
2021-02-11 12:12:09 +08:00
{
}
#[ cfg(feature = " bytemuck " ) ]
2021-08-03 00:41:46 +08:00
unsafe impl < T : Scalar , R : Dim , C : Dim , S : RawStorage < T , R , C > > bytemuck ::Pod for Matrix < T , R , C , S >
2021-02-11 12:12:09 +08:00
where
2021-02-25 21:10:34 +08:00
S : bytemuck ::Pod ,
Self : Copy ,
2021-02-11 12:12:09 +08:00
{
}
2021-05-07 08:59:02 +08:00
#[ cfg(feature = " rkyv-serialize-no-std " ) ]
2021-05-09 23:28:49 +08:00
mod rkyv_impl {
use super ::Matrix ;
use core ::marker ::PhantomData ;
use rkyv ::{ offset_of , project_struct , Archive , Deserialize , Fallible , Serialize } ;
impl < T : Archive , R : Archive , C : Archive , S : Archive > Archive for Matrix < T , R , C , S > {
type Archived = Matrix < T ::Archived , R ::Archived , C ::Archived , S ::Archived > ;
type Resolver = S ::Resolver ;
fn resolve (
& self ,
pos : usize ,
resolver : Self ::Resolver ,
2021-08-03 16:07:05 +08:00
out : & mut core ::mem ::MaybeUninit < Self ::Archived > ,
2021-05-09 23:28:49 +08:00
) {
self . data . resolve (
pos + offset_of! ( Self ::Archived , data ) ,
resolver ,
project_struct! ( out : Self ::Archived = > data ) ,
) ;
}
2021-05-07 08:59:02 +08:00
}
2021-05-09 23:28:49 +08:00
impl < T : Archive , R : Archive , C : Archive , S : Serialize < _S > , _S : Fallible + ? Sized > Serialize < _S >
for Matrix < T , R , C , S >
{
fn serialize ( & self , serializer : & mut _S ) -> Result < Self ::Resolver , _S ::Error > {
2021-07-04 11:19:07 +08:00
self . data . serialize ( serializer )
2021-05-09 23:28:49 +08:00
}
2021-05-07 08:59:02 +08:00
}
2021-05-09 23:28:49 +08:00
impl < T : Archive , R : Archive , C : Archive , S : Archive , D : Fallible + ? Sized >
Deserialize < Matrix < T , R , C , S > , D >
for Matrix < T ::Archived , R ::Archived , C ::Archived , S ::Archived >
where
S ::Archived : Deserialize < S , D > ,
{
fn deserialize ( & self , deserializer : & mut D ) -> Result < Matrix < T , R , C , S > , D ::Error > {
Ok ( Matrix {
data : self . data . deserialize ( deserializer ) ? ,
_phantoms : PhantomData ,
} )
}
2021-05-07 08:59:02 +08:00
}
}
2021-04-12 15:18:17 +08:00
impl < T , R , C , S > Matrix < T , R , C , S > {
2021-08-03 00:41:46 +08:00
/// Creates a new matrix with the given data without statically checking that the matrix
/// dimension matches the storage dimension.
2021-04-12 15:18:17 +08:00
#[ inline(always) ]
pub const unsafe fn from_data_statically_unchecked ( data : S ) -> Matrix < T , R , C , S > {
2016-12-05 05:44:42 +08:00
Matrix {
2020-10-11 16:57:26 +08:00
data ,
2018-02-02 19:26:35 +08:00
_phantoms : PhantomData ,
2016-12-05 05:44:42 +08:00
}
}
}
2021-05-07 15:00:29 +08:00
impl < T , const R : usize , const C : usize > SMatrix < T , R , C > {
2021-07-28 07:18:29 +08:00
/// Creates a new statically-allocated matrix from the given [`ArrayStorage`].
2021-05-07 15:00:29 +08:00
///
/// This method exists primarily as a workaround for the fact that `from_data` can not
/// work in `const fn` contexts.
#[ inline(always) ]
pub const fn from_array_storage ( storage : ArrayStorage < T , R , C > ) -> Self {
2021-08-03 00:41:46 +08:00
// This is sound because the row and column types are exactly the same as that of the
// storage, so there can be no mismatch
2021-05-07 15:00:29 +08:00
unsafe { Self ::from_data_statically_unchecked ( storage ) }
}
}
// TODO: Consider removing/deprecating `from_vec_storage` once we are able to make
// `from_data` const fn compatible
2021-05-07 15:09:10 +08:00
#[ cfg(any(feature = " std " , feature = " alloc " )) ]
2021-05-07 15:00:29 +08:00
impl < T > DMatrix < T > {
2021-07-28 07:18:29 +08:00
/// Creates a new heap-allocated matrix from the given [`VecStorage`].
2021-05-07 15:00:29 +08:00
///
/// This method exists primarily as a workaround for the fact that `from_data` can not
/// work in `const fn` contexts.
pub const fn from_vec_storage ( storage : VecStorage < T , Dynamic , Dynamic > ) -> Self {
2021-08-03 00:41:46 +08:00
// This is sound because the dimensions of the matrix and the storage are guaranteed
// to be the same
2021-05-07 15:00:29 +08:00
unsafe { Self ::from_data_statically_unchecked ( storage ) }
}
}
// TODO: Consider removing/deprecating `from_vec_storage` once we are able to make
// `from_data` const fn compatible
2021-05-07 15:09:10 +08:00
#[ cfg(any(feature = " std " , feature = " alloc " )) ]
2021-05-07 15:00:29 +08:00
impl < T > DVector < T > {
2021-07-28 07:18:29 +08:00
/// Creates a new heap-allocated matrix from the given [`VecStorage`].
2021-05-07 15:00:29 +08:00
///
/// This method exists primarily as a workaround for the fact that `from_data` can not
/// work in `const fn` contexts.
pub const fn from_vec_storage ( storage : VecStorage < T , Dynamic , U1 > ) -> Self {
2021-08-03 00:41:46 +08:00
// This is sound because the dimensions of the matrix and the storage are guaranteed
// to be the same
2021-05-07 15:00:29 +08:00
unsafe { Self ::from_data_statically_unchecked ( storage ) }
}
}
2021-08-26 17:07:00 +08:00
// TODO: Consider removing/deprecating `from_vec_storage` once we are able to make
// `from_data` const fn compatible
#[ cfg(any(feature = " std " , feature = " alloc " )) ]
impl < T > RowDVector < T > {
/// Creates a new heap-allocated matrix from the given [`VecStorage`].
///
/// This method exists primarily as a workaround for the fact that `from_data` can not
/// work in `const fn` contexts.
pub const fn from_vec_storage ( storage : VecStorage < T , U1 , Dynamic > ) -> Self {
// This is sound because the dimensions of the matrix and the storage are guaranteed
// to be the same
unsafe { Self ::from_data_statically_unchecked ( storage ) }
}
}
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim > UninitMatrix < T , R , C >
where
DefaultAllocator : Allocator < T , R , C > ,
{
/// Assumes a matrix's entries to be initialized. This operation should be near zero-cost.
///
/// # Safety
/// The user must make sure that every single entry of the buffer has been initialized,
2021-08-04 17:19:57 +08:00
/// or Undefined Behavior will immediately occur.
#[ inline(always) ]
2021-08-03 00:41:46 +08:00
pub unsafe fn assume_init ( self ) -> OMatrix < T , R , C > {
OMatrix ::from_data ( < DefaultAllocator as Allocator < T , R , C > > ::assume_init (
self . data ,
) )
}
}
impl < T , R : Dim , C : Dim , S : RawStorage < T , R , C > > Matrix < T , R , C , S > {
2016-12-05 05:44:42 +08:00
/// Creates a new matrix with the given data.
2021-04-12 15:18:17 +08:00
#[ inline(always) ]
2019-02-17 05:29:41 +08:00
pub fn from_data ( data : S ) -> Self {
2018-02-02 19:26:35 +08:00
unsafe { Self ::from_data_statically_unchecked ( data ) }
2016-12-05 05:44:42 +08:00
}
2017-02-13 01:17:09 +08:00
/// The shape of this matrix returned as the tuple (number of rows, number of columns).
2018-10-16 04:44:01 +08:00
///
2022-03-23 00:53:46 +08:00
/// # Example
2018-10-16 04:44:01 +08:00
/// ```
/// # use nalgebra::Matrix3x4;
/// let mat = Matrix3x4::<f32>::zeros();
/// assert_eq!(mat.shape(), (3, 4));
2022-03-23 00:53:46 +08:00
/// ```
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2016-12-05 05:44:42 +08:00
pub fn shape ( & self ) -> ( usize , usize ) {
2021-08-03 00:41:46 +08:00
let ( nrows , ncols ) = self . shape_generic ( ) ;
2016-12-05 05:44:42 +08:00
( nrows . value ( ) , ncols . value ( ) )
}
2021-08-03 23:26:56 +08:00
/// The shape of this matrix wrapped into their representative types (`Const` or `Dynamic`).
2021-08-03 00:41:46 +08:00
#[ inline ]
#[ must_use ]
pub fn shape_generic ( & self ) -> ( R , C ) {
self . data . shape ( )
}
2016-12-05 05:44:42 +08:00
/// The number of rows of this matrix.
2018-10-16 04:44:01 +08:00
///
2022-03-23 00:53:46 +08:00
/// # Example
2018-10-16 04:44:01 +08:00
/// ```
/// # use nalgebra::Matrix3x4;
/// let mat = Matrix3x4::<f32>::zeros();
/// assert_eq!(mat.nrows(), 3);
2022-03-23 00:53:46 +08:00
/// ```
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2016-12-05 05:44:42 +08:00
pub fn nrows ( & self ) -> usize {
self . shape ( ) . 0
}
/// The number of columns of this matrix.
2018-10-16 04:44:01 +08:00
///
2022-03-23 00:53:46 +08:00
/// # Example
2018-10-16 04:44:01 +08:00
/// ```
/// # use nalgebra::Matrix3x4;
/// let mat = Matrix3x4::<f32>::zeros();
/// assert_eq!(mat.ncols(), 4);
2022-03-23 00:53:46 +08:00
/// ```
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2016-12-05 05:44:42 +08:00
pub fn ncols ( & self ) -> usize {
self . shape ( ) . 1
}
/// The strides (row stride, column stride) of this matrix.
2018-10-16 04:44:01 +08:00
///
2022-03-23 00:53:46 +08:00
/// # Example
2018-10-16 04:44:01 +08:00
/// ```
/// # use nalgebra::DMatrix;
/// let mat = DMatrix::<f32>::zeros(10, 10);
/// let slice = mat.slice_with_steps((0, 0), (5, 3), (1, 2));
/// // The column strides is the number of steps (here 2) multiplied by the corresponding dimension.
/// assert_eq!(mat.strides(), (1, 10));
2022-03-23 00:53:46 +08:00
/// ```
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2016-12-05 05:44:42 +08:00
pub fn strides ( & self ) -> ( usize , usize ) {
let ( srows , scols ) = self . data . strides ( ) ;
( srows . value ( ) , scols . value ( ) )
}
/// Computes the row and column coordinates of the i-th element of this matrix seen as a
/// vector.
2020-04-03 17:39:49 +08:00
///
/// # Example
/// ```
/// # use nalgebra::Matrix2;
/// let m = Matrix2::new(1, 2,
/// 3, 4);
/// let i = m.vector_to_matrix_index(3);
/// assert_eq!(i, (1, 1));
/// assert_eq!(m[i], m[3]);
/// ```
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2016-12-05 05:44:42 +08:00
pub fn vector_to_matrix_index ( & self , i : usize ) -> ( usize , usize ) {
let ( nrows , ncols ) = self . shape ( ) ;
// Two most common uses that should be optimized by the compiler for statically-sized
// matrices.
if nrows = = 1 {
( 0 , i )
2018-02-02 19:26:35 +08:00
} else if ncols = = 1 {
2016-12-05 05:44:42 +08:00
( i , 0 )
2018-02-02 19:26:35 +08:00
} else {
2016-12-05 05:44:42 +08:00
( i % nrows , i / nrows )
}
}
2018-11-20 05:51:01 +08:00
/// Returns a pointer to the start of the matrix.
///
/// If the matrix is not empty, this pointer is guaranteed to be aligned
/// and non-null.
2020-04-03 17:56:08 +08:00
///
/// # Example
/// ```
/// # use nalgebra::Matrix2;
/// let m = Matrix2::new(1, 2,
/// 3, 4);
/// let ptr = m.as_ptr();
/// assert_eq!(unsafe { *ptr }, m[0]);
/// ```
2018-11-20 05:51:01 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn as_ptr ( & self ) -> * const T {
2018-11-20 05:51:01 +08:00
self . data . ptr ( )
}
2016-12-05 05:44:42 +08:00
/// Tests whether `self` and `rhs` are equal up to a given epsilon.
///
2018-05-19 21:41:58 +08:00
/// See `relative_eq` from the `RelativeEq` trait for more details.
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn relative_eq < R2 , C2 , SB > (
2018-02-02 19:26:35 +08:00
& self ,
2021-04-11 17:00:38 +08:00
other : & Matrix < T , R2 , C2 , SB > ,
eps : T ::Epsilon ,
max_relative : T ::Epsilon ,
2018-02-02 19:26:35 +08:00
) -> bool
where
2021-04-11 17:00:38 +08:00
T : RelativeEq ,
2021-08-03 00:41:46 +08:00
R2 : Dim ,
C2 : Dim ,
2021-04-11 17:00:38 +08:00
SB : Storage < T , R2 , C2 > ,
2021-08-04 23:34:25 +08:00
T ::Epsilon : Clone ,
2018-02-02 19:26:35 +08:00
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
{
assert! ( self . shape ( ) = = other . shape ( ) ) ;
self . iter ( )
. zip ( other . iter ( ) )
2021-08-04 23:34:25 +08:00
. all ( | ( a , b ) | a . relative_eq ( b , eps . clone ( ) , max_relative . clone ( ) ) )
2016-12-05 05:44:42 +08:00
}
2017-08-03 01:37:44 +08:00
/// Tests whether `self` and `rhs` are exactly equal.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-06-18 15:45:37 +08:00
#[ allow(clippy::should_implement_trait) ]
2021-08-03 00:41:46 +08:00
pub fn eq < R2 , C2 , SB > ( & self , other : & Matrix < T , R2 , C2 , SB > ) -> bool
2018-02-02 19:26:35 +08:00
where
2021-04-11 17:00:38 +08:00
T : PartialEq ,
2021-08-03 00:41:46 +08:00
R2 : Dim ,
C2 : Dim ,
SB : RawStorage < T , R2 , C2 > ,
2018-02-02 19:26:35 +08:00
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
{
assert! ( self . shape ( ) = = other . shape ( ) ) ;
self . iter ( ) . zip ( other . iter ( ) ) . all ( | ( a , b ) | * a = = * b )
2017-08-03 01:37:44 +08:00
}
/// Moves this matrix into one that owns its data.
#[ inline ]
2021-04-11 17:00:38 +08:00
pub fn into_owned ( self ) -> OMatrix < T , R , C >
2020-04-06 00:49:48 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
S : Storage < T , R , C > ,
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , R , C > ,
2020-04-06 00:49:48 +08:00
{
2021-08-03 00:41:46 +08:00
Matrix ::from_data ( self . data . into_owned ( ) )
2017-08-03 01:37:44 +08:00
}
2020-11-15 23:57:49 +08:00
// TODO: this could probably benefit from specialization.
2017-08-03 01:37:44 +08:00
// XXX: bad name.
/// Moves this matrix into one that owns its data. The actual type of the result depends on
/// matrix storage combination rules for addition.
#[ inline ]
2021-08-03 00:41:46 +08:00
pub fn into_owned_sum < R2 , C2 > ( self ) -> MatrixSum < T , R , C , R2 , C2 >
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
S : Storage < T , R , C > ,
R2 : Dim ,
C2 : Dim ,
2021-04-11 17:00:38 +08:00
DefaultAllocator : SameShapeAllocator < T , R , C , R2 , C2 > ,
2018-02-02 19:26:35 +08:00
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
{
2021-08-03 00:41:46 +08:00
if TypeId ::of ::< SameShapeStorage < T , R , C , R2 , C2 > > ( ) = = TypeId ::of ::< Owned < T , R , C > > ( ) {
// We can just return `self.into_owned()`.
2017-08-03 01:37:44 +08:00
unsafe {
2021-08-03 00:41:46 +08:00
// TODO: check that those copies are optimized away by the compiler.
let owned = self . into_owned ( ) ;
let res = mem ::transmute_copy ( & owned ) ;
mem ::forget ( owned ) ;
res
2017-08-03 01:37:44 +08:00
}
2018-02-02 19:26:35 +08:00
} else {
2017-08-03 01:37:44 +08:00
self . clone_owned_sum ( )
}
}
2017-08-14 01:53:02 +08:00
/// Clones this matrix to one that owns its data.
2017-08-03 01:37:44 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn clone_owned ( & self ) -> OMatrix < T , R , C >
2020-04-06 00:49:48 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
S : Storage < T , R , C > ,
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , R , C > ,
2020-04-06 00:49:48 +08:00
{
2021-08-03 00:41:46 +08:00
Matrix ::from_data ( self . data . clone_owned ( ) )
2017-08-03 01:37:44 +08:00
}
/// Clones this matrix into one that owns its data. The actual type of the result depends on
/// matrix storage combination rules for addition.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn clone_owned_sum < R2 , C2 > ( & self ) -> MatrixSum < T , R , C , R2 , C2 >
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
S : Storage < T , R , C > ,
R2 : Dim ,
C2 : Dim ,
2021-04-11 17:00:38 +08:00
DefaultAllocator : SameShapeAllocator < T , R , C , R2 , C2 > ,
2018-02-02 19:26:35 +08:00
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
{
2017-08-03 01:37:44 +08:00
let ( nrows , ncols ) = self . shape ( ) ;
let nrows : SameShapeR < R , R2 > = Dim ::from_usize ( nrows ) ;
let ncols : SameShapeC < C , C2 > = Dim ::from_usize ( ncols ) ;
2021-08-03 00:41:46 +08:00
let mut res = Matrix ::uninit ( nrows , ncols ) ;
2017-08-03 01:37:44 +08:00
2021-08-03 00:41:46 +08:00
unsafe {
// TODO: use copy_from?
for j in 0 .. res . ncols ( ) {
for i in 0 .. res . nrows ( ) {
2021-07-15 02:24:27 +08:00
* res . get_unchecked_mut ( ( i , j ) ) =
2021-08-04 23:34:25 +08:00
MaybeUninit ::new ( self . get_unchecked ( ( i , j ) ) . clone ( ) ) ;
2018-02-02 19:26:35 +08:00
}
2017-08-03 01:37:44 +08:00
}
2021-08-03 00:41:46 +08:00
// SAFETY: the output has been initialized above.
res . assume_init ( )
}
2017-08-03 01:37:44 +08:00
}
2021-08-03 00:41:46 +08:00
/// Transposes `self` and store the result into `out`.
2020-11-15 23:57:49 +08:00
#[ inline ]
2021-08-03 00:41:46 +08:00
fn transpose_to_uninit < Status , R2 , C2 , SB > (
& self ,
2021-09-26 03:05:31 +08:00
_status : Status ,
2021-08-03 00:41:46 +08:00
out : & mut Matrix < Status ::Value , R2 , C2 , SB > ,
) where
Status : InitStatus < T > ,
T : Scalar ,
R2 : Dim ,
C2 : Dim ,
SB : RawStorageMut < Status ::Value , R2 , C2 > ,
2020-11-15 23:57:49 +08:00
ShapeConstraint : SameNumberOfRows < R , C2 > + SameNumberOfColumns < C , R2 > ,
{
let ( nrows , ncols ) = self . shape ( ) ;
assert! (
( ncols , nrows ) = = out . shape ( ) ,
2021-08-03 00:41:46 +08:00
" Incompatible shape for transposition. "
2020-11-15 23:57:49 +08:00
) ;
// TODO: optimize that.
for i in 0 .. nrows {
for j in 0 .. ncols {
2021-08-03 00:41:46 +08:00
// Safety: the indices are in range.
2020-11-15 23:57:49 +08:00
unsafe {
2021-08-03 00:41:46 +08:00
Status ::init (
out . get_unchecked_mut ( ( j , i ) ) ,
2021-08-04 23:34:25 +08:00
self . get_unchecked ( ( i , j ) ) . clone ( ) ,
2021-08-03 00:41:46 +08:00
) ;
2020-11-15 23:57:49 +08:00
}
}
}
}
2021-08-03 00:41:46 +08:00
/// Transposes `self` and store the result into `out`.
#[ inline ]
pub fn transpose_to < R2 , C2 , SB > ( & self , out : & mut Matrix < T , R2 , C2 , SB > )
where
T : Scalar ,
R2 : Dim ,
C2 : Dim ,
SB : RawStorageMut < T , R2 , C2 > ,
ShapeConstraint : SameNumberOfRows < R , C2 > + SameNumberOfColumns < C , R2 > ,
{
self . transpose_to_uninit ( Init , out )
}
2020-11-15 23:57:49 +08:00
/// Transposes `self`.
#[ inline ]
#[ must_use = " Did you mean to use transpose_mut()? " ]
2021-04-11 17:00:38 +08:00
pub fn transpose ( & self ) -> OMatrix < T , C , R >
2020-11-15 23:57:49 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , C , R > ,
2020-11-15 23:57:49 +08:00
{
2021-08-03 00:41:46 +08:00
let ( nrows , ncols ) = self . shape_generic ( ) ;
2021-07-17 12:17:56 +08:00
2021-08-03 00:41:46 +08:00
let mut res = Matrix ::uninit ( ncols , nrows ) ;
self . transpose_to_uninit ( Uninit , & mut res ) ;
// Safety: res is now fully initialized.
unsafe { res . assume_init ( ) }
2020-11-15 23:57:49 +08:00
}
}
/// # Elementwise mapping and folding
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S : RawStorage < T , R , C > > Matrix < T , R , C , S > {
2017-08-03 01:37:44 +08:00
/// Returns a matrix containing the result of `f` applied to each of its entries.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn map < T2 : Scalar , F : FnMut ( T ) -> T2 > ( & self , mut f : F ) -> OMatrix < T2 , R , C >
2020-04-06 00:49:48 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T2 , R , C > ,
2020-04-06 00:49:48 +08:00
{
2021-08-03 00:41:46 +08:00
let ( nrows , ncols ) = self . shape_generic ( ) ;
let mut res = Matrix ::uninit ( nrows , ncols ) ;
2017-08-03 01:37:44 +08:00
2018-02-02 19:26:35 +08:00
for j in 0 .. ncols . value ( ) {
for i in 0 .. nrows . value ( ) {
2021-08-03 00:41:46 +08:00
// Safety: all indices are in range.
2017-08-03 01:37:44 +08:00
unsafe {
2021-08-04 23:34:25 +08:00
let a = self . data . get_unchecked ( i , j ) . clone ( ) ;
2021-07-15 02:24:27 +08:00
* res . data . get_unchecked_mut ( i , j ) = MaybeUninit ::new ( f ( a ) ) ;
2017-08-03 01:37:44 +08:00
}
}
}
2021-08-03 00:41:46 +08:00
// Safety: res is now fully initialized.
2021-07-15 02:24:27 +08:00
unsafe { res . assume_init ( ) }
2017-08-03 01:37:44 +08:00
}
2021-03-06 00:08:46 +08:00
/// Cast the components of `self` to another type.
///
/// # Example
/// ```
/// # use nalgebra::Vector3;
/// let q = Vector3::new(1.0f64, 2.0, 3.0);
/// let q2 = q.cast::<f32>();
/// assert_eq!(q2, Vector3::new(1.0f32, 2.0, 3.0));
/// ```
2021-08-03 00:41:46 +08:00
pub fn cast < T2 : Scalar > ( self ) -> OMatrix < T2 , R , C >
2021-03-06 00:08:46 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
2021-04-11 17:00:38 +08:00
OMatrix < T2 , R , C > : SupersetOf < Self > ,
DefaultAllocator : Allocator < T2 , R , C > ,
2021-03-06 00:08:46 +08:00
{
crate ::convert ( self )
}
2020-03-18 00:58:36 +08:00
/// Similar to `self.iter().fold(init, f)` except that `init` is replaced by a closure.
///
/// The initialization closure is given the first component of this matrix:
/// - If the matrix has no component (0 rows or 0 columns) then `init_f` is called with `None`
/// and its return value is the value returned by this method.
/// - If the matrix has has least one component, then `init_f` is called with the first component
/// to compute the initial value. Folding then continues on all the remaining components of the matrix.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn fold_with < T2 > (
2020-03-18 00:58:36 +08:00
& self ,
2021-04-11 17:00:38 +08:00
init_f : impl FnOnce ( Option < & T > ) -> T2 ,
f : impl FnMut ( T2 , & T ) -> T2 ,
2021-08-03 00:41:46 +08:00
) -> T2
where
T : Scalar ,
{
2020-03-18 00:58:36 +08:00
let mut it = self . iter ( ) ;
let init = init_f ( it . next ( ) ) ;
it . fold ( init , f )
}
2018-09-26 03:27:27 +08:00
/// Returns a matrix containing the result of `f` applied to each of its entries. Unlike `map`,
2019-01-30 09:31:10 +08:00
/// `f` also gets passed the row and column index, i.e. `f(row, col, value)`.
2018-09-26 03:27:27 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn map_with_location < T2 : Scalar , F : FnMut ( usize , usize , T ) -> T2 > (
2018-10-16 04:44:01 +08:00
& self ,
mut f : F ,
2021-04-11 17:00:38 +08:00
) -> OMatrix < T2 , R , C >
2018-09-26 03:27:27 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T2 , R , C > ,
2018-09-26 03:27:27 +08:00
{
2021-08-03 00:41:46 +08:00
let ( nrows , ncols ) = self . shape_generic ( ) ;
let mut res = Matrix ::uninit ( nrows , ncols ) ;
2018-09-26 03:27:27 +08:00
for j in 0 .. ncols . value ( ) {
for i in 0 .. nrows . value ( ) {
2021-08-03 00:41:46 +08:00
// Safety: all indices are in range.
2018-09-26 03:27:27 +08:00
unsafe {
2021-08-04 23:34:25 +08:00
let a = self . data . get_unchecked ( i , j ) . clone ( ) ;
2021-07-15 06:21:22 +08:00
* res . data . get_unchecked_mut ( i , j ) = MaybeUninit ::new ( f ( i , j , a ) ) ;
2018-09-26 03:27:27 +08:00
}
}
}
2021-08-03 00:41:46 +08:00
// Safety: res is now fully initialized.
2021-07-15 06:21:22 +08:00
unsafe { res . assume_init ( ) }
2018-09-26 03:27:27 +08:00
}
2017-08-03 01:37:44 +08:00
/// Returns a matrix containing the result of `f` applied to each entries of `self` and
/// `rhs`.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn zip_map < T2 , N3 , S2 , F > ( & self , rhs : & Matrix < T2 , R , C , S2 > , mut f : F ) -> OMatrix < N3 , R , C >
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
T2 : Scalar ,
N3 : Scalar ,
S2 : RawStorage < T2 , R , C > ,
2021-04-11 17:00:38 +08:00
F : FnMut ( T , T2 ) -> N3 ,
2018-02-02 19:26:35 +08:00
DefaultAllocator : Allocator < N3 , R , C > ,
{
2021-08-03 00:41:46 +08:00
let ( nrows , ncols ) = self . shape_generic ( ) ;
let mut res = Matrix ::uninit ( nrows , ncols ) ;
2017-08-03 01:37:44 +08:00
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
assert_eq! (
( nrows . value ( ) , ncols . value ( ) ) ,
rhs . shape ( ) ,
2018-02-02 19:26:35 +08:00
" Matrix simultaneous traversal error: dimension mismatch. "
) ;
2017-08-03 01:37:44 +08:00
2018-02-02 19:26:35 +08:00
for j in 0 .. ncols . value ( ) {
for i in 0 .. nrows . value ( ) {
2021-08-03 00:41:46 +08:00
// Safety: all indices are in range.
2017-08-03 01:37:44 +08:00
unsafe {
2021-08-04 23:34:25 +08:00
let a = self . data . get_unchecked ( i , j ) . clone ( ) ;
let b = rhs . data . get_unchecked ( i , j ) . clone ( ) ;
2021-08-03 00:41:46 +08:00
* res . data . get_unchecked_mut ( i , j ) = MaybeUninit ::new ( f ( a , b ) )
2017-08-03 01:37:44 +08:00
}
}
}
2021-08-03 00:41:46 +08:00
// Safety: res is now fully initialized.
2021-07-15 02:24:27 +08:00
unsafe { res . assume_init ( ) }
2017-08-03 01:37:44 +08:00
}
2018-09-22 21:38:33 +08:00
/// Returns a matrix containing the result of `f` applied to each entries of `self` and
/// `b`, and `c`.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn zip_zip_map < T2 , N3 , N4 , S2 , S3 , F > (
2018-10-16 04:44:01 +08:00
& self ,
2021-04-11 17:00:38 +08:00
b : & Matrix < T2 , R , C , S2 > ,
2018-10-16 04:44:01 +08:00
c : & Matrix < N3 , R , C , S3 > ,
mut f : F ,
2021-04-11 17:00:38 +08:00
) -> OMatrix < N4 , R , C >
2018-10-16 04:44:01 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
T2 : Scalar ,
N3 : Scalar ,
N4 : Scalar ,
S2 : RawStorage < T2 , R , C > ,
S3 : RawStorage < N3 , R , C > ,
2021-04-11 17:00:38 +08:00
F : FnMut ( T , T2 , N3 ) -> N4 ,
2018-10-16 04:44:01 +08:00
DefaultAllocator : Allocator < N4 , R , C > ,
2018-09-22 21:38:33 +08:00
{
2021-08-03 00:41:46 +08:00
let ( nrows , ncols ) = self . shape_generic ( ) ;
let mut res = Matrix ::uninit ( nrows , ncols ) ;
2018-09-22 21:38:33 +08:00
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
assert_eq! (
( nrows . value ( ) , ncols . value ( ) ) ,
b . shape ( ) ,
" Matrix simultaneous traversal error: dimension mismatch. "
) ;
assert_eq! (
( nrows . value ( ) , ncols . value ( ) ) ,
c . shape ( ) ,
2018-09-22 21:38:33 +08:00
" Matrix simultaneous traversal error: dimension mismatch. "
) ;
for j in 0 .. ncols . value ( ) {
for i in 0 .. nrows . value ( ) {
2021-08-03 00:41:46 +08:00
// Safety: all indices are in range.
2018-09-22 21:38:33 +08:00
unsafe {
2021-08-04 23:34:25 +08:00
let a = self . data . get_unchecked ( i , j ) . clone ( ) ;
let b = b . data . get_unchecked ( i , j ) . clone ( ) ;
let c = c . data . get_unchecked ( i , j ) . clone ( ) ;
2021-08-03 00:41:46 +08:00
* res . data . get_unchecked_mut ( i , j ) = MaybeUninit ::new ( f ( a , b , c ) )
2018-09-22 21:38:33 +08:00
}
}
}
2021-08-03 00:41:46 +08:00
// Safety: res is now fully initialized.
2021-07-15 02:24:27 +08:00
unsafe { res . assume_init ( ) }
2018-09-22 21:38:33 +08:00
}
2018-12-09 18:21:24 +08:00
/// Folds a function `f` on each entry of `self`.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn fold < Acc > ( & self , init : Acc , mut f : impl FnMut ( Acc , T ) -> Acc ) -> Acc
2021-07-14 17:25:16 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
2021-07-14 17:25:16 +08:00
{
2021-08-03 00:41:46 +08:00
let ( nrows , ncols ) = self . shape_generic ( ) ;
let mut res = init ;
2018-12-09 18:21:24 +08:00
for j in 0 .. ncols . value ( ) {
for i in 0 .. nrows . value ( ) {
2021-08-03 00:41:46 +08:00
// Safety: all indices are in range.
2018-12-09 18:21:24 +08:00
unsafe {
2021-08-04 23:34:25 +08:00
let a = self . data . get_unchecked ( i , j ) . clone ( ) ;
2021-08-03 00:41:46 +08:00
res = f ( res , a )
2018-12-09 18:21:24 +08:00
}
}
}
2021-08-03 00:41:46 +08:00
res
2018-12-09 18:21:24 +08:00
}
/// Folds a function `f` on each pairs of entries from `self` and `rhs`.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn zip_fold < T2 , R2 , C2 , S2 , Acc > (
2020-03-18 00:58:36 +08:00
& self ,
2021-04-11 17:00:38 +08:00
rhs : & Matrix < T2 , R2 , C2 , S2 > ,
2021-08-03 00:41:46 +08:00
init : Acc ,
2021-04-11 17:00:38 +08:00
mut f : impl FnMut ( Acc , T , T2 ) -> Acc ,
2020-03-18 00:58:36 +08:00
) -> Acc
where
2021-08-03 00:41:46 +08:00
T : Scalar ,
T2 : Scalar ,
R2 : Dim ,
C2 : Dim ,
S2 : RawStorage < T2 , R2 , C2 > ,
2020-03-18 00:58:36 +08:00
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
2018-12-09 18:21:24 +08:00
{
2021-08-03 00:41:46 +08:00
let ( nrows , ncols ) = self . shape_generic ( ) ;
let mut res = init ;
2018-12-09 18:21:24 +08:00
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
assert_eq! (
( nrows . value ( ) , ncols . value ( ) ) ,
rhs . shape ( ) ,
2018-12-09 18:21:24 +08:00
" Matrix simultaneous traversal error: dimension mismatch. "
) ;
for j in 0 .. ncols . value ( ) {
for i in 0 .. nrows . value ( ) {
unsafe {
2021-08-04 23:34:25 +08:00
let a = self . data . get_unchecked ( i , j ) . clone ( ) ;
let b = rhs . data . get_unchecked ( i , j ) . clone ( ) ;
2021-08-03 00:41:46 +08:00
res = f ( res , a , b )
2018-12-09 18:21:24 +08:00
}
}
}
2021-08-03 00:41:46 +08:00
res
2018-12-09 18:21:24 +08:00
}
2021-08-03 00:41:46 +08:00
/// Applies a closure `f` to modify each component of `self`.
2017-08-03 01:37:44 +08:00
#[ inline ]
2021-08-03 00:41:46 +08:00
pub fn apply < F : FnMut ( & mut T ) > ( & mut self , mut f : F )
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
S : RawStorageMut < T , R , C > ,
2020-11-15 23:57:49 +08:00
{
let ( nrows , ncols ) = self . shape ( ) ;
for j in 0 .. ncols {
for i in 0 .. nrows {
unsafe {
let e = self . data . get_unchecked_mut ( i , j ) ;
2021-08-03 00:41:46 +08:00
f ( e )
2020-11-15 23:57:49 +08:00
}
}
}
}
/// Replaces each component of `self` by the result of a closure `f` applied on its components
/// joined with the components from `rhs`.
#[ inline ]
2021-08-03 00:41:46 +08:00
pub fn zip_apply < T2 , R2 , C2 , S2 > (
2020-11-15 23:57:49 +08:00
& mut self ,
2021-04-11 17:00:38 +08:00
rhs : & Matrix < T2 , R2 , C2 , S2 > ,
2021-08-03 00:41:46 +08:00
mut f : impl FnMut ( & mut T , T2 ) ,
2020-11-15 23:57:49 +08:00
) where
2021-08-03 00:41:46 +08:00
S : RawStorageMut < T , R , C > ,
T2 : Scalar ,
R2 : Dim ,
C2 : Dim ,
S2 : RawStorage < T2 , R2 , C2 > ,
2020-11-15 23:57:49 +08:00
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
2018-02-02 19:26:35 +08:00
{
2017-08-03 01:37:44 +08:00
let ( nrows , ncols ) = self . shape ( ) ;
2020-11-15 23:57:49 +08:00
assert_eq! (
( nrows , ncols ) ,
rhs . shape ( ) ,
" Matrix simultaneous traversal error: dimension mismatch. "
2018-02-02 19:26:35 +08:00
) ;
2017-08-03 01:37:44 +08:00
2020-11-15 23:57:49 +08:00
for j in 0 .. ncols {
for i in 0 .. nrows {
2017-08-03 01:37:44 +08:00
unsafe {
2020-11-15 23:57:49 +08:00
let e = self . data . get_unchecked_mut ( i , j ) ;
2021-08-04 23:34:25 +08:00
let rhs = rhs . get_unchecked ( ( i , j ) ) . clone ( ) ;
2021-08-03 00:41:46 +08:00
f ( e , rhs )
2017-08-03 01:37:44 +08:00
}
}
}
}
2020-11-15 23:57:49 +08:00
/// Replaces each component of `self` by the result of a closure `f` applied on its components
/// joined with the components from `b` and `c`.
2017-08-03 01:37:44 +08:00
#[ inline ]
2021-08-03 00:41:46 +08:00
pub fn zip_zip_apply < T2 , R2 , C2 , S2 , N3 , R3 , C3 , S3 > (
2020-11-15 23:57:49 +08:00
& mut self ,
2021-04-11 17:00:38 +08:00
b : & Matrix < T2 , R2 , C2 , S2 > ,
2020-11-15 23:57:49 +08:00
c : & Matrix < N3 , R3 , C3 , S3 > ,
2021-08-03 00:41:46 +08:00
mut f : impl FnMut ( & mut T , T2 , N3 ) ,
2020-11-15 23:57:49 +08:00
) where
2021-08-03 00:41:46 +08:00
S : RawStorageMut < T , R , C > ,
T2 : Scalar ,
R2 : Dim ,
C2 : Dim ,
S2 : RawStorage < T2 , R2 , C2 > ,
N3 : Scalar ,
R3 : Dim ,
C3 : Dim ,
S3 : RawStorage < N3 , R3 , C3 > ,
2020-11-15 23:57:49 +08:00
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
2020-04-06 00:49:48 +08:00
{
2020-11-15 23:57:49 +08:00
let ( nrows , ncols ) = self . shape ( ) ;
2017-08-03 01:37:44 +08:00
2020-11-15 23:57:49 +08:00
assert_eq! (
( nrows , ncols ) ,
b . shape ( ) ,
" Matrix simultaneous traversal error: dimension mismatch. "
) ;
assert_eq! (
( nrows , ncols ) ,
c . shape ( ) ,
" Matrix simultaneous traversal error: dimension mismatch. "
) ;
2017-08-03 01:37:44 +08:00
2020-11-15 23:57:49 +08:00
for j in 0 .. ncols {
for i in 0 .. nrows {
unsafe {
let e = self . data . get_unchecked_mut ( i , j ) ;
2021-08-04 23:34:25 +08:00
let b = b . get_unchecked ( ( i , j ) ) . clone ( ) ;
let c = c . get_unchecked ( ( i , j ) ) . clone ( ) ;
2021-08-03 00:41:46 +08:00
f ( e , b , c )
2020-11-15 23:57:49 +08:00
}
}
2017-08-03 01:37:44 +08:00
}
}
2016-12-05 05:44:42 +08:00
}
2020-11-15 23:57:49 +08:00
/// # Iteration on components, rows, and columns
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S : RawStorage < T , R , C > > Matrix < T , R , C , S > {
2020-11-15 23:57:49 +08:00
/// Iterates through this matrix coordinates in column-major order.
///
2022-03-23 00:53:46 +08:00
/// # Example
2020-11-15 23:57:49 +08:00
/// ```
/// # use nalgebra::Matrix2x3;
/// let mat = Matrix2x3::new(11, 12, 13,
/// 21, 22, 23);
/// let mut it = mat.iter();
/// assert_eq!(*it.next().unwrap(), 11);
/// assert_eq!(*it.next().unwrap(), 21);
/// assert_eq!(*it.next().unwrap(), 12);
/// assert_eq!(*it.next().unwrap(), 22);
/// assert_eq!(*it.next().unwrap(), 13);
/// assert_eq!(*it.next().unwrap(), 23);
/// assert!(it.next().is_none());
2022-03-23 00:53:46 +08:00
/// ```
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-07-26 01:06:14 +08:00
pub fn iter ( & self ) -> MatrixIter < '_ , T , R , C , S > {
2020-11-15 23:57:49 +08:00
MatrixIter ::new ( & self . data )
2016-12-05 05:44:42 +08:00
}
2020-11-15 23:57:49 +08:00
/// Iterate through the rows of this matrix.
2018-11-20 05:51:01 +08:00
///
2020-11-15 23:57:49 +08:00
/// # Example
/// ```
/// # use nalgebra::Matrix2x3;
/// let mut a = Matrix2x3::new(1, 2, 3,
/// 4, 5, 6);
/// for (i, row) in a.row_iter().enumerate() {
/// assert_eq!(row, a.row(i))
/// }
/// ```
2018-11-20 05:51:01 +08:00
#[ inline ]
2021-07-26 01:06:14 +08:00
pub fn row_iter ( & self ) -> RowIter < '_ , T , R , C , S > {
2020-11-15 23:57:49 +08:00
RowIter ::new ( self )
}
/// Iterate through the columns of this matrix.
2022-03-24 11:08:20 +08:00
///
2020-11-15 23:57:49 +08:00
/// # Example
/// ```
/// # use nalgebra::Matrix2x3;
/// let mut a = Matrix2x3::new(1, 2, 3,
/// 4, 5, 6);
/// for (i, column) in a.column_iter().enumerate() {
/// assert_eq!(column, a.column(i))
/// }
/// ```
#[ inline ]
2021-07-26 01:06:14 +08:00
pub fn column_iter ( & self ) -> ColumnIter < '_ , T , R , C , S > {
2020-11-15 23:57:49 +08:00
ColumnIter ::new ( self )
}
/// Mutably iterates through this matrix coordinates.
#[ inline ]
2021-07-26 01:06:14 +08:00
pub fn iter_mut ( & mut self ) -> MatrixIterMut < '_ , T , R , C , S >
2020-11-15 23:57:49 +08:00
where
2021-08-03 00:41:46 +08:00
S : RawStorageMut < T , R , C > ,
2020-11-15 23:57:49 +08:00
{
MatrixIterMut ::new ( & mut self . data )
2018-11-20 05:51:01 +08:00
}
2018-12-09 22:21:05 +08:00
/// Mutably iterates through this matrix rows.
2019-02-03 18:17:09 +08:00
///
/// # Example
/// ```
/// # use nalgebra::Matrix2x3;
/// let mut a = Matrix2x3::new(1, 2, 3,
/// 4, 5, 6);
/// for (i, mut row) in a.row_iter_mut().enumerate() {
/// row *= (i + 1) * 10;
/// }
///
/// let expected = Matrix2x3::new(10, 20, 30,
/// 80, 100, 120);
/// assert_eq!(a, expected);
/// ```
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-07-26 01:06:14 +08:00
pub fn row_iter_mut ( & mut self ) -> RowIterMut < '_ , T , R , C , S >
2020-11-15 23:57:49 +08:00
where
2021-08-03 00:41:46 +08:00
S : RawStorageMut < T , R , C > ,
2020-11-15 23:57:49 +08:00
{
2018-12-09 22:21:05 +08:00
RowIterMut ::new ( self )
}
/// Mutably iterates through this matrix columns.
2019-02-03 18:17:09 +08:00
///
/// # Example
/// ```
/// # use nalgebra::Matrix2x3;
/// let mut a = Matrix2x3::new(1, 2, 3,
/// 4, 5, 6);
/// for (i, mut col) in a.column_iter_mut().enumerate() {
/// col *= (i + 1) * 10;
/// }
///
/// let expected = Matrix2x3::new(10, 40, 90,
/// 40, 100, 180);
/// assert_eq!(a, expected);
/// ```
2018-12-09 22:21:05 +08:00
#[ inline ]
2021-07-26 01:06:14 +08:00
pub fn column_iter_mut ( & mut self ) -> ColumnIterMut < '_ , T , R , C , S >
2020-11-15 23:57:49 +08:00
where
2021-08-03 00:41:46 +08:00
S : RawStorageMut < T , R , C > ,
2020-11-15 23:57:49 +08:00
{
2018-12-09 22:21:05 +08:00
ColumnIterMut ::new ( self )
2016-12-05 05:44:42 +08:00
}
2020-11-15 23:57:49 +08:00
}
2021-08-03 23:02:42 +08:00
impl < T , R : Dim , C : Dim , S : RawStorageMut < T , R , C > > Matrix < T , R , C , S > {
2020-11-15 23:57:49 +08:00
/// Returns a mutable pointer to the start of the matrix.
///
/// If the matrix is not empty, this pointer is guaranteed to be aligned
/// and non-null.
#[ inline ]
2021-04-11 17:00:38 +08:00
pub fn as_mut_ptr ( & mut self ) -> * mut T {
2020-11-15 23:57:49 +08:00
self . data . ptr_mut ( )
}
2016-12-05 05:44:42 +08:00
/// Swaps two entries without bound-checking.
#[ inline ]
pub unsafe fn swap_unchecked ( & mut self , row_cols1 : ( usize , usize ) , row_cols2 : ( usize , usize ) ) {
2017-08-03 01:37:44 +08:00
debug_assert! ( row_cols1 . 0 < self . nrows ( ) & & row_cols1 . 1 < self . ncols ( ) ) ;
debug_assert! ( row_cols2 . 0 < self . nrows ( ) & & row_cols2 . 1 < self . ncols ( ) ) ;
2016-12-05 05:44:42 +08:00
self . data . swap_unchecked ( row_cols1 , row_cols2 )
}
/// Swaps two entries.
#[ inline ]
pub fn swap ( & mut self , row_cols1 : ( usize , usize ) , row_cols2 : ( usize , usize ) ) {
let ( nrows , ncols ) = self . shape ( ) ;
2018-02-02 19:26:35 +08:00
assert! (
row_cols1 . 0 < nrows & & row_cols1 . 1 < ncols ,
" Matrix elements swap index out of bounds. "
) ;
assert! (
row_cols2 . 0 < nrows & & row_cols2 . 1 < ncols ,
" Matrix elements swap index out of bounds. "
) ;
2016-12-05 05:44:42 +08:00
unsafe { self . swap_unchecked ( row_cols1 , row_cols2 ) }
}
2018-09-13 11:38:57 +08:00
/// Fills this matrix with the content of a slice. Both must hold the same number of elements.
///
/// The components of the slice are assumed to be ordered in column-major order.
#[ inline ]
2021-08-03 23:02:42 +08:00
pub fn copy_from_slice ( & mut self , slice : & [ T ] )
where
T : Scalar ,
{
2018-09-13 11:38:57 +08:00
let ( nrows , ncols ) = self . shape ( ) ;
assert! (
nrows * ncols = = slice . len ( ) ,
" The slice must contain the same number of elements as the matrix. "
) ;
for j in 0 .. ncols {
for i in 0 .. nrows {
unsafe {
2021-08-04 23:34:25 +08:00
* self . get_unchecked_mut ( ( i , j ) ) = slice . get_unchecked ( i + j * nrows ) . clone ( ) ;
2018-09-13 11:38:57 +08:00
}
}
}
}
2021-08-03 00:41:46 +08:00
/// Fills this matrix with the content of another one. Both must have the same shape.
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-08-03 00:41:46 +08:00
pub fn copy_from < R2 , C2 , SB > ( & mut self , other : & Matrix < T , R2 , C2 , SB > )
2018-02-02 19:26:35 +08:00
where
2021-08-03 23:02:42 +08:00
T : Scalar ,
2021-08-03 00:41:46 +08:00
R2 : Dim ,
C2 : Dim ,
SB : RawStorage < T , R2 , C2 > ,
2021-07-16 12:56:58 +08:00
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
2018-02-02 19:26:35 +08:00
{
assert! (
self . shape ( ) = = other . shape ( ) ,
" Unable to copy from a matrix with a different shape. "
) ;
for j in 0 .. self . ncols ( ) {
for i in 0 .. self . nrows ( ) {
unsafe {
2021-08-04 23:34:25 +08:00
* self . get_unchecked_mut ( ( i , j ) ) = other . get_unchecked ( ( i , j ) ) . clone ( ) ;
2018-02-02 19:26:35 +08:00
}
2017-08-03 01:37:44 +08:00
}
2016-12-05 05:44:42 +08:00
}
}
2021-08-03 00:41:46 +08:00
/// Fills this matrix with the content of the transpose another one.
2021-07-16 12:56:58 +08:00
#[ inline ]
2021-08-03 00:41:46 +08:00
pub fn tr_copy_from < R2 , C2 , SB > ( & mut self , other : & Matrix < T , R2 , C2 , SB > )
2021-07-16 12:56:58 +08:00
where
2021-08-03 23:02:42 +08:00
T : Scalar ,
2021-08-03 00:41:46 +08:00
R2 : Dim ,
C2 : Dim ,
SB : RawStorage < T , R2 , C2 > ,
2021-07-16 12:56:58 +08:00
ShapeConstraint : DimEq < R , C2 > + SameNumberOfColumns < C , R2 > ,
2018-02-02 19:26:35 +08:00
{
2017-08-03 01:37:44 +08:00
let ( nrows , ncols ) = self . shape ( ) ;
2018-02-02 19:26:35 +08:00
assert! (
( ncols , nrows ) = = other . shape ( ) ,
" Unable to copy from a matrix with incompatible shape. "
) ;
2017-08-03 01:37:44 +08:00
2018-02-02 19:26:35 +08:00
for j in 0 .. ncols {
for i in 0 .. nrows {
unsafe {
2021-08-04 23:34:25 +08:00
* self . get_unchecked_mut ( ( i , j ) ) = other . get_unchecked ( ( j , i ) ) . clone ( ) ;
2018-02-02 19:26:35 +08:00
}
2017-08-03 01:37:44 +08:00
}
2016-12-05 05:44:42 +08:00
}
}
2017-05-18 04:37:18 +08:00
2020-11-15 23:57:49 +08:00
// TODO: rename `apply` to `apply_mut` and `apply_into` to `apply`?
2019-03-19 19:00:10 +08:00
/// Returns `self` with each of its components replaced by the result of a closure `f` applied on it.
#[ inline ]
2021-08-03 00:41:46 +08:00
pub fn apply_into < F : FnMut ( & mut T ) > ( mut self , f : F ) -> Self {
2019-03-19 19:00:10 +08:00
self . apply ( f ) ;
self
}
2016-12-05 05:44:42 +08:00
}
2017-05-18 04:37:18 +08:00
2021-08-03 00:41:46 +08:00
impl < T , D : Dim , S : RawStorage < T , D > > Vector < T , D , S > {
2017-08-03 01:37:44 +08:00
/// Gets a reference to the i-th element of this column vector without bound checking.
2017-05-18 04:37:18 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub unsafe fn vget_unchecked ( & self , i : usize ) -> & T {
2017-08-03 01:37:44 +08:00
debug_assert! ( i < self . nrows ( ) , " Vector index out of bounds. " ) ;
let i = i * self . strides ( ) . 0 ;
self . data . get_unchecked_linear ( i )
2017-05-18 04:37:18 +08:00
}
2016-12-05 05:44:42 +08:00
}
2021-08-03 00:41:46 +08:00
impl < T , D : Dim , S : RawStorageMut < T , D > > Vector < T , D , S > {
2017-08-03 01:37:44 +08:00
/// Gets a mutable reference to the i-th element of this column vector without bound checking.
#[ inline ]
2021-06-07 23:10:21 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub unsafe fn vget_unchecked_mut ( & mut self , i : usize ) -> & mut T {
2017-08-03 01:37:44 +08:00
debug_assert! ( i < self . nrows ( ) , " Vector index out of bounds. " ) ;
let i = i * self . strides ( ) . 0 ;
self . data . get_unchecked_linear_mut ( i )
}
}
2016-12-05 05:44:42 +08:00
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S : RawStorage < T , R , C > + IsContiguous > Matrix < T , R , C , S > {
2017-08-03 01:37:44 +08:00
/// Extracts a slice containing the entire matrix entries ordered column-by-columns.
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn as_slice ( & self ) -> & [ T ] {
2021-08-03 00:41:46 +08:00
// Safety: this is OK thanks to the IsContiguous trait.
unsafe { self . data . as_slice_unchecked ( ) }
2016-12-05 05:44:42 +08:00
}
2017-08-03 01:37:44 +08:00
}
2016-12-05 05:44:42 +08:00
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S : RawStorageMut < T , R , C > + IsContiguous > Matrix < T , R , C , S > {
2017-08-03 01:37:44 +08:00
/// Extracts a mutable slice containing the entire matrix entries ordered column-by-columns.
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 23:10:21 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn as_mut_slice ( & mut self ) -> & mut [ T ] {
2021-08-03 00:41:46 +08:00
// Safety: this is OK thanks to the IsContiguous trait.
unsafe { self . data . as_mut_slice_unchecked ( ) }
2016-12-05 05:44:42 +08:00
}
2017-08-03 01:37:44 +08:00
}
2016-12-05 05:44:42 +08:00
2021-08-03 00:41:46 +08:00
impl < T : Scalar , D : Dim , S : RawStorageMut < T , D , D > > Matrix < T , D , D , S > {
2017-08-03 01:37:44 +08:00
/// Transposes the square matrix `self` in-place.
pub fn transpose_mut ( & mut self ) {
2018-02-02 19:26:35 +08:00
assert! (
self . is_square ( ) ,
" Unable to transpose a non-square matrix in-place. "
) ;
2016-12-05 05:44:42 +08:00
2017-08-03 01:37:44 +08:00
let dim = self . shape ( ) . 0 ;
2016-12-05 05:44:42 +08:00
2018-02-02 19:26:35 +08:00
for i in 1 .. dim {
for j in 0 .. i {
2017-08-03 01:37:44 +08:00
unsafe { self . swap_unchecked ( ( i , j ) , ( j , i ) ) }
2016-12-05 05:44:42 +08:00
}
}
}
2017-08-03 01:37:44 +08:00
}
2016-12-05 05:44:42 +08:00
2021-08-03 00:41:46 +08:00
impl < T : SimdComplexField , R : Dim , C : Dim , S : RawStorage < T , R , C > > Matrix < T , R , C , S > {
2019-03-23 18:48:12 +08:00
/// Takes the adjoint (aka. conjugate-transpose) of `self` and store the result into `out`.
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-08-03 00:41:46 +08:00
fn adjoint_to_uninit < Status , R2 , C2 , SB > (
& self ,
2021-09-26 03:05:31 +08:00
_status : Status ,
2021-08-03 00:41:46 +08:00
out : & mut Matrix < Status ::Value , R2 , C2 , SB > ,
) where
Status : InitStatus < T > ,
R2 : Dim ,
C2 : Dim ,
SB : RawStorageMut < Status ::Value , R2 , C2 > ,
2018-02-02 19:26:35 +08:00
ShapeConstraint : SameNumberOfRows < R , C2 > + SameNumberOfColumns < C , R2 > ,
{
2017-08-03 01:37:44 +08:00
let ( nrows , ncols ) = self . shape ( ) ;
2018-02-02 19:26:35 +08:00
assert! (
( ncols , nrows ) = = out . shape ( ) ,
" Incompatible shape for transpose-copy. "
) ;
2016-12-05 05:44:42 +08:00
2020-11-15 23:57:49 +08:00
// TODO: optimize that.
2018-02-02 19:26:35 +08:00
for i in 0 .. nrows {
for j in 0 .. ncols {
2021-08-03 00:41:46 +08:00
// Safety: all indices are in range.
2017-08-03 01:37:44 +08:00
unsafe {
2021-08-03 00:41:46 +08:00
Status ::init (
out . get_unchecked_mut ( ( j , i ) ) ,
2021-08-04 23:34:25 +08:00
self . get_unchecked ( ( i , j ) ) . clone ( ) . simd_conjugate ( ) ,
2021-08-03 00:41:46 +08:00
) ;
2017-08-03 01:37:44 +08:00
}
2016-12-05 05:44:42 +08:00
}
}
}
2021-08-03 00:41:46 +08:00
/// Takes the adjoint (aka. conjugate-transpose) of `self` and store the result into `out`.
#[ inline ]
pub fn adjoint_to < R2 , C2 , SB > ( & self , out : & mut Matrix < T , R2 , C2 , SB > )
where
R2 : Dim ,
C2 : Dim ,
SB : RawStorageMut < T , R2 , C2 > ,
ShapeConstraint : SameNumberOfRows < R , C2 > + SameNumberOfColumns < C , R2 > ,
{
self . adjoint_to_uninit ( Init , out )
}
2019-03-23 18:48:12 +08:00
/// The adjoint (aka. conjugate-transpose) of `self`.
2016-12-05 05:44:42 +08:00
#[ inline ]
2019-06-06 05:04:04 +08:00
#[ must_use = " Did you mean to use adjoint_mut()? " ]
2021-04-11 17:00:38 +08:00
pub fn adjoint ( & self ) -> OMatrix < T , C , R >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , C , R > ,
2020-04-06 00:49:48 +08:00
{
2021-08-03 00:41:46 +08:00
let ( nrows , ncols ) = self . shape_generic ( ) ;
let mut res = Matrix ::uninit ( ncols , nrows ) ;
self . adjoint_to_uninit ( Uninit , & mut res ) ;
2016-12-05 05:44:42 +08:00
2021-08-03 00:41:46 +08:00
// Safety: res is now fully initialized.
2021-07-16 12:56:58 +08:00
unsafe { res . assume_init ( ) }
2016-12-05 05:44:42 +08:00
}
2019-03-03 02:33:49 +08:00
2019-03-23 18:48:12 +08:00
/// Takes the conjugate and transposes `self` and store the result into `out`.
#[ deprecated(note = " Renamed `self.adjoint_to(out)`. " ) ]
#[ inline ]
2021-08-03 00:41:46 +08:00
pub fn conjugate_transpose_to < R2 , C2 , SB > ( & self , out : & mut Matrix < T , R2 , C2 , SB > )
where
R2 : Dim ,
C2 : Dim ,
SB : RawStorageMut < T , R2 , C2 > ,
2020-03-18 00:58:36 +08:00
ShapeConstraint : SameNumberOfRows < R , C2 > + SameNumberOfColumns < C , R2 > ,
2019-03-23 18:48:12 +08:00
{
self . adjoint_to ( out )
}
/// The conjugate transposition of `self`.
#[ deprecated(note = " Renamed `self.adjoint()`. " ) ]
#[ inline ]
2021-04-11 17:00:38 +08:00
pub fn conjugate_transpose ( & self ) -> OMatrix < T , C , R >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , C , R > ,
2020-04-06 00:49:48 +08:00
{
2019-03-23 18:48:12 +08:00
self . adjoint ( )
}
2019-03-03 02:33:49 +08:00
/// The conjugate of `self`.
#[ inline ]
2019-06-06 05:04:04 +08:00
#[ must_use = " Did you mean to use conjugate_mut()? " ]
2021-04-11 17:00:38 +08:00
pub fn conjugate ( & self ) -> OMatrix < T , R , C >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , R , C > ,
2020-04-06 00:49:48 +08:00
{
2020-03-18 00:58:36 +08:00
self . map ( | e | e . simd_conjugate ( ) )
2019-03-03 02:33:49 +08:00
}
2019-03-23 21:13:00 +08:00
/// Divides each component of the complex matrix `self` by the given real.
2019-03-03 02:33:49 +08:00
#[ inline ]
2019-06-06 05:04:04 +08:00
#[ must_use = " Did you mean to use unscale_mut()? " ]
2021-04-11 17:00:38 +08:00
pub fn unscale ( & self , real : T ::SimdRealField ) -> OMatrix < T , R , C >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , R , C > ,
2020-04-06 00:49:48 +08:00
{
2021-08-04 23:34:25 +08:00
self . map ( | e | e . simd_unscale ( real . clone ( ) ) )
2019-03-03 02:33:49 +08:00
}
2019-03-23 21:13:00 +08:00
/// Multiplies each component of the complex matrix `self` by the given real.
2019-03-03 02:33:49 +08:00
#[ inline ]
2019-06-06 05:04:04 +08:00
#[ must_use = " Did you mean to use scale_mut()? " ]
2021-04-11 17:00:38 +08:00
pub fn scale ( & self , real : T ::SimdRealField ) -> OMatrix < T , R , C >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , R , C > ,
2020-04-06 00:49:48 +08:00
{
2021-08-04 23:34:25 +08:00
self . map ( | e | e . simd_scale ( real . clone ( ) ) )
2019-03-03 02:33:49 +08:00
}
}
2021-08-03 00:41:46 +08:00
impl < T : SimdComplexField , R : Dim , C : Dim , S : RawStorageMut < T , R , C > > Matrix < T , R , C , S > {
2019-03-23 21:13:00 +08:00
/// The conjugate of the complex matrix `self` computed in-place.
2019-03-03 02:33:49 +08:00
#[ inline ]
pub fn conjugate_mut ( & mut self ) {
2021-08-04 23:34:25 +08:00
self . apply ( | e | * e = e . clone ( ) . simd_conjugate ( ) )
2019-03-03 02:33:49 +08:00
}
2019-03-23 21:13:00 +08:00
/// Divides each component of the complex matrix `self` by the given real.
2019-03-03 02:33:49 +08:00
#[ inline ]
2021-04-11 17:00:38 +08:00
pub fn unscale_mut ( & mut self , real : T ::SimdRealField ) {
2021-08-04 23:34:25 +08:00
self . apply ( | e | * e = e . clone ( ) . simd_unscale ( real . clone ( ) ) )
2019-03-03 02:33:49 +08:00
}
2019-03-23 21:13:00 +08:00
/// Multiplies each component of the complex matrix `self` by the given real.
2019-03-03 02:33:49 +08:00
#[ inline ]
2021-04-11 17:00:38 +08:00
pub fn scale_mut ( & mut self , real : T ::SimdRealField ) {
2021-08-04 23:34:25 +08:00
self . apply ( | e | * e = e . clone ( ) . simd_scale ( real . clone ( ) ) )
2019-03-03 02:33:49 +08:00
}
2016-12-05 05:44:42 +08:00
}
2021-08-03 00:41:46 +08:00
impl < T : SimdComplexField , D : Dim , S : RawStorageMut < T , D , D > > Matrix < T , D , D , S > {
2019-03-23 21:13:00 +08:00
/// Sets `self` to its adjoint.
#[ deprecated(note = " Renamed to `self.adjoint_mut()`. " ) ]
pub fn conjugate_transform_mut ( & mut self ) {
self . adjoint_mut ( )
}
/// Sets `self` to its adjoint (aka. conjugate-transpose).
pub fn adjoint_mut ( & mut self ) {
2018-02-02 19:26:35 +08:00
assert! (
self . is_square ( ) ,
" Unable to transpose a non-square matrix in-place. "
) ;
2016-12-05 05:44:42 +08:00
let dim = self . shape ( ) . 0 ;
2019-03-12 20:15:02 +08:00
for i in 0 .. dim {
2018-02-02 19:26:35 +08:00
for j in 0 .. i {
2017-08-03 01:37:44 +08:00
unsafe {
2021-08-04 23:34:25 +08:00
let ref_ij = self . get_unchecked ( ( i , j ) ) . clone ( ) ;
let ref_ji = self . get_unchecked ( ( j , i ) ) . clone ( ) ;
let conj_ij = ref_ij . simd_conjugate ( ) ;
let conj_ji = ref_ji . simd_conjugate ( ) ;
* self . get_unchecked_mut ( ( i , j ) ) = conj_ji ;
* self . get_unchecked_mut ( ( j , i ) ) = conj_ij ;
2017-08-03 01:37:44 +08:00
}
2016-12-05 05:44:42 +08:00
}
2019-03-23 21:13:00 +08:00
{
let diag = unsafe { self . get_unchecked_mut ( ( i , i ) ) } ;
2021-08-04 23:34:25 +08:00
* diag = diag . clone ( ) . simd_conjugate ( ) ;
2019-03-23 21:13:00 +08:00
}
2016-12-05 05:44:42 +08:00
}
}
}
2021-08-03 00:41:46 +08:00
impl < T : Scalar , D : Dim , S : RawStorage < T , D , D > > SquareMatrix < T , D , S > {
2019-03-19 21:22:59 +08:00
/// The diagonal of this matrix.
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn diagonal ( & self ) -> OVector < T , D >
2020-04-06 00:49:48 +08:00
where
2021-07-15 02:24:27 +08:00
DefaultAllocator : Allocator < T , D > ,
2020-04-06 00:49:48 +08:00
{
2019-03-19 21:22:59 +08:00
self . map_diagonal ( | e | e )
}
/// Apply the given function to this matrix's diagonal and returns it.
///
/// This is a more efficient version of `self.diagonal().map(f)` since this
/// allocates only once.
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn map_diagonal < T2 : Scalar > ( & self , mut f : impl FnMut ( T ) -> T2 ) -> OVector < T2 , D >
2020-04-06 00:49:48 +08:00
where
2021-07-15 02:24:27 +08:00
DefaultAllocator : Allocator < T2 , D > ,
2020-04-06 00:49:48 +08:00
{
2018-02-02 19:26:35 +08:00
assert! (
self . is_square ( ) ,
" Unable to get the diagonal of a non-square matrix. "
) ;
2016-12-05 05:44:42 +08:00
2021-08-03 00:41:46 +08:00
let dim = self . shape_generic ( ) . 0 ;
let mut res = Matrix ::uninit ( dim , Const ::< 1 > ) ;
2016-12-05 05:44:42 +08:00
2018-02-02 19:26:35 +08:00
for i in 0 .. dim . value ( ) {
2021-08-03 00:41:46 +08:00
// Safety: all indices are in range.
2018-02-02 19:26:35 +08:00
unsafe {
2021-07-14 17:25:16 +08:00
* res . vget_unchecked_mut ( i ) =
2021-08-04 23:34:25 +08:00
MaybeUninit ::new ( f ( self . get_unchecked ( ( i , i ) ) . clone ( ) ) ) ;
2018-02-02 19:26:35 +08:00
}
2016-12-05 05:44:42 +08:00
}
2021-08-03 00:41:46 +08:00
// Safety: res is now fully initialized.
2021-07-16 12:56:58 +08:00
unsafe { res . assume_init ( ) }
2016-12-05 05:44:42 +08:00
}
2017-03-20 05:33:01 +08:00
/// Computes a trace of a square matrix, i.e., the sum of its diagonal elements.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn trace ( & self ) -> T
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
T : Scalar + Zero + ClosedAdd ,
2020-04-06 00:49:48 +08:00
{
2018-02-02 19:26:35 +08:00
assert! (
self . is_square ( ) ,
" Cannot compute the trace of non-square matrix. "
) ;
2017-03-20 05:33:01 +08:00
2021-08-03 00:41:46 +08:00
let dim = self . shape_generic ( ) . 0 ;
2021-04-11 17:00:38 +08:00
let mut res = T ::zero ( ) ;
2017-03-20 05:33:01 +08:00
2018-02-02 19:26:35 +08:00
for i in 0 .. dim . value ( ) {
2021-08-04 23:34:25 +08:00
res + = unsafe { self . get_unchecked ( ( i , i ) ) . clone ( ) } ;
2017-03-20 05:33:01 +08:00
}
res
}
2016-12-05 05:44:42 +08:00
}
2021-04-11 17:00:38 +08:00
impl < T : SimdComplexField , D : Dim , S : Storage < T , D , D > > SquareMatrix < T , D , S > {
2019-03-03 02:33:49 +08:00
/// The symmetric part of `self`, i.e., `0.5 * (self + self.transpose())`.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn symmetric_part ( & self ) -> OMatrix < T , D , D >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , D , D > ,
2020-04-06 00:49:48 +08:00
{
2020-03-18 00:58:36 +08:00
assert! (
self . is_square ( ) ,
" Cannot compute the symmetric part of a non-square matrix. "
) ;
2019-03-03 02:33:49 +08:00
let mut tr = self . transpose ( ) ;
tr + = self ;
2021-04-11 17:00:38 +08:00
tr * = crate ::convert ::< _ , T > ( 0.5 ) ;
2019-03-03 02:33:49 +08:00
tr
}
2019-03-23 18:48:12 +08:00
/// The hermitian part of `self`, i.e., `0.5 * (self + self.adjoint())`.
2019-03-03 02:33:49 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn hermitian_part ( & self ) -> OMatrix < T , D , D >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , D , D > ,
2020-04-06 00:49:48 +08:00
{
2020-03-18 00:58:36 +08:00
assert! (
self . is_square ( ) ,
" Cannot compute the hermitian part of a non-square matrix. "
) ;
2019-03-03 02:33:49 +08:00
2019-03-23 18:48:12 +08:00
let mut tr = self . adjoint ( ) ;
2019-03-23 18:46:56 +08:00
tr + = self ;
2021-04-11 17:00:38 +08:00
tr * = crate ::convert ::< _ , T > ( 0.5 ) ;
2019-03-23 18:46:56 +08:00
tr
2019-03-03 02:33:49 +08:00
}
}
2021-08-03 00:41:46 +08:00
impl < T : Scalar + Zero + One , D : DimAdd < U1 > + IsNotStaticOne , S : RawStorage < T , D , D > >
2021-04-11 17:00:38 +08:00
Matrix < T , D , D , S >
2020-03-18 00:58:36 +08:00
{
2018-11-18 23:51:40 +08:00
/// Yields the homogeneous matrix for this matrix, i.e., appending an additional dimension and
/// and setting the diagonal element to `1`.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn to_homogeneous ( & self ) -> OMatrix < T , DimSum < D , U1 > , DimSum < D , U1 > >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , DimSum < D , U1 > , DimSum < D , U1 > > ,
2020-04-06 00:49:48 +08:00
{
2020-03-18 00:58:36 +08:00
assert! (
self . is_square ( ) ,
" Only square matrices can currently be transformed to homogeneous coordinates. "
) ;
2018-11-20 04:21:25 +08:00
let dim = DimSum ::< D , U1 > ::from_usize ( self . nrows ( ) + 1 ) ;
2021-04-11 17:00:38 +08:00
let mut res = OMatrix ::identity_generic ( dim , dim ) ;
2021-08-03 00:41:46 +08:00
res . generic_slice_mut ::< D , D > ( ( 0 , 0 ) , self . shape_generic ( ) )
2021-07-04 11:19:07 +08:00
. copy_from ( self ) ;
2018-11-18 23:51:40 +08:00
res
}
}
2021-08-03 00:41:46 +08:00
impl < T : Scalar + Zero , D : DimAdd < U1 > , S : RawStorage < T , D > > Vector < T , D , S > {
2016-12-05 05:44:42 +08:00
/// Computes the coordinates in projective space of this vector, i.e., appends a `0` to its
/// coordinates.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn to_homogeneous ( & self ) -> OVector < T , DimSum < D , U1 > >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , DimSum < D , U1 > > ,
2020-04-06 00:49:48 +08:00
{
2021-04-11 17:00:38 +08:00
self . push ( T ::zero ( ) )
2016-12-05 05:44:42 +08:00
}
2017-02-13 01:17:09 +08:00
/// Constructs a vector from coordinates in projective space, i.e., removes a `0` at the end of
/// `self`. Returns `None` if this last component is not zero.
#[ inline ]
2021-04-11 17:00:38 +08:00
pub fn from_homogeneous < SB > ( v : Vector < T , DimSum < D , U1 > , SB > ) -> Option < OVector < T , D > >
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
SB : RawStorage < T , DimSum < D , U1 > > ,
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , D > ,
2018-02-02 19:26:35 +08:00
{
2017-02-13 01:17:09 +08:00
if v [ v . len ( ) - 1 ] . is_zero ( ) {
let nrows = D ::from_usize ( v . len ( ) - 1 ) ;
2021-01-03 22:20:34 +08:00
Some ( v . generic_slice ( ( 0 , 0 ) , ( nrows , Const ::< 1 > ) ) . into_owned ( ) )
2018-02-02 19:26:35 +08:00
} else {
2017-02-13 01:17:09 +08:00
None
}
}
2016-12-05 05:44:42 +08:00
}
2021-08-03 00:41:46 +08:00
impl < T : Scalar , D : DimAdd < U1 > , S : RawStorage < T , D > > Vector < T , D , S > {
2019-01-10 10:38:58 +08:00
/// Constructs a new vector of higher dimension by appending `element` to the end of `self`.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn push ( & self , element : T ) -> OVector < T , DimSum < D , U1 > >
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator : Allocator < T , DimSum < D , U1 > > ,
2020-04-06 00:49:48 +08:00
{
2019-01-10 10:38:58 +08:00
let len = self . len ( ) ;
let hnrows = DimSum ::< D , U1 > ::from_usize ( len + 1 ) ;
2021-08-03 00:41:46 +08:00
let mut res = Matrix ::uninit ( hnrows , Const ::< 1 > ) ;
// This is basically a copy_from except that we warp the copied
// values into MaybeUninit.
res . generic_slice_mut ( ( 0 , 0 ) , self . shape_generic ( ) )
. zip_apply ( self , | out , e | * out = MaybeUninit ::new ( e ) ) ;
2021-07-16 12:56:58 +08:00
res [ ( len , 0 ) ] = MaybeUninit ::new ( element ) ;
2019-01-10 10:38:58 +08:00
2021-08-03 00:41:46 +08:00
// Safety: res has been fully initialized.
2021-07-16 12:56:58 +08:00
unsafe { res . assume_init ( ) }
2019-01-10 10:38:58 +08:00
}
}
2021-04-11 17:00:38 +08:00
impl < T , R : Dim , C : Dim , S > AbsDiffEq for Matrix < T , R , C , S >
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar + AbsDiffEq ,
S : RawStorage < T , R , C > ,
2021-08-04 23:34:25 +08:00
T ::Epsilon : Clone ,
2018-02-02 19:26:35 +08:00
{
2021-04-11 17:00:38 +08:00
type Epsilon = T ::Epsilon ;
2016-12-05 05:44:42 +08:00
#[ inline ]
fn default_epsilon ( ) -> Self ::Epsilon {
2021-04-11 17:00:38 +08:00
T ::default_epsilon ( )
2016-12-05 05:44:42 +08:00
}
#[ inline ]
2018-05-19 21:41:58 +08:00
fn abs_diff_eq ( & self , other : & Self , epsilon : Self ::Epsilon ) -> bool {
self . iter ( )
. zip ( other . iter ( ) )
2021-08-04 23:34:25 +08:00
. all ( | ( a , b ) | a . abs_diff_eq ( b , epsilon . clone ( ) ) )
2016-12-05 05:44:42 +08:00
}
2018-05-19 21:41:58 +08:00
}
2016-12-05 05:44:42 +08:00
2021-04-11 17:00:38 +08:00
impl < T , R : Dim , C : Dim , S > RelativeEq for Matrix < T , R , C , S >
2018-05-19 21:41:58 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar + RelativeEq ,
2021-04-11 17:00:38 +08:00
S : Storage < T , R , C > ,
2021-08-04 23:34:25 +08:00
T ::Epsilon : Clone ,
2018-05-19 21:41:58 +08:00
{
2016-12-05 05:44:42 +08:00
#[ inline ]
2018-05-19 21:41:58 +08:00
fn default_max_relative ( ) -> Self ::Epsilon {
2021-04-11 17:00:38 +08:00
T ::default_max_relative ( )
2016-12-05 05:44:42 +08:00
}
#[ inline ]
2018-02-02 19:26:35 +08:00
fn relative_eq (
& self ,
other : & Self ,
epsilon : Self ::Epsilon ,
max_relative : Self ::Epsilon ,
2020-04-06 00:49:48 +08:00
) -> bool {
2016-12-05 05:44:42 +08:00
self . relative_eq ( other , epsilon , max_relative )
}
2018-05-19 21:41:58 +08:00
}
2021-04-11 17:00:38 +08:00
impl < T , R : Dim , C : Dim , S > UlpsEq for Matrix < T , R , C , S >
2018-05-19 21:41:58 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar + UlpsEq ,
S : RawStorage < T , R , C > ,
2021-08-04 23:34:25 +08:00
T ::Epsilon : Clone ,
2018-05-19 21:41:58 +08:00
{
#[ inline ]
fn default_max_ulps ( ) -> u32 {
2021-04-11 17:00:38 +08:00
T ::default_max_ulps ( )
2018-05-19 21:41:58 +08:00
}
2016-12-05 05:44:42 +08:00
#[ inline ]
fn ulps_eq ( & self , other : & Self , epsilon : Self ::Epsilon , max_ulps : u32 ) -> bool {
2018-02-02 19:26:35 +08:00
assert! ( self . shape ( ) = = other . shape ( ) ) ;
self . iter ( )
. zip ( other . iter ( ) )
2021-08-16 00:12:17 +08:00
. all ( | ( a , b ) | a . ulps_eq ( b , epsilon . clone ( ) , max_ulps ) )
2016-12-05 05:44:42 +08:00
}
}
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S > PartialOrd for Matrix < T , R , C , S >
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar + PartialOrd ,
S : RawStorage < T , R , C > ,
2018-02-02 19:26:35 +08:00
{
2016-12-05 05:44:42 +08:00
#[ inline ]
fn partial_cmp ( & self , other : & Self ) -> Option < Ordering > {
2018-10-13 17:32:49 +08:00
if self . shape ( ) ! = other . shape ( ) {
2018-10-13 17:24:11 +08:00
return None ;
}
2018-02-02 19:26:35 +08:00
2018-10-13 17:32:49 +08:00
if self . nrows ( ) = = 0 | | self . ncols ( ) = = 0 {
return Some ( Ordering ::Equal ) ;
}
2018-10-13 17:24:11 +08:00
let mut first_ord = unsafe {
2018-02-02 19:26:35 +08:00
self . data
. get_unchecked_linear ( 0 )
. partial_cmp ( other . data . get_unchecked_linear ( 0 ) )
} ;
2016-12-05 05:44:42 +08:00
2018-10-13 17:24:11 +08:00
if let Some ( first_ord ) = first_ord . as_mut ( ) {
2016-12-05 05:44:42 +08:00
let mut it = self . iter ( ) . zip ( other . iter ( ) ) ;
2017-02-13 01:17:09 +08:00
let _ = it . next ( ) ; // Drop the first elements (we already tested it).
2016-12-05 05:44:42 +08:00
for ( left , right ) in it {
if let Some ( ord ) = left . partial_cmp ( right ) {
match ord {
2018-02-02 19:26:35 +08:00
Ordering ::Equal = > { /* Does not change anything. */ }
Ordering ::Less = > {
2018-10-13 17:24:11 +08:00
if * first_ord = = Ordering ::Greater {
2016-12-05 05:44:42 +08:00
return None ;
}
2018-10-13 17:24:11 +08:00
* first_ord = ord
2018-02-02 19:26:35 +08:00
}
2016-12-05 05:44:42 +08:00
Ordering ::Greater = > {
2018-10-13 17:24:11 +08:00
if * first_ord = = Ordering ::Less {
2016-12-05 05:44:42 +08:00
return None ;
}
2018-10-13 17:24:11 +08:00
* first_ord = ord
2018-02-02 19:26:35 +08:00
}
2016-12-05 05:44:42 +08:00
}
2018-02-02 19:26:35 +08:00
} else {
return None ;
2016-12-05 05:44:42 +08:00
}
}
}
2018-09-24 11:58:58 +08:00
first_ord
2016-12-05 05:44:42 +08:00
}
#[ inline ]
fn lt ( & self , right : & Self ) -> bool {
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
assert_eq! (
self . shape ( ) ,
right . shape ( ) ,
2018-02-02 19:26:35 +08:00
" Matrix comparison error: dimensions mismatch. "
) ;
2016-12-05 05:44:42 +08:00
self . iter ( ) . zip ( right . iter ( ) ) . all ( | ( a , b ) | a . lt ( b ) )
}
#[ inline ]
fn le ( & self , right : & Self ) -> bool {
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
assert_eq! (
self . shape ( ) ,
right . shape ( ) ,
2018-02-02 19:26:35 +08:00
" Matrix comparison error: dimensions mismatch. "
) ;
2016-12-05 05:44:42 +08:00
self . iter ( ) . zip ( right . iter ( ) ) . all ( | ( a , b ) | a . le ( b ) )
}
#[ inline ]
fn gt ( & self , right : & Self ) -> bool {
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
assert_eq! (
self . shape ( ) ,
right . shape ( ) ,
2018-02-02 19:26:35 +08:00
" Matrix comparison error: dimensions mismatch. "
) ;
2016-12-05 05:44:42 +08:00
self . iter ( ) . zip ( right . iter ( ) ) . all ( | ( a , b ) | a . gt ( b ) )
}
#[ inline ]
fn ge ( & self , right : & Self ) -> bool {
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
assert_eq! (
self . shape ( ) ,
right . shape ( ) ,
2018-02-02 19:26:35 +08:00
" Matrix comparison error: dimensions mismatch. "
) ;
2016-12-05 05:44:42 +08:00
self . iter ( ) . zip ( right . iter ( ) ) . all ( | ( a , b ) | a . ge ( b ) )
}
}
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S > Eq for Matrix < T , R , C , S >
where
T : Scalar + Eq ,
S : RawStorage < T , R , C > ,
{
}
2016-12-05 05:44:42 +08:00
2021-08-03 00:41:46 +08:00
impl < T , R , R2 , C , C2 , S , S2 > PartialEq < Matrix < T , R2 , C2 , S2 > > for Matrix < T , R , C , S >
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar + PartialEq ,
C : Dim ,
C2 : Dim ,
R : Dim ,
R2 : Dim ,
S : RawStorage < T , R , C > ,
S2 : RawStorage < T , R2 , C2 > ,
2018-02-02 19:26:35 +08:00
{
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-04-11 17:00:38 +08:00
fn eq ( & self , right : & Matrix < T , R2 , C2 , S2 > ) -> bool {
2020-01-25 19:58:05 +08:00
self . shape ( ) = = right . shape ( ) & & self . iter ( ) . zip ( right . iter ( ) ) . all ( | ( l , r ) | l = = r )
2016-12-05 05:44:42 +08:00
}
}
2019-08-21 21:53:23 +08:00
macro_rules ! impl_fmt {
( $trait : path , $fmt_str_without_precision : expr , $fmt_str_with_precision : expr ) = > {
2021-04-11 17:00:38 +08:00
impl < T , R : Dim , C : Dim , S > $trait for Matrix < T , R , C , S >
2019-08-21 21:53:23 +08:00
where
2021-04-11 17:00:38 +08:00
T : Scalar + $trait ,
2021-08-03 00:41:46 +08:00
S : RawStorage < T , R , C > ,
2019-08-21 21:53:23 +08:00
{
2021-07-26 01:06:14 +08:00
fn fmt ( & self , f : & mut fmt ::Formatter < '_ > ) -> fmt ::Result {
2019-08-21 21:53:23 +08:00
#[ cfg(feature = " std " ) ]
2021-07-26 01:06:14 +08:00
fn val_width < T : Scalar + $trait > ( val : & T , f : & mut fmt ::Formatter < '_ > ) -> usize {
2019-08-21 21:53:23 +08:00
match f . precision ( ) {
2020-03-18 00:58:36 +08:00
Some ( precision ) = > format! ( $fmt_str_with_precision , val , precision )
. chars ( )
. count ( ) ,
2019-08-21 21:53:23 +08:00
None = > format! ( $fmt_str_without_precision , val ) . chars ( ) . count ( ) ,
}
2017-08-03 01:37:44 +08:00
}
2019-08-19 21:15:14 +08:00
2019-08-21 21:53:23 +08:00
#[ cfg(not(feature = " std " )) ]
2021-07-26 01:06:14 +08:00
fn val_width < T : Scalar + $trait > ( _ : & T , _ : & mut fmt ::Formatter < '_ > ) -> usize {
2019-08-21 21:53:23 +08:00
4
}
2019-08-19 21:15:14 +08:00
2021-07-08 20:18:22 +08:00
let ( nrows , ncols ) = self . shape ( ) ;
2019-08-19 21:15:14 +08:00
2021-07-08 20:18:22 +08:00
if nrows = = 0 | | ncols = = 0 {
2019-08-21 21:53:23 +08:00
return write! ( f , " [ ] " ) ;
}
2019-08-19 21:15:14 +08:00
2019-08-21 21:53:23 +08:00
let mut max_length = 0 ;
2019-08-19 21:15:14 +08:00
2019-08-21 21:53:23 +08:00
for i in 0 .. nrows {
for j in 0 .. ncols {
2021-07-08 20:18:22 +08:00
max_length = crate ::max ( max_length , val_width ( & self [ ( i , j ) ] , f ) ) ;
2019-08-21 21:53:23 +08:00
}
}
2019-08-19 21:15:14 +08:00
2019-08-21 21:53:23 +08:00
let max_length_with_space = max_length + 1 ;
writeln! ( f ) ? ;
writeln! (
f ,
" ┌ {:>width$} ┐ " ,
" " ,
width = max_length_with_space * ncols - 1
) ? ;
for i in 0 .. nrows {
write! ( f , " │ " ) ? ;
for j in 0 .. ncols {
2021-07-08 20:18:22 +08:00
let number_length = val_width ( & self [ ( i , j ) ] , f ) + 1 ;
2019-08-21 21:53:23 +08:00
let pad = max_length_with_space - number_length ;
write! ( f , " {:>thepad$} " , " " , thepad = pad ) ? ;
match f . precision ( ) {
2020-03-18 00:58:36 +08:00
Some ( precision ) = > {
write! ( f , $fmt_str_with_precision , ( * self ) [ ( i , j ) ] , precision ) ?
}
2019-08-21 21:53:23 +08:00
None = > write! ( f , $fmt_str_without_precision , ( * self ) [ ( i , j ) ] ) ? ,
}
}
writeln! ( f , " │ " ) ? ;
2019-08-19 21:15:14 +08:00
}
2019-08-21 21:53:23 +08:00
writeln! (
f ,
" └ {:>width$} ┘ " ,
" " ,
width = max_length_with_space * ncols - 1
) ? ;
writeln! ( f )
2019-08-19 21:15:14 +08:00
}
}
2019-08-21 21:53:23 +08:00
} ;
2019-08-19 21:15:14 +08:00
}
2019-08-21 21:53:23 +08:00
impl_fmt! ( fmt ::Display , " {} " , " {:.1$} " ) ;
impl_fmt! ( fmt ::LowerExp , " {:e} " , " {:.1$e} " ) ;
impl_fmt! ( fmt ::UpperExp , " {:E} " , " {:.1$E} " ) ;
impl_fmt! ( fmt ::Octal , " {:o} " , " {:1$o} " ) ;
impl_fmt! ( fmt ::LowerHex , " {:x} " , " {:1$x} " ) ;
impl_fmt! ( fmt ::UpperHex , " {:X} " , " {:1$X} " ) ;
impl_fmt! ( fmt ::Binary , " {:b} " , " {:.1$b} " ) ;
impl_fmt! ( fmt ::Pointer , " {:p} " , " {:.1$p} " ) ;
2019-08-19 21:15:14 +08:00
2021-07-08 20:18:22 +08:00
#[ cfg(test) ]
mod tests {
#[ test ]
fn empty_display ( ) {
let vec : Vec < f64 > = Vec ::new ( ) ;
let dvector = crate ::DVector ::from_vec ( vec ) ;
assert_eq! ( format! ( " {} " , dvector ) , " [ ] " )
}
#[ test ]
fn lower_exp ( ) {
let test = crate ::Matrix2 ::new ( 1e6 , 2e5 , 2e-5 , 1. ) ;
assert_eq! (
format! ( " {:e} " , test ) ,
r "
2019-08-19 21:15:14 +08:00
┌ ┐
│ 1e6 2e5 │
│ 2e-5 1e0 │
└ ┘
2020-03-18 00:58:36 +08:00
"
2021-07-08 20:18:22 +08:00
)
}
2019-08-19 21:15:14 +08:00
}
2020-11-15 23:57:49 +08:00
/// # Cross product
2021-08-03 00:41:46 +08:00
impl < T : Scalar + ClosedAdd + ClosedSub + ClosedMul , R : Dim , C : Dim , S : RawStorage < T , R , C > >
2021-04-11 17:00:38 +08:00
Matrix < T , R , C , S >
2020-03-21 19:16:46 +08:00
{
2016-12-05 05:44:42 +08:00
/// The perpendicular product between two 2D column vectors, i.e. `a.x * b.y - a.y * b.x`.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn perp < R2 , C2 , SB > ( & self , b : & Matrix < T , R2 , C2 , SB > ) -> T
2018-02-02 19:26:35 +08:00
where
R2 : Dim ,
C2 : Dim ,
2021-08-03 00:41:46 +08:00
SB : RawStorage < T , R2 , C2 > ,
2018-02-02 19:26:35 +08:00
ShapeConstraint : SameNumberOfRows < R , U2 >
+ SameNumberOfColumns < C , U1 >
+ SameNumberOfRows < R2 , U2 >
+ SameNumberOfColumns < C2 , U1 > ,
{
2021-09-19 22:00:49 +08:00
let shape = self . shape ( ) ;
assert_eq! (
shape ,
b . shape ( ) ,
" 2D vector perpendicular product dimension mismatch. "
) ;
assert_eq! (
shape ,
( 2 , 1 ) ,
" 2D perpendicular product requires (2, 1) vectors {:?} " ,
shape
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
) ;
2016-12-05 05:44:42 +08:00
2021-09-19 22:00:49 +08:00
// SAFETY: assertion above ensures correct shape
let ax = unsafe { self . get_unchecked ( ( 0 , 0 ) ) . clone ( ) } ;
let ay = unsafe { self . get_unchecked ( ( 1 , 0 ) ) . clone ( ) } ;
let bx = unsafe { b . get_unchecked ( ( 0 , 0 ) ) . clone ( ) } ;
let by = unsafe { b . get_unchecked ( ( 1 , 0 ) ) . clone ( ) } ;
ax * by - ay * bx
2016-12-05 05:44:42 +08:00
}
2020-11-15 23:57:49 +08:00
// TODO: use specialization instead of an assertion.
2016-12-05 05:44:42 +08:00
/// The 3D cross product between two vectors.
///
/// Panics if the shape is not 3D vector. In the future, this will be implemented only for
/// dynamically-sized matrices and statically-sized 3D matrices.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-08-03 00:41:46 +08:00
pub fn cross < R2 , C2 , SB > ( & self , b : & Matrix < T , R2 , C2 , SB > ) -> MatrixCross < T , R , C , R2 , C2 >
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
R2 : Dim ,
C2 : Dim ,
SB : RawStorage < T , R2 , C2 > ,
2021-04-11 17:00:38 +08:00
DefaultAllocator : SameShapeAllocator < T , R , C , R2 , C2 > ,
2018-02-02 19:26:35 +08:00
ShapeConstraint : SameNumberOfRows < R , R2 > + SameNumberOfColumns < C , C2 > ,
{
2016-12-05 05:44:42 +08:00
let shape = self . shape ( ) ;
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
assert_eq! ( shape , b . shape ( ) , " Vector cross product dimension mismatch. " ) ;
2018-02-02 19:26:35 +08:00
assert! (
2021-09-19 22:00:49 +08:00
shape = = ( 3 , 1 ) | | shape = = ( 1 , 3 ) ,
More verbose DMatrix dim asserts where possible.
Previously, most dimension mismatch asserts used raw `assert!` and did
not include the mismatching dimensions in the panic message. When using
dynamic matrices, this led to somewhat-opaque panics such as:
```rust
let m1 = DMatrix::<f32>::zeros(2, 3);
let m2 = DMatrix::<f32>::zeros(5, 10);
m1 + m2 // panic: Matrix addition/subtraction dimensions mismatch.
```
This patch adds dimension information in the panic messages wherever
doing so did not add additional bounds checks, mostly by simply changing
`assert!(a == b, ...)` cases to `assert_eq!`. After:
```rust
// panic: assertion failed: `(left == right)`
// left: `(2, 3)`,
// right: `(5, 10)`: Matrix addition/subtraction dimensions mismatch.
```
Note that the `gemv` and `ger` were not updated, as they are called from
within other functions on subset matricies -- e.g., `gemv` is called
from `gemm` which is called from `mul_to` . Including dimension
information in the `gemv` panic messages would be confusing to
`mul` / `mul_to` users, because it would include dimensions of the column
vectors that `gemm` passes to `gemv` rather than of the original `mul`
arguments. A fix would be to add bounds checks to `mul_to`, but that may
have performance and redundancy implications, so is left to another
patch.
2020-06-23 06:29:13 +08:00
" Vector cross product dimension mismatch: must be (3, 1) or (1, 3) but found {:?}. " ,
shape
2018-02-02 19:26:35 +08:00
) ;
2016-12-05 05:44:42 +08:00
if shape . 0 = = 3 {
unsafe {
2021-09-19 22:00:49 +08:00
let mut res = Matrix ::uninit ( Dim ::from_usize ( 3 ) , Dim ::from_usize ( 1 ) ) ;
2016-12-05 05:44:42 +08:00
2019-12-06 06:54:17 +08:00
let ax = self . get_unchecked ( ( 0 , 0 ) ) ;
let ay = self . get_unchecked ( ( 1 , 0 ) ) ;
let az = self . get_unchecked ( ( 2 , 0 ) ) ;
2016-12-05 05:44:42 +08:00
2019-12-06 06:54:17 +08:00
let bx = b . get_unchecked ( ( 0 , 0 ) ) ;
let by = b . get_unchecked ( ( 1 , 0 ) ) ;
let bz = b . get_unchecked ( ( 2 , 0 ) ) ;
2016-12-05 05:44:42 +08:00
2021-08-04 23:34:25 +08:00
* res . get_unchecked_mut ( ( 0 , 0 ) ) =
MaybeUninit ::new ( ay . clone ( ) * bz . clone ( ) - az . clone ( ) * by . clone ( ) ) ;
* res . get_unchecked_mut ( ( 1 , 0 ) ) =
MaybeUninit ::new ( az . clone ( ) * bx . clone ( ) - ax . clone ( ) * bz . clone ( ) ) ;
* res . get_unchecked_mut ( ( 2 , 0 ) ) =
MaybeUninit ::new ( ax . clone ( ) * by . clone ( ) - ay . clone ( ) * bx . clone ( ) ) ;
2021-07-16 12:56:58 +08:00
2021-08-03 00:41:46 +08:00
// Safety: res is now fully initialized.
2021-07-16 12:56:58 +08:00
res . assume_init ( )
2016-12-05 05:44:42 +08:00
}
2018-02-02 19:26:35 +08:00
} else {
2016-12-05 05:44:42 +08:00
unsafe {
2021-09-19 22:00:49 +08:00
let mut res = Matrix ::uninit ( Dim ::from_usize ( 1 ) , Dim ::from_usize ( 3 ) ) ;
2016-12-05 05:44:42 +08:00
2019-12-06 06:54:17 +08:00
let ax = self . get_unchecked ( ( 0 , 0 ) ) ;
let ay = self . get_unchecked ( ( 0 , 1 ) ) ;
let az = self . get_unchecked ( ( 0 , 2 ) ) ;
2016-12-05 05:44:42 +08:00
2019-12-06 06:54:17 +08:00
let bx = b . get_unchecked ( ( 0 , 0 ) ) ;
let by = b . get_unchecked ( ( 0 , 1 ) ) ;
let bz = b . get_unchecked ( ( 0 , 2 ) ) ;
2016-12-05 05:44:42 +08:00
2021-08-04 23:34:25 +08:00
* res . get_unchecked_mut ( ( 0 , 0 ) ) =
MaybeUninit ::new ( ay . clone ( ) * bz . clone ( ) - az . clone ( ) * by . clone ( ) ) ;
* res . get_unchecked_mut ( ( 0 , 1 ) ) =
MaybeUninit ::new ( az . clone ( ) * bx . clone ( ) - ax . clone ( ) * bz . clone ( ) ) ;
* res . get_unchecked_mut ( ( 0 , 2 ) ) =
MaybeUninit ::new ( ax . clone ( ) * by . clone ( ) - ay . clone ( ) * bx . clone ( ) ) ;
2021-07-16 12:56:58 +08:00
2021-08-03 00:41:46 +08:00
// Safety: res is now fully initialized.
2021-07-16 12:56:58 +08:00
res . assume_init ( )
2016-12-05 05:44:42 +08:00
}
}
}
}
2021-08-03 00:41:46 +08:00
impl < T : Scalar + Field , S : RawStorage < T , U3 > > Vector < T , U3 , S > {
2018-02-02 19:26:31 +08:00
/// Computes the matrix `M` such that for all vector `v` we have `M * v == self.cross(&v)`.
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn cross_matrix ( & self ) -> OMatrix < T , U3 , U3 > {
OMatrix ::< T , U3 , U3 > ::new (
T ::zero ( ) ,
2021-08-04 23:34:25 +08:00
- self [ 2 ] . clone ( ) ,
self [ 1 ] . clone ( ) ,
self [ 2 ] . clone ( ) ,
2021-04-11 17:00:38 +08:00
T ::zero ( ) ,
2021-08-04 23:34:25 +08:00
- self [ 0 ] . clone ( ) ,
- self [ 1 ] . clone ( ) ,
self [ 0 ] . clone ( ) ,
2021-04-11 17:00:38 +08:00
T ::zero ( ) ,
2018-02-02 19:26:35 +08:00
)
2018-02-02 19:26:31 +08:00
}
}
2021-04-11 17:00:38 +08:00
impl < T : SimdComplexField , R : Dim , C : Dim , S : Storage < T , R , C > > Matrix < T , R , C , S > {
2017-08-03 01:37:44 +08:00
/// The smallest angle between two vectors.
2016-12-05 05:44:42 +08:00
#[ inline ]
2021-06-07 22:34:03 +08:00
#[ must_use ]
2021-04-11 17:00:38 +08:00
pub fn angle < R2 : Dim , C2 : Dim , SB > ( & self , other : & Matrix < T , R2 , C2 , SB > ) -> T ::SimdRealField
2018-02-02 19:26:35 +08:00
where
2021-04-11 17:00:38 +08:00
SB : Storage < T , R2 , C2 > ,
2018-02-02 19:26:35 +08:00
ShapeConstraint : DimEq < R , R2 > + DimEq < C , C2 > ,
{
2019-03-23 18:48:12 +08:00
let prod = self . dotc ( other ) ;
2018-02-02 19:26:35 +08:00
let n1 = self . norm ( ) ;
let n2 = other . norm ( ) ;
2016-12-05 05:44:42 +08:00
if n1 . is_zero ( ) | | n2 . is_zero ( ) {
2021-04-11 17:00:38 +08:00
T ::SimdRealField ::zero ( )
2018-02-02 19:26:35 +08:00
} else {
2020-03-18 00:58:36 +08:00
let cang = prod . simd_real ( ) / ( n1 * n2 ) ;
2021-04-11 17:00:38 +08:00
cang . simd_clamp ( - T ::SimdRealField ::one ( ) , T ::SimdRealField ::one ( ) )
2020-03-18 00:58:36 +08:00
. simd_acos ( )
2016-12-05 05:44:42 +08:00
}
}
2017-08-03 01:37:44 +08:00
}
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S > AbsDiffEq for Unit < Matrix < T , R , C , S > >
2018-02-02 19:26:35 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar + AbsDiffEq ,
S : RawStorage < T , R , C > ,
2021-08-04 23:34:25 +08:00
T ::Epsilon : Clone ,
2018-02-02 19:26:35 +08:00
{
2021-04-11 17:00:38 +08:00
type Epsilon = T ::Epsilon ;
2017-02-13 01:17:09 +08:00
#[ inline ]
fn default_epsilon ( ) -> Self ::Epsilon {
2021-04-11 17:00:38 +08:00
T ::default_epsilon ( )
2017-02-13 01:17:09 +08:00
}
#[ inline ]
2018-05-19 21:41:58 +08:00
fn abs_diff_eq ( & self , other : & Self , epsilon : Self ::Epsilon ) -> bool {
self . as_ref ( ) . abs_diff_eq ( other . as_ref ( ) , epsilon )
2017-02-13 01:17:09 +08:00
}
2018-05-19 21:41:58 +08:00
}
2017-02-13 01:17:09 +08:00
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S > RelativeEq for Unit < Matrix < T , R , C , S > >
2018-05-19 21:41:58 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar + RelativeEq ,
2021-04-11 17:00:38 +08:00
S : Storage < T , R , C > ,
2021-08-04 23:34:25 +08:00
T ::Epsilon : Clone ,
2018-05-19 21:41:58 +08:00
{
2017-02-13 01:17:09 +08:00
#[ inline ]
2018-05-19 21:41:58 +08:00
fn default_max_relative ( ) -> Self ::Epsilon {
2021-04-11 17:00:38 +08:00
T ::default_max_relative ( )
2017-02-13 01:17:09 +08:00
}
#[ inline ]
2018-02-02 19:26:35 +08:00
fn relative_eq (
& self ,
other : & Self ,
epsilon : Self ::Epsilon ,
max_relative : Self ::Epsilon ,
2020-04-06 00:49:48 +08:00
) -> bool {
2018-02-02 19:26:35 +08:00
self . as_ref ( )
. relative_eq ( other . as_ref ( ) , epsilon , max_relative )
2017-02-13 01:17:09 +08:00
}
2018-05-19 21:41:58 +08:00
}
2021-08-03 00:41:46 +08:00
impl < T , R : Dim , C : Dim , S > UlpsEq for Unit < Matrix < T , R , C , S > >
2018-05-19 21:41:58 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar + UlpsEq ,
S : RawStorage < T , R , C > ,
2021-08-04 23:34:25 +08:00
T ::Epsilon : Clone ,
2018-05-19 21:41:58 +08:00
{
#[ inline ]
fn default_max_ulps ( ) -> u32 {
2021-04-11 17:00:38 +08:00
T ::default_max_ulps ( )
2018-05-19 21:41:58 +08:00
}
2017-02-13 01:17:09 +08:00
#[ inline ]
fn ulps_eq ( & self , other : & Self , epsilon : Self ::Epsilon , max_ulps : u32 ) -> bool {
self . as_ref ( ) . ulps_eq ( other . as_ref ( ) , epsilon , max_ulps )
}
}
2018-12-28 01:05:25 +08:00
2021-08-03 00:41:46 +08:00
impl < T , R , C , S > Hash for Matrix < T , R , C , S >
2018-12-28 01:05:25 +08:00
where
2021-08-03 00:41:46 +08:00
T : Scalar + Hash ,
R : Dim ,
C : Dim ,
S : RawStorage < T , R , C > ,
2018-12-28 01:05:25 +08:00
{
fn hash < H : Hasher > ( & self , state : & mut H ) {
let ( nrows , ncols ) = self . shape ( ) ;
( nrows , ncols ) . hash ( state ) ;
for j in 0 .. ncols {
for i in 0 .. nrows {
unsafe {
2020-03-18 00:58:36 +08:00
self . get_unchecked ( ( i , j ) ) . hash ( state ) ;
2018-12-28 01:05:25 +08:00
}
}
}
}
}