more generic static<->dynamic conversation

This commit is contained in:
Lishen 2024-05-05 02:26:03 +03:00
parent 02d66f28d0
commit 62e7d8a43e
2 changed files with 57 additions and 37 deletions

View File

@ -1,5 +1,7 @@
#[cfg(all(feature = "alloc", not(feature = "std")))] #[cfg(all(feature = "alloc", not(feature = "std")))]
use alloc::vec::Vec; use alloc::vec::Vec;
use core::convert::TryFrom;
use core::panic;
use simba::scalar::{SubsetOf, SupersetOf}; use simba::scalar::{SubsetOf, SupersetOf};
use std::borrow::{Borrow, BorrowMut}; use std::borrow::{Borrow, BorrowMut};
use std::convert::{AsMut, AsRef, From, Into}; use std::convert::{AsMut, AsRef, From, Into};
@ -19,12 +21,8 @@ use crate::base::{
ArrayStorage, DVectorView, DVectorViewMut, DefaultAllocator, Matrix, MatrixView, MatrixViewMut, ArrayStorage, DVectorView, DVectorViewMut, DefaultAllocator, Matrix, MatrixView, MatrixViewMut,
OMatrix, Scalar, OMatrix, Scalar,
}; };
//TODO: move near From implementation?
#[cfg(any(feature = "std", feature = "alloc"))] #[cfg(any(feature = "std", feature = "alloc"))]
use crate::base::{DMatrixView, SMatrixView, SVectorView}; use crate::base::{DVector, RowDVector, VecStorage};
#[cfg(any(feature = "std", feature = "alloc"))]
use crate::base::{DMatrix, DVector, RowDVector, VecStorage};
use crate::base::{ViewStorage, ViewStorageMut}; use crate::base::{ViewStorage, ViewStorageMut};
use crate::constraint::DimEq; use crate::constraint::DimEq;
use crate::{IsNotStaticOne, RowSVector, SMatrix, SVector, VectorView, VectorViewMut}; use crate::{IsNotStaticOne, RowSVector, SMatrix, SVector, VectorView, VectorViewMut};
@ -433,34 +431,65 @@ where
} }
#[cfg(any(feature = "std", feature = "alloc"))] #[cfg(any(feature = "std", feature = "alloc"))]
impl<'a, T: Scalar, const R: usize, const C: usize> From<&'a DMatrix<T>> for SMatrix<T, R, C> { impl<'a, T: Scalar, const R0: usize, const C0: usize, R1: Dim, C1: Dim>
fn from(m: &'a DMatrix<T>) -> Self { TryFrom<&VecStorage<T, R1, C1>> for ArrayStorage<T, R0, C0>
let v: DMatrixView<'a, T> = m.as_view(); {
SMatrixView::<T, R, C>::from(&v).clone_owned() // TODO: where to put Error enum ?
type Error = ();
fn try_from(vstorage: &VecStorage<T, R1, C1>) -> Result<Self, Self::Error> {
if R0 * C0 != vstorage.len() {
Err(())
} else {
let mut data = MaybeUninit::<[[T; R0]; C0]>::uninit();
unsafe { core::slice::from_raw_parts_mut(data.as_mut_ptr() as *mut T, R0 * C0) }
.clone_from_slice(vstorage.as_slice());
Ok(Self(unsafe { data.assume_init() }))
}
} }
} }
#[cfg(any(feature = "std", feature = "alloc"))] #[cfg(any(feature = "std", feature = "alloc"))]
impl<'a, T: Scalar, const R: usize, const C: usize> From<&'a SMatrix<T, R, C>> for DMatrix<T> { impl<'a, T: Scalar, const R0: usize, const C0: usize, R1: Dim, C1: Dim>
fn from(m: &'a SMatrix<T, R, C>) -> Self { From<&ArrayStorage<T, R0, C0>> for VecStorage<T, R1, C1>
let v: SMatrixView<'a, T, R, C> = m.as_view(); {
DMatrixView::<T>::from(&v).clone_owned() fn from(astorage: &ArrayStorage<T, R0, C0>) -> Self {
VecStorage::new(
R1::from_usize(R0),
C1::from_usize(C0),
Vec::from(astorage.as_slice()),
)
} }
} }
#[cfg(any(feature = "std", feature = "alloc"))] #[cfg(any(feature = "std", feature = "alloc"))]
impl<'a, T: Scalar, const R: usize> From<&'a DVector<T>> for SVector<T, R> { impl<'a, T: Scalar, const R0: usize, const C0: usize, R1: Dim, C1: Dim>
fn from(m: &'a DVector<T>) -> Self { From<&'a Matrix<T, R1, C1, VecStorage<T, R1, C1>>>
let v: DVectorView<'a, T> = m.as_view(); for Matrix<T, Const<R0>, Const<C0>, ArrayStorage<T, R0, C0>>
SVectorView::<T, R>::from(&v).clone_owned() where
VecStorage<T, R1, C1>: RawStorage<T, R1, C1>,
{
fn from(m: &'a Matrix<T, R1, C1, VecStorage<T, R1, C1>>) -> Self {
match ArrayStorage::<T, R0, C0>::try_from(&m.data) {
Ok(s) => {
assert_eq!(R0, m.nrows());
assert_eq!(C0, m.ncols());
Self::from_array_storage(s)
}
Err(_) => panic!("storage dimensions mismatch"),
}
} }
} }
#[cfg(any(feature = "std", feature = "alloc"))] #[cfg(any(feature = "std", feature = "alloc"))]
impl<'a, T: Scalar, const R: usize> From<&'a SVector<T, R>> for DVector<T> { impl<'a, T: Scalar, const R0: usize, const C0: usize, R1: Dim, C1: Dim>
fn from(m: &'a SVector<T, R>) -> Self { From<&'a Matrix<T, Const<R0>, Const<C0>, ArrayStorage<T, R0, C0>>>
let v: SVectorView<'a, T, R> = m.as_view(); for Matrix<T, R1, C1, VecStorage<T, R1, C1>>
DVectorView::<T>::from(&v).clone_owned() where
VecStorage<T, R1, C1>: RawStorage<T, R1, C1>,
{
fn from(m: &'a Matrix<T, Const<R0>, Const<C0>, ArrayStorage<T, R0, C0>>) -> Self {
Self::from_data(VecStorage::from(&m.data))
} }
} }

View File

@ -1366,22 +1366,6 @@ fn dvector_from_svector() {
assert_eq!(d, s); assert_eq!(d, s);
} }
#[test]
fn svector3_from_dvector() {
type T = f64;
let d = DVector::<T>::from_fn(3, |i, j| (i * 3) as T + j as T);
let s = Vector3::<T>::from(&d);
assert_eq!(d, s);
}
#[test]
fn dvector_from_svector3() {
type T = f64;
let s = Vector3::<T>::from_fn(|i, j| (i * 3) as T + j as T);
let d = DVector::<T>::from(&s);
assert_eq!(d, s);
}
#[test] #[test]
fn svector3_from_dmatrix3x1() { fn svector3_from_dmatrix3x1() {
type T = f64; type T = f64;
@ -1398,3 +1382,10 @@ fn svector3_from_dmatrix3x3() {
let s = Vector3::<T>::from(&d); let s = Vector3::<T>::from(&d);
assert_eq!(d, s); assert_eq!(d, s);
} }
#[test]
fn dmatrix_from_svector3() {
type T = f64;
let s = SVector::<T, 3>::from_fn(|i, j| (i * 3) as T + j as T);
let d = DMatrix::<T>::from(&s);
assert_eq!(d, s);
}