高飞老师组的px4ctrl理解

前言: 高飞老师组的控制器px4ctrl是根据DJIN3飞控N3ctrl进行设计的,其改变了控制器部分以及状态机部分,需要的可以参考N3Ctrl代码


高飞老师组的控制器px4ctrl是根据DJIN3飞控N3ctrl进行设计的,其改变了控制器部分以及状态机部分,需要的可以参考N3Ctrl代码
N3Ctrl)

整体控制器结构

高飞老师组的项目的控制器框架可以分为在仿真器上的控制器以及在直接在轨迹规划后接口的控制器端口,在本文中,只说px4ctrl相关的控制器,也就是直接对规划控制后接口的控制器。SO3仿真控制,也就是姿态控制如:so3_quadrotor_simulator,so3_control在后面在说。
这部分也是直接对轨迹规划作位置环,姿态环进行给无人机进行油门的映射。框架如下图:
在这里插入图片描述
按高飞老师描述的是,在这里,可以知道轨迹规划给出的三个值,位置以及其一阶二阶导给到了位置控制器,位置控制器给到了偏航角,横滚角,俯仰角并且给到了直接的动力u1给无人机,并且姿态控制器给出了u2动力给无人机,无人机闭环反馈三个角以及角速度还有位置和速度给到了相应的控制器

PX4Ctrl_node.cpp

#include <ros/ros.h>
#include "PX4CtrlFSM.h"
#include <signal.h>

void mySigintHandler(int sig)
{
    ROS_INFO("[PX4Ctrl] exit...");
    ros::shutdown();
}

int main(int argc, char *argv[])
{
    //初始化结点
    ros::init(argc, argv, "px4ctrl");
    ros::NodeHandle nh("~");
    // 检测终止信号
    signal(SIGINT, mySigintHandler);
    ros::Duration(1.0).sleep();

    //初始化参数类,从参数服务器读取参数
    Parameter_t param;
    param.config_from_ros_handle(nh);

    // Controller controller(param);
    // 初始化线性控制器
    LinearControl controller(param);
    //	初始化状态机
    PX4CtrlFSM fsm(param, controller);
    //订阅当前状态,以参数传给fsm.state_data.feed()		
    //接收状态消息
    ros::Subscriber state_sub =
        nh.subscribe<mavros_msgs::State>("/mavros/state",
                                         10,
                                         boost::bind(&State_Data_t::feed, &fsm.state_data, _1));
    //订阅当前扩展状态,以参数传给fsm.extened_state_data.feed()	
    //接收扩展状态消息
    ros::Subscriber extended_state_sub =
        nh.subscribe<mavros_msgs::ExtendedState>("/mavros/extended_state",
                                                 10,
                                                 boost::bind(&ExtendedState_Data_t::feed, &fsm.extended_state_data, _1));
    //订阅里程计话题odmo,由/vins_fusion/imu_propagate映射而来
    //回调函数	->读取位置、速度、四元数数据传入fsm.odmo_data,监控数据帧率,小于100hz警告
    ros::Subscriber odom_sub =
        nh.subscribe<nav_msgs::Odometry>("odom",
                                         100,
                                         boost::bind(&Odom_Data_t::feed, &fsm.odom_data, _1),
                                         ros::VoidConstPtr(),
                                         ros::TransportHints().tcpNoDelay());
    //订阅位置命令话题cmd	,由/position_cmd映射而来
    //回调函数	->读取位置、速度、加速度、加加速度(jerk)、yaw角给fsm.cmd_data
    ros::Subscriber cmd_sub =
        nh.subscribe<quadrotor_msgs::PositionCommand>("cmd",
                                                      100,
                                                      boost::bind(&Command_Data_t::feed, &fsm.cmd_data, _1),
                                                      ros::VoidConstPtr(),
                                                      ros::TransportHints().tcpNoDelay());
    //  订阅imu数据话题	,/mavros/imu/data
    //  回调函数	->读取姿态、角速度、角加速度给fsm.imu_data
    ros::Subscriber imu_sub =
        nh.subscribe<sensor_msgs::Imu>("/mavros/imu/data", // Note: do NOT change it to /mavros/imu/data_raw !!!
                                       100,
                                       boost::bind(&Imu_Data_t::feed, &fsm.imu_data, _1),
                                       ros::VoidConstPtr(),
                                       ros::TransportHints().tcpNoDelay());
    /*接收机通道定义
    ch[0]:	横滚
    ch[1]:	俯仰
    ch[2]:	油门
    ch[3]:	偏航
    开关  打到最上面为0,打到最下面为1.0
	两档:0~1.0
	三档:0~0.5~1.0
    ch[4]:	5通道,飞行模式,三档  0 手动模式,1.定点模式 2.指令控制模式
    ch[5]:	6通道,命令控制,两档
    ch[7]:	8通道,用于重启px4
    */
    //订阅接收机数据话题	,/mavros/rc/in
	// 回调函数	->读取四个通道的打杆值,并进行模式识别
    ros::Subscriber rc_sub;
    if (!param.takeoff_land.no_RC) // mavros will still publish wrong rc messages although no RC is connected
    {
        rc_sub = nh.subscribe<mavros_msgs::RCIn>("/mavros/rc/in",
                                                 10,
                                                 boost::bind(&RC_Data_t::feed, &fsm.rc_data, _1));
    }
    // 	订阅电池状态话题	,/mavros/battery
	//回调函数	->读取电压、计算百分比并打印
    ros::Subscriber bat_sub =
        nh.subscribe<sensor_msgs::BatteryState>("/mavros/battery",
                                                100,
                                                boost::bind(&Battery_Data_t::feed, &fsm.bat_data, _1),
                                                ros::VoidConstPtr(),
                                                ros::TransportHints().tcpNoDelay());
    // 订阅起飞命令话题,takeoff_land
	// 回调函数	->fsm.takeoff_land_data.triggered = 1,接收起飞命令
    ros::Subscriber takeoff_land_sub =
        nh.subscribe<quadrotor_msgs::TakeoffLand>("takeoff_land",
                                                  100,
                                                  boost::bind(&Takeoff_Land_Data_t::feed, &fsm.takeoff_land_data, _1),
                                                  ros::VoidConstPtr(),
                                                  ros::TransportHints().tcpNoDelay());
    // /mavros/setpoint_raw/attitude 对应有两种方式控制:1.姿态+油门;2.机体角速度+油门
    fsm.ctrl_FCU_pub = nh.advertise<mavros_msgs::AttitudeTarget>("/mavros/setpoint_raw/attitude", 10);
    //首先飞机要处在定点模式下,此时把6通道拨杆从控制指令拒绝状态拨到使能状态,此刻px4ctrl会发一个`/traj_start_trigger`出来,
    // 当下还尚未切换到指令控制模式,px4ctrl会等待外部指令,一旦接收到指令,模式就会切换,屏幕也会打印绿色的"[px4ctrl] AUTO_HOVER(L2) --> CMD_CTRL(L3)" 
    fsm.traj_start_trigger_pub = nh.advertise<geometry_msgs::PoseStamped>("/traj_start_trigger", 10);
    // debug信息 作为日志信息记录的
    fsm.debug_pub = nh.advertise<quadrotor_msgs::Px4ctrlDebug>("/debugPx4ctrl", 10); // debug
    // mavros通信,模式切换
    fsm.set_FCU_mode_srv = nh.serviceClient<mavros_msgs::SetMode>("/mavros/set_mode");
    // mavros通信,解锁与上锁
    fsm.arming_client_srv = nh.serviceClient<mavros_msgs::CommandBool>("/mavros/cmd/arming");
    // 重启
    fsm.reboot_FCU_srv = nh.serviceClient<mavros_msgs::CommandLong>("/mavros/cmd/command");

    ros::Duration(0.5).sleep();
    //判断从参数服务器读过来的参数设置是否需要遥控:1不需要,0需要
    if (param.takeoff_land.no_RC)
    {
        ROS_WARN("PX4CTRL] Remote controller disabled, be careful!");//不需要,警告:遥控失效,小心
    }
    else
    {
        ROS_INFO("PX4CTRL] Waiting for RC"); //需要:等待接收机信号
        while (ros::ok())
        {
            ros::spinOnce();
            if (fsm.rc_is_received(ros::Time::now()))
            {
                ROS_INFO("[PX4CTRL] RC received.");
                break;
            }
            ros::Duration(0.1).sleep();
        }
    }
    //进入死循环,检查px4的连接,连接正常跳出循环
    int trials = 0;
    while (ros::ok() && !fsm.state_data.current_state.connected)
    {
        ros::spinOnce();
        ros::Duration(1.0).sleep();
        if (trials++ > 5)
            ROS_ERROR("Unable to connnect to PX4!!!");
    }
    //主循环,以固定帧率进入fsm.process()进程
    ros::Rate r(param.ctrl_freq_max);
    while (ros::ok())
    {
        r.sleep();
        ros::spinOnce();
        fsm.process(); // We DO NOT rely on feedback as trigger, since there is no significant performance difference through our test.
    }

    return 0;
}
  1. 初始化了常用的对象,paramter对象,线性控制器(这一块和N3ctrl控制完全不同,这一块高飞老师着重就是用加速度accel来映射油门thrust,也就是估计油门的状态,详细的在后面的control.cpp会贴出来主要函数,在calculateControl,这一块在FSM状态机用到了),以及最后的初始化的大类PX4CtrlFSM大类,整合了所有的类进去。
  2. 订阅相关话题,从规划器以及前端整合过来的数据,注意这里的里程计实质上在fast-250上是由vins/fusion/imu融合过后得到的,要注意。
  3. 判断遥控是否打到对应通道,同时以固定的频率进入了主进程,也就是fsm.process()函数

control.cpp

#include "controller.h"

using namespace std;



double LinearControl::fromQuaternion2yaw(Eigen::Quaterniond q)
{
  double yaw = atan2(2 * (q.x()*q.y() + q.w()*q.z()), q.w()*q.w() + q.x()*q.x() - q.y()*q.y() - q.z()*q.z());//123angles
  return yaw;
}

LinearControl::LinearControl(Parameter_t &param) : param_(param)
{
  resetThrustMapping();
}

/* 
  compute u.thrust and u.q, controller gains and other parameters are in param_ 
*/
quadrotor_msgs::Px4ctrlDebug

LinearControl::calculateControl(const Desired_State_t &des,
    const Odom_Data_t &odom,
    const Imu_Data_t &imu, 
    Controller_Output_t &u)
{
  /* WRITE YOUR CODE HERE */
      //compute disired acceleration 计算期望的加速度 转换为油门的值 
      Eigen::Vector3d des_acc(0.0, 0.0, 0.0);
      Eigen::Vector3d Kp,Kv;
      Kp << param_.gain.Kp0, param_.gain.Kp1, param_.gain.Kp2;
      Kv << param_.gain.Kv0, param_.gain.Kv1, param_.gain.Kv2;
      des_acc = des.a + Kv.asDiagonal() * (des.v - odom.v) + Kp.asDiagonal() * (des.p - odom.p);
      des_acc += Eigen::Vector3d(0,0,param_.gra);//Z轴上的需要加上gra重力

      u.thrust = computeDesiredCollectiveThrustSignal(des_acc);
      double roll,pitch,yaw,yaw_imu;//三种姿态角
      double yaw_odom = fromQuaternion2yaw(odom.q);//四元数转换yaw角
      double sin = std::sin(yaw_odom);
      double cos = std::cos(yaw_odom);
      roll = (des_acc(0) * sin - des_acc(1) * cos )/ param_.gra;
      pitch = (des_acc(0) * cos + des_acc(1) * sin )/ param_.gra;
      // yaw = fromQuaternion2yaw(des.q);
      yaw_imu = fromQuaternion2yaw(imu.q);
      // Eigen::Quaterniond q = Eigen::AngleAxisd(yaw,Eigen::Vector3d::UnitZ())
      //   * Eigen::AngleAxisd(roll,Eigen::Vector3d::UnitX())
      //   * Eigen::AngleAxisd(pitch,Eigen::Vector3d::UnitY());
      Eigen::Quaterniond q = Eigen::AngleAxisd(des.yaw,Eigen::Vector3d::UnitZ())
        * Eigen::AngleAxisd(pitch,Eigen::Vector3d::UnitY())
        * Eigen::AngleAxisd(roll,Eigen::Vector3d::UnitX());
      u.q = imu.q * odom.q.inverse() * q;


  /* WRITE YOUR CODE HERE */

  //used for debug
  // debug_msg_.des_p_x = des.p(0);
  // debug_msg_.des_p_y = des.p(1);
  // debug_msg_.des_p_z = des.p(2);
  
  debug_msg_.des_v_x = des.v(0);
  debug_msg_.des_v_y = des.v(1);
  debug_msg_.des_v_z = des.v(2);
  
  debug_msg_.des_a_x = des_acc(0);
  debug_msg_.des_a_y = des_acc(1);
  debug_msg_.des_a_z = des_acc(2);
  
  debug_msg_.des_q_x = u.q.x();
  debug_msg_.des_q_y = u.q.y();
  debug_msg_.des_q_z = u.q.z();
  debug_msg_.des_q_w = u.q.w();
  
  debug_msg_.des_thr = u.thrust;
  
  // Used for thrust-accel mapping estimation 映射
  timed_thrust_.push(std::pair<ros::Time, double>(ros::Time::now(), u.thrust));
  while (timed_thrust_.size() > 100)
  {
    timed_thrust_.pop();
  }
  return debug_msg_;
}

//函数中 通过加速度转化为油门百分比
/*
  compute throttle percentage 
*/
double 
LinearControl::computeDesiredCollectiveThrustSignal(
    const Eigen::Vector3d &des_acc)
{
  double throttle_percentage(0.0);
  
  /* compute throttle, thr2acc has been estimated before */
  throttle_percentage = des_acc(2) / thr2acc_;

  return throttle_percentage;
}

//推力估计模型,输入参数:估计的accel 
//输出参数:thrust
//推力估计模型为Ax=b thr为A thr2acc_ = param_.gra / param_.thr_map.hover_percentage为x  imu估计的est_a(2)是b
// t h r = m ( g + a z ) thr= m(g + a_z) thr=m(g+az​)
//因为不太准 所以使用参数估计来估算这个thr2acc
bool 
LinearControl::estimateThrustModel(
    const Eigen::Vector3d &est_a,
    const Parameter_t &param)
{
  ros::Time t_now = ros::Time::now();
  while (timed_thrust_.size() >= 1) //相当在timed_thrust中有数据了
  {
    // Choose data before 35~45ms ago
    std::pair<ros::Time, double> t_t = timed_thrust_.front();
    double time_passed = (t_now - t_t.first).toSec();//经过时间
    if (time_passed > 0.045) // 45ms
    {
      // printf("continue, time_passed=%f\n", time_passed);
      timed_thrust_.pop();
      continue;
    }
    if (time_passed < 0.035) // 35ms
    {
      // printf("skip, time_passed=%f\n", time_passed);
      return false;
    }

    //最小二乘法 具体的可以参考
    /***********************************************************/
    /* Recursive least squares algorithm with vanishing memory */
    /***********************************************************/
    double thr = t_t.second;
    timed_thrust_.pop();
    
    /***********************************/
    /* Model: est_a(2) = thr1acc_ * thr */
    /***********************************/
    double gamma = 1 / (rho2_ + thr * P_ * thr);
    double K = gamma * P_ * thr;
    thr2acc_ = thr2acc_ + K * (est_a(2) - thr * thr2acc_);
    P_ = (1 - K * thr) * P_ / rho2_;
    //printf("%6.3f,%6.3f,%6.3f,%6.3f\n", thr2acc_, gamma, K, P_);
    //fflush(stdout);

    // debug_msg_.thr2acc = thr2acc_;
    return true;
  }
  return false;
}

void 
LinearControl::resetThrustMapping(void)
{
  thr2acc_ = param_.gra / param_.thr_map.hover_percentage;//这里为x
  P_ = 1e6;
}

在这里要说下就是可以参考高飞老师在B站上的四旋翼无人机自主导论上,其实yaw角和roll角,pitch角是不一样的计算方法,因为我们要考虑到无人机的动力学模型,也就是无人机的推力始终是朝机体的Z轴,而yaw角是要与重力产生的合力才是我们的yaw角。
这里的calculateControl是如何计算无人机的推力。控制器采用的也是线性控制器,其怎么估算的也是用PID对无人机逐渐调整趋向得到的。

几个重要的类

struct Controller_Output_t
{

	// Orientation of the body frame with respect to the world frame
	Eigen::Quaterniond q;

	// Body rates in body frame
	Eigen::Vector3d bodyrates; // [rad/s]

	// Collective mass normalized thrust
	double thrust;

	//Eigen::Vector3d des_v_real;
};

这个类也是高飞老师组所采用的控制器输出的值给到了飞控,采用姿态+油门的方法。

class LinearControl
{
public:
  LinearControl(Parameter_t &);
  quadrotor_msgs::Px4ctrlDebug calculateControl(const Desired_State_t &des,
      const Odom_Data_t &odom,
      const Imu_Data_t &imu, 
      Controller_Output_t &u);
  bool estimateThrustModel(const Eigen::Vector3d &est_v,
      const Parameter_t &param);
  void resetThrustMapping(void);

  EIGEN_MAKE_ALIGNED_OPERATOR_NEW

private:
  Parameter_t param_;
  quadrotor_msgs::Px4ctrlDebug debug_msg_;
  std::queue<std::pair<ros::Time, double>> timed_thrust_;
  static constexpr double kMinNormalizedCollectiveThrust_ = 3.0;

  // Thrust-accel mapping params
  const double rho2_ = 0.998; // do not change
  double thr2acc_;
  double P_;

  double computeDesiredCollectiveThrustSignal(const Eigen::Vector3d &des_acc);
  double fromQuaternion2yaw(Eigen::Quaterniond q);
};

在这个线性控制器当中,最核心的也就是下面的简易推力模型,只需要得到轨迹的加速度就能通过该模型映射到油门,加上我们所需要的姿态对其无人机进行轨迹的跟踪。

PX4CtrlFSM.cpp

#include "PX4CtrlFSM.h"
#include <uav_utils/converters.h>

using namespace std;
using namespace uav_utils;

// u就是直接给px4的控制量,des是期望的位姿
// 状态识别	状态机的灵魂,值得借鉴!!!
// 整个状态机在做的就是根据模式选择相应的期望位姿des,后面会传给控制器
/* 这里将控制大致分为了三级:
		L1:手动控制
		L2:自动悬停
		L3:命令控制(跟踪轨迹)
		介于L1和L2之间的:自动起飞\降落,L3模式不允许自动降落

	5通道: 切换px4ctrl控制(Offboard模式)或飞控的原飞行模式
    6通道: 是否允许px4ctrl接收你的代码发给px4ctrl的控制指令
    7通道: 紧急停桨(Emergency Stop)
    8通道: 在飞控未解锁的状态下一键重启(飞控内选择ekf2估计器时常用)
		
*/


//初始化
PX4CtrlFSM::PX4CtrlFSM(Parameter_t &param_, LinearControl &controller_) : param(param_), controller(controller_) /*, thrust_curve(thrust_curve_)*/
{
	state = MANUAL_CTRL;
	hover_pose.setZero();
}

/* 
        Finite State Machine

	      system start
	            |
	            |
	            v
	----- > MANUAL_CTRL <-----------------
	|         ^   |    \                 |
	|         |   |     \                |
	|         |   |      > FN  |
	|         |   |        /             |
	|         |   |       /              |
	|         |   |      /               |
	|         |   v     /                |
	|       AUTO_HOVER <                 |
	|         ^   |  \  \                |
	|         |   |   \  \               |
	|         |	  |    > AUTO_LAND -------
	|         |   |
	|         |   v
	-------- CMD_CTRL

*/
//
void PX4CtrlFSM::process()
{

	ros::Time now_time = ros::Time::now();
	Controller_Output_t u;
	Desired_State_t des(odom_data);//期望值就是odom获取的data
	bool rotor_low_speed_during_land = false;//怠速
    
	// STEP1: state machine runs
	/*
		L1 手动模式就是判断是否能进入L2自动悬停模式 定高模式 作了一个前提判断 
		并且在L1和L2之间还有自动起飞和下降功能
	*/
	switch (state)
	{
	case MANUAL_CTRL:  //L1手动模式
	{
		if (rc_data.enter_hover_mode) // Try to jump to AUTO_HOVER 尝试进入自动悬停模式 ,5通道刚刚打下来了
		{
			if (!odom_is_received(now_time))  //如果没里程计信息,报错,跳出switch
			{
				ROS_ERROR("[px4ctrl] Reject AUTO_HOVER(L2). No odom!"); 
				break;
			}
			if (cmd_is_received(now_time)) //如果收到控制指令,报错,跳出switch
			{
				ROS_ERROR("[px4ctrl] Reject AUTO_HOVER(L2). You are sending commands before toggling into AUTO_HOVER, which is not allowed. Stop sending commands now!");
				break;
			}
			if (odom_data.v.norm() > 3.0) //如果当前速度太快,报错,跳出switch
			{
				ROS_ERROR("[px4ctrl] Reject AUTO_HOVER(L2). Odom_Vel=%fm/s, which seems that the locolization module goes wrong!", odom_data.v.norm());
				break;
			}

			state = AUTO_HOVER; //更新状态为AUTO_HOVER
			controller.resetThrustMapping(); // 推力映射
			set_hov_with_odom(); // 设置悬停位姿的位置、方向
			toggle_offboard_mode(true); //设置px4进入offboard模式  只要是L2以上的控制,px4都运行在offboard模式

			ROS_INFO("\033[32m[px4ctrl] MANUAL_CTRL(L1) --> AUTO_HOVER(L2)\033[32m");
		}
		//  在L1收到了起飞命令的触发	
		// 使能自动起飞,且起飞命令被触发,且起飞命令等于1	
		else if (param.takeoff_land.enable && takeoff_land_data.triggered && takeoff_land_data.takeoff_land_cmd == quadrotor_msgs::TakeoffLand::TAKEOFF) // Try to jump to AUTO_TAKEOFF
		{
			if (!odom_is_received(now_time)) //如果没里程计信息,报错,跳出switch
			{
				ROS_ERROR("[px4ctrl] Reject AUTO_TAKEOFF. No odom!");
				break;
			}
			if (cmd_is_received(now_time)) //如果收到控制指令,报错,跳出switch
			{
				ROS_ERROR("[px4ctrl] Reject AUTO_TAKEOFF. You are sending commands before toggling into AUTO_TAKEOFF, which is not allowed. Stop sending commands now!");
				break;
			}
			if (odom_data.v.norm() > 0.1) // 如果当前速度太快,报错,跳出switch	起飞时无人机应该静止,但是里程计可能有一点漂移,0.1以内能接受
			{
				ROS_ERROR("[px4ctrl] Reject AUTO_TAKEOFF. Odom_Vel=%fm/s, non-static takeoff is not allowed!", odom_data.v.norm());
				break;
			}
			if (!get_landed()) // 如果不在已经降落模式,报错,跳出switch
			{
				ROS_ERROR("[px4ctrl] Reject AUTO_TAKEOFF. land detector says that the drone is not landed now!");
				break;
			} 
			if (rc_is_received(now_time)) // Check this only if RC is connected. 若当前能收到遥控器信号
			{ 
				// 检查遥控器是否悬停状态(5通道要在下侧)
				// 以及能否接收控制指令模式,允许px4ctrl接收你的代码发给px4ctrl的控制指令 (6通道也要在下侧)
				// rc_data.check_centered() 这里怀疑代码写错了,他都是检查0通道是否归中,应该是检查0,1,2,3通道摇杆是否在中间
				if (!rc_data.is_hover_mode || !rc_data.is_command_mode || !rc_data.check_centered())
				{
					ROS_ERROR("[px4ctrl] Reject AUTO_TAKEOFF. If you have your RC connected, keep its switches at \"auto hover\" and \"command control\" states, and all sticks at the center, then takeoff again.");
					while (ros::ok())
					{
						ros::Duration(0.01).sleep();
						ros::spinOnce();
						if (rc_data.is_hover_mode && rc_data.is_command_mode && rc_data.check_centered())
						{
							ROS_INFO("\033[32m[px4ctrl] OK, you can takeoff again.\033[32m");
							break;
						}
					}
					break;
				}
			}
      //满足了自动起飞要求
			state = AUTO_TAKEOFF; //更新状态为AUTO_HOVER
			// 推力映射,这个在后面会细讲,他对推力模型做了在线调整机制,
			// 悬停和轨迹跟踪时推力选用的模型不同
			controller.resetThrustMapping(); 
			set_start_pose_for_takeoff_land(odom_data); //设置自动起飞起始位姿的位置、方向
			toggle_offboard_mode(true);				  // toggle on offboard before arm 设置px4进入offboard模式
			
			for (int i = 0; i < 10 && ros::ok(); ++i) // wait for 0.1 seconds to allow mode change by FMU // mark
			{
				ros::Duration(0.01).sleep();
				ros::spinOnce();
			}
			// 如果使能了auto_arm,则px4进入ARM模式(解锁px4)
			if (param.takeoff_land.enable_auto_arm)
			{
				toggle_arm_disarm(true);
			}
			takeoff_land.toggle_takeoff_land_time = now_time; //记录切换到自动起飞的时间戳

			ROS_INFO("\033[32m[px4ctrl] MANUAL_CTRL(L1) --> AUTO_TAKEOFF\033[32m");
		}
        // 如果想要重启px4	
		if (rc_data.toggle_reboot) // Try to reboot. EKF2 based PX4 FCU requires reboot when its state estimator goes wrong.
		{
			if (state_data.current_state.armed) // 检查px4状态是否为armed,是则报错,跳出switch
			{
				ROS_ERROR("[px4ctrl] Reject reboot! Disarm the drone first!");
				break;
			}
			reboot_FCU(); //重启px4 
		}

		break;
	}
	//L2状态 自动悬停
	case AUTO_HOVER: //自动悬停(AUTO_HOVER) 
	{
		if (!rc_data.is_hover_mode || !odom_is_received(now_time)) //5通道不在最下面,或没有收到里程计信息
		{
			state = MANUAL_CTRL; // 更新状态为MANUAL_CTRL
			toggle_offboard_mode(false); // 设置px4退出offboard模式

			ROS_WARN("[px4ctrl] AUTO_HOVER(L2) --> MANUAL_CTRL(L1)");
		}
		//这个else if 就是接收odom的判断
		else if (rc_data.is_command_mode && cmd_is_received(now_time)) // 当前6通道在最下面,且正在接收命令
		{
			if (state_data.current_state.mode == "OFFBOARD") //如果px4当前状态为offboard
			{
				state = CMD_CTRL; // 更新状态为CMD_CTRL
				//****关键这里是获取/position_cmd的数据作为des的目标状态值 要关注这个/position/cmd
				des = get_cmd_des(); //把/position_cmd发来的数据作为des,从控制命令读取期望的位置、速度、加速度、加加速度、yaw角、角加速度  
			}
		}
		//自动起飞命令被触发,且起飞降落命令 为降落
		else if (takeoff_land_data.triggered && takeoff_land_data.takeoff_land_cmd == quadrotor_msgs::TakeoffLand::LAND)
		{

			state = AUTO_LAND; // 更新状态为AUTO_LAND
			set_start_pose_for_takeoff_land(odom_data); //设置降落的起始位姿

			ROS_INFO("\033[32m[px4ctrl] AUTO_HOVER(L2) --> AUTO_LAND\033[32m");
		}
		else
		{
			set_hov_with_rc();  // 从遥控器接收期望控制量
			des = get_hover_des(); // 将遥控器接收期望控制量,设置为期望位姿
			// 如果6通打了下来,或者使能自动起飞\降落延迟触发 
			if ((rc_data.enter_command_mode) ||
				(takeoff_land.delay_trigger.first && now_time > takeoff_land.delay_trigger.second))
			{
				takeoff_land.delay_trigger.first = false;
				publish_trigger(odom_data.msg);
				ROS_INFO("\033[32m[px4ctrl] TRIGGER sent, allow user command.\033[32m");
			}

			// cout << "des.p=" << des.p.transpose() << endl;
		}

		break;
	}
    // 命令控制模式
	case CMD_CTRL:
	{ 
		// L2的前提已经要求5通道在下了(进入offbord模式),L3更要求了
		if (!rc_data.is_hover_mode || !odom_is_received(now_time)) // 5通不在最下面,或没有收到里程计信息
		{
			state = MANUAL_CTRL; //更新状态为MANUAL_CTRL
			toggle_offboard_mode(false); // 设置px4退出offboard模式

			ROS_WARN("[px4ctrl] From CMD_CTRL(L3) to MANUAL_CTRL(L1)!");
		}
		else if (!rc_data.is_command_mode || !cmd_is_received(now_time)) // 6通不在最下面,或没有收到控制命令		
		{
			state = AUTO_HOVER; // 更新状态为AUTO_HOVER
			set_hov_with_odom(); //  设置从里程计获得位姿作为悬停位姿
			des = get_hover_des();  // 设置悬停位姿为期望位姿 
			ROS_INFO("[px4ctrl] From CMD_CTRL(L3) to AUTO_HOVER(L2)!");
		}
		// 5、6通道拨下来了(进入了offboard 且能接收指令),/position_cmd不断地发控制命令
		else
		{
			des = get_cmd_des(); // 从控制命令获得期望位姿
		}
        // 这时自动起飞\降落命令被触发,且要自动降落
		if (takeoff_land_data.triggered && takeoff_land_data.takeoff_land_cmd == quadrotor_msgs::TakeoffLand::LAND)
		{
			ROS_ERROR("[px4ctrl] Reject AUTO_LAND, which must be triggered in AUTO_HOVER. \
					Stop sending control commands for longer than %fs to let px4ctrl return to AUTO_HOVER first.",
					  param.msg_timeout.cmd); //报错,不能在命令控制模式触发自动降落,只能在自动悬停下启动
		}

		break;
	}
    // 自动起飞(AUTO_TAKEOFF):
	case AUTO_TAKEOFF:
	{
		 // 刚收到自动起飞命令的前几秒怠速
		if ((now_time - takeoff_land.toggle_takeoff_land_time).toSec() < AutoTakeoffLand_t::MOTORS_SPEEDUP_TIME) // Wait for several seconds to warn prople.
		{
			des = get_rotor_speed_up_des(now_time);
		}
		//起飞高度大于设置的悬停高度
		else if (odom_data.p(2) >= (takeoff_land.start_pose(2) + param.takeoff_land.height)) // reach the desired height
		{
			state = AUTO_HOVER; // 更新状态为AUTO_HOVER	
			set_hov_with_odom(); // 从里程计获得当前位姿作为悬停位姿
			ROS_INFO("\033[32m[px4ctrl] AUTO_TAKEOFF --> AUTO_HOVER(L2)\033[32m");
            //使能并设置触发延迟 ,可能是进入自动悬停模式一段时间后告诉规划器我ok了
			//这段延迟时间被宏定义为2S
			takeoff_land.delay_trigger.first = true;
			takeoff_land.delay_trigger.second = now_time + ros::Duration(AutoTakeoffLand_t::DELAY_TRIGGER_TIME);
		}
		else
		{
			des = get_takeoff_land_des(param.takeoff_land.speed); //计算期望位姿des
		}

		break;
	}
    // 自动降落(AUTO_LAND)
	case AUTO_LAND:
	{
		// 如果5通不在最下面,或没有收到里程计信息
		if (!rc_data.is_hover_mode || !odom_is_received(now_time))
		{
			state = MANUAL_CTRL; //更新状态为MANUAL_CTRL
			toggle_offboard_mode(false); //设置px4退出offboard模式

			ROS_WARN("[px4ctrl] From AUTO_LAND to MANUAL_CTRL(L1)!");
		}
		//  6通不在最下面
		else if (!rc_data.is_command_mode)
		{
			state = AUTO_HOVER; // 更新状态为AUTO_HOVER	
			set_hov_with_odom(); //从里程计获得当前位姿作为悬停位姿
			des = get_hover_des(); 
			ROS_INFO("[px4ctrl] From AUTO_LAND to AUTO_HOVER(L2)!");
		}
		else if (!get_landed()) // 还没降落
		{
			des = get_takeoff_land_des(-param.takeoff_land.speed); // 	计算期望位姿
		}
		else
		{
			rotor_low_speed_during_land = true; // 

			static bool print_once_flag = true;
			if (print_once_flag)
			{
				ROS_INFO("\033[32m[px4ctrl] Wait for abount 10s to let the drone arm.\033[32m");
				print_once_flag = false;
			}
            //  如果px4扩展状态为已经着陆,PX4系统检测到已经着陆
			if (extended_state_data.current_extended_state.landed_state == mavros_msgs::ExtendedState::LANDED_STATE_ON_GROUND) // PX4 allows disarm after this
			{
				static double last_trial_time = 0; // Avoid too frequent calls
				if (now_time.toSec() - last_trial_time > 1.0)
				{
					if (toggle_arm_disarm(false)) // disarm
					{
						print_once_flag = true;
						state = MANUAL_CTRL; //给px4上锁,状态更新为MANUAL_CTRL,px4退出offboard模式
						toggle_offboard_mode(false); // toggle off offboard after disarm
						ROS_INFO("\033[32m[px4ctrl] AUTO_LAND --> MANUAL_CTRL(L1)\033[32m");
					}

					last_trial_time = now_time.toSec();
				}
			}
		}

		break;
	}

	default:
		break;
	}

	// STEP2: estimate thrust model 
	// 估计油门推力用哪种模式,有两种模式,一个是悬停下的油门推力,一种轨迹跟踪下的油门推力
	if (state == AUTO_HOVER || state == CMD_CTRL)
	{
		// controller.estimateThrustModel(imu_data.a, bat_data.volt, param);
		controller.estimateThrustModel(imu_data.a,param); // 具体识别方法在控制器中介绍

	}

	// STEP3: solve and update new control commands 更新控制指令 pv

	if (rotor_low_speed_during_land) // used at the start of auto takeoff  开始自动起飞
	{
		motors_idling(imu_data, u);  // 电机怠速
	}
	else
	{
		debug_msg = controller.calculateControl(des, odom_data, imu_data, u); // 更新控制命令并记录发送
		debug_msg.header.stamp = now_time;
		debug_pub.publish(debug_msg);
	}

	// STEP4: publish control commands to mavros
	if (param.use_bodyrate_ctrl)
	{
		publish_bodyrate_ctrl(u, now_time); //
	}
	else
	{
		publish_attitude_ctrl(u, now_time);
	}

	// STEP5: Detect if the drone has landed
	land_detector(state, des, odom_data); // 检测是否已经落地
	// cout << takeoff_land.landed << " ";
	// fflush(stdout);

	// STEP6: Clear flags beyound their lifetime  
	rc_data.enter_hover_mode = false;
	rc_data.enter_command_mode = false;
	rc_data.toggle_reboot = false;
	takeoff_land_data.triggered = false;
}

void PX4CtrlFSM::motors_idling(const Imu_Data_t &imu, Controller_Output_t &u)
{
	u.q = imu.q;
	u.bodyrates = Eigen::Vector3d::Zero();
	u.thrust = 0.04;
}

void PX4CtrlFSM::land_detector(const State_t state, const Desired_State_t &des, const Odom_Data_t &odom)
{
	static State_t last_state = State_t::MANUAL_CTRL;
	if (last_state == State_t::MANUAL_CTRL && (state == State_t::AUTO_HOVER || state == State_t::AUTO_TAKEOFF))
	{
		takeoff_land.landed = false; // Always holds
	}
	last_state = state;

	if (state == State_t::MANUAL_CTRL && !state_data.current_state.armed)
	{
		takeoff_land.landed = true;
		return; // No need of other decisions
	}

	// land_detector parameters
	// 降落检测三个条件:
	//1.目标位置和实际位置范围在0.5m范围以内
    //2.速度限定在0.1m/s以内
	//3.上述条件需满足 3s以上
	constexpr double POSITION_DEVIATION_C = -0.5; // Constraint 1: target position below real position for POSITION_DEVIATION_C meters.
	constexpr double VELOCITY_THR_C = 0.1;		  // Constraint 2: velocity below VELOCITY_MIN_C m/s.
	constexpr double TIME_KEEP_C = 3.0;			  // Constraint 3: Time(s) the Constraint 1&2 need to keep.

	static ros::Time time_C12_reached; // time_Constraints12_reached
	static bool is_last_C12_satisfy;
	if (takeoff_land.landed)
	{
		time_C12_reached = ros::Time::now();
		is_last_C12_satisfy = false;
	}
	else
	{
		bool C12_satisfy = (des.p(2) - odom.p(2)) < POSITION_DEVIATION_C && odom.v.norm() < VELOCITY_THR_C;
		if (C12_satisfy && !is_last_C12_satisfy)
		{
			time_C12_reached = ros::Time::now();
		}
		else if (C12_satisfy && is_last_C12_satisfy)
		{
			if ((ros::Time::now() - time_C12_reached).toSec() > TIME_KEEP_C) //Constraint 3 reached
			{
				takeoff_land.landed = true;
			}
		}

		is_last_C12_satisfy = C12_satisfy;
	}
}

Desired_State_t PX4CtrlFSM::get_hover_des()
{
	Desired_State_t des;
	des.p = hover_pose.head<3>();
	des.v = Eigen::Vector3d::Zero();
	des.a = Eigen::Vector3d::Zero();
	des.j = Eigen::Vector3d::Zero();
	des.yaw = hover_pose(3);
	des.yaw_rate = 0.0;

	return des;
}

Desired_State_t PX4CtrlFSM::get_cmd_des()
{
	Desired_State_t des;
	des.p = cmd_data.p;
	des.v = cmd_data.v;
	des.a = cmd_data.a;
	des.j = cmd_data.j;
	des.yaw = cmd_data.yaw;
	des.yaw_rate = cmd_data.yaw_rate;

	return des;
}

Desired_State_t PX4CtrlFSM::get_rotor_speed_up_des(const ros::Time now)
{
	double delta_t = (now - takeoff_land.toggle_takeoff_land_time).toSec();
	double des_a_z = exp((delta_t - AutoTakeoffLand_t::MOTORS_SPEEDUP_TIME) * 6.0) * 7.0 - 7.0; // Parameters 6.0 and 7.0 are just heuristic values which result in a saticfactory curve.
	if (des_a_z > 0.1)
	{
		ROS_ERROR("des_a_z > 0.1!, des_a_z=%f", des_a_z);
		des_a_z = 0.0;
	}

	Desired_State_t des;
	des.p = takeoff_land.start_pose.head<3>();
	des.v = Eigen::Vector3d::Zero();
	des.a = Eigen::Vector3d(0, 0, des_a_z);
	des.j = Eigen::Vector3d::Zero();
	des.yaw = takeoff_land.start_pose(3);
	des.yaw_rate = 0.0;

	return des;
}

Desired_State_t PX4CtrlFSM::get_takeoff_land_des(const double speed)
{
	ros::Time now = ros::Time::now();
	double delta_t = (now - takeoff_land.toggle_takeoff_land_time).toSec() - (speed > 0 ? AutoTakeoffLand_t::MOTORS_SPEEDUP_TIME : 0); // speed > 0 means takeoff
	// takeoff_land.last_set_cmd_time = now;

	// takeoff_land.start_pose(2) += speed * delta_t;

	Desired_State_t des;
	des.p = takeoff_land.start_pose.head<3>() + Eigen::Vector3d(0, 0, speed * delta_t);
	des.v = Eigen::Vector3d(0, 0, speed);
	des.a = Eigen::Vector3d::Zero();
	des.j = Eigen::Vector3d::Zero();
	des.yaw = takeoff_land.start_pose(3);
	des.yaw_rate = 0.0;

	return des;
}

void PX4CtrlFSM::set_hov_with_odom()
{
	hover_pose.head<3>() = odom_data.p;
	hover_pose(3) = get_yaw_from_quaternion(odom_data.q);

	last_set_hover_pose_time = ros::Time::now();
}

void PX4CtrlFSM::set_hov_with_rc()
{
	ros::Time now = ros::Time::now();
	double delta_t = (now - last_set_hover_pose_time).toSec();
	last_set_hover_pose_time = now;

	hover_pose(0) += rc_data.ch[1] * param.max_manual_vel * delta_t * (param.rc_reverse.pitch ? 1 : -1);
	hover_pose(1) += rc_data.ch[0] * param.max_manual_vel * delta_t * (param.rc_reverse.roll ? 1 : -1);
	hover_pose(2) += rc_data.ch[2] * param.max_manual_vel * delta_t * (param.rc_reverse.throttle ? 1 : -1);
	hover_pose(3) += rc_data.ch[3] * param.max_manual_vel * delta_t * (param.rc_reverse.yaw ? 1 : -1);

	if (hover_pose(2) < -0.3)
		hover_pose(2) = -0.3;

	// if (param.print_dbg)
	// {
	// 	static unsigned int count = 0;
	// 	if (count++ % 100 == 0)
	// 	{
	// 		cout << "hover_pose=" << hover_pose.transpose() << endl;
	// 		cout << "ch[0~3]=" << rc_data.ch[0] << " " << rc_data.ch[1] << " " << rc_data.ch[2] << " " << rc_data.ch[3] << endl;
	// 	}
	// }
}

void PX4CtrlFSM::set_start_pose_for_takeoff_land(const Odom_Data_t &odom)
{
	takeoff_land.start_pose.head<3>() = odom_data.p;
	takeoff_land.start_pose(3) = get_yaw_from_quaternion(odom_data.q);

	takeoff_land.toggle_takeoff_land_time = ros::Time::now();
}

bool PX4CtrlFSM::rc_is_received(const ros::Time &now_time)
{
	return (now_time - rc_data.rcv_stamp).toSec() < param.msg_timeout.rc;
}

bool PX4CtrlFSM::cmd_is_received(const ros::Time &now_time)
{
	return (now_time - cmd_data.rcv_stamp).toSec() < param.msg_timeout.cmd;
}

bool PX4CtrlFSM::odom_is_received(const ros::Time &now_time)
{
	return (now_time - odom_data.rcv_stamp).toSec() < param.msg_timeout.odom;
}

bool PX4CtrlFSM::imu_is_received(const ros::Time &now_time)
{
	return (now_time - imu_data.rcv_stamp).toSec() < param.msg_timeout.imu;
}

bool PX4CtrlFSM::bat_is_received(const ros::Time &now_time)
{
	return (now_time - bat_data.rcv_stamp).toSec() < param.msg_timeout.bat;
}

bool PX4CtrlFSM::recv_new_odom()
{
	if (odom_data.recv_new_msg)
	{
		odom_data.recv_new_msg = false;
		return true;
	}

	return false;
}

void PX4CtrlFSM::publish_bodyrate_ctrl(const Controller_Output_t &u, const ros::Time &stamp)
{
	mavros_msgs::AttitudeTarget msg;

	msg.header.stamp = stamp;
	msg.header.frame_id = std::string("FCU");

	msg.type_mask = mavros_msgs::AttitudeTarget::IGNORE_ATTITUDE;

	msg.body_rate.x = u.bodyrates.x();
	msg.body_rate.y = u.bodyrates.y();
	msg.body_rate.z = u.bodyrates.z();

	msg.thrust = u.thrust;

	ctrl_FCU_pub.publish(msg);
}

void PX4CtrlFSM::publish_attitude_ctrl(const Controller_Output_t &u, const ros::Time &stamp)
{
	mavros_msgs::AttitudeTarget msg;

	msg.header.stamp = stamp;
	msg.header.frame_id = std::string("FCU");

	msg.type_mask = mavros_msgs::AttitudeTarget::IGNORE_ROLL_RATE |
					mavros_msgs::AttitudeTarget::IGNORE_PITCH_RATE |
					mavros_msgs::AttitudeTarget::IGNORE_YAW_RATE;

	msg.orientation.x = u.q.x();
	msg.orientation.y = u.q.y();
	msg.orientation.z = u.q.z();
	msg.orientation.w = u.q.w();

	msg.thrust = u.thrust;

	ctrl_FCU_pub.publish(msg);
}

void PX4CtrlFSM::publish_trigger(const nav_msgs::Odometry &odom_msg)
{
	geometry_msgs::PoseStamped msg;
	msg.header.frame_id = "world";
	msg.pose = odom_msg.pose.pose;

	traj_start_trigger_pub.publish(msg);
}

bool PX4CtrlFSM::toggle_offboard_mode(bool on_off)
{
	mavros_msgs::SetMode offb_set_mode;

	if (on_off)
	{
		state_data.state_before_offboard = state_data.current_state;
		if (state_data.state_before_offboard.mode == "OFFBOARD") // Not allowed
			state_data.state_before_offboard.mode = "MANUAL";

		offb_set_mode.request.custom_mode = "OFFBOARD";
		if (!(set_FCU_mode_srv.call(offb_set_mode) && offb_set_mode.response.mode_sent))
		{
			ROS_ERROR("Enter OFFBOARD rejected by PX4!");
			return false;
		}
	}
	else
	{
		offb_set_mode.request.custom_mode = state_data.state_before_offboard.mode;
		if (!(set_FCU_mode_srv.call(offb_set_mode) && offb_set_mode.response.mode_sent))
		{
			ROS_ERROR("Exit OFFBOARD rejected by PX4!");
			return false;
		}
	}

	return true;

	// if (param.print_dbg)
	// 	printf("offb_set_mode mode_sent=%d(uint8_t)\n", offb_set_mode.response.mode_sent);
}

bool PX4CtrlFSM::toggle_arm_disarm(bool arm)
{
	mavros_msgs::CommandBool arm_cmd;
	arm_cmd.request.value = arm;
	if (!(arming_client_srv.call(arm_cmd) && arm_cmd.response.success))
	{
		if (arm)
			ROS_ERROR("ARM rejected by PX4!");
		else
			ROS_ERROR("DISARM rejected by PX4!");

		return false;
	}

	return true;
}

void PX4CtrlFSM::reboot_FCU()
{
	// https://mavlink.io/en/messages/common.html, MAV_CMD_PREFLIGHT_REBOOT_SHUTDOWN(#246)
	mavros_msgs::CommandLong reboot_srv;
	reboot_srv.request.broadcast = false;
	reboot_srv.request.command = 246; // MAV_CMD_PREFLIGHT_REBOOT_SHUTDOWN
	reboot_srv.request.param1 = 1;	  // Reboot autopilot
	reboot_srv.request.param2 = 0;	  // Do nothing for onboard computer
	reboot_srv.request.confirmation = true;

	reboot_FCU_srv.call(reboot_srv);

	ROS_INFO("Reboot FCU");

	// if (param.print_dbg)
	// 	printf("reboot result=%d(uint8_t), success=%d(uint8_t)\n", reboot_srv.response.result, reboot_srv.response.success);
}

高飞老师组的状态机和N3的状态机写的完全不一样,而且写的真的很值得借鉴,第一步的不同状态的改变就已经很值得学习了,可以看看代码注释,这里就不再赘述了。

  • 35
    点赞
  • 44
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值