mirror of
https://github.com/isledecomp/isle.git
synced 2024-11-22 15:48:09 -05:00
85ce10ab50
* Tidy up `operator[]` code * Add weird index operator for `FUN_1002ddc0` * Cleanup * Add Matrix4 BETA10 annotations, fix typo --------- Co-authored-by: jonschz <jonschz@users.noreply.github.com>
357 lines
8.8 KiB
C++
357 lines
8.8 KiB
C++
#ifndef MATRIX_H
|
|
#define MATRIX_H
|
|
|
|
#include "vector.h"
|
|
|
|
#include <memory.h>
|
|
|
|
struct UnknownMatrixType {
|
|
float m_data[4][4];
|
|
};
|
|
|
|
// Note: Many functions most likely take const references/pointers instead of non-const.
|
|
// The class needs to undergo a very careful refactoring to fix that (no matches should break).
|
|
|
|
// VTABLE: LEGO1 0x100d4350
|
|
// VTABLE: BETA10 0x101b8340
|
|
// SIZE 0x08
|
|
class Matrix4 {
|
|
public:
|
|
// FUNCTION: LEGO1 0x10004500
|
|
// FUNCTION: BETA10 0x1000fc70
|
|
Matrix4(float (*p_data)[4]) { SetData(p_data); }
|
|
|
|
// Note: virtual function overloads appear in the virtual table
|
|
// in reverse order of appearance.
|
|
|
|
// FUNCTION: LEGO1 0x10002320
|
|
// FUNCTION: BETA10 0x1000fcb0
|
|
virtual void Equals(float (*p_data)[4]) { memcpy(m_data, p_data, sizeof(float) * 4 * 4); } // vtable+0x04
|
|
|
|
// FUNCTION: LEGO1 0x10002340
|
|
// FUNCTION: BETA10 0x1000fcf0
|
|
virtual void Equals(const Matrix4& p_matrix)
|
|
{
|
|
memcpy(m_data, p_matrix.m_data, sizeof(float) * 4 * 4);
|
|
} // vtable+0x00
|
|
|
|
// FUNCTION: LEGO1 0x10002360
|
|
// FUNCTION: BETA10 0x1000fd30
|
|
virtual void SetData(float (*p_data)[4]) { m_data = p_data; } // vtable+0x0c
|
|
|
|
// FUNCTION: LEGO1 0x10002370
|
|
// FUNCTION: BETA10 0x1000fd60
|
|
virtual void SetData(UnknownMatrixType& p_matrix) { m_data = p_matrix.m_data; } // vtable+0x08
|
|
|
|
// FUNCTION: LEGO1 0x10002380
|
|
// FUNCTION: BETA10 0x1000fd90
|
|
virtual float (*GetData())[4] { return m_data; } // vtable+0x14
|
|
|
|
// FUNCTION: LEGO1 0x10002390
|
|
// FUNCTION: BETA10 0x1000fdc0
|
|
virtual float (*GetData() const)[4] { return m_data; } // vtable+0x10
|
|
|
|
// FUNCTION: LEGO1 0x100023a0
|
|
// FUNCTION: BETA10 0x1000fdf0
|
|
virtual float* Element(int p_row, int p_col) { return &m_data[p_row][p_col]; } // vtable+0x1c
|
|
|
|
// FUNCTION: LEGO1 0x100023c0
|
|
// FUNCTION: BETA10 0x1000fe30
|
|
virtual const float* Element(int p_row, int p_col) const { return &m_data[p_row][p_col]; } // vtable+0x18
|
|
|
|
// FUNCTION: LEGO1 0x100023e0
|
|
// FUNCTION: BETA10 0x1000fe70
|
|
virtual void Clear() { memset(m_data, 0, 16 * sizeof(float)); } // vtable+0x20
|
|
|
|
// FUNCTION: LEGO1 0x100023f0
|
|
// FUNCTION: BETA10 0x1000feb0
|
|
virtual void SetIdentity()
|
|
{
|
|
Clear();
|
|
m_data[0][0] = 1.0f;
|
|
m_data[1][1] = 1.0f;
|
|
m_data[2][2] = 1.0f;
|
|
m_data[3][3] = 1.0f;
|
|
} // vtable+0x24
|
|
|
|
// FUNCTION: LEGO1 0x10002420
|
|
// FUNCTION: BETA10 0x1000ff20
|
|
virtual void operator=(const Matrix4& p_matrix) { Equals(p_matrix); } // vtable+0x28
|
|
|
|
// FUNCTION: LEGO1 0x10002430
|
|
// FUNCTION: BETA10 0x1000ff50
|
|
virtual Matrix4& operator+=(float (*p_data)[4])
|
|
{
|
|
for (int i = 0; i < 16; i++) {
|
|
((float*) m_data)[i] += ((float*) p_data)[i];
|
|
}
|
|
return *this;
|
|
} // vtable+0x2c
|
|
|
|
// FUNCTION: LEGO1 0x10002460
|
|
// FUNCTION: BETA10 0x1000ffc0
|
|
virtual void TranslateBy(const float& p_x, const float& p_y, const float& p_z)
|
|
{
|
|
m_data[3][0] += p_x;
|
|
m_data[3][1] += p_y;
|
|
m_data[3][2] += p_z;
|
|
} // vtable+0x30
|
|
|
|
// FUNCTION: LEGO1 0x100024a0
|
|
// FUNCTION: BETA10 0x10010040
|
|
virtual void SetTranslation(const float& p_x, const float& p_y, const float& p_z)
|
|
{
|
|
m_data[3][0] = p_x;
|
|
m_data[3][1] = p_y;
|
|
m_data[3][2] = p_z;
|
|
} // vtable+0x34
|
|
|
|
// FUNCTION: LEGO1 0x100024d0
|
|
// FUNCTION: BETA10 0x100100a0
|
|
virtual void Product(float (*p_a)[4], float (*p_b)[4])
|
|
{
|
|
float* cur = (float*) m_data;
|
|
for (int row = 0; row < 4; row++) {
|
|
for (int col = 0; col < 4; col++) {
|
|
*cur = 0.0f;
|
|
for (int k = 0; k < 4; k++) {
|
|
*cur += p_a[row][k] * p_b[k][col];
|
|
}
|
|
cur++;
|
|
}
|
|
}
|
|
} // vtable+0x3c
|
|
|
|
// FUNCTION: LEGO1 0x10002530
|
|
// FUNCTION: BETA10 0x10010180
|
|
virtual void Product(const Matrix4& p_a, const Matrix4& p_b) { Product(p_a.m_data, p_b.m_data); } // vtable+0x38
|
|
|
|
inline virtual void ToQuaternion(Vector4& p_resultQuat); // vtable+0x40
|
|
inline virtual int FromQuaternion(const Vector4& p_vec); // vtable+0x44
|
|
|
|
// FUNCTION: LEGO1 0x100a0ff0
|
|
void Scale(const float& p_x, const float& p_y, const float& p_z)
|
|
{
|
|
for (int i = 0; i < 4; i++) {
|
|
m_data[i][0] *= p_x;
|
|
m_data[i][1] *= p_y;
|
|
m_data[i][2] *= p_z;
|
|
}
|
|
}
|
|
|
|
// FUNCTION: BETA10 0x1001c6a0
|
|
void RotateX(const float& p_angle)
|
|
{
|
|
float s = sin(p_angle);
|
|
float c = cos(p_angle);
|
|
float matrix[4][4];
|
|
memcpy(matrix, m_data, sizeof(float) * 16);
|
|
for (int i = 0; i < 4; i++) {
|
|
m_data[i][1] = matrix[i][1] * c - matrix[i][2] * s;
|
|
m_data[i][2] = matrix[i][2] * c + matrix[i][1] * s;
|
|
}
|
|
}
|
|
|
|
// FUNCTION: BETA10 0x1001fd60
|
|
void RotateY(const float& p_angle)
|
|
{
|
|
float s = sin(p_angle);
|
|
float c = cos(p_angle);
|
|
float matrix[4][4];
|
|
memcpy(matrix, m_data, sizeof(float) * 16);
|
|
for (int i = 0; i < 4; i++) {
|
|
m_data[i][0] = matrix[i][0] * c + matrix[i][2] * s;
|
|
m_data[i][2] = matrix[i][2] * c - matrix[i][0] * s;
|
|
}
|
|
}
|
|
|
|
// FUNCTION: BETA10 0x1006ab10
|
|
void RotateZ(const float& p_angle)
|
|
{
|
|
float s = sin(p_angle);
|
|
float c = cos(p_angle);
|
|
float matrix[4][4];
|
|
memcpy(matrix, m_data, sizeof(float) * 16);
|
|
for (int i = 0; i < 4; i++) {
|
|
m_data[i][0] = matrix[i][0] * c - matrix[i][1] * s;
|
|
m_data[i][1] = matrix[i][1] * c + matrix[i][0] * s;
|
|
}
|
|
}
|
|
|
|
inline int Unknown(Matrix4& p_mat);
|
|
|
|
// FUNCTION: LEGO1 0x1006b500
|
|
void Swap(int p_d1, int p_d2)
|
|
{
|
|
for (int i = 0; i < 4; i++) {
|
|
float e = m_data[p_d1][i];
|
|
m_data[p_d1][i] = m_data[p_d2][i];
|
|
m_data[p_d2][i] = e;
|
|
}
|
|
}
|
|
|
|
float* operator[](int idx) { return m_data[idx]; }
|
|
const float* operator[](int idx) const { return m_data[idx]; }
|
|
|
|
protected:
|
|
float (*m_data)[4];
|
|
};
|
|
|
|
// FUNCTION: LEGO1 0x10002550
|
|
// FUNCTION: BETA10 0x100101c0
|
|
inline void Matrix4::ToQuaternion(Vector4& p_outQuat)
|
|
{
|
|
float trace = m_data[0][0] + m_data[1][1] + m_data[2][2];
|
|
if (trace > 0) {
|
|
trace = sqrt(trace + 1.0);
|
|
p_outQuat[3] = trace * 0.5f;
|
|
trace = 0.5f / trace;
|
|
p_outQuat[0] = (m_data[2][1] - m_data[1][2]) * trace;
|
|
p_outQuat[1] = (m_data[0][2] - m_data[2][0]) * trace;
|
|
p_outQuat[2] = (m_data[1][0] - m_data[0][1]) * trace;
|
|
}
|
|
else {
|
|
|
|
// GLOBAL: LEGO1 0x100d4090
|
|
static int rotateIndex[] = {1, 2, 0};
|
|
|
|
// Largest element along the trace
|
|
int largest = 0;
|
|
if (m_data[0][0] < m_data[1][1]) {
|
|
largest = 1;
|
|
}
|
|
if (*Element(largest, largest) < m_data[2][2]) {
|
|
largest = 2;
|
|
}
|
|
|
|
int next = rotateIndex[largest];
|
|
int nextNext = rotateIndex[next];
|
|
|
|
float trace = sqrt(*Element(largest, largest) - (*Element(nextNext, nextNext) + *Element(next, next)) + 1.0);
|
|
|
|
p_outQuat[largest] = trace * 0.5f;
|
|
trace = 0.5f / trace;
|
|
|
|
p_outQuat[3] = (*Element(nextNext, next) - *Element(next, nextNext)) * trace;
|
|
p_outQuat[next] = (*Element(largest, next) + *Element(next, largest)) * trace;
|
|
p_outQuat[nextNext] = (*Element(largest, nextNext) + *Element(nextNext, largest)) * trace;
|
|
}
|
|
}
|
|
|
|
// FUNCTION: LEGO1 0x10002710
|
|
// FUNCTION: BETA10 0x10010550
|
|
inline int Matrix4::FromQuaternion(const Vector4& p_vec)
|
|
{
|
|
float len = p_vec.LenSquared();
|
|
|
|
if (len > 0.0f) {
|
|
float v7 = 2.0f / len;
|
|
|
|
float v9 = p_vec[0] * v7;
|
|
float v11 = p_vec[1] * v7;
|
|
float v12 = p_vec[2] * v7;
|
|
|
|
float v13 = p_vec[3] * v9;
|
|
float v14 = p_vec[3] * v11;
|
|
float v16 = p_vec[3] * v12;
|
|
|
|
float v17 = p_vec[0] * v9;
|
|
float v22 = p_vec[0] * v11;
|
|
float v23 = p_vec[0] * v12;
|
|
|
|
float v18 = p_vec[1] * v11;
|
|
float v24 = p_vec[1] * v12;
|
|
float v19 = p_vec[2] * v12;
|
|
|
|
m_data[0][0] = 1.0f - (v18 + v19);
|
|
m_data[1][0] = v22 + v16;
|
|
m_data[2][0] = v23 - v14;
|
|
|
|
m_data[0][1] = v22 - v16;
|
|
m_data[1][1] = 1.0f - (v17 + v19);
|
|
m_data[2][1] = v24 + v13;
|
|
|
|
m_data[0][2] = v14 + v23;
|
|
m_data[1][2] = v24 - v13;
|
|
m_data[2][2] = 1.0f - (v18 + v17);
|
|
|
|
m_data[3][0] = 0;
|
|
m_data[3][1] = 0;
|
|
m_data[3][2] = 0;
|
|
m_data[3][3] = 1.0f;
|
|
|
|
m_data[0][3] = 0;
|
|
m_data[1][3] = 0;
|
|
m_data[2][3] = 0;
|
|
return 0;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
// FUNCTION: BETA10 0x1005a590
|
|
inline int Matrix4::Unknown(Matrix4& p_mat)
|
|
{
|
|
float local5c[4][4];
|
|
Matrix4 localc(local5c);
|
|
|
|
((Matrix4&) localc) = *this;
|
|
p_mat.SetIdentity();
|
|
|
|
for (int i = 0; i < 4; i++) {
|
|
int local1c = i;
|
|
int local10;
|
|
|
|
for (local10 = i + 1; local10 < 4; local10++) {
|
|
if (fabs(localc[local1c][i]) < fabs(localc[local10][i])) {
|
|
local1c = local10;
|
|
}
|
|
}
|
|
|
|
if (local1c != i) {
|
|
localc.Swap(local1c, i);
|
|
p_mat.Swap(local1c, i);
|
|
}
|
|
|
|
if (localc[i][i] < 0.001f && localc[i][i] > -0.001f) {
|
|
return -1;
|
|
}
|
|
|
|
float local60 = localc[i][i];
|
|
int local18;
|
|
|
|
for (local18 = 0; local18 < 4; local18++) {
|
|
p_mat[i][local18] /= local60;
|
|
}
|
|
|
|
for (local18 = 0; local18 < 4; local18++) {
|
|
localc[i][local18] /= local60;
|
|
}
|
|
|
|
for (local10 = 0; local10 < 4; local10++) {
|
|
if (i != local10) {
|
|
float afStack70[4];
|
|
|
|
for (local18 = 0; local18 < 4; local18++) {
|
|
afStack70[local18] = p_mat[i][local18] * localc[local10][i];
|
|
}
|
|
|
|
for (local18 = 0; local18 < 4; local18++) {
|
|
p_mat[local10][local18] -= afStack70[local18];
|
|
}
|
|
|
|
for (local18 = 0; local18 < 4; local18++) {
|
|
afStack70[local18] = localc[i][local18] * localc[local10][i];
|
|
}
|
|
|
|
for (local18 = 0; local18 < 4; local18++) {
|
|
localc[local10][local18] -= afStack70[local18];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
#endif // MATRIX_H
|