Added support for 2017 WPILib.
Change-Id: Ib22e31d0999b123c65498c31c654e3fa7bab4798
diff --git a/frc971/wpilib/wpilib_interface.h b/frc971/wpilib/wpilib_interface.h
index 216bf09..1c55cc7 100644
--- a/frc971/wpilib/wpilib_interface.h
+++ b/frc971/wpilib/wpilib_interface.h
@@ -3,13 +3,22 @@
#include <stdint.h>
+#ifdef WPILIB2017
+namespace frc {
class DriverStation;
+} // namespace frc
+#else
+class DriverStation;
+namespace frc {
+using ::DriverStation;
+} // namespace frc
+#endif
namespace frc971 {
namespace wpilib {
// Sends out a message on ::aos::robot_state.
-void SendRobotState(int32_t my_pid, DriverStation *ds);
+void SendRobotState(int32_t my_pid, ::frc::DriverStation *ds);
} // namespace wpilib
} // namespace frc971