Guest User

Untitled

a guest
Apr 20th, 2018
91
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C 3.80 KB | None | 0 0
  1. void FreeIMU::AHRSupdate(float gx, float gy, float gz, float ax, float ay, float az, float mx, float my, float mz) {
  2.   float recipNorm;
  3.   float q0q0, q0q1, q0q2, q0q3, q1q1, q1q2, q1q3, q2q2, q2q3, q3q3;
  4.   float halfex = 0.0f, halfey = 0.0f, halfez = 0.0f;
  5.   float qa, qb, qc;
  6.  
  7.   // Auxiliary variables to avoid repeated arithmetic
  8.   q0q0 = q0 * q0;
  9.   q0q1 = q0 * q1;
  10.   q0q2 = q0 * q2;
  11.   q0q3 = q0 * q3;
  12.   q1q1 = q1 * q1;
  13.   q1q2 = q1 * q2;
  14.   q1q3 = q1 * q3;
  15.   q2q2 = q2 * q2;
  16.   q2q3 = q2 * q3;
  17.   q3q3 = q3 * q3;
  18.  
  19.   // Use magnetometer measurement only when valid (avoids NaN in magnetometer normalisation)
  20.   if((mx != 0.0f) && (my != 0.0f) && (mz != 0.0f)) {
  21.     float hx, hy, bx, bz;
  22.     float halfwx, halfwy, halfwz;
  23.    
  24.     // Normalise magnetometer measurement
  25.     recipNorm = invSqrt(mx * mx + my * my + mz * mz);
  26.     mx *= recipNorm;
  27.     my *= recipNorm;
  28.     mz *= recipNorm;
  29.    
  30.     // Reference direction of Earth's magnetic field
  31.     hx = 2.0f * (mx * (0.5f - q2q2 - q3q3) + my * (q1q2 - q0q3) + mz * (q1q3 + q0q2));
  32.     hy = 2.0f * (mx * (q1q2 + q0q3) + my * (0.5f - q1q1 - q3q3) + mz * (q2q3 - q0q1));
  33.     bx = sqrt(hx * hx + hy * hy);
  34.     bz = 2.0f * (mx * (q1q3 - q0q2) + my * (q2q3 + q0q1) + mz * (0.5f - q1q1 - q2q2));
  35.    
  36.     // Estimated direction of magnetic field
  37.     halfwx = bx * (0.5f - q2q2 - q3q3) + bz * (q1q3 - q0q2);
  38.     halfwy = bx * (q1q2 - q0q3) + bz * (q0q1 + q2q3);
  39.     halfwz = bx * (q0q2 + q1q3) + bz * (0.5f - q1q1 - q2q2);
  40.    
  41.     // Error is sum of cross product between estimated direction and measured direction of field vectors
  42.     halfex = (my * halfwz - mz * halfwy);
  43.     halfey = (mz * halfwx - mx * halfwz);
  44.     halfez = (mx * halfwy - my * halfwx);
  45.   }
  46.  
  47.   // Compute feedback only if accelerometer measurement valid (avoids NaN in accelerometer normalisation)
  48.   if((ax != 0.0f) && (ay != 0.0f) && (az != 0.0f)) {
  49.     float halfvx, halfvy, halfvz;
  50.    
  51.     // Normalise accelerometer measurement
  52.     recipNorm = invSqrt(ax * ax + ay * ay + az * az);
  53.     ax *= recipNorm;
  54.     ay *= recipNorm;
  55.     az *= recipNorm;
  56.    
  57.     // Estimated direction of gravity
  58.     halfvx = q1q3 - q0q2;
  59.     halfvy = q0q1 + q2q3;
  60.     halfvz = q0q0 - 0.5f + q3q3;
  61.  
  62.     // Error is sum of cross product between estimated direction and measured direction of field vectors
  63.     halfex += (ay * halfvz - az * halfvy);
  64.     halfey += (az * halfvx - ax * halfvz);
  65.     halfez += (ax * halfvy - ay * halfvx);
  66.   }
  67.  
  68.   // Apply feedback only when valid data has been gathered from the accelerometer or magnetometer
  69.   if(halfex != 0.0f && halfey != 0.0f && halfez != 0.0f) {
  70.     // Compute and apply integral feedback if enabled
  71.     if(twoKi > 0.0f) {
  72.       integralFBx += twoKi * halfex * (1.0f / sampleFreq);  // integral error scaled by Ki
  73.       integralFBy += twoKi * halfey * (1.0f / sampleFreq);
  74.       integralFBz += twoKi * halfez * (1.0f / sampleFreq);
  75.       gx += integralFBx;  // apply integral feedback
  76.       gy += integralFBy;
  77.       gz += integralFBz;
  78.     }
  79.     else {
  80.       integralFBx = 0.0f; // prevent integral windup
  81.       integralFBy = 0.0f;
  82.       integralFBz = 0.0f;
  83.     }
  84.  
  85.     // Apply proportional feedback
  86.     gx += twoKp * halfex;
  87.     gy += twoKp * halfey;
  88.     gz += twoKp * halfez;
  89.   }
  90.  
  91.   // Integrate rate of change of quaternion
  92.   gx *= (0.5f * (1.0f / sampleFreq));   // pre-multiply common factors
  93.   gy *= (0.5f * (1.0f / sampleFreq));
  94.   gz *= (0.5f * (1.0f / sampleFreq));
  95.   qa = q0;
  96.   qb = q1;
  97.   qc = q2;
  98.   q0 += (-qb * gx - qc * gy - q3 * gz);
  99.   q1 += (qa * gx + qc * gz - q3 * gy);
  100.   q2 += (qa * gy - qb * gz + q3 * gx);
  101.   q3 += (qa * gz + qb * gy - qc * gx);
  102.  
  103.   // Normalise quaternion
  104.   recipNorm = invSqrt(q0 * q0 + q1 * q1 + q2 * q2 + q3 * q3);
  105.   q0 *= recipNorm;
  106.   q1 *= recipNorm;
  107.   q2 *= recipNorm;
  108.   q3 *= recipNorm;
  109. }
Add Comment
Please, Sign In to add comment