mirror of
https://github.com/bulletphysics/bullet3
synced 2024-12-14 05:40:05 +00:00
79d78a325a
Use btCylinderShapeZ for URDF cylinder, instead of converting it to a btConvexHullShape. Implement PyBullet.getCollisionShapeData Extend PyBullet.getDynamicsInfo / b3GetDynamicsInfo, remove flag (don't rely on API returning a fixed number of elements in a list!) Extend PyBullet.getJointInfo: add parentIndex
112 lines
3.4 KiB
C++
112 lines
3.4 KiB
C++
#ifndef BODY_JOINT_INFO_UTILITY_H
|
|
#define BODY_JOINT_INFO_UTILITY_H
|
|
|
|
#include "Bullet3Common/b3Logging.h"
|
|
|
|
namespace Bullet
|
|
{
|
|
class btMultiBodyDoubleData;
|
|
class btMultiBodyFloatData;
|
|
};
|
|
|
|
inline char* strDup(const char* const str)
|
|
{
|
|
#ifdef _WIN32
|
|
return _strdup(str);
|
|
#else
|
|
return strdup(str);
|
|
#endif
|
|
}
|
|
|
|
template <typename T, typename U> void addJointInfoFromMultiBodyData(const T* mb, U* bodyJoints, bool verboseOutput)
|
|
{
|
|
if (mb->m_baseName)
|
|
{
|
|
if (verboseOutput)
|
|
{
|
|
b3Printf("mb->m_baseName = %s\n", mb->m_baseName);
|
|
}
|
|
}
|
|
int qOffset = 7;
|
|
int uOffset = 6;
|
|
|
|
for (int link = 0; link < mb->m_numLinks; link++)
|
|
{
|
|
{
|
|
b3JointInfo info;
|
|
info.m_jointName[0] = 0;
|
|
info.m_linkName[0] = 0;
|
|
info.m_flags = 0;
|
|
info.m_jointIndex = link;
|
|
info.m_qIndex =
|
|
(0 < mb->m_links[link].m_posVarCount) ? qOffset : -1;
|
|
info.m_uIndex =
|
|
(0 < mb->m_links[link].m_dofCount) ? uOffset : -1;
|
|
|
|
if (mb->m_links[link].m_linkName) {
|
|
if (verboseOutput) {
|
|
b3Printf("mb->m_links[%d].m_linkName = %s\n", link,
|
|
mb->m_links[link].m_linkName);
|
|
}
|
|
strcpy(info.m_linkName,mb->m_links[link].m_linkName);
|
|
|
|
}
|
|
if (mb->m_links[link].m_jointName) {
|
|
if (verboseOutput) {
|
|
b3Printf("mb->m_links[%d].m_jointName = %s\n", link,
|
|
mb->m_links[link].m_jointName);
|
|
}
|
|
strcpy(info.m_jointName,mb->m_links[link].m_jointName);
|
|
//info.m_jointName = strDup(mb->m_links[link].m_jointName);
|
|
}
|
|
|
|
info.m_jointType = mb->m_links[link].m_jointType;
|
|
info.m_jointDamping = mb->m_links[link].m_jointDamping;
|
|
info.m_jointFriction = mb->m_links[link].m_jointFriction;
|
|
info.m_jointLowerLimit = mb->m_links[link].m_jointLowerLimit;
|
|
info.m_jointUpperLimit = mb->m_links[link].m_jointUpperLimit;
|
|
info.m_jointMaxForce = mb->m_links[link].m_jointMaxForce;
|
|
info.m_jointMaxVelocity = mb->m_links[link].m_jointMaxVelocity;
|
|
|
|
info.m_parentFrame[0] = mb->m_links[link].m_parentComToThisComOffset.m_floats[0];
|
|
info.m_parentFrame[1] = mb->m_links[link].m_parentComToThisComOffset.m_floats[1];
|
|
info.m_parentFrame[2] = mb->m_links[link].m_parentComToThisComOffset.m_floats[2];
|
|
info.m_parentFrame[3] = mb->m_links[link].m_zeroRotParentToThis.m_floats[0];
|
|
info.m_parentFrame[4] = mb->m_links[link].m_zeroRotParentToThis.m_floats[1];
|
|
info.m_parentFrame[5] = mb->m_links[link].m_zeroRotParentToThis.m_floats[2];
|
|
info.m_parentFrame[6] = mb->m_links[link].m_zeroRotParentToThis.m_floats[3];
|
|
|
|
info.m_jointAxis[0] = 0;
|
|
info.m_jointAxis[1] = 0;
|
|
info.m_jointAxis[2] = 0;
|
|
info.m_parentIndex = mb->m_links[link].m_parentIndex;
|
|
|
|
if (info.m_jointType == eRevoluteType)
|
|
{
|
|
info.m_jointAxis[0] = mb->m_links[link].m_jointAxisTop[0].m_floats[0];
|
|
info.m_jointAxis[1] = mb->m_links[link].m_jointAxisTop[0].m_floats[1];
|
|
info.m_jointAxis[2] = mb->m_links[link].m_jointAxisTop[0].m_floats[2];
|
|
}
|
|
if (info.m_jointType == ePrismaticType)
|
|
{
|
|
info.m_jointAxis[0] = mb->m_links[link].m_jointAxisBottom[0].m_floats[0];
|
|
info.m_jointAxis[1] = mb->m_links[link].m_jointAxisBottom[0].m_floats[1];
|
|
info.m_jointAxis[2] = mb->m_links[link].m_jointAxisBottom[0].m_floats[2];
|
|
}
|
|
|
|
if ((mb->m_links[link].m_jointType == eRevoluteType) ||
|
|
(mb->m_links[link].m_jointType == ePrismaticType)) {
|
|
info.m_flags |= JOINT_HAS_MOTORIZED_POWER;
|
|
}
|
|
bodyJoints->m_jointInfo.push_back(info);
|
|
}
|
|
qOffset += mb->m_links[link].m_posVarCount;
|
|
uOffset += mb->m_links[link].m_dofCount;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif //BODY_JOINT_INFO_UTILITY_H
|