Pass voltage through to the trajectory python wrapper.
Add dependencies on python_init at suggestion of Brian.
Change-Id: I4fc7394bbe891115cf05f8604f2dde398f40e75f
diff --git a/frc971/control_loops/python/libspline.py b/frc971/control_loops/python/libspline.py
index fd44cb4..7a77637 100644
--- a/frc971/control_loops/python/libspline.py
+++ b/frc971/control_loops/python/libspline.py
@@ -193,6 +193,16 @@
"""
libSpline.TrajectoryPlan(self.__trajectory)
+ def Voltage(self, distance):
+ """
+ Returns a pair of voltages for a given distance.
+ Order is left-right.
+ """
+ result = np.zeros(2)
+ libSpline.TrajectoryVoltage(self.__trajectory, ct.c_double(distance),
+ np.ctypeslib.as_ctypes(result))
+ return result
+
def Length(self):
return libSpline.TrajectoryLength(self.__trajectory)