1use crate::{
4 euler::{FromEuler, ToEuler},
5 f64::math,
6 swizzles::*,
7 DMat2, DMat4, DQuat, DVec2, DVec3, EulerRot, Mat3,
8};
9use core::fmt;
10use core::iter::{Product, Sum};
11use core::ops::{Add, AddAssign, Div, DivAssign, Mul, MulAssign, Neg, Sub, SubAssign};
12
13#[inline(always)]
15#[must_use]
16pub const fn dmat3(x_axis: DVec3, y_axis: DVec3, z_axis: DVec3) -> DMat3 {
17 DMat3::from_cols(x_axis, y_axis, z_axis)
18}
19
20#[derive(Clone, Copy)]
45#[repr(C)]
46pub struct DMat3 {
47 pub x_axis: DVec3,
48 pub y_axis: DVec3,
49 pub z_axis: DVec3,
50}
51
52impl DMat3 {
53 pub const ZERO: Self = Self::from_cols(DVec3::ZERO, DVec3::ZERO, DVec3::ZERO);
55
56 pub const IDENTITY: Self = Self::from_cols(DVec3::X, DVec3::Y, DVec3::Z);
58
59 pub const NAN: Self = Self::from_cols(DVec3::NAN, DVec3::NAN, DVec3::NAN);
61
62 #[allow(clippy::too_many_arguments)]
63 #[inline(always)]
64 #[must_use]
65 const fn new(
66 m00: f64,
67 m01: f64,
68 m02: f64,
69 m10: f64,
70 m11: f64,
71 m12: f64,
72 m20: f64,
73 m21: f64,
74 m22: f64,
75 ) -> Self {
76 Self {
77 x_axis: DVec3::new(m00, m01, m02),
78 y_axis: DVec3::new(m10, m11, m12),
79 z_axis: DVec3::new(m20, m21, m22),
80 }
81 }
82
83 #[inline(always)]
85 #[must_use]
86 pub const fn from_cols(x_axis: DVec3, y_axis: DVec3, z_axis: DVec3) -> Self {
87 Self {
88 x_axis,
89 y_axis,
90 z_axis,
91 }
92 }
93
94 #[inline]
98 #[must_use]
99 pub const fn from_cols_array(m: &[f64; 9]) -> Self {
100 Self::new(m[0], m[1], m[2], m[3], m[4], m[5], m[6], m[7], m[8])
101 }
102
103 #[inline]
106 #[must_use]
107 pub const fn to_cols_array(&self) -> [f64; 9] {
108 [
109 self.x_axis.x,
110 self.x_axis.y,
111 self.x_axis.z,
112 self.y_axis.x,
113 self.y_axis.y,
114 self.y_axis.z,
115 self.z_axis.x,
116 self.z_axis.y,
117 self.z_axis.z,
118 ]
119 }
120
121 #[inline]
125 #[must_use]
126 pub const fn from_cols_array_2d(m: &[[f64; 3]; 3]) -> Self {
127 Self::from_cols(
128 DVec3::from_array(m[0]),
129 DVec3::from_array(m[1]),
130 DVec3::from_array(m[2]),
131 )
132 }
133
134 #[inline]
137 #[must_use]
138 pub const fn to_cols_array_2d(&self) -> [[f64; 3]; 3] {
139 [
140 self.x_axis.to_array(),
141 self.y_axis.to_array(),
142 self.z_axis.to_array(),
143 ]
144 }
145
146 #[doc(alias = "scale")]
148 #[inline]
149 #[must_use]
150 pub const fn from_diagonal(diagonal: DVec3) -> Self {
151 Self::new(
152 diagonal.x, 0.0, 0.0, 0.0, diagonal.y, 0.0, 0.0, 0.0, diagonal.z,
153 )
154 }
155
156 #[inline]
158 #[must_use]
159 pub fn from_mat4(m: DMat4) -> Self {
160 Self::from_cols(
161 DVec3::from_vec4(m.x_axis),
162 DVec3::from_vec4(m.y_axis),
163 DVec3::from_vec4(m.z_axis),
164 )
165 }
166
167 #[inline]
174 #[must_use]
175 pub fn from_mat4_minor(m: DMat4, i: usize, j: usize) -> Self {
176 match (i, j) {
177 (0, 0) => Self::from_cols(m.y_axis.yzw(), m.z_axis.yzw(), m.w_axis.yzw()),
178 (0, 1) => Self::from_cols(m.y_axis.xzw(), m.z_axis.xzw(), m.w_axis.xzw()),
179 (0, 2) => Self::from_cols(m.y_axis.xyw(), m.z_axis.xyw(), m.w_axis.xyw()),
180 (0, 3) => Self::from_cols(m.y_axis.xyz(), m.z_axis.xyz(), m.w_axis.xyz()),
181 (1, 0) => Self::from_cols(m.x_axis.yzw(), m.z_axis.yzw(), m.w_axis.yzw()),
182 (1, 1) => Self::from_cols(m.x_axis.xzw(), m.z_axis.xzw(), m.w_axis.xzw()),
183 (1, 2) => Self::from_cols(m.x_axis.xyw(), m.z_axis.xyw(), m.w_axis.xyw()),
184 (1, 3) => Self::from_cols(m.x_axis.xyz(), m.z_axis.xyz(), m.w_axis.xyz()),
185 (2, 0) => Self::from_cols(m.x_axis.yzw(), m.y_axis.yzw(), m.w_axis.yzw()),
186 (2, 1) => Self::from_cols(m.x_axis.xzw(), m.y_axis.xzw(), m.w_axis.xzw()),
187 (2, 2) => Self::from_cols(m.x_axis.xyw(), m.y_axis.xyw(), m.w_axis.xyw()),
188 (2, 3) => Self::from_cols(m.x_axis.xyz(), m.y_axis.xyz(), m.w_axis.xyz()),
189 (3, 0) => Self::from_cols(m.x_axis.yzw(), m.y_axis.yzw(), m.z_axis.yzw()),
190 (3, 1) => Self::from_cols(m.x_axis.xzw(), m.y_axis.xzw(), m.z_axis.xzw()),
191 (3, 2) => Self::from_cols(m.x_axis.xyw(), m.y_axis.xyw(), m.z_axis.xyw()),
192 (3, 3) => Self::from_cols(m.x_axis.xyz(), m.y_axis.xyz(), m.z_axis.xyz()),
193 _ => panic!("index out of bounds"),
194 }
195 }
196
197 #[inline]
203 #[must_use]
204 pub fn from_quat(rotation: DQuat) -> Self {
205 glam_assert!(rotation.is_normalized());
206
207 let x2 = rotation.x + rotation.x;
208 let y2 = rotation.y + rotation.y;
209 let z2 = rotation.z + rotation.z;
210 let xx = rotation.x * x2;
211 let xy = rotation.x * y2;
212 let xz = rotation.x * z2;
213 let yy = rotation.y * y2;
214 let yz = rotation.y * z2;
215 let zz = rotation.z * z2;
216 let wx = rotation.w * x2;
217 let wy = rotation.w * y2;
218 let wz = rotation.w * z2;
219
220 Self::from_cols(
221 DVec3::new(1.0 - (yy + zz), xy + wz, xz - wy),
222 DVec3::new(xy - wz, 1.0 - (xx + zz), yz + wx),
223 DVec3::new(xz + wy, yz - wx, 1.0 - (xx + yy)),
224 )
225 }
226
227 #[inline]
234 #[must_use]
235 pub fn from_axis_angle(axis: DVec3, angle: f64) -> Self {
236 glam_assert!(axis.is_normalized());
237
238 let (sin, cos) = math::sin_cos(angle);
239 let (xsin, ysin, zsin) = axis.mul(sin).into();
240 let (x, y, z) = axis.into();
241 let (x2, y2, z2) = axis.mul(axis).into();
242 let omc = 1.0 - cos;
243 let xyomc = x * y * omc;
244 let xzomc = x * z * omc;
245 let yzomc = y * z * omc;
246 Self::from_cols(
247 DVec3::new(x2 * omc + cos, xyomc + zsin, xzomc - ysin),
248 DVec3::new(xyomc - zsin, y2 * omc + cos, yzomc + xsin),
249 DVec3::new(xzomc + ysin, yzomc - xsin, z2 * omc + cos),
250 )
251 }
252
253 #[inline]
256 #[must_use]
257 pub fn from_euler(order: EulerRot, a: f64, b: f64, c: f64) -> Self {
258 Self::from_euler_angles(order, a, b, c)
259 }
260
261 #[inline]
270 #[must_use]
271 pub fn to_euler(&self, order: EulerRot) -> (f64, f64, f64) {
272 glam_assert!(
273 self.x_axis.is_normalized()
274 && self.y_axis.is_normalized()
275 && self.z_axis.is_normalized()
276 );
277 self.to_euler_angles(order)
278 }
279
280 #[inline]
282 #[must_use]
283 pub fn from_rotation_x(angle: f64) -> Self {
284 let (sina, cosa) = math::sin_cos(angle);
285 Self::from_cols(
286 DVec3::X,
287 DVec3::new(0.0, cosa, sina),
288 DVec3::new(0.0, -sina, cosa),
289 )
290 }
291
292 #[inline]
294 #[must_use]
295 pub fn from_rotation_y(angle: f64) -> Self {
296 let (sina, cosa) = math::sin_cos(angle);
297 Self::from_cols(
298 DVec3::new(cosa, 0.0, -sina),
299 DVec3::Y,
300 DVec3::new(sina, 0.0, cosa),
301 )
302 }
303
304 #[inline]
306 #[must_use]
307 pub fn from_rotation_z(angle: f64) -> Self {
308 let (sina, cosa) = math::sin_cos(angle);
309 Self::from_cols(
310 DVec3::new(cosa, sina, 0.0),
311 DVec3::new(-sina, cosa, 0.0),
312 DVec3::Z,
313 )
314 }
315
316 #[inline]
321 #[must_use]
322 pub fn from_translation(translation: DVec2) -> Self {
323 Self::from_cols(
324 DVec3::X,
325 DVec3::Y,
326 DVec3::new(translation.x, translation.y, 1.0),
327 )
328 }
329
330 #[inline]
336 #[must_use]
337 pub fn from_angle(angle: f64) -> Self {
338 let (sin, cos) = math::sin_cos(angle);
339 Self::from_cols(
340 DVec3::new(cos, sin, 0.0),
341 DVec3::new(-sin, cos, 0.0),
342 DVec3::Z,
343 )
344 }
345
346 #[inline]
352 #[must_use]
353 pub fn from_scale_angle_translation(scale: DVec2, angle: f64, translation: DVec2) -> Self {
354 let (sin, cos) = math::sin_cos(angle);
355 Self::from_cols(
356 DVec3::new(cos * scale.x, sin * scale.x, 0.0),
357 DVec3::new(-sin * scale.y, cos * scale.y, 0.0),
358 DVec3::new(translation.x, translation.y, 1.0),
359 )
360 }
361
362 #[inline]
371 #[must_use]
372 pub fn from_scale(scale: DVec2) -> Self {
373 glam_assert!(scale.cmpne(DVec2::ZERO).any());
375
376 Self::from_cols(
377 DVec3::new(scale.x, 0.0, 0.0),
378 DVec3::new(0.0, scale.y, 0.0),
379 DVec3::Z,
380 )
381 }
382
383 #[inline]
388 pub fn from_mat2(m: DMat2) -> Self {
389 Self::from_cols((m.x_axis, 0.0).into(), (m.y_axis, 0.0).into(), DVec3::Z)
390 }
391
392 #[inline]
398 #[must_use]
399 pub const fn from_cols_slice(slice: &[f64]) -> Self {
400 Self::new(
401 slice[0], slice[1], slice[2], slice[3], slice[4], slice[5], slice[6], slice[7],
402 slice[8],
403 )
404 }
405
406 #[inline]
412 pub fn write_cols_to_slice(self, slice: &mut [f64]) {
413 slice[0] = self.x_axis.x;
414 slice[1] = self.x_axis.y;
415 slice[2] = self.x_axis.z;
416 slice[3] = self.y_axis.x;
417 slice[4] = self.y_axis.y;
418 slice[5] = self.y_axis.z;
419 slice[6] = self.z_axis.x;
420 slice[7] = self.z_axis.y;
421 slice[8] = self.z_axis.z;
422 }
423
424 #[inline]
430 #[must_use]
431 pub fn col(&self, index: usize) -> DVec3 {
432 match index {
433 0 => self.x_axis,
434 1 => self.y_axis,
435 2 => self.z_axis,
436 _ => panic!("index out of bounds"),
437 }
438 }
439
440 #[inline]
446 pub fn col_mut(&mut self, index: usize) -> &mut DVec3 {
447 match index {
448 0 => &mut self.x_axis,
449 1 => &mut self.y_axis,
450 2 => &mut self.z_axis,
451 _ => panic!("index out of bounds"),
452 }
453 }
454
455 #[inline]
461 #[must_use]
462 pub fn row(&self, index: usize) -> DVec3 {
463 match index {
464 0 => DVec3::new(self.x_axis.x, self.y_axis.x, self.z_axis.x),
465 1 => DVec3::new(self.x_axis.y, self.y_axis.y, self.z_axis.y),
466 2 => DVec3::new(self.x_axis.z, self.y_axis.z, self.z_axis.z),
467 _ => panic!("index out of bounds"),
468 }
469 }
470
471 #[inline]
474 #[must_use]
475 pub fn is_finite(&self) -> bool {
476 self.x_axis.is_finite() && self.y_axis.is_finite() && self.z_axis.is_finite()
477 }
478
479 #[inline]
481 #[must_use]
482 pub fn is_nan(&self) -> bool {
483 self.x_axis.is_nan() || self.y_axis.is_nan() || self.z_axis.is_nan()
484 }
485
486 #[inline]
488 #[must_use]
489 pub fn transpose(&self) -> Self {
490 Self {
491 x_axis: DVec3::new(self.x_axis.x, self.y_axis.x, self.z_axis.x),
492 y_axis: DVec3::new(self.x_axis.y, self.y_axis.y, self.z_axis.y),
493 z_axis: DVec3::new(self.x_axis.z, self.y_axis.z, self.z_axis.z),
494 }
495 }
496
497 #[inline]
499 #[must_use]
500 pub fn determinant(&self) -> f64 {
501 self.z_axis.dot(self.x_axis.cross(self.y_axis))
502 }
503
504 #[inline]
512 #[must_use]
513 pub fn inverse(&self) -> Self {
514 let tmp0 = self.y_axis.cross(self.z_axis);
515 let tmp1 = self.z_axis.cross(self.x_axis);
516 let tmp2 = self.x_axis.cross(self.y_axis);
517 let det = self.z_axis.dot(tmp2);
518 glam_assert!(det != 0.0);
519 let inv_det = DVec3::splat(det.recip());
520 Self::from_cols(tmp0.mul(inv_det), tmp1.mul(inv_det), tmp2.mul(inv_det)).transpose()
521 }
522
523 #[inline]
533 #[must_use]
534 pub fn transform_point2(&self, rhs: DVec2) -> DVec2 {
535 glam_assert!(self.row(2).abs_diff_eq(DVec3::Z, 1e-6));
536 DMat2::from_cols(self.x_axis.xy(), self.y_axis.xy()) * rhs + self.z_axis.xy()
537 }
538
539 #[inline]
549 #[must_use]
550 pub fn transform_vector2(&self, rhs: DVec2) -> DVec2 {
551 glam_assert!(self.row(2).abs_diff_eq(DVec3::Z, 1e-6));
552 DMat2::from_cols(self.x_axis.xy(), self.y_axis.xy()) * rhs
553 }
554
555 #[inline]
557 #[must_use]
558 pub fn mul_vec3(&self, rhs: DVec3) -> DVec3 {
559 let mut res = self.x_axis.mul(rhs.x);
560 res = res.add(self.y_axis.mul(rhs.y));
561 res = res.add(self.z_axis.mul(rhs.z));
562 res
563 }
564
565 #[inline]
567 #[must_use]
568 pub fn mul_mat3(&self, rhs: &Self) -> Self {
569 Self::from_cols(
570 self.mul(rhs.x_axis),
571 self.mul(rhs.y_axis),
572 self.mul(rhs.z_axis),
573 )
574 }
575
576 #[inline]
578 #[must_use]
579 pub fn add_mat3(&self, rhs: &Self) -> Self {
580 Self::from_cols(
581 self.x_axis.add(rhs.x_axis),
582 self.y_axis.add(rhs.y_axis),
583 self.z_axis.add(rhs.z_axis),
584 )
585 }
586
587 #[inline]
589 #[must_use]
590 pub fn sub_mat3(&self, rhs: &Self) -> Self {
591 Self::from_cols(
592 self.x_axis.sub(rhs.x_axis),
593 self.y_axis.sub(rhs.y_axis),
594 self.z_axis.sub(rhs.z_axis),
595 )
596 }
597
598 #[inline]
600 #[must_use]
601 pub fn mul_scalar(&self, rhs: f64) -> Self {
602 Self::from_cols(
603 self.x_axis.mul(rhs),
604 self.y_axis.mul(rhs),
605 self.z_axis.mul(rhs),
606 )
607 }
608
609 #[inline]
611 #[must_use]
612 pub fn div_scalar(&self, rhs: f64) -> Self {
613 let rhs = DVec3::splat(rhs);
614 Self::from_cols(
615 self.x_axis.div(rhs),
616 self.y_axis.div(rhs),
617 self.z_axis.div(rhs),
618 )
619 }
620
621 #[inline]
631 #[must_use]
632 pub fn abs_diff_eq(&self, rhs: Self, max_abs_diff: f64) -> bool {
633 self.x_axis.abs_diff_eq(rhs.x_axis, max_abs_diff)
634 && self.y_axis.abs_diff_eq(rhs.y_axis, max_abs_diff)
635 && self.z_axis.abs_diff_eq(rhs.z_axis, max_abs_diff)
636 }
637
638 #[inline]
640 #[must_use]
641 pub fn abs(&self) -> Self {
642 Self::from_cols(self.x_axis.abs(), self.y_axis.abs(), self.z_axis.abs())
643 }
644
645 #[inline]
646 pub fn as_mat3(&self) -> Mat3 {
647 Mat3::from_cols(
648 self.x_axis.as_vec3(),
649 self.y_axis.as_vec3(),
650 self.z_axis.as_vec3(),
651 )
652 }
653}
654
655impl Default for DMat3 {
656 #[inline]
657 fn default() -> Self {
658 Self::IDENTITY
659 }
660}
661
662impl Add<DMat3> for DMat3 {
663 type Output = Self;
664 #[inline]
665 fn add(self, rhs: Self) -> Self::Output {
666 self.add_mat3(&rhs)
667 }
668}
669
670impl AddAssign<DMat3> for DMat3 {
671 #[inline]
672 fn add_assign(&mut self, rhs: Self) {
673 *self = self.add_mat3(&rhs);
674 }
675}
676
677impl Sub<DMat3> for DMat3 {
678 type Output = Self;
679 #[inline]
680 fn sub(self, rhs: Self) -> Self::Output {
681 self.sub_mat3(&rhs)
682 }
683}
684
685impl SubAssign<DMat3> for DMat3 {
686 #[inline]
687 fn sub_assign(&mut self, rhs: Self) {
688 *self = self.sub_mat3(&rhs);
689 }
690}
691
692impl Neg for DMat3 {
693 type Output = Self;
694 #[inline]
695 fn neg(self) -> Self::Output {
696 Self::from_cols(self.x_axis.neg(), self.y_axis.neg(), self.z_axis.neg())
697 }
698}
699
700impl Mul<DMat3> for DMat3 {
701 type Output = Self;
702 #[inline]
703 fn mul(self, rhs: Self) -> Self::Output {
704 self.mul_mat3(&rhs)
705 }
706}
707
708impl MulAssign<DMat3> for DMat3 {
709 #[inline]
710 fn mul_assign(&mut self, rhs: Self) {
711 *self = self.mul_mat3(&rhs);
712 }
713}
714
715impl Mul<DVec3> for DMat3 {
716 type Output = DVec3;
717 #[inline]
718 fn mul(self, rhs: DVec3) -> Self::Output {
719 self.mul_vec3(rhs)
720 }
721}
722
723impl Mul<DMat3> for f64 {
724 type Output = DMat3;
725 #[inline]
726 fn mul(self, rhs: DMat3) -> Self::Output {
727 rhs.mul_scalar(self)
728 }
729}
730
731impl Mul<f64> for DMat3 {
732 type Output = Self;
733 #[inline]
734 fn mul(self, rhs: f64) -> Self::Output {
735 self.mul_scalar(rhs)
736 }
737}
738
739impl MulAssign<f64> for DMat3 {
740 #[inline]
741 fn mul_assign(&mut self, rhs: f64) {
742 *self = self.mul_scalar(rhs);
743 }
744}
745
746impl Div<DMat3> for f64 {
747 type Output = DMat3;
748 #[inline]
749 fn div(self, rhs: DMat3) -> Self::Output {
750 rhs.div_scalar(self)
751 }
752}
753
754impl Div<f64> for DMat3 {
755 type Output = Self;
756 #[inline]
757 fn div(self, rhs: f64) -> Self::Output {
758 self.div_scalar(rhs)
759 }
760}
761
762impl DivAssign<f64> for DMat3 {
763 #[inline]
764 fn div_assign(&mut self, rhs: f64) {
765 *self = self.div_scalar(rhs);
766 }
767}
768
769impl Sum<Self> for DMat3 {
770 fn sum<I>(iter: I) -> Self
771 where
772 I: Iterator<Item = Self>,
773 {
774 iter.fold(Self::ZERO, Self::add)
775 }
776}
777
778impl<'a> Sum<&'a Self> for DMat3 {
779 fn sum<I>(iter: I) -> Self
780 where
781 I: Iterator<Item = &'a Self>,
782 {
783 iter.fold(Self::ZERO, |a, &b| Self::add(a, b))
784 }
785}
786
787impl Product for DMat3 {
788 fn product<I>(iter: I) -> Self
789 where
790 I: Iterator<Item = Self>,
791 {
792 iter.fold(Self::IDENTITY, Self::mul)
793 }
794}
795
796impl<'a> Product<&'a Self> for DMat3 {
797 fn product<I>(iter: I) -> Self
798 where
799 I: Iterator<Item = &'a Self>,
800 {
801 iter.fold(Self::IDENTITY, |a, &b| Self::mul(a, b))
802 }
803}
804
805impl PartialEq for DMat3 {
806 #[inline]
807 fn eq(&self, rhs: &Self) -> bool {
808 self.x_axis.eq(&rhs.x_axis) && self.y_axis.eq(&rhs.y_axis) && self.z_axis.eq(&rhs.z_axis)
809 }
810}
811
812#[cfg(not(target_arch = "spirv"))]
813impl AsRef<[f64; 9]> for DMat3 {
814 #[inline]
815 fn as_ref(&self) -> &[f64; 9] {
816 unsafe { &*(self as *const Self as *const [f64; 9]) }
817 }
818}
819
820#[cfg(not(target_arch = "spirv"))]
821impl AsMut<[f64; 9]> for DMat3 {
822 #[inline]
823 fn as_mut(&mut self) -> &mut [f64; 9] {
824 unsafe { &mut *(self as *mut Self as *mut [f64; 9]) }
825 }
826}
827
828impl fmt::Debug for DMat3 {
829 fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
830 fmt.debug_struct(stringify!(DMat3))
831 .field("x_axis", &self.x_axis)
832 .field("y_axis", &self.y_axis)
833 .field("z_axis", &self.z_axis)
834 .finish()
835 }
836}
837
838impl fmt::Display for DMat3 {
839 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
840 if let Some(p) = f.precision() {
841 write!(
842 f,
843 "[{:.*}, {:.*}, {:.*}]",
844 p, self.x_axis, p, self.y_axis, p, self.z_axis
845 )
846 } else {
847 write!(f, "[{}, {}, {}]", self.x_axis, self.y_axis, self.z_axis)
848 }
849 }
850}