/*
 * far蛇模组
 */

#pragma once

#ifdef __cplusplus
extern "C" {
#endif

#include "main.h"
#include <stdbool.h>
#include "component/pid.h"
#include "device/motor_rm.h"


/* Exported constants ------------------------------------------------------- */
#define SHOOT_OK (0)        /* 运行正常 */
#define SHOOT_ERR (-1)      /* 运行时发现了其他错误 */
#define SHOOT_ERR_NULL (-2) /* 运行时发现NULL指针 */
#define SHOOT_ERR_MODE (-3) /* 运行时配置了错误的CMD_ChassisMode_t */
#define SHOOT_ERR_TYPE (-4) /* 运行时配置了错误的Chassis_Type_t */

#define SHOOT_FRIC_NUM (2) /* 摩擦轮数量 */
#define MAX_FRIC_RPM 7000.0f
#define MAX_TRIG_RPM 1000.0f
/* Exported macro ----------------------------------------------------------- */
/* Exported types ----------------------------------------------------------- */

typedef enum {
    SHOOT_STATE_IDLE = 0,      // 熄火
    SHOOT_STATE_READY,         // 准备射击
    SHOOT_STATE_FIRE           // 射击
} Shoot_State_t;

typedef enum {
    SHOOT_MODE_SAFE = 0,      // 安全模式
    SHOOT_MODE_SINGLE,         // 单发模式
    SHOOT_MODE_BURST,          // 多发模式
    SHOOT_MODE_CONTINUE         // 连发模式
} Shoot_Mode_t;
typedef struct {
  bool online;	
	
  bool ready;                    /* 准备射击 */
  bool firecmd;                  /* 射击指令 */

  bool last_firecmd;

} Shoot_CMD_t;
typedef struct {
  MOTOR_Feedback_t fric[SHOOT_FRIC_NUM]; /* 摩擦轮电机反馈 */
  MOTOR_RM_t *trig; /* 拨弹电机反馈 */

  float fil_fric_rpm[SHOOT_FRIC_NUM]; /* 滤波后的摩擦轮转速 */
  float fil_trig_rpm; /* 滤波后的拨弹电机转速*/

  float fric_rpm[SHOOT_FRIC_NUM]; /* 归一化摩擦轮转速 */
  float fric_avgrpm; /* 归一化摩擦轮平均转速*/
  float trig_rpm; /* 归一化拨弹电机转速*/

}Shoot_Feedback_t;

typedef struct {
  float out_follow[SHOOT_FRIC_NUM];
  float out_err[SHOOT_FRIC_NUM];
  float out_fric[SHOOT_FRIC_NUM];
  float lpfout_fric[SHOOT_FRIC_NUM];

  
  float out_trig;
  float lpfout_trig;
}Shoot_Output_t;


/* 底盘参数的结构体，包含所有初始化用的参数，通常是const，存好几组 */
typedef struct {
  float trig_step_angle; /* 每次拨弹电机转动的角度 */

  MOTOR_RM_Param_t fric_motor_param[SHOOT_FRIC_NUM];
  MOTOR_RM_Param_t trig_motor_param;
  

  KPID_Params_t fric_follow;               /* 摩擦轮电机PID控制参数，用于跟随目标速度 */
  KPID_Params_t fric_err;               /* 摩擦轮电机PID控制参数，用于消除转速误差 */
  KPID_Params_t trig;              /* 拨弹电机PID控制参数 */


  /* 低通滤波器截止频率 */
  struct {
    struct{
    float in;  /* 反馈值滤波器 */
    float out; /* 输出值滤波器 */
    }fric;
    struct{
    float in;  /* 反馈值滤波器 */
    float out; /* 输出值滤波器 */
    }trig;
  } filter;
} Shoot_Params_t;

/*
 * 运行的主结构体，所有这个文件里的函数都在操作这个结构体
 * 包含了初始化参数，中间变量，输出变量
 */
typedef struct {
  bool online;

  uint32_t lask_wakeup;
  float dt;

  Shoot_Params_t *param; /* */
  /* 模块通用 */
  Shoot_State_t running_state; /* 运行状态机 */
  Shoot_Mode_t  mode;
  /* 反馈信息 */
  Shoot_Feedback_t feedback;
  /* 控制信息*/
  Shoot_Output_t output;
  /* 目标控制量 */
  struct {
    float target_rpm; /* 目标摩擦轮转速 */
    float target_angle; /* 目标拨弹位置 */
  }target_variable;

  /* 反馈控制用的PID */
  struct {
    KPID_t fric_follow[SHOOT_FRIC_NUM]; /* */
    KPID_t fric_err[SHOOT_FRIC_NUM];    /*  */
    KPID_t trig;
  } pid;

  /* 滤波器 */
  struct {
        struct{
    LowPassFilter2p_t in[SHOOT_FRIC_NUM];  /* 反馈值滤波器 */
    LowPassFilter2p_t out[SHOOT_FRIC_NUM]; /* 输出值滤波器 */
    }fric;
    struct{
    LowPassFilter2p_t in;  /* 反馈值滤波器 */
    LowPassFilter2p_t out; /* 输出值滤波器 */
    }trig;
  } filter;

  float errtosee; /*调试用*/

} shoot_t;

/* Exported functions prototypes -------------------------------------------- */

/**
 * \brief 初始化发射
 *
 * \param c 包含发射数据的结构体
 * \param param 包含发射参数的结构体指针
 * \param target_freq 任务预期的运行频率
 *
 * \return 函数运行结果
 */
int8_t Shoot_Init(shoot_t *c, Shoot_Params_t *param, float target_freq);

/**
 * \brief 更新反馈
 *
 * \param c 包含发射数据的结构体
 *
 * \return 函数运行结果
 */
int8_t Chassis_UpdateFeedback(shoot_t *c);

/**
 * \brief 初始化发射
 *
 * \param c 包含发射数据的结构体
 * \param cmd 包含发射命令的结构体
 *
 * \return 函数运行结果
 */
int8_t Shoot_Control(shoot_t *c, const Shoot_CMD_t *cmd);



#ifdef __cplusplus
}
#endif