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