aboutsummaryrefslogtreecommitdiff
path: root/src/common/quaternion.h
blob: 5bb5f2af083d3acaedc05acf9d27857481e15414 (plain) (blame)
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
// SPDX-FileCopyrightText: 2016 Citra Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later

#pragma once

#include "common/vector_math.h"

namespace Common {

template <typename T>
class Quaternion {
public:
    Vec3<T> xyz;
    T w{};

    [[nodiscard]] Quaternion<decltype(-T{})> Inverse() const {
        return {-xyz, w};
    }

    [[nodiscard]] Quaternion<decltype(T{} + T{})> operator+(const Quaternion& other) const {
        return {xyz + other.xyz, w + other.w};
    }

    [[nodiscard]] Quaternion<decltype(T{} - T{})> operator-(const Quaternion& other) const {
        return {xyz - other.xyz, w - other.w};
    }

    [[nodiscard]] Quaternion<decltype(T{} * T{} - T{} * T{})> operator*(
        const Quaternion& other) const {
        return {xyz * other.w + other.xyz * w + Cross(xyz, other.xyz),
                w * other.w - Dot(xyz, other.xyz)};
    }

    [[nodiscard]] Quaternion<T> Normalized() const {
        T length = std::sqrt(xyz.Length2() + w * w);
        return {xyz / length, w / length};
    }

    [[nodiscard]] std::array<decltype(-T{}), 16> ToMatrix() const {
        const T x2 = xyz[0] * xyz[0];
        const T y2 = xyz[1] * xyz[1];
        const T z2 = xyz[2] * xyz[2];

        const T xy = xyz[0] * xyz[1];
        const T wz = w * xyz[2];
        const T xz = xyz[0] * xyz[2];
        const T wy = w * xyz[1];
        const T yz = xyz[1] * xyz[2];
        const T wx = w * xyz[0];

        return {1.0f - 2.0f * (y2 + z2),
                2.0f * (xy + wz),
                2.0f * (xz - wy),
                0.0f,
                2.0f * (xy - wz),
                1.0f - 2.0f * (x2 + z2),
                2.0f * (yz + wx),
                0.0f,
                2.0f * (xz + wy),
                2.0f * (yz - wx),
                1.0f - 2.0f * (x2 + y2),
                0.0f,
                0.0f,
                0.0f,
                0.0f,
                1.0f};
    }
};

template <typename T>
[[nodiscard]] auto QuaternionRotate(const Quaternion<T>& q, const Vec3<T>& v) {
    return v + 2 * Cross(q.xyz, Cross(q.xyz, v) + v * q.w);
}

[[nodiscard]] inline Quaternion<float> MakeQuaternion(const Vec3<float>& axis, float angle) {
    return {axis * std::sin(angle / 2), std::cos(angle / 2)};
}

} // namespace Common