pxmlw6n2f/Gazebo_Distributed_MPI/test/regression/1124_collision_pose.cc

125 lines
3.8 KiB
C++

/*
* Copyright (C) 2014 Open Source Robotics Foundation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
#include "gazebo/test/ServerFixture.hh"
using namespace gazebo;
class Issue1124Test : public ServerFixture
{
};
/////////////////////////////////////////////////
// \brief Test for issue #1124 by directly setting a model pose
TEST_F(Issue1124Test, SetModelPose)
{
Load("worlds/box_plane_low_friction_test.world", true);
physics::WorldPtr world = physics::get_world("default");
ASSERT_TRUE(world != NULL);
physics::ModelPtr model = world->GetModel("box");
ASSERT_TRUE(model != NULL);
physics::LinkPtr link = model->GetLink("link");
ASSERT_TRUE(link != NULL);
physics::CollisionPtr coll = link->GetCollision("collision");
ASSERT_TRUE(coll != NULL);
// The start pose should be centered at the origin
EXPECT_EQ(coll->GetWorldPose(), math::Pose(0, 0, 0.5, 0, 0, 0));
// The world->steps are not necessary. There are included to
// err on the side of caution.
world->Step(1);
model->SetWorldPose(math::Pose(2, 2, 0.5, 0, 0, 0));
world->Step(1);
EXPECT_EQ(model->GetWorldPose(), math::Pose(2, 2, 0.5, 0, 0, 0));
world->Step(1);
EXPECT_EQ(link->GetWorldPose(), math::Pose(2, 2, 0.5, 0, 0, 0));
world->Step(1);
// The new pose should be centered a the new model location
EXPECT_EQ(coll->GetWorldPose(), math::Pose(2, 2, 0.5, 0, 0, 0));
world->Step(1);
}
/////////////////////////////////////////////////
// \brief Test for issue #1124 by directly setting a link pose
TEST_F(Issue1124Test, SetLinkPose)
{
Load("worlds/box_plane_low_friction_test.world");
physics::WorldPtr world = physics::get_world("default");
ASSERT_TRUE(world != NULL);
physics::ModelPtr model = world->GetModel("box");
ASSERT_TRUE(model != NULL);
physics::LinkPtr link = model->GetLink("link");
ASSERT_TRUE(link != NULL);
physics::CollisionPtr coll = link->GetCollision("collision");
ASSERT_TRUE(coll != NULL);
// The start pose should be centered at the origin
EXPECT_EQ(coll->GetWorldPose(), math::Pose(0, 0, 0.5, 0, 0, 0));
link->SetWorldPose(math::Pose(-4, 5, 0.5, 0, 0, 0));
// The new pose should be centered a the new model location
EXPECT_EQ(coll->GetWorldPose(), math::Pose(-4, 5, 0.5, 0, 0, 0));
}
/////////////////////////////////////////////////
// \brief Test for issue #1124 using the physics engine
TEST_F(Issue1124Test, MovingPose)
{
Load("worlds/box_plane_low_friction_test.world", true);
physics::WorldPtr world = physics::get_world("default");
ASSERT_TRUE(world != NULL);
physics::ModelPtr model = world->GetModel("box");
ASSERT_TRUE(model != NULL);
physics::LinkPtr link = model->GetLink("link");
ASSERT_TRUE(link != NULL);
physics::CollisionPtr coll = link->GetCollision("collision");
ASSERT_TRUE(coll != NULL);
// The start pose should be centered at the origin
EXPECT_EQ(coll->GetWorldPose(), math::Pose(0, 0, 0.5, 0, 0, 0));
// Move the model by adding a force. This will exercise the physics engine
// pose change callbacks.
link->AddForce(math::Vector3(100, 0, 0.0));
world->Step(5000);
EXPECT_GT(model->GetWorldPose().pos.x, 1.0);
EXPECT_EQ(model->GetWorldPose(), coll->GetWorldPose());
}
/////////////////////////////////////////////////
/// Main
int main(int argc, char **argv)
{
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}