WPILibC++  unspecified
 All Classes Files Functions Variables Typedefs Enumerations Enumerator Pages
DriverStation.h
1 /*----------------------------------------------------------------------------*/
2 /* Copyright (c) FIRST 2008-2016. All Rights Reserved. */
3 /* Open Source Software - may be modified and shared by FRC teams. The code */
4 /* must be accompanied by the FIRST BSD license file in the root directory of */
5 /* the project. */
6 /*----------------------------------------------------------------------------*/
7 
8 #pragma once
9 
10 #include <atomic>
11 #include <condition_variable>
12 #include <memory>
13 #include <string>
14 #include <thread>
15 
16 #include "HAL/DriverStation.h"
17 #include "HAL/cpp/priority_condition_variable.h"
18 #include "HAL/cpp/priority_mutex.h"
19 #include "RobotState.h"
20 #include "SensorBase.h"
21 #include "llvm/StringRef.h"
22 
23 namespace frc {
24 
30  public:
31  enum Alliance { kRed, kBlue, kInvalid };
32 
33  virtual ~DriverStation();
34  static DriverStation& GetInstance();
35  static void ReportError(std::string error);
36  static void ReportWarning(std::string error);
37  static void ReportError(bool is_error, int code, const std::string& error,
38  const std::string& location,
39  const std::string& stack);
40 
41  static const int kJoystickPorts = 6;
42 
43  double GetStickAxis(int stick, int axis);
44  int GetStickPOV(int stick, int pov);
45  int GetStickButtons(int stick) const;
46  bool GetStickButton(int stick, int button);
47 
48  int GetStickAxisCount(int stick) const;
49  int GetStickPOVCount(int stick) const;
50  int GetStickButtonCount(int stick) const;
51 
52  bool GetJoystickIsXbox(int stick) const;
53  int GetJoystickType(int stick) const;
54  std::string GetJoystickName(int stick) const;
55  int GetJoystickAxisType(int stick, int axis) const;
56 
57  bool IsEnabled() const override;
58  bool IsDisabled() const override;
59  bool IsAutonomous() const override;
60  bool IsOperatorControl() const override;
61  bool IsTest() const override;
62  bool IsDSAttached() const;
63  bool IsNewControlData() const;
64  bool IsFMSAttached() const;
65  bool IsSysActive() const;
66  bool IsBrownedOut() const;
67 
68  Alliance GetAlliance() const;
69  int GetLocation() const;
70  void WaitForData();
71  bool WaitForData(double timeout);
72  double GetMatchTime() const;
73  double GetBatteryVoltage() const;
74 
79  void InDisabled(bool entering) { m_userInDisabled = entering; }
84  void InAutonomous(bool entering) { m_userInAutonomous = entering; }
89  void InOperatorControl(bool entering) { m_userInTeleop = entering; }
93  void InTest(bool entering) { m_userInTest = entering; }
94 
95  protected:
96  void GetData();
97 
98  private:
99  DriverStation();
100  void ReportJoystickUnpluggedError(llvm::StringRef message);
101  void ReportJoystickUnpluggedWarning(llvm::StringRef message);
102  void Run();
103  void UpdateControlWord(bool force, HAL_ControlWord& controlWord) const;
104 
105  // Joystick User Data
106  std::unique_ptr<HAL_JoystickAxes[]> m_joystickAxes;
107  std::unique_ptr<HAL_JoystickPOVs[]> m_joystickPOVs;
108  std::unique_ptr<HAL_JoystickButtons[]> m_joystickButtons;
109  std::unique_ptr<HAL_JoystickDescriptor[]> m_joystickDescriptor;
110 
111  // Joystick Cached Data
112  std::unique_ptr<HAL_JoystickAxes[]> m_joystickAxesCache;
113  std::unique_ptr<HAL_JoystickPOVs[]> m_joystickPOVsCache;
114  std::unique_ptr<HAL_JoystickButtons[]> m_joystickButtonsCache;
115  std::unique_ptr<HAL_JoystickDescriptor[]> m_joystickDescriptorCache;
116 
117  // Internal Driver Station thread
118  std::thread m_dsThread;
119  std::atomic<bool> m_isRunning{false};
120 
121  // WPILib WaitForData control variables
122  bool m_waitForDataPredicate = false;
123  std::condition_variable_any m_waitForDataCond;
124  priority_mutex m_waitForDataMutex;
125 
126  mutable std::atomic<bool> m_newControlData{false};
127 
128  mutable priority_mutex m_joystickDataMutex;
129 
130  // Robot state status variables
131  bool m_userInDisabled = false;
132  bool m_userInAutonomous = false;
133  bool m_userInTeleop = false;
134  bool m_userInTest = false;
135 
136  // Control word variables
137  mutable HAL_ControlWord m_controlWordCache;
138  mutable std::chrono::steady_clock::time_point m_lastControlWordUpdate;
139  mutable priority_mutex m_controlWordMutex;
140 
141  double m_nextMessageTime = 0;
142 };
143 
144 } // namespace frc
int GetStickPOV(int stick, int pov)
Get the state of a POV on the joystick.
Definition: DriverStation.cpp:108
Base class for all sensors.
Definition: SensorBase.h:22
void InOperatorControl(bool entering)
Only to be used to tell the Driver Station what code you claim to be executing for diagnostic purpose...
Definition: DriverStation.h:89
int GetJoystickType(int stick) const
Returns the type of joystick at a given port.
Definition: DriverStation.cpp:239
int GetStickAxisCount(int stick) const
Returns the number of axes on a given joystick port.
Definition: DriverStation.cpp:179
void GetData()
Copy data from the DS task for the user.
Definition: DriverStation.cpp:535
Provide access to the network communication data to / from the Driver Station.
Definition: DriverStation.h:29
static void ReportWarning(std::string error)
Report a warning to the DriverStation messages window.
Definition: DriverStation.cpp:56
int GetStickPOVCount(int stick) const
Returns the number of POVs on a given joystick port.
Definition: DriverStation.cpp:194
int GetStickButtons(int stick) const
The state of the buttons on the joystick.
Definition: DriverStation.cpp:134
bool IsAutonomous() const override
Check if the DS is commanding autonomous mode.
Definition: DriverStation.cpp:305
void InDisabled(bool entering)
Only to be used to tell the Driver Station what code you claim to be executing for diagnostic purpose...
Definition: DriverStation.h:79
static DriverStation & GetInstance()
Return a pointer to the singleton DriverStation.
Definition: DriverStation.cpp:37
bool IsEnabled() const override
Check if the DS has enabled the robot.
Definition: DriverStation.cpp:283
double GetStickAxis(int stick, int axis)
Get the value of the axis on a joystick.
Definition: DriverStation.cpp:82
Definition: priority_mutex.h:53
void WaitForData()
Wait until a new packet comes from the driver station.
Definition: DriverStation.cpp:453
double GetBatteryVoltage() const
Read the battery voltage.
Definition: DriverStation.cpp:521
bool IsNewControlData() const
Has a new control packet from the driver station arrived since the last time this function was called...
Definition: DriverStation.cpp:353
bool IsBrownedOut() const
Check if the system is browned out.
Definition: DriverStation.cpp:389
bool IsTest() const override
Check if the DS is commanding test mode.
Definition: DriverStation.cpp:327
bool GetJoystickIsXbox(int stick) const
Returns a boolean indicating if the controller is an xbox controller.
Definition: DriverStation.cpp:224
bool GetStickButton(int stick, int button)
The state of one joystick button.
Definition: DriverStation.cpp:150
Definition: RobotState.h:14
Alliance GetAlliance() const
Return the alliance that the driver station says it is on.
Definition: DriverStation.cpp:403
int GetStickButtonCount(int stick) const
Returns the number of buttons on a given joystick port.
Definition: DriverStation.cpp:209
bool IsDSAttached() const
Check if the DS is attached.
Definition: DriverStation.cpp:338
bool IsDisabled() const override
Check if the robot is disabled.
Definition: DriverStation.cpp:294
void InTest(bool entering)
Only to be used to tell the Driver Station what code you claim to be executing for diagnostic purpose...
Definition: DriverStation.h:93
int GetLocation() const
Return the driver station location on the field.
Definition: DriverStation.cpp:427
std::string GetJoystickName(int stick) const
Returns the name of the joystick at the given port.
Definition: DriverStation.cpp:254
Definition: DriverStation.h:31
int GetJoystickAxisType(int stick, int axis) const
Returns the types of Axes on a given joystick port.
Definition: DriverStation.cpp:269
bool IsSysActive() const
Check if the FPGA outputs are enabled.
Definition: DriverStation.cpp:377
double GetMatchTime() const
Return the approximate match time.
Definition: DriverStation.cpp:511
void InAutonomous(bool entering)
Only to be used to tell the Driver Station what code you claim to be executing for diagnostic purpose...
Definition: DriverStation.h:84
bool IsFMSAttached() const
Is the driver station attached to a Field Management System?
Definition: DriverStation.cpp:363
bool IsOperatorControl() const override
Check if the DS is commanding teleop mode.
Definition: DriverStation.cpp:316
static void ReportError(std::string error)
Report an error to the DriverStation messages window.
Definition: DriverStation.cpp:47