1 | |
---|
2 | #include <stdio.h> |
---|
3 | #include <math.h> |
---|
4 | #include "matrix.h" |
---|
5 | |
---|
6 | |
---|
7 | Vector Matrix::getEigenValues() const |
---|
8 | { |
---|
9 | Vector eigVl; |
---|
10 | |
---|
11 | float a = 0; |
---|
12 | float b = 0; |
---|
13 | |
---|
14 | float c[3]; |
---|
15 | |
---|
16 | // c[0] is the determinante of mat |
---|
17 | c[0] = this->m11 * this->m22 * this->m33 + |
---|
18 | 2* this->m12 * this->m13 * this->m23 - |
---|
19 | this->m11 * this->m23 * this->m23 - |
---|
20 | this->m22 * this->m13 * this->m13 - |
---|
21 | this->m33 * this->m12 * this->m12; |
---|
22 | |
---|
23 | // c[1] is the trace of a |
---|
24 | c[1] = this->m11 * this->m22 - |
---|
25 | this->m12 * this->m12 + |
---|
26 | this->m11 * this->m33 - |
---|
27 | this->m13 * this->m13 + |
---|
28 | this->m22 * this->m33 - |
---|
29 | this->m23 * this->m23; |
---|
30 | |
---|
31 | // c[2] is the sum of the diagonal elements |
---|
32 | c[2] = this->m11 + |
---|
33 | this->m22 + |
---|
34 | this->m33; |
---|
35 | |
---|
36 | |
---|
37 | // Computing the roots: |
---|
38 | a = (3.0*c[1] - c[2]*c[2]) / 3.0; |
---|
39 | b = (-2.0*c[2]*c[2]*c[2] + 9.0*c[1]*c[2] - 27.0*c[0]) / 27.0; |
---|
40 | |
---|
41 | float Q = b*b/4.0 + a*a*a/27.0; |
---|
42 | |
---|
43 | // 3 distinct Roots |
---|
44 | if (Q < 0) |
---|
45 | { |
---|
46 | float psi = atan2(sqrt(-Q), -b/2.0); |
---|
47 | float p = sqrt((b/2.0)*(b/2.0) - Q); |
---|
48 | |
---|
49 | eigVl.x = c[2]/3.0 + 2 * pow(p, 1/3.0) * cos(psi/3.0); |
---|
50 | eigVl.y = c[2]/3.0 - pow(p, 1/3.0) * (cos(psi/3.0) |
---|
51 | + sqrt(3.0) * sin(psi/3.0)); |
---|
52 | eigVl.z = c[2]/3.0 - pow(p, 1/3.0) * (cos(psi/3.0) |
---|
53 | - sqrt(3.0) * sin(psi/3.0)); |
---|
54 | |
---|
55 | } |
---|
56 | // 2 Distinct Roots |
---|
57 | else if (Q == 0) |
---|
58 | { |
---|
59 | eigVl.x = c[2]/3.0 + pow(b/2.0, 1.0/3.0); |
---|
60 | eigVl.y = c[2]/3.0 + pow(b/2.0, 1.0/3.0); |
---|
61 | eigVl.z = c[2]/3.0 + 2* pow(b/2.0, 1.0/3.0); |
---|
62 | } |
---|
63 | // 1 Root (not calculating anything.) |
---|
64 | else if (Q > 0) |
---|
65 | { |
---|
66 | printf("This Matrix is a multiple of the Identity matrix (lambda * I3))\n"); |
---|
67 | eigVl.x = eigVl.y = eigVl.z = 1; |
---|
68 | } |
---|
69 | return eigVl; |
---|
70 | } |
---|
71 | |
---|
72 | void Matrix::getEigenVectors(Vector& a, Vector& b, Vector& c) const |
---|
73 | { |
---|
74 | Vector eigVl = this->getEigenValues(); |
---|
75 | |
---|
76 | float eigVal[3] = { eigVl.x, eigVl.y, eigVl.z }; |
---|
77 | |
---|
78 | Vector eigVc[3]; |
---|
79 | /* eigenvec test */ |
---|
80 | for(int i = 0; i < 2; i++) |
---|
81 | { |
---|
82 | eigVc[i].x = -1/this->m13*(this->m33 - eigVal[i]) + (this->m32*(-this->m31*this->m32 + this->m12*this->m33 - this->m12*eigVal[i])) / |
---|
83 | this->m13*(-this->m13*this->m22 - this->m12*this->m23 + this->m13*eigVal[i]); |
---|
84 | |
---|
85 | eigVc[i].y = -( -this->m13*this->m23 + this->m12*this->m33 - this->m12*eigVal[i]) / |
---|
86 | (-this->m31*this->m22 + this->m12*this->m23 + this->m13*eigVal[i]); |
---|
87 | |
---|
88 | eigVc[i].z = 1.0f; |
---|
89 | |
---|
90 | eigVc[i] /= eigVc[i].len(); |
---|
91 | } |
---|
92 | eigVc[2] = eigVc[0].cross(eigVc[1]); |
---|
93 | |
---|
94 | a = eigVc[0]; |
---|
95 | b = eigVc[1]; |
---|
96 | c = eigVc[2]; |
---|
97 | } |
---|
98 | |
---|
99 | void Matrix::debug() const |
---|
100 | { |
---|
101 | printf("| %f | %f | %f |\n", this->m11, this->m12, this->m13 ); |
---|
102 | printf("| %f | %f | %f |\n", this->m21, this->m22, this->m23 ); |
---|
103 | printf("| %f | %f | %f |\n", this->m31, this->m32, this->m33 ); |
---|
104 | |
---|
105 | } |
---|
106 | |
---|