1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137

use crate::linear::bivec::Bivector2;
use crate::linear::from_float;
use crate::linear::mat::Matrix;
use crate::linear::scalar::{Scalar, Two};
use crate::linear::vec::Vector;
use crate::numeric::float::Float;
use crate::numeric::identity::Zero;
use crate::numeric::mul_add::MulAdd;
use crate::numeric::trig::Trig;

#[derive(Debug, Copy, Clone)]
pub struct Rotor2<S> {
    pub scalar: S,
    pub bivector: Bivector2<S>,
}

impl<S> Rotor2<S> {
    #[inline]
    pub const fn new(scalar: S, bivector: Bivector2<S>) -> Self {
        Self { scalar, bivector }
    }

    #[inline]
    pub fn layout() -> std::alloc::Layout {
        std::alloc::Layout::from_size_align(std::mem::size_of::<Self>(), std::mem::align_of::<S>())
            .unwrap()
    }
}

impl<S: Scalar> Rotor2<S> {
    #[inline]
    pub fn identity() -> Self {
        Self {
            scalar: <S>::one(),
            bivector: Bivector2::zero(),
        }
    }

    #[inline]
    pub fn reverse(&mut self) {
        self.bivector = -self.bivector;
    }

    #[inline]
    pub fn reversed(&self) -> Self {
        let mut s = *self;
        s.reverse();
        s
    }

    #[inline]
    pub fn dot(&self, rhs: Self) -> S {
        self.scalar * rhs.scalar + self.bivector.dot(rhs.bivector)
    }
}

impl<S: Scalar + MulAdd<Output = S>> Rotor2<S> {
    #[inline]
    pub fn rotate_by(&mut self, other: Self) {
        let b = *self;
        let a = other;
        let sa2_plus_baxy2 = a
            .scalar
            .mul_add(a.scalar, a.bivector.data * a.bivector.data);

        self.scalar =
            (a.scalar - b.scalar) * a.bivector.data * b.bivector.data + b.scalar * sa2_plus_baxy2;
        self.bivector.data = b.bivector.data * sa2_plus_baxy2;
    }

    #[inline]
    pub fn rotated_by(mut self, other: Self) -> Self {
        self.rotate_by(other);
        self
    }
}

impl<S: Scalar + Two> Rotor2<S> {
    #[inline]
    pub fn rotate_vec(self, vec: &mut Vector<S, 2>) {
        let fx = self.scalar * vec[0] + self.bivector.data * vec[1];
        let fy = self.scalar * vec[1] - (self.bivector.data * vec[0]);

        vec[0] = self.scalar * fx + self.bivector.data * fy;
        vec[1] = self.scalar * fy - (self.bivector.data * fx);
    }

    #[inline]
    pub fn into_matrix(self) -> Matrix<S, 2, 2> {
        let s2_minus_bxy2 = self.scalar * self.scalar - self.bivector.data * self.bivector.data;
        let two_s_bxy = S::two() * self.scalar * self.bivector.data;

        Matrix::new([
            Vector::new([s2_minus_bxy2, -two_s_bxy]),
            Vector::new([two_s_bxy, s2_minus_bxy2]),
        ])
    }
}

impl<S: Scalar + Float> Rotor2<S> {
    #[inline]
    pub fn from_rotation_between(from: Vector<S, 2>, rhs: Vector<S, 2>) -> Self {
        Self::new(<S>::one() + rhs.dot(from), rhs.wedge(from)).normalized()
    }

    #[inline]
    pub fn magnitude_squared(&self) -> S {
        self.scalar * self.scalar + self.bivector.magnitude_squared()
    }
    #[inline]
    pub fn magnitude(&self) -> S {
        self.magnitude_squared().sqrt()
    }

    #[inline]
    pub fn normalize(&mut self) {
        let mag = self.magnitude();
        self.scalar /= mag;
        self.bivector.data /= mag;
    }

    #[inline]
    pub fn normalized(&self) -> Self {
        let mut rotor = *self;
        rotor.normalize();
        rotor
    }
}

impl<S: Scalar + Two + Trig> Rotor2<S> {
    #[inline]
    pub fn from_angle(angle: S) -> Self {
        let half_angle = angle / <S>::two();
        let (sin, cos) = half_angle.sin_cos();
        Self::new(cos, Bivector2::new(-sin))
    }
}