From 1591426b51e4cc2f285b82de21024af0b14b0bae Mon Sep 17 00:00:00 2001 From: Elliot Watkins Date: Tue, 16 Jan 2024 19:58:36 -0600 Subject: [PATCH] allianceTransformUtils.py is not giving errors on my laptop now... --- utils/allianceTransformUtils.py | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/utils/allianceTransformUtils.py b/utils/allianceTransformUtils.py index 481d93a..27e963a 100644 --- a/utils/allianceTransformUtils.py +++ b/utils/allianceTransformUtils.py @@ -21,35 +21,35 @@ def transformX(in_): def transformY(self,in_): return in_ + # Rotation def transformRotation(self,in_): if wpilib._wpilib.DriverStation.Alliance == wpilib._wpilib.DriverStation.Alliance.kRed: return (Rotation2d.fromDegrees(180) - in_) else: return in_ - def transformTranslation(self,in_): + # Translation + def transform(self,in_): if wpilib._wpilib.DriverStation.Alliance == wpilib._wpilib.DriverStation.Alliance.kRed: - #return Translation2d(self, AllianceTransformUtils.transformX(x value), AllianceTranformUtils.transformY(y value)) - #We need to get an x to transform it, but we don't know where to get said x from. Same with y - return in_ + return Translation2d(self, AllianceTransformUtils.transformX(in_.getx()), in_.gety()) else: return in_ - def transformTransform(self,in_): + # Transform + def transform(self,in_): if wpilib._wpilib.DriverStation.Alliance == wpilib._wpilib.DriverStation.Alliance.kRed: - #translation = Transform2d((self, AllianceTransformUtils.transformX(x value), AllianceTranformUtils.transformY(y value))) - #rotation = AllianceTransformUtils.transformRotation(rotation value) - #return Transform2d(self, translation, rotation) - return in_ + translation = Transform2d((self, AllianceTransformUtils.transformX(in_.X()), in_.Y())) + rotation = AllianceTransformUtils.transformRotation(in_.rotation()) + return Transform2d(self, translation, rotation) else: return in_ - def transformPose(self,in_): + # Pose2d + def transform(self,in_): if wpilib._wpilib.DriverStation.Alliance == wpilib._wpilib.DriverStation.Alliance.kRed: - #trans = AllianceTransformUtils.transformTranlation(translation value) - #rot = AllianceTransformUtils.transformPose(rotation) - #return Pose2d(trans, rot) - return in_ + trans = AllianceTransformUtils.transformTranlation(in_.translation()) + rot = AllianceTransformUtils.transformPose(in_.rotation()) + return Pose2d(trans, rot) else: return in_