52 lines
1.6 KiB
Rust
52 lines
1.6 KiB
Rust
|
#[cfg(feature = "arbitrary")]
|
||
|
|
use quickcheck::{Arbitrary, Gen};
|
||
|
|
#[cfg(feature = "arbitrary")]
|
||
|
|
use core::storage::Owned;
|
||
|
|
|
||
|
|
use num_complex::Complex;
|
||
|
|
use alga::general::Real;
|
||
|
|
use core::{DefaultAllocator, MatrixN};
|
||
|
|
use core::dimension::{Dim, Dynamic, U2};
|
||
|
|
use core::allocator::Allocator;
|
||
|
|
use geometry::UnitComplex;
|
||
|
|
|
||
|
|
/// A random orthogonal matrix.
|
||
|
|
#[derive(Clone, Debug)]
|
||
|
|
pub struct RandomOrthogonal<N: Real, D: Dim = Dynamic>
|
||
|
|
where DefaultAllocator: Allocator<N, D, D> {
|
||
|
|
m: MatrixN<N, D>
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
impl<N: Real, D: Dim> RandomOrthogonal<N, D>
|
||
|
|
where DefaultAllocator: Allocator<N, D, D> {
|
||
|
|
/// Retrieve the generated matrix.
|
||
|
|
pub fn unwrap(self) -> MatrixN<N, D> {
|
||
|
|
self.m
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Creates a new random orthogonal matrix from its dimension and a random reals generators.
|
||
|
|
pub fn new<Rand: FnMut() -> N>(dim: D, mut rand: Rand) -> Self {
|
||
|
|
let mut res = MatrixN::identity_generic(dim, dim);
|
||
|
|
|
||
|
|
// Create an orthogonal matrix by compositing planar 2D rotations.
|
||
|
|
for i in 0 .. dim.value() - 1 {
|
||
|
|
let c = Complex::new(rand(), rand());
|
||
|
|
let rot: UnitComplex<N> = UnitComplex::from_complex(c);
|
||
|
|
rot.rotate(&mut res.fixed_rows_mut::<U2>(i));
|
||
|
|
}
|
||
|
|
|
||
|
|
RandomOrthogonal { m: res }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(feature = "arbitrary")]
|
||
|
|
impl<N: Real + Arbitrary + Send, D: Dim> Arbitrary for RandomOrthogonal<N, D>
|
||
|
|
where DefaultAllocator: Allocator<N, D, D>,
|
||
|
|
Owned<N, D, D>: Clone + Send {
|
||
|
|
fn arbitrary<G: Gen>(g: &mut G) -> Self {
|
||
|
|
let dim = D::try_to_usize().unwrap_or(g.gen_range(1, 50));
|
||
|
|
Self::new(D::from_usize(dim), || N::arbitrary(g))
|
||
|
|
}
|
||
|
|
}
|