112 lines
2.8 KiB
C++
112 lines
2.8 KiB
C++
#include <UnitTest++.h>
|
|
|
|
#include <iostream>
|
|
#include <cmath>
|
|
|
|
#include "Fixtures.h"
|
|
#include "Human36Fixture.h"
|
|
#include "rbdl/rbdl_mathutils.h"
|
|
#include "rbdl/Logging.h"
|
|
|
|
#include "rbdl/Model.h"
|
|
#include "rbdl/Kinematics.h"
|
|
#include "rbdl/Dynamics.h"
|
|
#include "rbdl/Constraints.h"
|
|
|
|
using namespace std;
|
|
using namespace RigidBodyDynamics;
|
|
using namespace RigidBodyDynamics::Math;
|
|
|
|
struct ScrewJoint1DofFixedBase {
|
|
ScrewJoint1DofFixedBase() {
|
|
using namespace RigidBodyDynamics;
|
|
using namespace RigidBodyDynamics::Math;
|
|
|
|
ClearLogOutput();
|
|
model = new Model;
|
|
|
|
/* Single screw joint with a fixed base. Rotation about Z, translation along X.
|
|
* A rolling log.
|
|
*/
|
|
|
|
Body body = Body (1., Vector3d (0., 0, 0.), Vector3d (1., 1., 1.));
|
|
|
|
Joint joint = Joint (SpatialVector (0., 0., 1., 1., 0., 0.));
|
|
|
|
roller = model->AppendBody (Xtrans (Vector3d (0., 0., 0.)), joint, body, "Roller");
|
|
|
|
q = VectorNd::Constant ((size_t) model->dof_count, 0.);
|
|
qdot = VectorNd::Constant ((size_t) model->dof_count, 0.);
|
|
qddot = VectorNd::Constant ((size_t) model->dof_count, 0.);
|
|
tau = VectorNd::Constant ((size_t) model->dof_count, 0.);
|
|
|
|
epsilon = 1e-8;
|
|
|
|
ClearLogOutput();
|
|
}
|
|
~ScrewJoint1DofFixedBase() {
|
|
delete model;
|
|
}
|
|
|
|
RigidBodyDynamics::Model *model;
|
|
|
|
RigidBodyDynamics::Math::VectorNd q;
|
|
RigidBodyDynamics::Math::VectorNd qdot;
|
|
RigidBodyDynamics::Math::VectorNd qddot;
|
|
RigidBodyDynamics::Math::VectorNd tau;
|
|
|
|
unsigned int roller;
|
|
double epsilon;
|
|
};
|
|
|
|
|
|
TEST_FIXTURE ( ScrewJoint1DofFixedBase, UpdateKinematics ) {
|
|
q[0] = 1;
|
|
qdot[0] = 2;
|
|
qddot[0] = 0;
|
|
|
|
UpdateKinematics (*model, q, qdot, qddot);
|
|
|
|
CHECK_ARRAY_EQUAL (Xrot(1,Vector3d(0,0,1)).E.data(), model->X_base[roller].E.data(), 9);
|
|
CHECK_ARRAY_EQUAL (Vector3d(1.,0.,0.).data(), model->X_base[roller].r.data(), 3);
|
|
CHECK_ARRAY_EQUAL (SpatialVector(0.,0.,2.,cos(q[0])*2,-sin(q[0])*2.,0.).data(), model->v[roller].data(), 6);
|
|
|
|
SpatialVector a0(model->a[roller]);
|
|
SpatialVector v0(model->v[roller]);
|
|
|
|
q[0] = 1+2*epsilon;
|
|
qdot[0] = 2;
|
|
qddot[0] = 0;
|
|
|
|
UpdateKinematics (*model, q, qdot, qddot);
|
|
|
|
v0 = model->v[roller] - v0;
|
|
v0 /= epsilon;
|
|
|
|
CHECK_ARRAY_CLOSE (a0.data(),v0.data(), 6, 1e-5); //finite diff vs. analytical derivative
|
|
|
|
}
|
|
|
|
TEST_FIXTURE ( ScrewJoint1DofFixedBase, Jacobians ) {
|
|
q[0] = 1;
|
|
qdot[0] = 0;
|
|
qddot[0] = 9;
|
|
|
|
Vector3d refPt = Vector3d(1,0,3);
|
|
MatrixNd GrefPt = MatrixNd::Constant(3,1,0.);
|
|
MatrixNd Gexpected = MatrixNd::Constant(3,1,0.);
|
|
Vector3d refPtBaseCoord = Vector3d();
|
|
|
|
refPtBaseCoord = CalcBodyToBaseCoordinates(*model, q, roller, refPt);
|
|
|
|
CHECK_ARRAY_EQUAL (Vector3d(1+cos(1), sin(1), 3).data(), refPtBaseCoord.data(), 3);
|
|
|
|
CalcPointJacobian(*model, q, roller, refPt, GrefPt);
|
|
|
|
Gexpected(0,0) = 1 - sin(1);
|
|
Gexpected(1,0) = cos(1);
|
|
Gexpected(2,0) = 0;
|
|
|
|
CHECK_ARRAY_EQUAL (Gexpected.data(), GrefPt.data(), 3);
|
|
}
|