ziggi/stm32uav
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added original Mahony alg for comparation; impr IMU calbration

Browse Source
This commit is contained in:
Oleg Borodin
2022-09-07 03:13:22 +02:00
parent 9a973c054d
commit f85bc21e21
3 changed files with 171 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

170
mculoop/main.c
View File

@@ -60,7 +60,7 @@ static void usart_setup(void) {
usart_enable(USART1); usart_enable(USART1);
} }
const uint32_t systic_freq = 50 * 1000; const uint32_t systic_freq = 100 * 1000;
static void systick_setup(void) { static void systick_setup(void) {
g_sys_tick_counter = 0; g_sys_tick_counter = 0;
@@ -114,7 +114,100 @@ double invSqrt(double x) {
return 1.0f / sqrt(x); return 1.0f / sqrt(x);
} }
void ahrs_update(double dt, quaternion_t* q, mpu_value_t* m) { volatile float twoKp = (2.0f * 0.5f); // 2 * proportional gain (Kp)
volatile float twoKi = (2.0f * 0.0f); // 2 * integral gain (Ki)
volatile float integralFBx = 0.0f;
volatile float integralFBy = 0.0f;
volatile float integralFBz = 0.0f;
void quaternion_update_mahony(double dt, quaternion_t* q, mpu_value_t* m) {
double q0 = q->w;
double q1 = q->x;
double q2 = q->y;
double q3 = q->z; // quaternion of sensor frame relative to auxiliary frame
double gx = m->gx;
double gy = m->gy;
double gz = m->gz;
double ax = m->ax;
double ay = m->ay;
double az = m->az;
double recipNorm;
double halfvx, halfvy, halfvz;
double halfex, halfey, halfez;
double qa, qb, qc;
// Compute feedback only if accelerometer measurement valid (avoids NaN in accelerometer normalisation)
if(!((ax == 0.0f) && (ay == 0.0f) && (az == 0.0f))) {
// Normalise accelerometer measurement
recipNorm = invSqrt(ax*ax + ay*ay + az*az);
ax *= recipNorm;
ay *= recipNorm;
az *= recipNorm;
// Estimated direction of gravity and vector perpendicular to magnetic flux
halfvx = q1*q3 - q0*q2;
halfvy = q0*q1 + q2*q3;
halfvz = q0*q0 - 0.5f + q3*q3;
// Error is sum of cross product between estimated and measured direction of gravity
halfex = (ay*halfvz - az*halfvy);
halfey = (az*halfvx - ax*halfvz);
halfez = (ax*halfvy - ay*halfvx);
// Compute and apply integral feedback if enabled
if(twoKi > 0.0f) {
integralFBx += twoKi*halfex*dt; // integral error scaled by Ki
integralFBy += twoKi*halfey*dt;
integralFBz += twoKi*halfez*dt;
gx += integralFBx; // apply integral feedback
gy += integralFBy;
gz += integralFBz;
}
else {
integralFBx = 0.0f; // prevent integral windup
integralFBy = 0.0f;
integralFBz = 0.0f;
}
// Apply proportional feedback
gx += twoKp*halfex;
gy += twoKp*halfey;
gz += twoKp*halfez;
}
// Integrate rate of change of quaternion
gx *= (0.5f*dt); // pre-multiply common factors
gy *= (0.5f*dt);
gz *= (0.5f*dt);
qa = q0;
qb = q1;
qc = q2;
q0 += (-qb*gx - qc*gy - q3*gz);
q1 += ( qa*gx + qc*gz - q3*gy);
q2 += ( qa*gy - qb*gz + q3*gx);
q3 += ( qa*gz + qb*gy - qc*gx);
// Normalise quaternion
recipNorm = invSqrt(q0*q0 + q1*q1 + q2*q2 + q3*q3);
q0 *= recipNorm;
q1 *= recipNorm;
q2 *= recipNorm;
q3 *= recipNorm;
q->w = q0;
q->x = q1;
q->y = q2;
q->z = q3;
}
void quaternion_update(double dt, quaternion_t* q, mpu_value_t* m) {
double q0 = q->w; double q0 = q->w;
double q1 = q->x; double q1 = q->x;
@@ -133,16 +226,15 @@ void ahrs_update(double dt, quaternion_t* q, mpu_value_t* m) {
double recipNorm; double recipNorm;
double s0, s1, s2, s3; double s0, s1, s2, s3;
double qDot1, qDot2, qDot3, qDot4; double qDot1, qDot2, qDot3, qDot4;
double _2q0, _2q1, _2q2, _2q3, _4q0, _4q1, _4q2, _8q1, _8q2, q0q0, q1q1, q2q2, q3q3;
// Rate of change of quaternion from gyroscope // Rate of change of quaternion from gyroscope
qDot1 = 0.5f * (-q1*gx - q2*gy - q3*gz); qDot1 = 0.5 * (-q1*gx - q2*gy - q3*gz);
qDot2 = 0.5f * ( q0*gx + q2*gz - q3*gy); qDot2 = 0.5 * ( q0*gx + q2*gz - q3*gy);
qDot3 = 0.5f * ( q0*gy - q1*gz + q3*gx); qDot3 = 0.5 * ( q0*gy - q1*gz + q3*gx);
qDot4 = 0.5f * ( q0*gz + q1*gy - q2*gx); qDot4 = 0.5 * ( q0*gz + q1*gy - q2*gx);
// Compute feedback only if accelerometer measurement valid (avoids NaN in accelerometer normalisation) // Compute feedback only if accelerometer measurement valid (avoids NaN in accelerometer normalisation)
if (!((ax == 0.0f) && (ay == 0.0f) && (az == 0.0f))) { if (!((ax == 0.0) && (ay == 0.0) && (az == 0.0))) {
// Normalise accelerometer measurement // Normalise accelerometer measurement
recipNorm = invSqrt(ax*ax + ay*ay + az*az); recipNorm = invSqrt(ax*ax + ay*ay + az*az);
@@ -150,28 +242,14 @@ void ahrs_update(double dt, quaternion_t* q, mpu_value_t* m) {
ay *= recipNorm; ay *= recipNorm;
az *= recipNorm; az *= recipNorm;
// Auxiliary variables to avoid repeated arithmetic
_2q0 = 2.0f * q0;
_2q1 = 2.0f * q1;
_2q2 = 2.0f * q2;
_2q3 = 2.0f * q3;
_4q0 = 4.0f * q0;
_4q1 = 4.0f * q1;
_4q2 = 4.0f * q2;
_8q1 = 8.0f * q1;
_8q2 = 8.0f * q2;
q0q0 = q0 * q0;
q1q1 = q1 * q1;
q2q2 = q2 * q2;
q3q3 = q3 * q3;
// Gradient decent algorithm corrective step // Gradient decent algorithm corrective step
s0 = _4q0*q2q2 + _2q2*ax + _4q0*q1q1 - _2q1*ay; s0 = 4.0*q0*q2*q2 + 2.0*q2*ax + 4.0*q0*q1*q1 - 2.0*q1*ay;
s1 = _4q1*q3q3 - _2q3*ax + 4.0f*q0q0*q1 - _2q0*ay - _4q1 + _8q1*q1q1 + _8q1*q2q2 + _4q1*az; s1 = 4.0*q1*q3*q3 - 2.0*q3*ax + 4.0*q0*q0*q1 - 2.0*q0*ay - 4.0*q1 + 8.0*q1*q1*q1 + 8.0*q1*q2*q2 + 4.0*q1*az;
s2 = 4.0f*q0q0*q2 + _2q0*ax + _4q2*q3q3 - _2q3*ay - _4q2 + _8q2*q1q1 + _8q2*q2q2 + _4q2*az; s2 = 4.0*q0*q0*q2 + 2.0*q0*ax + 4.0*q2*q3*q3 - 2.0*q3*ay - 4.0*q2 + 8.0*q2*q1*q1 + 8.0*q2*q2*q2 + 4.0*q2*az;
s3 = 4.0f*q1q1*q3 - _2q1*ax + 4.0f*q2q2*q3 - _2q2*ay; s3 = 4.0*q1*q1*q3 - 2.0*q1*ax + 4.0*q2*q2*q3 - 2.0*q2*ay;
recipNorm = invSqrt(s0*s0 + s1*s1 + s2*s2 + s3*s3); // normalise step magnitude // Normalise step magnitude
recipNorm = invSqrt(s0*s0 + s1*s1 + s2*s2 + s3*s3);
s0 *= recipNorm; s0 *= recipNorm;
s1 *= recipNorm; s1 *= recipNorm;
s2 *= recipNorm; s2 *= recipNorm;
@@ -203,6 +281,7 @@ void ahrs_update(double dt, quaternion_t* q, mpu_value_t* m) {
q->z = q3; q->z = q3;
} }
double radian2degrees(double radians) { double radian2degrees(double radians) {
return radians * (180.0f / M_PI); return radians * (180.0f / M_PI);
} }
@@ -245,12 +324,12 @@ angle_t quaternion2xyz(quaternion_t* q) {
double t0 = (x + z)*(x - z); // x^2-z^2 double t0 = (x + z)*(x - z); // x^2-z^2
double t1 = (w + y)*(w - y); // w^2-y^2 double t1 = (w + y)*(w - y); // w^2-y^2
double xx = 0.5f * (t0 + t1); // 1/2 x of x' double xx = 0.5 * (t0 + t1); // 1/2 x of x'
double xy = x*y + w*z; // 1/2 y of x' double xy = x*y + w*z; // 1/2 y of x'
double xz = w*y - x*z; // 1/2 z of x' double xz = w*y - x*z; // 1/2 z of x'
double t = xx*xx + xy*xy; // cos(theta)^2 double t = xx*xx + xy*xy; // cos(theta)^2
double yz = 2.0f * (y*z + w*x); // z of y' double yz = 2.0 * (y*z + w*x); // z of y'
a.z = atan2(xy, xx); // yaw (psi) a.z = atan2(xy, xx); // yaw (psi)
a.y = atan(xz /sqrt(t)); // pitch (theta) a.y = atan(xz /sqrt(t)); // pitch (theta)
@@ -263,6 +342,13 @@ angle_t quaternion2xyz(quaternion_t* q) {
return a; return a;
} }
void angle_norm(angle_t* a) {
double n = sqrt(a->x*a->x + a->y*a->y + a->z*a->z);
a->x *= n;
a->y *= n;
a->z *= n;
}
int main(void) { int main(void) {
@@ -273,35 +359,29 @@ int main(void) {
i2c_setup(); i2c_setup();
systick_setup(); systick_setup();
printf("==== start ====\r\n");
mpu_t mpu; mpu_t mpu;
printf("==== mpu initialize ====\r\n");
mpu_setup(&mpu, I2C1, 0x68); mpu_setup(&mpu, I2C1, 0x68);
_delay(10); _delay(100);
mpu_calibrate(&mpu, 20000); mpu_calibrate(&mpu, 20000);
printf("==== mpu ====\r\n"); printf("==== mpu started ====\r\n");
mpu_value_t mval; mpu_value_t mval;
uint32_t prev_ts = 0, last_ts = 0;
quaternion_t q; quaternion_t q;
quaternion_init(&q); quaternion_init(&q);
int i = 0; uint32_t prev_ts = 0;
uint32_t last_ts = 0;
printf("==== main loop ====\r\n");
while (true) { while (true) {
mpu_read(&mpu, &mval); mpu_read(&mpu, &mval);
gpio_toggle(GPIOB, GPIO6);
//if ((i % 350) == 0) {
// printf("gx=%8.4f gy=%8.4f gz=%8.4f ax=%8.4f ay=%8.4f az=%8.4f \r\n", mval.gx, mval.gy, mval.gz, mval.ax, mval.ay, mval.az);
//}
i++;
last_ts = g_sys_tick_counter; last_ts = g_sys_tick_counter;
float dt = (float)(last_ts - prev_ts) / (float)systic_freq; double dt = (float)(last_ts - prev_ts) / (float)systic_freq;
prev_ts = last_ts; prev_ts = last_ts;
ahrs_update(dt, &q, &mval); quaternion_update(dt, &q, &mval);
angle_t a = quaternion2xyz(&q); angle_t a = quaternion2xyz(&q);
angle_degress(&a); angle_degress(&a);

65
mculoop/mpu6050.c
View File

@@ -94,6 +94,9 @@
#define MPU_PWR2_STBY_YG_BIT 1 #define MPU_PWR2_STBY_YG_BIT 1
#define MPU_PWR2_STBY_ZG_BIT 0 #define MPU_PWR2_STBY_ZG_BIT 0
//static void i2cdev_reg_setbits(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t mask);
//static void i2cdev_reg_cleanbits(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t mask);
//static uint8_t i2cdev_read_reg8(uint32_t i2c, uint8_t addr, uint8_t reg); //static uint8_t i2cdev_read_reg8(uint32_t i2c, uint8_t addr, uint8_t reg);
static void i2cdev_write_reg8(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t value); static void i2cdev_write_reg8(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t value);
static void i2cdev_read_seq8(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t* buffer, uint8_t size); static void i2cdev_read_seq8(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t* buffer, uint8_t size);
@@ -105,7 +108,13 @@ static void i2cdev_write_reg8(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t v
i2c_transfer7(i2c, addr, buffer, 2, NULL, 0); i2c_transfer7(i2c, addr, buffer, 2, NULL, 0);
} }
//static void i2cdev_reg_setbit(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t mask) { //static uint8_t i2cdev_read_reg8(uint32_t i2c, uint8_t addr, uint8_t reg) {
// uint8_t val;
// i2c_transfer7(i2c, addr, &reg, 1, &val, 1);
// return val;
//}
//static void i2cdev_reg_setbits(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t mask) {
// uint8_t buffer[2]; // uint8_t buffer[2];
// buffer[0] = reg; // buffer[0] = reg;
// buffer[1] = 0x00; // buffer[1] = 0x00;
@@ -114,23 +123,25 @@ static void i2cdev_write_reg8(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t v
// i2c_transfer7(i2c, addr, buffer, 2, NULL, 0); // i2c_transfer7(i2c, addr, buffer, 2, NULL, 0);
//} //}
//static uint8_t i2cdev_read_reg8(uint32_t i2c, uint8_t addr, uint8_t reg) { //static void i2cdev_reg_cleanbits(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t mask) {
// uint8_t val; // uint8_t buffer[2];
// i2c_transfer7(i2c, addr, &reg, 1, &val, 1); // buffer[0] = reg;
// return val; // buffer[1] = 0x00;
// i2c_transfer7(i2c, addr, &buffer[0], 1, &buffer[1], 1);
// buffer[1] &= ~(mask);
// i2c_transfer7(i2c, addr, buffer, 2, NULL, 0);
//} //}
static void i2cdev_read_seq8(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t* buffer, uint8_t size) { static void i2cdev_read_seq8(uint32_t i2c, uint8_t addr, uint8_t reg, uint8_t* buffer, uint8_t size) {
i2c_transfer7(i2c, addr, &reg, 1, buffer, size); i2c_transfer7(i2c, addr, &reg, 1, buffer, size);
} }
#define MPU_GYRO_LSB MPU_GYRO_LSB_1000 #define MPU_GYRO_LSB MPU_GYRO_LSB_1000
#define MPU_GYRO_FS MPU_GYRO_FS_1000 #define MPU_GYRO_FS MPU_GYRO_FS_1000
#define MPU_ACCEL_LSB MPU_ACCEL_LSB_16 #define MPU_ACCEL_LSB MPU_ACCEL_LSB_16
#define MPU_ACCEL_FS MPU_ACCEL_FS_16 #define MPU_ACCEL_FS MPU_ACCEL_FS_16
void mpu_setup(mpu_t* mpu, uint32_t i2c, uint8_t addr) { void mpu_setup(mpu_t* mpu, uint32_t i2c, uint8_t addr) {
@@ -145,8 +156,7 @@ void mpu_setup(mpu_t* mpu, uint32_t i2c, uint8_t addr) {
mpu->err.gz = 0; mpu->err.gz = 0;
//i2cdev_write_reg8(i2c, addr, MPU_REG_PWR_MGMT_1, 1 << MPU_PWR1_DEVICE_RESET_BIT); //i2cdev_write_reg8(i2c, addr, MPU_REG_PWR_MGMT_1, 1 << MPU_PWR1_DEVICE_RESET_BIT);
//for (int i = 0; i < 10000; i++) __asm__("nop");
for (int i = 0; i < 10000; i++) __asm__("nop");
i2cdev_write_reg8(i2c, addr, MPU_REG_PWR_MGMT_1, 0x00); i2cdev_write_reg8(i2c, addr, MPU_REG_PWR_MGMT_1, 0x00);
@@ -169,17 +179,17 @@ static void mpu_rawread(mpu_t* mpu, mpu_value_t* val) {
int16_t gy = (((int16_t)buffer[10]) << 8) | buffer[11]; int16_t gy = (((int16_t)buffer[10]) << 8) | buffer[11];
int16_t gz = (((int16_t)buffer[12]) << 8) | buffer[13]; int16_t gz = (((int16_t)buffer[12]) << 8) | buffer[13];
val->ax = (float)ax / (float)MPU_ACCEL_LSB; val->ax = (double)ax / (double)MPU_ACCEL_LSB;
val->ay = (float)ay / (float)MPU_ACCEL_LSB; val->ay = (double)ay / (double)MPU_ACCEL_LSB;
val->az = (float)az / (float)MPU_ACCEL_LSB; val->az = (double)az / (double)MPU_ACCEL_LSB;
val->gx = (float)gx / (float)MPU_GYRO_LSB; val->gx = (double)gx / (double)MPU_GYRO_LSB;
val->gy = (float)gy / (float)MPU_GYRO_LSB; val->gy = (double)gy / (double)MPU_GYRO_LSB;
val->gz = (float)gz / (float)MPU_GYRO_LSB; val->gz = (double)gz / (double)MPU_GYRO_LSB;
val->gx *= M_PI / 180.0f; val->gx *= M_PI / 180.0;
val->gy *= M_PI / 180.0f; val->gy *= M_PI / 180.0;
val->gz *= M_PI / 180.0f; val->gz *= M_PI / 180.0;
} }
@@ -195,13 +205,15 @@ void mpu_calibrate(mpu_t* mpu, int count) {
for (int i = 0; i < count; i++) { for (int i = 0; i < count; i++) {
mpu_rawread(mpu, &val); mpu_rawread(mpu, &val);
mpu->err.gx += (float)val.gx / count; mpu->err.gx += val.gx / (double)count;
mpu->err.gy += (float)val.gy / count; mpu->err.gy += val.gy / (double)count;
mpu->err.gz += (float)val.gz / count; mpu->err.gz += val.gz / (double)count;
mpu->err.ax += val.ax / (double)count;
mpu->err.ay += val.ay / (double)count;
} }
} }
void mpu_read(mpu_t* mpu, mpu_value_t* val) { void mpu_read(mpu_t* mpu, mpu_value_t* val) {
mpu_rawread(mpu, val); mpu_rawread(mpu, val);
@@ -209,4 +221,7 @@ void mpu_read(mpu_t* mpu, mpu_value_t* val) {
val->gy -= mpu->err.gy; val->gy -= mpu->err.gy;
val->gz -= mpu->err.gz; val->gz -= mpu->err.gz;
val->ax -= mpu->err.ax;
val->ay -= mpu->err.ay;
val->az -= mpu->err.az;
} }

12
mculoop/mpu6050.h
View File

@@ -6,12 +6,12 @@
#include <stdint.h> #include <stdint.h>
typedef struct { typedef struct {
float ax; double ax;
float ay; double ay;
float az; double az;
float gx; double gx;
float gy; double gy;
float gz; double gz;
} mpu_value_t; } mpu_value_t;
typedef struct { typedef struct {