dynamics2d_velocity_control.cpp

Go to the documentation of this file.

#include "dynamics2d_velocity_control.h"
#include <argos3/plugins/simulator/physics_engines/dynamics2d/dynamics2d_engine.h>

namespace argos {

   /****************************************/
   /****************************************/

   CDynamics2DVelocityControl::CDynamics2DVelocityControl(CDynamics2DEngine& c_engine,
                                                          Real f_max_force,
                                                          Real f_max_torque,
                                                          TConfigurationNode* t_node) :
      m_cDyn2DEngine(c_engine),
      m_ptControlBody(NULL),
      m_ptControlledBody(NULL),
      m_ptLinearConstraint(NULL),
      m_ptAngularConstraint(NULL),
      m_fMaxForce(f_max_force),
      m_fMaxTorque(f_max_torque) {
      m_ptControlBody = cpBodyNew(INFINITY, INFINITY);
      if(t_node &&
         NodeExists(*t_node, "dynamics2d")) {
         TConfigurationNode& tNode = GetNode(*t_node, "dynamics2d");
         if(NodeExists(tNode, "differential_steering")) {
            TConfigurationNode& tDiffSteer = GetNode(tNode, "differential_steering");
            GetNodeAttributeOrDefault(tDiffSteer, "max_force", m_fMaxForce, m_fMaxForce);
            GetNodeAttributeOrDefault(tDiffSteer, "max_torque", m_fMaxTorque, m_fMaxTorque);
         }
      }
   }

   /****************************************/
   /****************************************/

   CDynamics2DVelocityControl::~CDynamics2DVelocityControl() {
      cpBodyFree(m_ptControlBody);
   }

   /****************************************/
   /****************************************/

   void CDynamics2DVelocityControl::AttachTo(cpBody* pt_body) {
      /* If we are already controlling a body, detach from it first */
      if(m_ptControlledBody != NULL) {
         Detach();
      }
      /* Set the wanted body as the new controlled one */
      m_ptControlledBody = pt_body;
      /* Add linear constraint */
      m_ptLinearConstraint =
         cpSpaceAddConstraint(m_cDyn2DEngine.GetPhysicsSpace(),
                              cpPivotJointNew2(m_ptControlledBody,
                                               m_ptControlBody,
                                               cpvzero,
                                               cpvzero));
      m_ptLinearConstraint->maxBias = 0.0f; /* disable joint correction */
      m_ptLinearConstraint->maxForce = m_fMaxForce; /* limit the dragging force */
      /* Add angular constraint */
      m_ptAngularConstraint =
         cpSpaceAddConstraint(m_cDyn2DEngine.GetPhysicsSpace(),
                              cpGearJointNew(m_ptControlledBody,
                                             m_ptControlBody,
                                             0.0f,
                                             1.0f));
      m_ptAngularConstraint->maxBias = 0.0f; /* disable joint correction */
      m_ptAngularConstraint->maxForce = m_fMaxTorque; /* limit the torque */
   }

   /****************************************/
   /****************************************/

   void CDynamics2DVelocityControl::Detach() {
      if(m_ptControlledBody != NULL) {
         /* Remove constraints */
         cpSpaceRemoveConstraint(m_cDyn2DEngine.GetPhysicsSpace(), m_ptLinearConstraint);
         cpSpaceRemoveConstraint(m_cDyn2DEngine.GetPhysicsSpace(), m_ptAngularConstraint);
         /* Free memory up */
         cpConstraintFree(m_ptLinearConstraint);
         cpConstraintFree(m_ptAngularConstraint);
         /* Erase pointer to controlled body */
         m_ptControlledBody = NULL;
      }
   }

   /****************************************/
   /****************************************/

   void CDynamics2DVelocityControl::Reset() {
      m_ptControlBody->v.x = 0;
      m_ptControlBody->v.y = 0;
      m_ptControlBody->w = 0;
   }

   /****************************************/
   /****************************************/

   CVector2 CDynamics2DVelocityControl::GetLinearVelocity() const {
      return CVector2(m_ptControlledBody->v.x,
                      m_ptControlledBody->v.y);
   }

   /****************************************/
   /****************************************/

   void CDynamics2DVelocityControl::SetLinearVelocity(const CVector2& c_velocity) {
      m_ptControlBody->v.x = c_velocity.GetX();
      m_ptControlBody->v.y = c_velocity.GetY();
   }

   /****************************************/
   /****************************************/

   Real CDynamics2DVelocityControl::GetAngularVelocity() const {
      return m_ptControlBody->w;
   }

   /****************************************/
   /****************************************/

   void CDynamics2DVelocityControl::SetAngularVelocity(Real f_velocity) {
      m_ptControlBody->w = f_velocity;
   }

   /****************************************/
   /****************************************/

}