contacts/contact-6d.hpp

Go to the documentation of this file.

//
// Copyright (c) 2017 CNRS, NYU, MPI Tübingen
//
// This file is part of tsid
// tsid is free software: you can redistribute it
// and/or modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation, either version
// 3 of the License, or (at your option) any later version.
// tsid is distributed in the hope that it will be
// useful, but WITHOUT ANY WARRANTY; without even the implied warranty
// of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// General Lesser Public License for more details. You should have
// received a copy of the GNU Lesser General Public License along with
// tsid If not, see
// <http://www.gnu.org/licenses/>.
//

#ifndef __invdyn_contact_6d_hpp__
#define __invdyn_contact_6d_hpp__

#include "tsid/deprecated.hh"
#include "tsid/contacts/contact-base.hpp"
#include "tsid/tasks/task-se3-equality.hpp"
#include "tsid/math/constraint-inequality.hpp"
#include "tsid/math/constraint-equality.hpp"

namespace tsid {
namespace contacts {
class Contact6d : public ContactBase {
 public:
  EIGEN_MAKE_ALIGNED_OPERATOR_NEW

  typedef math::ConstRefMatrix ConstRefMatrix;
  typedef math::ConstRefVector ConstRefVector;
  typedef math::Matrix3x Matrix3x;
  typedef math::Vector6 Vector6;
  typedef math::Vector3 Vector3;
  typedef math::Vector Vector;
  typedef tasks::TaskSE3Equality TaskSE3Equality;
  typedef math::ConstraintInequality ConstraintInequality;
  typedef math::ConstraintEquality ConstraintEquality;
  typedef pinocchio::SE3 SE3;

  Contact6d(const std::string& name, RobotWrapper& robot,
            const std::string& frameName, ConstRefMatrix contactPoints,
            ConstRefVector contactNormal, const double frictionCoefficient,
            const double minNormalForce, const double maxNormalForce);

  TSID_DEPRECATED Contact6d(const std::string& name, RobotWrapper& robot,
                            const std::string& frameName,
                            ConstRefMatrix contactPoints,
                            ConstRefVector contactNormal,
                            const double frictionCoefficient,
                            const double minNormalForce,
                            const double maxNormalForce,
                            const double forceRegWeight);

  virtual ~Contact6d() {}

  virtual unsigned int n_motion() const;

  virtual unsigned int n_force() const;

  virtual const ConstraintBase& computeMotionTask(const double t,
                                                  ConstRefVector q,
                                                  ConstRefVector v, Data& data);

  virtual const ConstraintInequality& computeForceTask(const double t,
                                                       ConstRefVector q,
                                                       ConstRefVector v,
                                                       const Data& data);

  virtual const Matrix& getForceGeneratorMatrix();

  virtual const ConstraintEquality& computeForceRegularizationTask(
      const double t, ConstRefVector q, ConstRefVector v, const Data& data);

  const TaskSE3Equality& getMotionTask() const;
  const ConstraintBase& getMotionConstraint() const;
  const ConstraintInequality& getForceConstraint() const;
  const ConstraintEquality& getForceRegularizationTask() const;

  double getNormalForce(ConstRefVector f) const;
  double getMinNormalForce() const;
  double getMaxNormalForce() const;
  const Matrix3x& getContactPoints() const;

  const Vector& Kp() const;
  const Vector& Kd() const;
  void Kp(ConstRefVector Kp);
  void Kd(ConstRefVector Kp);

  bool setContactPoints(ConstRefMatrix contactPoints);
  bool setContactNormal(ConstRefVector contactNormal);

  bool setFrictionCoefficient(const double frictionCoefficient);
  bool setMinNormalForce(const double minNormalForce);
  bool setMaxNormalForce(const double maxNormalForce);
  void setReference(const SE3& ref);
  void setForceReference(ConstRefVector& f_ref);
  void setRegularizationTaskWeightVector(ConstRefVector& w);

 private:
  void init();

 protected:
  void updateForceInequalityConstraints();
  void updateForceRegularizationTask();
  void updateForceGeneratorMatrix();

  TaskSE3Equality m_motionTask;
  ConstraintInequality m_forceInequality;
  ConstraintEquality m_forceRegTask;
  Matrix3x m_contactPoints;
  Vector3 m_contactNormal;
  Vector6 m_fRef;
  Vector6 m_weightForceRegTask;
  double m_mu;
  double m_fMin;
  double m_fMax;
  Matrix m_forceGenMat;
};
}  // namespace contacts
}  // namespace tsid

#endif  // ifndef __invdyn_contact_6d_hpp__