245 lines
		
	
	
		
			8.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			245 lines
		
	
	
		
			8.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
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.
 | 
						|
 | 
						|
    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];
 | 
						|
    
 | 
						|
    //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_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_appliedForce.setValue( 0, 0, 0);
 | 
						|
		m_appliedTorque.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]);
 | 
						|
 | 
						|
		switch(m_jointType)
 | 
						|
		{
 | 
						|
			case eRevolute:
 | 
						|
			{
 | 
						|
				m_cachedRotParentToThis = btQuaternion(getAxisTop(0),-pJointPos[0]) * m_zeroRotParentToThis;
 | 
						|
				m_cachedRVector = m_dVector + quatRotate(m_cachedRotParentToThis,m_eVector);
 | 
						|
 | 
						|
				break;
 | 
						|
			}
 | 
						|
			case ePrismatic:
 | 
						|
			{
 | 
						|
				// m_cachedRotParentToThis never changes, so no need to update
 | 
						|
				m_cachedRVector = m_dVector + quatRotate(m_cachedRotParentToThis,m_eVector) + pJointPos[0] * getAxisBottom(0);
 | 
						|
 | 
						|
				break;
 | 
						|
			}
 | 
						|
			case eSpherical:
 | 
						|
			{
 | 
						|
				m_cachedRotParentToThis = btQuaternion(pJointPos[0], pJointPos[1], pJointPos[2], -pJointPos[3]) * m_zeroRotParentToThis;
 | 
						|
				m_cachedRVector = m_dVector + quatRotate(m_cachedRotParentToThis,m_eVector);
 | 
						|
 | 
						|
				break;
 | 
						|
			}
 | 
						|
			case ePlanar:
 | 
						|
			{
 | 
						|
				m_cachedRotParentToThis = btQuaternion(getAxisTop(0),-pJointPos[0]) * m_zeroRotParentToThis;				
 | 
						|
				m_cachedRVector = quatRotate(btQuaternion(getAxisTop(0),-pJointPos[0]), pJointPos[1] * getAxisBottom(1) + pJointPos[2] * getAxisBottom(2)) + quatRotate(m_cachedRotParentToThis,m_eVector);				
 | 
						|
 | 
						|
				break;
 | 
						|
			}
 | 
						|
			case eFixed:
 | 
						|
			{
 | 
						|
				m_cachedRotParentToThis = m_zeroRotParentToThis;
 | 
						|
				m_cachedRVector = m_dVector + quatRotate(m_cachedRotParentToThis,m_eVector);
 | 
						|
 | 
						|
				break;
 | 
						|
			}
 | 
						|
			default:
 | 
						|
			{
 | 
						|
				//invalid type
 | 
						|
				btAssert(0);
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
};
 | 
						|
 | 
						|
 | 
						|
#endif //BT_MULTIBODY_LINK_H
 |