Skip to content

Commit

Permalink
Lint
Browse files Browse the repository at this point in the history
Signed-off-by: Michael Carroll <[email protected]>
  • Loading branch information
mjcarroll committed Jan 11, 2023
1 parent 30d6b26 commit 8c9fc92
Show file tree
Hide file tree
Showing 3 changed files with 24 additions and 6 deletions.
6 changes: 4 additions & 2 deletions dartsim/src/AddedMassFeatures.cc
Original file line number Diff line number Diff line change
Expand Up @@ -72,11 +72,13 @@ void AddedMassFeatures::SetLinkAddedMass(const Identity &_link,
}
}

gz::math::Matrix6d AddedMassFeatures::GetLinkAddedMass(const Identity &_link) const
gz::math::Matrix6d
AddedMassFeatures::GetLinkAddedMass(const Identity &_link) const
{
auto linkInfo = this->ReferenceInterface<LinkInfo>(_link);

if (linkInfo->inertial.has_value() && linkInfo->inertial->FluidAddedMass().has_value())
if (linkInfo->inertial.has_value() &&
linkInfo->inertial->FluidAddedMass().has_value())
{
return linkInfo->inertial->FluidAddedMass().value();
}
Expand Down
3 changes: 2 additions & 1 deletion dartsim/src/AddedMassFeatures.hh
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,8 @@ class AddedMassFeatures :
public: void SetLinkAddedMass(const Identity &_link,
const gz::math::Matrix6d &_addedMass) override;

public: gz::math::Matrix6d GetLinkAddedMass(const Identity &_link) const override;
public: gz::math::Matrix6d GetLinkAddedMass(
const Identity &_link) const override;
};
} // namespace gz::physics::dartsim

Expand Down
21 changes: 18 additions & 3 deletions dartsim/src/AddedMassFeatures_TEST.cc
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ TEST(AddedMassFeatures, AddedMass)
ASSERT_EQ(3u, dartWorld->getNumSkeletons());

{
const dart::dynamics::SkeletonPtr skeleton = dartWorld->getSkeleton("body_no_added_mass");
const auto skeleton = dartWorld->getSkeleton("body_no_added_mass");
ASSERT_NE(skeleton, nullptr);

ASSERT_EQ(1u, skeleton->getNumBodyNodes());
Expand All @@ -139,10 +139,15 @@ TEST(AddedMassFeatures, AddedMass)

const Eigen::Matrix6d spatialInertia = link->getSpatialInertia();
ASSERT_TRUE(expectedZeroSpatialInertia.isApprox(spatialInertia));

const auto linkAddedMass =
world->GetModel("body_no_added_mass")->GetLink("link")->GetAddedMass();
ASSERT_TRUE(Eigen::Matrix6d::Zero().isApprox(
gz::math::eigen3::convert(linkAddedMass)));
}

{
const dart::dynamics::SkeletonPtr skeleton = dartWorld->getSkeleton("body_zero_added_mass");
const auto skeleton = dartWorld->getSkeleton("body_zero_added_mass");
ASSERT_NE(skeleton, nullptr);

ASSERT_EQ(1u, skeleton->getNumBodyNodes());
Expand All @@ -151,10 +156,15 @@ TEST(AddedMassFeatures, AddedMass)

const Eigen::Matrix6d spatialInertia = link->getSpatialInertia();
ASSERT_TRUE(expectedZeroSpatialInertia.isApprox(spatialInertia));

auto linkAddedMass =
world->GetModel("body_zero_added_mass")->GetLink("link")->GetAddedMass();
ASSERT_TRUE(Eigen::Matrix6d::Zero().isApprox(
gz::math::eigen3::convert(linkAddedMass)));
}

{
const dart::dynamics::SkeletonPtr skeleton = dartWorld->getSkeleton("body_added_mass");
const auto skeleton = dartWorld->getSkeleton("body_added_mass");
ASSERT_NE(skeleton, nullptr);

ASSERT_EQ(1u, skeleton->getNumBodyNodes());
Expand All @@ -163,5 +173,10 @@ TEST(AddedMassFeatures, AddedMass)

const Eigen::Matrix6d spatialInertia = link->getSpatialInertia();
ASSERT_TRUE(expectedSpatialInertia.isApprox(spatialInertia));

auto linkAddedMass =
world->GetModel("body_added_mass")->GetLink("link")->GetAddedMass();
ASSERT_FALSE(Eigen::Matrix6d::Zero().isApprox(
gz::math::eigen3::convert(linkAddedMass)));
}
}

0 comments on commit 8c9fc92

Please sign in to comment.