mirror of
https://github.com/betaflight/betaflight.git
synced 2025-07-13 11:29:58 +03:00
d262c6e66c
git-svn-id: https://afrodevices.googlecode.com/svn/trunk/baseflight@86 7c89a4a9-59b9-e629-4cfe-3a2d53b20e61
343 lines
12 KiB
C++
Executable file
343 lines
12 KiB
C++
Executable file
|
|
void computeIMU()
|
|
{
|
|
uint8_t axis;
|
|
static int16_t gyroADCprevious[3] = { 0, 0, 0 };
|
|
int16_t gyroADCp[3];
|
|
int16_t gyroADCinter[3];
|
|
static uint32_t timeInterleave = 0;
|
|
#if defined(TRI)
|
|
static int16_t gyroYawSmooth = 0;
|
|
#endif
|
|
|
|
if (MAG)
|
|
Mag_getADC();
|
|
if (BARO)
|
|
Baro_update();
|
|
|
|
//we separate the 2 situations because reading gyro values with a gyro only setup can be acchieved at a higher rate
|
|
//gyro+nunchuk: we must wait for a quite high delay betwwen 2 reads to get both WM+ and Nunchuk data. It works with 3ms
|
|
//gyro only: the delay to read 2 consecutive values can be reduced to only 0.65ms
|
|
if (!ACC && nunchuk) {
|
|
annexCode();
|
|
while ((micros() - timeInterleave) < INTERLEAVING_DELAY); //interleaving delay between 2 consecutive reads
|
|
timeInterleave = micros();
|
|
WMP_getRawADC();
|
|
getEstimatedAttitude(); // computation time must last less than one interleaving delay
|
|
#if BARO
|
|
getEstimatedAltitude();
|
|
#endif
|
|
while ((micros() - timeInterleave) < INTERLEAVING_DELAY); //interleaving delay between 2 consecutive reads
|
|
timeInterleave = micros();
|
|
while (WMP_getRawADC() != 1); // For this interleaving reading, we must have a gyro update at this point (less delay)
|
|
|
|
for (axis = 0; axis < 3; axis++) {
|
|
// empirical, we take a weighted value of the current and the previous values
|
|
// /4 is to average 4 values, note: overflow is not possible for WMP gyro here
|
|
gyroData[axis] = (gyroADC[axis] * 3 + gyroADCprevious[axis] + 2) / 4;
|
|
gyroADCprevious[axis] = gyroADC[axis];
|
|
}
|
|
} else {
|
|
if (ACC) {
|
|
ACC_getADC();
|
|
getEstimatedAttitude();
|
|
if (BARO)
|
|
getEstimatedAltitude();
|
|
}
|
|
if (GYRO)
|
|
Gyro_getADC();
|
|
else
|
|
WMP_getRawADC();
|
|
for (axis = 0; axis < 3; axis++)
|
|
gyroADCp[axis] = gyroADC[axis];
|
|
timeInterleave = micros();
|
|
annexCode();
|
|
if ((micros() - timeInterleave) > 650) {
|
|
annex650_overrun_count++;
|
|
} else {
|
|
while ((micros() - timeInterleave) < 650); //empirical, interleaving delay between 2 consecutive reads
|
|
}
|
|
if (GYRO)
|
|
Gyro_getADC();
|
|
else
|
|
WMP_getRawADC();
|
|
for (axis = 0; axis < 3; axis++) {
|
|
gyroADCinter[axis] = gyroADC[axis] + gyroADCp[axis];
|
|
// empirical, we take a weighted value of the current and the previous values
|
|
gyroData[axis] = (gyroADCinter[axis] + gyroADCprevious[axis] + 1) / 3;
|
|
gyroADCprevious[axis] = gyroADCinter[axis] / 2;
|
|
if (!ACC)
|
|
accADC[axis] = 0;
|
|
}
|
|
}
|
|
#if defined(TRI)
|
|
gyroData[YAW] = (gyroYawSmooth * 2 + gyroData[YAW] + 1) / 3;
|
|
gyroYawSmooth = gyroData[YAW];
|
|
#endif
|
|
}
|
|
|
|
// **************************************************
|
|
// Simplified IMU based on "Complementary Filter"
|
|
// Inspired by http://starlino.com/imu_guide.html
|
|
//
|
|
// adapted by ziss_dm : http://www.multiwii.com/forum/viewtopic.php?f=8&t=198
|
|
//
|
|
// The following ideas was used in this project:
|
|
// 1) Rotation matrix: http://en.wikipedia.org/wiki/Rotation_matrix
|
|
// 2) Small-angle approximation: http://en.wikipedia.org/wiki/Small-angle_approximation
|
|
// 3) C. Hastings approximation for atan2()
|
|
// 4) Optimization tricks: http://www.hackersdelight.org/
|
|
//
|
|
// Currently Magnetometer uses separate CF which is used only
|
|
// for heading approximation.
|
|
//
|
|
// Modified: 19/04/2011 by ziss_dm
|
|
// Version: V1.1
|
|
//
|
|
// code size deduction and tmp vector intermediate step for vector rotation computation: October 2011 by Alex
|
|
// **************************************************
|
|
|
|
//****** advanced users settings *******************
|
|
/* Set the Low Pass Filter factor for ACC */
|
|
/* Increasing this value would reduce ACC noise (visible in GUI), but would increase ACC lag time*/
|
|
/* Comment this if you do not want filter at all.*/
|
|
/* Default WMC value: 8*/
|
|
#define ACC_LPF_FACTOR 4
|
|
|
|
/* Set the Low Pass Filter factor for Magnetometer */
|
|
/* Increasing this value would reduce Magnetometer noise (not visible in GUI), but would increase Magnetometer lag time*/
|
|
/* Comment this if you do not want filter at all.*/
|
|
/* Default WMC value: n/a*/
|
|
//#define MG_LPF_FACTOR 4
|
|
|
|
/* Set the Gyro Weight for Gyro/Acc complementary filter */
|
|
/* Increasing this value would reduce and delay Acc influence on the output of the filter*/
|
|
/* Default WMC value: 300*/
|
|
#define GYR_CMPF_FACTOR 310.0f
|
|
|
|
/* Set the Gyro Weight for Gyro/Magnetometer complementary filter */
|
|
/* Increasing this value would reduce and delay Magnetometer influence on the output of the filter*/
|
|
/* Default WMC value: n/a*/
|
|
#define GYR_CMPFM_FACTOR 200.0f
|
|
|
|
//****** end of advanced users settings *************
|
|
|
|
#define INV_GYR_CMPF_FACTOR (1.0f / (GYR_CMPF_FACTOR + 1.0f))
|
|
#define INV_GYR_CMPFM_FACTOR (1.0f / (GYR_CMPFM_FACTOR + 1.0f))
|
|
#if GYRO
|
|
#define GYRO_SCALE ((2380 * PI)/((32767.0f / 4.0f ) * 180.0f * 1000000.0f)) //should be 2279.44 but 2380 gives better result
|
|
// +-2000/sec deg scale
|
|
//#define GYRO_SCALE ((200.0f * PI)/((32768.0f / 5.0f / 4.0f ) * 180.0f * 1000000.0f) * 1.5f)
|
|
// +- 200/sec deg scale
|
|
// 1.5 is emperical, not sure what it means
|
|
// should be in rad/sec
|
|
#else
|
|
#define GYRO_SCALE (1.0f/200e6f)
|
|
// empirical, depends on WMP on IDG datasheet, tied of deg/ms sensibility
|
|
// !!!!should be adjusted to the rad/sec
|
|
#endif
|
|
// Small angle approximation
|
|
#define ssin(val) (val)
|
|
#define scos(val) 1.0f
|
|
|
|
typedef struct fp_vector {
|
|
float X;
|
|
float Y;
|
|
float Z;
|
|
} t_fp_vector_def;
|
|
|
|
typedef union {
|
|
float A[3];
|
|
t_fp_vector_def V;
|
|
} t_fp_vector;
|
|
|
|
int16_t _atan2(float y, float x)
|
|
{
|
|
#define fp_is_neg(val) ((((byte*)&val)[3] & 0x80) != 0)
|
|
float z = y / x;
|
|
int16_t zi = abs(int16_t(z * 100));
|
|
int8_t y_neg = fp_is_neg(y);
|
|
if (zi < 100) {
|
|
if (zi > 10)
|
|
z = z / (1.0f + 0.28f * z * z);
|
|
if (fp_is_neg(x)) {
|
|
if (y_neg)
|
|
z -= PI;
|
|
else
|
|
z += PI;
|
|
}
|
|
} else {
|
|
z = (PI / 2.0f) - z / (z * z + 0.28f);
|
|
if (y_neg)
|
|
z -= PI;
|
|
}
|
|
z *= (180.0f / PI * 10);
|
|
return z;
|
|
}
|
|
|
|
// Rotate Estimated vector(s) with small angle approximation, according to the gyro data
|
|
void rotateV(struct fp_vector *v, float *delta)
|
|
{
|
|
fp_vector v_tmp = *v;
|
|
v->Z -= delta[ROLL] * v_tmp.X + delta[PITCH] * v_tmp.Y;
|
|
v->X += delta[ROLL] * v_tmp.Z - delta[YAW] * v_tmp.Y;
|
|
v->Y += delta[PITCH] * v_tmp.Z + delta[YAW] * v_tmp.X;
|
|
}
|
|
|
|
void getEstimatedAttitude()
|
|
{
|
|
uint8_t axis;
|
|
int32_t accMag = 0;
|
|
static t_fp_vector EstG;
|
|
#if MAG
|
|
static t_fp_vector EstM;
|
|
#endif
|
|
#if defined(MG_LPF_FACTOR)
|
|
static int16_t mgSmooth[3];
|
|
#endif
|
|
#if defined(ACC_LPF_FACTOR)
|
|
static int16_t accTemp[3]; //projection of smoothed and normalized magnetic vector on x/y/z axis, as measured by magnetometer
|
|
#endif
|
|
static uint16_t previousT;
|
|
uint16_t currentT = micros();
|
|
float scale, deltaGyroAngle[3];
|
|
|
|
scale = (currentT - previousT) * GYRO_SCALE;
|
|
previousT = currentT;
|
|
|
|
// Initialization
|
|
for (axis = 0; axis < 3; axis++) {
|
|
deltaGyroAngle[axis] = gyroADC[axis] * scale;
|
|
#if defined(ACC_LPF_FACTOR)
|
|
accTemp[axis] = (accTemp[axis] - (accTemp[axis] >> ACC_LPF_FACTOR)) + accADC[axis];
|
|
accSmooth[axis] = accTemp[axis] >> ACC_LPF_FACTOR;
|
|
#define ACC_VALUE accSmooth[axis]
|
|
#else
|
|
accSmooth[axis] = accADC[axis];
|
|
#define ACC_VALUE accADC[axis]
|
|
#endif
|
|
// accMag += (ACC_VALUE * 10 / (int16_t)acc_1G) * (ACC_VALUE * 10 / (int16_t)acc_1G);
|
|
accMag += (int32_t) ACC_VALUE *ACC_VALUE;
|
|
#if MAG
|
|
#if defined(MG_LPF_FACTOR)
|
|
mgSmooth[axis] = (mgSmooth[axis] * (MG_LPF_FACTOR - 1) + magADC[axis]) / MG_LPF_FACTOR; // LPF for Magnetometer values
|
|
#define MAG_VALUE mgSmooth[axis]
|
|
#else
|
|
#define MAG_VALUE magADC[axis]
|
|
#endif
|
|
#endif
|
|
}
|
|
accMag = accMag * 100 / ((int32_t) acc_1G * acc_1G);
|
|
|
|
rotateV(&EstG.V, deltaGyroAngle);
|
|
#if MAG
|
|
rotateV(&EstM.V, deltaGyroAngle);
|
|
#endif
|
|
|
|
if (abs(accSmooth[ROLL]) < acc_25deg && abs(accSmooth[PITCH]) < acc_25deg && accSmooth[YAW] > 0)
|
|
smallAngle25 = 1;
|
|
else
|
|
smallAngle25 = 0;
|
|
|
|
// Apply complimentary filter (Gyro drift correction)
|
|
// If accel magnitude >1.4G or <0.6G and ACC vector outside of the limit range => we neutralize the effect of accelerometers in the angle estimation.
|
|
// To do that, we just skip filter, as EstV already rotated by Gyro
|
|
if ((36 < accMag && accMag < 196) || smallAngle25)
|
|
for (axis = 0; axis < 3; axis++) {
|
|
int16_t acc = ACC_VALUE;
|
|
#if not defined(TRUSTED_ACCZ)
|
|
if (smallAngle25 && axis == YAW)
|
|
//We consider ACCZ = acc_1G when the acc on other axis is small.
|
|
//It's a tweak to deal with some configs where ACC_Z tends to a value < acc_1G when high throttle is applied.
|
|
//This tweak applies only when the multi is not in inverted position
|
|
acc = acc_1G;
|
|
#endif
|
|
EstG.A[axis] = (EstG.A[axis] * GYR_CMPF_FACTOR + acc) * INV_GYR_CMPF_FACTOR;
|
|
}
|
|
#if MAG
|
|
for (axis = 0; axis < 3; axis++)
|
|
EstM.A[axis] = (EstM.A[axis] * GYR_CMPFM_FACTOR + MAG_VALUE) * INV_GYR_CMPFM_FACTOR;
|
|
#endif
|
|
|
|
// Attitude of the estimated vector
|
|
angle[ROLL] = _atan2(EstG.V.X, EstG.V.Z);
|
|
angle[PITCH] = _atan2(EstG.V.Y, EstG.V.Z);
|
|
#if MAG
|
|
// Attitude of the cross product vector GxM
|
|
heading = _atan2(EstG.V.X * EstM.V.Z - EstG.V.Z * EstM.V.X, EstG.V.Z * EstM.V.Y - EstG.V.Y * EstM.V.Z) / 10;
|
|
#endif
|
|
}
|
|
|
|
float InvSqrt(float x)
|
|
{
|
|
union {
|
|
int32_t i;
|
|
float f;
|
|
} conv;
|
|
conv.f = x;
|
|
conv.i = 0x5f3759df - (conv.i >> 1);
|
|
return 0.5f * conv.f * (3.0f - x * conv.f * conv.f);
|
|
}
|
|
|
|
int32_t isq(int32_t x)
|
|
{
|
|
return x * x;
|
|
}
|
|
|
|
#define UPDATE_INTERVAL 25000 // 40hz update rate (20hz LPF on acc)
|
|
#define INIT_DELAY 4000000 // 4 sec initialization delay
|
|
#define Kp1 5.5f // PI observer velocity gain
|
|
#define Kp2 10.0f // PI observer position gain
|
|
#define Ki 0.01f // PI observer integral gain (bias cancellation)
|
|
#define dt (UPDATE_INTERVAL / 1000000.0f)
|
|
|
|
void getEstimatedAltitude()
|
|
{
|
|
static uint8_t inited = 0;
|
|
static int16_t AltErrorI = 0;
|
|
static float AccScale;
|
|
static uint32_t deadLine = INIT_DELAY;
|
|
int16_t AltError;
|
|
int16_t InstAcc;
|
|
static int32_t tmpAlt;
|
|
static int16_t EstVelocity = 0;
|
|
static uint32_t velTimer;
|
|
static int16_t lastAlt;
|
|
|
|
if (currentTime < deadLine)
|
|
return;
|
|
deadLine = currentTime + UPDATE_INTERVAL;
|
|
// Soft start
|
|
|
|
if (!inited) {
|
|
inited = 1;
|
|
tmpAlt = BaroAlt * 10;
|
|
AccScale = 100 * 9.80665f / acc_1G;
|
|
}
|
|
// Estimation Error
|
|
AltError = BaroAlt - EstAlt;
|
|
AltErrorI += AltError;
|
|
AltErrorI = constrain(AltErrorI, -2500, +2500);
|
|
// Gravity vector correction and projection to the local Z
|
|
//InstAcc = (accADC[YAW] * (1 - acc_1G * InvSqrt(isq(accADC[ROLL]) + isq(accADC[PITCH]) + isq(accADC[YAW])))) * AccScale + (Ki) * AltErrorI;
|
|
#if defined(TRUSTED_ACCZ)
|
|
InstAcc = (accADC[YAW] * (1 - acc_1G * InvSqrt(isq(accADC[ROLL]) + isq(accADC[PITCH]) + isq(accADC[YAW])))) * AccScale + AltErrorI / 100;
|
|
#else
|
|
InstAcc = AltErrorI / 100;
|
|
#endif
|
|
|
|
// Integrators
|
|
tmpAlt += EstVelocity * (dt * dt) + (Kp2 * dt) * AltError;
|
|
EstVelocity += InstAcc + Kp1 * AltError;
|
|
EstVelocity = constrain(EstVelocity, -10000, +10000);
|
|
|
|
EstAlt = tmpAlt / 10;
|
|
|
|
if (currentTime < velTimer)
|
|
return;
|
|
velTimer = currentTime + 500000;
|
|
zVelocity = tmpAlt - lastAlt;
|
|
lastAlt = tmpAlt;
|
|
|
|
debug4 = zVelocity;
|
|
}
|