Minor plotting tool improvements
-Add trajectory drivetrain velocity goals to drivetrain plot
-Handle unpopulated values in flatbuffers.
Change-Id: I92308ab82eb034fd7d1878af1da5795c7b5a8efd
diff --git a/frc971/analysis/plot.py b/frc971/analysis/plot.py
index 124968f..40feea3 100644
--- a/frc971/analysis/plot.py
+++ b/frc971/analysis/plot.py
@@ -53,7 +53,13 @@
monotonic_time.append(entry[0] * 1e-9)
value = entry[2]
for name in field_path:
- value = value[name]
+ # If the value wasn't populated in a given message, fill in
+ # NaN rather than crashing.
+ if name in value:
+ value = value[name]
+ else:
+ value = float("nan")
+ break
# Catch NaNs and convert them to floats.
value = float(value)
signal_data.append(value)