btMultiBodyLink.h

Go to the documentation of this file.

/*
Bullet Continuous Collision Detection and Physics Library
Copyright (c) 2013 Erwin Coumans  http://bulletphysics.org
 
This software is provided 'as-is', without any express or implied warranty.
In no event will the authors be held liable for any damages arising from the use of this software.
Permission is granted to anyone to use this software for any purpose, 
including commercial applications, and to alter it and redistribute it freely, 
subject to the following restrictions:
 
1. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required.
2. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.
3. This notice may not be removed or altered from any source distribution.
*/
 
#ifndef BT_MULTIBODY_LINK_H
#define BT_MULTIBODY_LINK_H
 
#include "LinearMath/btQuaternion.h"
#include "LinearMath/btVector3.h"
#include "BulletCollision/CollisionDispatch/btCollisionObject.h"
 
enum btMultiBodyLinkFlags
{
        BT_MULTIBODYLINKFLAGS_DISABLE_PARENT_COLLISION = 1,
        BT_MULTIBODYLINKFLAGS_DISABLE_ALL_PARENT_COLLISION = 2,
};
 
//both defines are now permanently enabled
#define BT_MULTIBODYLINK_INCLUDE_PLANAR_JOINTS
#define TEST_SPATIAL_ALGEBRA_LAYER
 
//
// Various spatial helper functions
//
 
//namespace {
 
#include "LinearMath/btSpatialAlgebra.h"
 
//}
 
//
// Link struct
//
 
struct btMultibodyLink
{
        BT_DECLARE_ALIGNED_ALLOCATOR();
 
        btScalar m_mass;           // mass of link
        btVector3 m_inertiaLocal;  // inertia of link (local frame; diagonal)
 
        int m_parent;  // index of the parent link (assumed to be < index of this link), or -1 if parent is the base link.
 
        btQuaternion m_zeroRotParentToThis;  // rotates vectors in parent-frame to vectors in local-frame (when q=0). constant.
 
        btVector3 m_dVector;  // vector from the inboard joint pos to this link's COM. (local frame.) constant.
                                                  //this is set to zero for planar joint (see also m_eVector comment)
 
        // m_eVector is constant, but depends on the joint type:
        // revolute, fixed, prismatic, spherical: vector from parent's COM to the pivot point, in PARENT's frame.
        // planar: vector from COM of parent to COM of this link, WHEN Q = 0. (local frame.)
        // todo: fix the planar so it is consistent with the other joints
 
        btVector3 m_eVector;
 
        btSpatialMotionVector m_absFrameTotVelocity, m_absFrameLocVelocity;
 
        enum eFeatherstoneJointType
        {
                eRevolute = 0,
                ePrismatic = 1,
                eSpherical = 2,
                ePlanar = 3,
                eFixed = 4,
                eInvalid
        };
 
        // "axis" = spatial joint axis (Mirtich Defn 9 p104). (expressed in local frame.) constant.
        // for prismatic: m_axesTop[0] = zero;
        //                m_axesBottom[0] = unit vector along the joint axis.
        // for revolute: m_axesTop[0] = unit vector along the rotation axis (u);
        //               m_axesBottom[0] = u cross m_dVector (i.e. COM linear motion due to the rotation at the joint)
        //
        // for spherical: m_axesTop[0][1][2] (u1,u2,u3) form a 3x3 identity matrix (3 rotation axes)
        //                                m_axesBottom[0][1][2] cross u1,u2,u3 (i.e. COM linear motion due to the rotation at the joint)
        //
        // for planar: m_axesTop[0] = unit vector along the rotation axis (u); defines the plane of motion
        //                         m_axesTop[1][2] = zero
        //                         m_axesBottom[0] = zero
        //                         m_axesBottom[1][2] = unit vectors along the translational axes on that plane
        btSpatialMotionVector m_axes[6];
        void setAxisTop(int dof, const btVector3 &axis) { m_axes[dof].m_topVec = axis; }
        void setAxisBottom(int dof, const btVector3 &axis)
        {
                m_axes[dof].m_bottomVec = axis;
        }
        void setAxisTop(int dof, const btScalar &x, const btScalar &y, const btScalar &z)
        {
                m_axes[dof].m_topVec.setValue(x, y, z);
        }
        void setAxisBottom(int dof, const btScalar &x, const btScalar &y, const btScalar &z)
        {
                m_axes[dof].m_bottomVec.setValue(x, y, z);
        }
        const btVector3 &getAxisTop(int dof) const { return m_axes[dof].m_topVec; }
        const btVector3 &getAxisBottom(int dof) const { return m_axes[dof].m_bottomVec; }
 
        int m_dofOffset, m_cfgOffset;
 
        btQuaternion m_cachedRotParentToThis;  // rotates vectors in parent frame to vectors in local frame
        btVector3 m_cachedRVector;             // vector from COM of parent to COM of this link, in local frame.
    
    // predicted verstion
    btQuaternion m_cachedRotParentToThis_interpolate;  // rotates vectors in parent frame to vectors in local frame
    btVector3 m_cachedRVector_interpolate;             // vector from COM of parent to COM of this link, in local frame.
 
        btVector3 m_appliedForce;   // In WORLD frame
        btVector3 m_appliedTorque;  // In WORLD frame
 
        btVector3 m_appliedConstraintForce;   // In WORLD frame
        btVector3 m_appliedConstraintTorque;  // In WORLD frame
 
        btScalar m_jointPos[7];
    btScalar m_jointPos_interpolate[7];
 
        //m_jointTorque is the joint torque applied by the user using 'addJointTorque'.
        //It gets set to zero after each internal stepSimulation call
        btScalar m_jointTorque[6];
 
        class btMultiBodyLinkCollider *m_collider;
        int m_flags;
 
        int m_dofCount, m_posVarCount;  //redundant but handy
 
        eFeatherstoneJointType m_jointType;
 
        struct btMultiBodyJointFeedback *m_jointFeedback;
 
        btTransform m_cachedWorldTransform;  //this cache is updated when calling btMultiBody::forwardKinematics
 
        const char *m_linkName;   //m_linkName memory needs to be managed by the developer/user!
        const char *m_jointName;  //m_jointName memory needs to be managed by the developer/user!
        const void *m_userPtr;    //m_userPtr ptr needs to be managed by the developer/user!
 
        btScalar m_jointDamping;      //todo: implement this internally. It is unused for now, it is set by a URDF loader. User can apply manual damping.
        btScalar m_jointFriction;     //todo: implement this internally. It is unused for now, it is set by a URDF loader. User can apply manual friction using a velocity motor.
        btScalar m_jointLowerLimit;   //todo: implement this internally. It is unused for now, it is set by a URDF loader.
        btScalar m_jointUpperLimit;   //todo: implement this internally. It is unused for now, it is set by a URDF loader.
        btScalar m_jointMaxForce;     //todo: implement this internally. It is unused for now, it is set by a URDF loader.
        btScalar m_jointMaxVelocity;  //todo: implement this internally. It is unused for now, it is set by a URDF loader.
 
        // ctor: set some sensible defaults
        btMultibodyLink()
                : m_mass(1),
                  m_parent(-1),
                  m_zeroRotParentToThis(0, 0, 0, 1),
                  m_cachedRotParentToThis(0, 0, 0, 1),
          m_cachedRotParentToThis_interpolate(0, 0, 0, 1),
                  m_collider(0),
                  m_flags(0),
                  m_dofCount(0),
                  m_posVarCount(0),
                  m_jointType(btMultibodyLink::eInvalid),
                  m_jointFeedback(0),
                  m_linkName(0),
                  m_jointName(0),
                  m_userPtr(0),
                  m_jointDamping(0),
                  m_jointFriction(0),
                  m_jointLowerLimit(0),
                  m_jointUpperLimit(0),
                  m_jointMaxForce(0),
                  m_jointMaxVelocity(0)
        {
                m_inertiaLocal.setValue(1, 1, 1);
                setAxisTop(0, 0., 0., 0.);
                setAxisBottom(0, 1., 0., 0.);
                m_dVector.setValue(0, 0, 0);
                m_eVector.setValue(0, 0, 0);
                m_cachedRVector.setValue(0, 0, 0);
        m_cachedRVector_interpolate.setValue(0, 0, 0);
                m_appliedForce.setValue(0, 0, 0);
                m_appliedTorque.setValue(0, 0, 0);
                m_appliedConstraintForce.setValue(0, 0, 0);
                m_appliedConstraintTorque.setValue(0, 0, 0);
                //
                m_jointPos[0] = m_jointPos[1] = m_jointPos[2] = m_jointPos[4] = m_jointPos[5] = m_jointPos[6] = 0.f;
                m_jointPos[3] = 1.f;  //"quat.w"
                m_jointTorque[0] = m_jointTorque[1] = m_jointTorque[2] = m_jointTorque[3] = m_jointTorque[4] = m_jointTorque[5] = 0.f;
                m_cachedWorldTransform.setIdentity();
        }
 
        // routine to update m_cachedRotParentToThis and m_cachedRVector
        void updateCacheMultiDof(btScalar *pq = 0)
        {
        btScalar *pJointPos = (pq ? pq : &m_jointPos[0]);
        btQuaternion& cachedRot = m_cachedRotParentToThis;
        btVector3& cachedVector = m_cachedRVector;
                switch (m_jointType)
                {
                        case eRevolute:
                        {
                                cachedRot = btQuaternion(getAxisTop(0), -pJointPos[0]) * m_zeroRotParentToThis;
                                cachedVector = m_dVector + quatRotate(m_cachedRotParentToThis, m_eVector);
 
                                break;
                        }
                        case ePrismatic:
                        {
                                // m_cachedRotParentToThis never changes, so no need to update
                                cachedVector = m_dVector + quatRotate(m_cachedRotParentToThis, m_eVector) + pJointPos[0] * getAxisBottom(0);
 
                                break;
                        }
                        case eSpherical:
                        {
                                cachedRot = btQuaternion(pJointPos[0], pJointPos[1], pJointPos[2], -pJointPos[3]) * m_zeroRotParentToThis;
                                cachedVector = m_dVector + quatRotate(cachedRot, m_eVector);
 
                                break;
                        }
                        case ePlanar:
                        {
                                cachedRot = btQuaternion(getAxisTop(0), -pJointPos[0]) * m_zeroRotParentToThis;
                                cachedVector = quatRotate(btQuaternion(getAxisTop(0), -pJointPos[0]), pJointPos[1] * getAxisBottom(1) + pJointPos[2] * getAxisBottom(2)) + quatRotate(cachedRot, m_eVector);
 
                                break;
                        }
                        case eFixed:
                        {
                                cachedRot = m_zeroRotParentToThis;
                                cachedVector = m_dVector + quatRotate(cachedRot, m_eVector);
 
                                break;
                        }
                        default:
                        {
                                //invalid type
                                btAssert(0);
                        }
                }
        m_cachedRotParentToThis_interpolate = m_cachedRotParentToThis;
        m_cachedRVector_interpolate = m_cachedRVector;
        }
    
    void updateInterpolationCacheMultiDof()
    {
        btScalar *pJointPos = &m_jointPos_interpolate[0];
        
        btQuaternion& cachedRot = m_cachedRotParentToThis_interpolate;
        btVector3& cachedVector = m_cachedRVector_interpolate;
        switch (m_jointType)
        {
            case eRevolute:
            {
                cachedRot = btQuaternion(getAxisTop(0), -pJointPos[0]) * m_zeroRotParentToThis;
                cachedVector = m_dVector + quatRotate(m_cachedRotParentToThis, m_eVector);
                
                break;
            }
            case ePrismatic:
            {
                // m_cachedRotParentToThis never changes, so no need to update
                cachedVector = m_dVector + quatRotate(m_cachedRotParentToThis, m_eVector) + pJointPos[0] * getAxisBottom(0);
                
                break;
            }
            case eSpherical:
            {
                cachedRot = btQuaternion(pJointPos[0], pJointPos[1], pJointPos[2], -pJointPos[3]) * m_zeroRotParentToThis;
                cachedVector = m_dVector + quatRotate(cachedRot, m_eVector);
                
                break;
            }
            case ePlanar:
            {
                cachedRot = btQuaternion(getAxisTop(0), -pJointPos[0]) * m_zeroRotParentToThis;
                cachedVector = quatRotate(btQuaternion(getAxisTop(0), -pJointPos[0]), pJointPos[1] * getAxisBottom(1) + pJointPos[2] * getAxisBottom(2)) + quatRotate(cachedRot, m_eVector);
                
                break;
            }
            case eFixed:
            {
                cachedRot = m_zeroRotParentToThis;
                cachedVector = m_dVector + quatRotate(cachedRot, m_eVector);
                
                break;
            }
            default:
            {
                //invalid type
                btAssert(0);
            }
        }
    }
 
 
 
};
 
#endif  //BT_MULTIBODY_LINK_H