VehicleAckermann.h

Go to the documentation of this file.

/*+-------------------------------------------------------------------------+
  |                       MultiVehicle simulator (libmvsim)                 |
  |                                                                         |
  | Copyright (C) 2014-2020  Jose Luis Blanco Claraco                       |
  | Copyright (C) 2017  Borys Tymchenko (Odessa Polytechnic University)     |
  | Distributed under 3-clause BSD License                                  |
  |   See COPYING                                                           |
  +-------------------------------------------------------------------------+ */

#pragma once

#include <mrpt/img/TColor.h>
#include <mvsim/PID_Controller.h>
#include <mvsim/VehicleBase.h>

namespace mvsim
{
class DynamicsAckermann : public VehicleBase
{
        DECLARES_REGISTER_VEHICLE_DYNAMICS(DynamicsAckermann)
   public:
        // Wheels: [0]:rear-left, [1]:rear-right, [2]: front-left, [3]: front-right
        enum
        {
                WHEEL_RL = 0,
                WHEEL_RR = 1,
                WHEEL_FL = 2,
                WHEEL_FR = 3
        };

        DynamicsAckermann(World* parent);

        double getMaxSteeringAngle() const { return m_max_steer_ang; }
        void setMaxSteeringAngle(double val) { m_max_steer_ang = val; }
        struct TControllerInput
        {
                TSimulContext context;
        };
        struct TControllerOutput
        {
                double fl_torque, fr_torque, rl_torque, rr_torque;
                double steer_ang;  
                TControllerOutput()
                        : fl_torque(0),
                          fr_torque(0),
                          rl_torque(0),
                          rr_torque(0),
                          steer_ang(0)
                {
                }
        };

        typedef ControllerBaseTempl<DynamicsAckermann> ControllerBase;
        typedef std::shared_ptr<ControllerBase> ControllerBasePtr;

        class ControllerRawForces : public ControllerBase
        {
           public:
                ControllerRawForces(DynamicsAckermann& veh);
                static const char* class_name() { return "raw"; }
                double setpoint_wheel_torque_l, setpoint_wheel_torque_r,
                        setpoint_steer_ang;
                virtual void control_step(
                        const DynamicsAckermann::TControllerInput& ci,
                        DynamicsAckermann::TControllerOutput& co) override;
                virtual void load_config(const rapidxml::xml_node<char>& node) override;
                virtual void teleop_interface(
                        const TeleopInput& in, TeleopOutput& out) override;
        };

        class ControllerTwistFrontSteerPID : public ControllerBase
        {
           public:
                ControllerTwistFrontSteerPID(DynamicsAckermann& veh);
                static const char* class_name() { return "twist_front_steer_pid"; }
                double setpoint_lin_speed,
                        setpoint_ang_speed;      
                virtual void control_step(
                        const DynamicsAckermann::TControllerInput& ci,
                        DynamicsAckermann::TControllerOutput& co) override;
                virtual void load_config(const rapidxml::xml_node<char>& node) override;
                virtual void teleop_interface(
                        const TeleopInput& in, TeleopOutput& out) override;

                double KP, KI, KD;      
                double max_torque;      

                // See base docs.
                virtual bool setTwistCommand(const double vx, const double wz) override
                {
                        setpoint_lin_speed = vx;
                        setpoint_ang_speed = wz;
                        return true;
                }

           private:
                double m_dist_fWheels, m_r2f_L;
                PID_Controller m_PID[2];  //<! [0]:fl, [1]: fr
        };

        class ControllerFrontSteerPID : public ControllerBase
        {
           public:
                ControllerFrontSteerPID(DynamicsAckermann& veh);
                static const char* class_name() { return "front_steer_pid"; }
                double setpoint_lin_speed, setpoint_steer_ang;  
                virtual void control_step(
                        const DynamicsAckermann::TControllerInput& ci,
                        DynamicsAckermann::TControllerOutput& co) override;
                virtual void load_config(const rapidxml::xml_node<char>& node) override;
                virtual void teleop_interface(
                        const TeleopInput& in, TeleopOutput& out) override;

                double KP, KI, KD;      
                double max_torque;      
           private:
                ControllerTwistFrontSteerPID m_twist_control;
                double m_r2f_L;
        };

        const ControllerBasePtr& getController() const { return m_controller; }
        ControllerBasePtr& getController() { return m_controller; }
        virtual ControllerBaseInterface* getControllerInterface() override
        {
                return m_controller.get();
        }
  // end controllers

        virtual mrpt::math::TTwist2D getVelocityLocalOdoEstimate() const override;

        void computeFrontWheelAngles(
                const double desired_equiv_steer_ang, double& out_fl_ang,
                double& out_fr_ang) const;

   protected:
        // See base class docs
        virtual void dynamics_load_params_from_xml(
                const rapidxml::xml_node<char>* xml_node) override;
        // See base class doc
        virtual void invoke_motor_controllers(
                const TSimulContext& context,
                std::vector<double>& out_force_per_wheel) override;

   private:
        ControllerBasePtr m_controller;  

        double m_max_steer_ang = mrpt::DEG2RAD(30);
};
}  // namespace mvsim