JointController.hh

Go to the documentation of this file.

/*
 * Copyright 2012 Nate Koenig & Andrew Howard
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
*/
#ifndef _JOINTCONTROLLER_HH_
#define _JOINTCONTROLLER_HH_

#include <map>
#include <string>
#include <vector>

#include "gazebo/common/PID.hh"
#include "gazebo/common/Time.hh"
#include "gazebo/physics/PhysicsTypes.hh"
#include "gazebo/transport/TransportTypes.hh"
#include "gazebo/msgs/msgs.hh"

namespace gazebo
{
  namespace physics
  {

    class JointController
    {
      public: explicit JointController(ModelPtr _model);

      public: void AddJoint(JointPtr _joint);

      public: void Update();

      public: void Reset();

      public: void SetJointPosition(const std::string &_name, double _position);

      public: void SetJointPositions(
                  const std::map<std::string, double> &_jointPositions);

      private: void OnJointCmd(ConstJointCmdPtr &_msg);

      public: void SetJointPosition(JointPtr _joint, double _position);

      private: void MoveLinks(JointPtr _joint, LinkPtr _link,
                   const math::Vector3 &_anchor, const math::Vector3 &_axis,
                   double _dposition, bool _updateChildren = false);

      private: void ComputeAndSetLinkTwist(LinkPtr _link,
                    const math::Pose &_old, const math::Pose &_new, double dt);

      private: void AddConnectedLinks(std::vector<LinkPtr> &_linksOut,
                                      const LinkPtr &_link,
                                      bool _checkParentTree = false);

      private: template<class InputVector, class T>
                 bool ContainsLink(InputVector _vector, const T &_value)
                 {
                   typename InputVector::iterator iter = _vector.begin();
                   for (; iter != _vector.end(); ++iter)
                     if ((*iter)->GetScopedName() == _value->GetScopedName())
                       return true;
                   return false;
                 }

      private: ModelPtr model;

      private: Link_V updatedLinks;

      private: std::map<std::string, JointPtr> joints;

      private: std::map<std::string, common::PID> posPids;

      private: std::map<std::string, common::PID> velPids;

      private: std::map<std::string, double> forces;

      private: std::map<std::string, double> positions;

      private: std::map<std::string, double> velocities;

      private: transport::NodePtr node;

      private: transport::SubscriberPtr jointCmdSub;

      private: common::Time prevUpdateTime;
    };
  }
}
#endif