Merge pull request #472 from thedropbears/we-are-a-team-that-logs

We-are-a-team-that-logs

Author: L9-bms

components/vision.py

@@ -72,6 +72,8 @@ class VisualLocalizer(HasPerLoopCache):
 
     reproj_error_threshold = tunable(2.0)
 
+    tags_state = tunable("none")
+
     def __init__(
         self,
         # The name of the camera in PhotonVision.
@@ -323,6 +325,8 @@ def execute(self) -> None:
         for results in all_results:
             # if results didn't see any targets
             if not results.getTargets():
+                self.tags_state = "none"
+
                 return
 
             # if we have already processed these results
@@ -353,6 +357,9 @@ def execute(self) -> None:
                             self.rotation_vision_uncertainty_multi_tag,
                         ),
                     )
+                    self.tags_state = "multi"
+                else:
+                    self.tags_state = "none"
 
                 if self.should_log:
                     # Multitag results don't have best and alternates
@@ -369,6 +376,7 @@ def execute(self) -> None:
                         or target.getBestCameraToTarget().translation().norm()
                         > self.CAMERA_MAX_RANGE
                     ):
+                        self.tags_state = "none"
                         continue
 
                     heading = self.heading_buffer.sample(results.getTimestampSeconds())
@@ -396,6 +404,8 @@ def execute(self) -> None:
                         ),
                     )
 
+                    self.tags_state = "single"
+
                     if self.should_log:
                         self.best_log.setPose(pose)
 

robot.py

@@ -60,6 +60,9 @@ def createObjects(self) -> None:
         self.event_loop = wpilib.event.EventLoop()
         self.data_log = wpilib.DataLogManager.getLog()
 
+        # Log driver station data
+        wpilib.DriverStation.startDataLog(self.data_log)
+
         # Log deploy info to show in AdvantageScope.
         meta_table = ntcore.NetworkTableInstance.getDefault().getTable("Metadata")
         deploy_info = wpilib.deployinfo.getDeployData()