[wpimath] Add RotateBy() function to pose classes (#5491)

Fixes #5472.
This commit is contained in:
Tyler Veness
2023-07-31 21:16:44 -07:00
committed by GitHub
parent 26d6e68c8f
commit 35a8b129d9
10 changed files with 139 additions and 0 deletions

View File

@@ -11,6 +11,27 @@
using namespace frc;
TEST(Pose3dTest, RotateBy) {
constexpr auto x = 1_m;
constexpr auto y = 2_m;
const Pose3d initial{x, y, 0_m, Rotation3d{0_deg, 0_deg, 45_deg}};
constexpr units::radian_t yaw = 5_deg;
const Rotation3d rotation{0_deg, 0_deg, yaw};
const auto rotated = initial.RotateBy(rotation);
// Translation is rotated by CCW rotation matrix
double c = std::cos(yaw.value());
double s = std::sin(yaw.value());
EXPECT_DOUBLE_EQ(c * x.value() - s * y.value(), rotated.X().value());
EXPECT_DOUBLE_EQ(s * x.value() + c * y.value(), rotated.Y().value());
EXPECT_DOUBLE_EQ(0.0, rotated.Z().value());
EXPECT_DOUBLE_EQ(0.0, rotated.Rotation().X().value());
EXPECT_DOUBLE_EQ(0.0, rotated.Rotation().Y().value());
EXPECT_DOUBLE_EQ(initial.Rotation().Z().value() + rotation.Z().value(),
rotated.Rotation().Z().value());
}
TEST(Pose3dTest, TestTransformByRotations) {
const double kEpsilon = 1E-9;