Avoid unnecessary copies after of mju_mulQuat.

Since fdbbc8bbe0, mju_mulQuat allows in-place computations.
There were a few places where the result was unnecessarily assigned to a temporary variable.

PiperOrigin-RevId: 477746005
Change-Id: I7351ef4796e7d70a4a54b930037bfaadee781f12
This commit is contained in:
Nimrod Gileadi
2022-09-29 09:33:12 -07:00
committed by Copybara-Service
parent 48f2ed99a5
commit 8a8305d3ca
5 changed files with 11 additions and 15 deletions
+2 -3
View File
@@ -35,7 +35,7 @@
// forward kinematics
void mj_kinematics(const mjModel* m, mjData* d) {
mjtNum pos[3], quat[4], *bodypos, *bodyquat;
mjtNum qloc[4], qtmp[4], vec[3], vec1[3], xanchor[3], xaxis[3];
mjtNum qloc[4], vec[3], vec1[3], xanchor[3], xaxis[3];
// set world position and orientation
mju_zero3(d->xpos);
@@ -124,8 +124,7 @@ void mj_kinematics(const mjModel* m, mjData* d) {
}
// apply rotation
mju_mulQuat(qtmp, quat, qloc);
mju_copy4(quat, qtmp);
mju_mulQuat(quat, quat, qloc);
// correct for off-center rotation
mju_sub3(vec, xanchor, pos);
+1 -2
View File
@@ -90,8 +90,7 @@ static void add_noise(const mjModel* m, mjData* d, mjtStage stage) {
// quaternion
else if (m->sensor_datatype[i]==mjDATATYPE_QUATERNION) {
// apply quaternion rotation to quaternion, assign
mju_mulQuat(res, d->sensordata+adr, quat);
mju_copy4(d->sensordata+adr, res);
mju_mulQuat(d->sensordata+adr, d->sensordata+adr, quat);
}
// unknown datatype
+2 -4
View File
@@ -493,8 +493,7 @@ int mju_eig3(mjtNum* eigval, mjtNum* eigvec, mjtNum quat[4], const mjtNum mat[9]
mju_normalize4(tmp);
// accumulate quaternion rotation
mju_mulQuat(tmp+4, quat, tmp);
mju_copy4(quat, tmp+4);
mju_mulQuat(quat, quat, tmp);
mju_normalize4(quat);
}
@@ -512,8 +511,7 @@ int mju_eig3(mjtNum* eigval, mjtNum* eigvec, mjtNum quat[4], const mjtNum mat[9]
tmp[0] = 0.707106781186548; // mju_cos(pi/4) = mju_sin(pi/4)
tmp[1] = tmp[2] = tmp[3] = 0;
tmp[(j1+2)%3+1] = tmp[0];
mju_mulQuat(tmp+4, quat, tmp);
mju_copy4(quat, tmp+4);
mju_mulQuat(quat, quat, tmp);
mju_normalize4(quat);
}
}
+2 -3
View File
@@ -240,9 +240,8 @@ void mju_quatIntegrate(mjtNum quat[4], const mjtNum vel[3], mjtNum scale) {
mju_copy3(tmp, vel);
angle = scale * mju_normalize3(tmp);
mju_axisAngle2Quat(qrot, tmp, angle);
mju_mulQuat(tmp, quat, qrot);
mju_normalize4(tmp);
mju_copy4(quat, tmp);
mju_mulQuat(quat, quat, qrot);
mju_normalize4(quat);
}
+4 -3
View File
@@ -374,7 +374,7 @@ void mjv_moveCamera(const mjModel* m, int action, mjtNum reldx, mjtNum reldy,
void mjv_movePerturb(const mjModel* m, const mjData* d, int action, mjtNum reldx,
mjtNum reldy, const mjvScene* scn, mjvPerturb* pert) {
int sel = pert->select;
mjtNum forward[3], vec[3], dif[3], scl, q1[4], q2[4], xiquat[4];
mjtNum forward[3], vec[3], scl, q1[4], xiquat[4];
// get camera info and align
mjv_cameraInModel(NULL, forward, NULL, scn);
@@ -394,8 +394,7 @@ void mjv_movePerturb(const mjModel* m, const mjData* d, int action, mjtNum reldx
// make quaternion and apply
mju_axisAngle2Quat(q1, vec, scl*mjPI*2);
mju_mulQuat(q2, q1, pert->refquat);
mju_copy4(pert->refquat, q2);
mju_mulQuat(pert->refquat, q1, pert->refquat);
mju_normalize4(pert->refquat);
// compute xiquat
@@ -404,10 +403,12 @@ void mjv_movePerturb(const mjModel* m, const mjData* d, int action, mjtNum reldx
// limit rotation relative to selected body
if (sel>0 && sel<m->nbody) {
// q2 = neg(selbody) * refquat
mjtNum q2[4];
mju_negQuat(q1, xiquat);
mju_mulQuat(q2, q1, pert->refquat);
// convert q2 to axis-angle
mjtNum dif[3];
mju_quat2Vel(dif, q2, 1);
scl = mju_normalize3(dif);