mirror of
https://github.com/PhotonVision/photonvision
synced 2026-06-22 01:11:40 +00:00
Add python simulation (#1532)
This commit is contained in:
25
photon-lib/py/photonlibpy/estimation/cameraTargetRelation.py
Normal file
25
photon-lib/py/photonlibpy/estimation/cameraTargetRelation.py
Normal file
@@ -0,0 +1,25 @@
|
||||
import math
|
||||
|
||||
from wpimath.geometry import Pose3d, Rotation2d, Transform3d
|
||||
from wpimath.units import meters
|
||||
|
||||
|
||||
class CameraTargetRelation:
|
||||
def __init__(self, cameraPose: Pose3d, targetPose: Pose3d):
|
||||
self.camPose = cameraPose
|
||||
self.camToTarg = Transform3d(cameraPose, targetPose)
|
||||
self.camToTargDist = self.camToTarg.translation().norm()
|
||||
self.camToTargDistXY: meters = math.hypot(
|
||||
self.camToTarg.translation().X(), self.camToTarg.translation().Y()
|
||||
)
|
||||
self.camToTargYaw = Rotation2d(self.camToTarg.X(), self.camToTarg.Y())
|
||||
self.camToTargPitch = Rotation2d(self.camToTargDistXY, -self.camToTarg.Z())
|
||||
self.camToTargAngle = Rotation2d(
|
||||
math.hypot(self.camToTargYaw.radians(), self.camToTargPitch.radians())
|
||||
)
|
||||
self.targToCam = Transform3d(targetPose, cameraPose)
|
||||
self.targToCamYaw = Rotation2d(self.targToCam.X(), self.targToCam.Y())
|
||||
self.targToCamPitch = Rotation2d(self.camToTargDistXY, -self.targToCam.Z())
|
||||
self.targtoCamAngle = Rotation2d(
|
||||
math.hypot(self.targToCamYaw.radians(), self.targToCamPitch.radians())
|
||||
)
|
||||
Reference in New Issue
Block a user