[wpimath] Add 2D to 3D geometry constructors (#7380)

This commit is contained in:
Joseph Eng
2024-11-12 15:18:37 -08:00
committed by GitHub
parent 6adad7bad7
commit 425bf83036
8 changed files with 70 additions and 0 deletions

View File

@@ -83,6 +83,8 @@ public class Pose3d implements Interpolatable<Pose3d>, ProtobufSerializable, Str
* Constructs a 3D pose from a 2D pose in the X-Y plane.
*
* @param pose The 2D pose.
* @see Rotation3d#Rotation3d(Rotation2d)
* @see Translation3d#Translation3d(Translation2d)
*/
public Pose3d(Pose2d pose) {
m_translation = new Translation3d(pose.getX(), pose.getY(), 0.0);

View File

@@ -273,6 +273,17 @@ public class Rotation3d
}
}
/**
* Constructs a 3D rotation from a 2D rotation in the X-Y plane.
*
* @param rotation The 2D rotation.
* @see Pose3d#Pose3d(Pose2d)
* @see Transform3d#Transform3d(Transform2d)
*/
public Rotation3d(Rotation2d rotation) {
this(0.0, 0.0, rotation.getRadians());
}
/**
* Adds two rotations together.
*

View File

@@ -86,6 +86,18 @@ public class Transform3d implements ProtobufSerializable, StructSerializable {
m_rotation = Rotation3d.kZero;
}
/**
* Constructs a 3D transform from a 2D transform in the X-Y plane.
*
* @param transform The 2D transform.
* @see Rotation3d#Rotation3d(Rotation2d)
* @see Translation3d#Translation3d(Translation2d)
*/
public Transform3d(Transform2d transform) {
m_translation = new Translation3d(transform.getTranslation());
m_rotation = new Rotation3d(transform.getRotation());
}
/**
* Multiplies the transform by the scalar.
*

View File

@@ -92,6 +92,17 @@ public class Translation3d
this(x.in(Meters), y.in(Meters), z.in(Meters));
}
/**
* Constructs a 3D translation from a 2D translation in the X-Y plane.
*
* @param translation The 2D translation.
* @see Pose3d#Pose3d(Pose2d)
* @see Transform3d#Transform3d(Transform2d)
*/
public Translation3d(Translation2d translation) {
this(translation.getX(), translation.getY(), 0.0);
}
/**
* Constructs a Translation3d from the provided translation vector's X, Y, and Z components. The
* values are assumed to be in meters.