#pragma once

#ifndef IMU_H
#define IMU_H

#include "stm32g431xx.h"

#define ICM_ADDR				0x68
#define REG_PWR_MGMT_1			0x06
#define REG_BANK_SEL			0x7F
#define REG_GYRO_CONFIG_1		0x01
#define REG_ACCEL_CONFIG		0x14
#define IMU_DT					0.002f   // 2 ms

#define GYRO_FS_SEL_2000		3 << 1
#define GYRO_DLPFCFG_73			3 << 3
#define GYRO_FCHOICE_ON			1
#define GYRO_FCHOICE_OFF		0

#define ACCEL_FS_SEL_8			2 << 1
#define ACCEL_DLPFCFG_69		3 << 3
#define ACCEL_FCHOICE_ON		1
#define ACCEL_FCHOICE_OFF		0

typedef struct
{
	int16_t ax, ay, az; // lsb
	int16_t gx, gy, gz; // lsb
} imu_raw_t;

void imu_pow_init();

void i2c_gpio_init();

void i2c1_init();

void imu_init();

void imu_tim6_init();

void i2c_read(uint8_t addr, uint8_t reg, uint8_t* buf, uint8_t len);

void i2c_write(uint8_t addr, uint8_t reg, uint8_t data);

void imu_read_raw(imu_raw_t* data);

static void i2c_wait_idle(I2C_TypeDef* i2c);

#endif