Files
allwpilib/robotpyExamples/MecanumDrivePoseEstimator/exampleglobalmeasurementsensor.py
PJ Reiniger 8f9fc4d1b6 [copybara] Import robotpy examples (#8608)
GitOrigin-RevId: 9ba4bc3040fa7e772f5a594039e78fc6c43d114e
2026-02-20 15:30:35 -08:00

35 lines
1.0 KiB
Python

#
# Copyright (c) FIRST and other WPILib contributors.
# Open Source Software; you can modify and/or share it under the terms of
# the WPILib BSD license file in the root directory of this project.
#
import random
import wpimath
import wpimath.units
class ExampleGlobalMeasurementSensor:
"""This dummy class represents a global measurement sensor, such as a computer vision
solution.
"""
def __init__(self) -> None:
raise RuntimeError("Utility class")
@staticmethod
def getEstimatedGlobalPose(estimatedRobotPose: wpimath.Pose2d) -> wpimath.Pose2d:
"""Get a "noisy" fake global pose reading.
:param estimatedRobotPose: The robot pose.
"""
rand_x = random.gauss(0.0, 0.5)
rand_y = random.gauss(0.0, 0.5)
rand_rot = random.gauss(0.0, wpimath.units.degreesToRadians(30))
return wpimath.Pose2d(
estimatedRobotPose.x + rand_x,
estimatedRobotPose.y + rand_y,
estimatedRobotPose.rotation().rotateBy(wpimath.Rotation2d(rand_rot)),
)