#ifndef __SHAKE_DETECT_H
#define __SHAKE_DETECT_H

#include <stdint.h>

#define LIS3DSH_I2C_MASTER_NUM              0
#define LIS3DSH_I2C_SEL_IO                  11          // 地址选择
#define LIS3DSH_I2C_MASTER_SDA_IO           12   
#define LIS3DSH_I2C_MASTER_SCL_IO           13        
#define LIS3DSH_I2C_MASTER_FREQ_HZ          400000                     
#define LIS3DSH_I2C_SALVE_ADDR_L            0x18        // 7bit地址0b0011000
#define LIS3DSH_I2C_MASTER_TIMEOUT_MS       1000         
#define ACK_CHECK_EN                        0x1                        
#define ACK_CHECK_DIS                       0x0                                           

#define LIS3DSH_INT1_IO                     9   // 数据就绪信号
#define LIS3DSH_CS_IO                       10  // SPI I2C模式选择，1-I2C
#define LIS3DSH_LED_IO                      3
#define BOOT_TIME                           10


#define N_SAMPLES                           1024
#define ANGLE_N_SAMPLES                     128     // 需能整除N_SAMPLES

#define LIS3DSH_CONFIG_FS_NVS_KAY           "l_cfg_fs"
#define LIS3DSH_CONFIG_ODR_NVS_KAY          "l_cfg_odr"
#define SHAKE_CONFIG_FMIN_NVS_KAY           "s_cfg_fmin"
#define SHAKE_CONFIG_FMAX_NVS_KAY           "s_cfg_fmax"
#define SHAKE_CONFIG_INATH_NVS_KAY          "s_cfg_inath"
#define SHAKE_CONFIG_OUTATH_NVS_KAY         "s_cfg_outath"
#define SHAKE_CONFIG_CH1_NVS_KAY            "s_cfg_ch1"
#define SHAKE_CONFIG_CH2_NVS_KAY            "s_cfg_ch2"
#define SHAKE_CONFIG_SEL_NVS_KAY            "s_cfg_sel"

typedef float fft_type_t;

typedef struct
{
    uint8_t fscale; // 寄存器配置的参数
    uint8_t odr;

    int fscale_v;   // 实际值
    int odr_v;
} lis3dsh_config_t;

// 三轴加速度，经过单位转换,g
typedef struct
{
    fft_type_t a[3]; // x,y,z
} lis3dsh_adata_t;

typedef struct
{
    uint16_t feq_min_id; // 注意，单位并不是HZ，为: 1600/N_SAMPLES HZ
    uint16_t feq_max_id;
    fft_type_t enter_amplitude_th;
    fft_type_t exit_amplitude_th;
    uint8_t channel1;   // FFT可以同时计算两轴的频谱图，选择[1,2,3]->[x,y,z]
    uint8_t channel2;   
    uint8_t select;     // 选择进行震动检测的是频谱图，选择[1,2,3]->[ch1,ch2,both]
} shake_detect_cfg_t;

typedef struct
{
    uint16_t  count;
    uint8_t   is_shake;
    float wind[N_SAMPLES];
    fft_type_t y_cf[N_SAMPLES * 2];
    fft_type_t *y1_cf;
    fft_type_t *y2_cf;
    fft_type_t sum_y[N_SAMPLES / 2];
} fft_data_t;

typedef struct
{
    uint16_t count;
    float x;
    float y;
    float z;
} angle_data_t;


extern int buf_r;
extern shake_detect_cfg_t shake_cfg;
extern lis3dsh_adata_t lis3dsh_data[N_SAMPLES * 2];
extern fft_data_t fft_data;
extern angle_data_t angle_data;

void    shake_detect_init(void);
void    lis3dsh_calculating_angle(float ax, float ay, float az, float *pitch, float *roll, float *yaw);

int     lis3dsh_set_fscale(uint8_t fscale);
int     lis3dsh_set_odr(uint8_t odr);
uint8_t lis3dsh_get_fscale(void);
uint8_t lis3dsh_get_odr(void);
int     lis3dsh_config_restore_default(void);
int     lis3dsh_config_nvs_load(void);
int     lis3dsh_config_nvs_save(void);

int     shake_config_restore_default(void);
int     shake_config_nvs_load(void);
int     shake_config_nvs_save(void);

#endif