WPILibC++  unspecified
DriverStation.h
1 /*----------------------------------------------------------------------------*/
2 /* Copyright (c) 2008-2017 FIRST. 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 <memory>
12 #include <mutex>
13 #include <string>
14 #include <thread>
15 
16 #include <HAL/DriverStation.h>
17 #include <llvm/StringRef.h>
18 
19 #include "RobotState.h"
20 #include "SensorBase.h"
21 
22 namespace frc {
23 
29  public:
30  enum Alliance { kRed, kBlue, kInvalid };
31 
32  virtual ~DriverStation();
33  static DriverStation& GetInstance();
34  static void ReportError(llvm::StringRef error);
35  static void ReportWarning(llvm::StringRef error);
36  static void ReportError(bool is_error, int code, llvm::StringRef error,
37  llvm::StringRef location, llvm::StringRef stack);
38 
39  static const int kJoystickPorts = 6;
40 
41  double GetStickAxis(int stick, int axis);
42  int GetStickPOV(int stick, int pov);
43  int GetStickButtons(int stick) const;
44  bool GetStickButton(int stick, int button);
45 
46  int GetStickAxisCount(int stick) const;
47  int GetStickPOVCount(int stick) const;
48  int GetStickButtonCount(int stick) const;
49 
50  bool GetJoystickIsXbox(int stick) const;
51  int GetJoystickType(int stick) const;
52  std::string GetJoystickName(int stick) const;
53  int GetJoystickAxisType(int stick, int axis) const;
54 
55  bool IsEnabled() const override;
56  bool IsDisabled() const override;
57  bool IsAutonomous() const override;
58  bool IsOperatorControl() const override;
59  bool IsTest() const override;
60  bool IsDSAttached() const;
61  bool IsNewControlData() const;
62  bool IsFMSAttached() const;
63  bool IsSysActive() const;
64  bool IsBrownedOut() const;
65 
66  Alliance GetAlliance() const;
67  int GetLocation() const;
68  void WaitForData();
69  bool WaitForData(double timeout);
70  double GetMatchTime() const;
71  double GetBatteryVoltage() const;
72 
77  void InDisabled(bool entering) { m_userInDisabled = entering; }
82  void InAutonomous(bool entering) { m_userInAutonomous = entering; }
87  void InOperatorControl(bool entering) { m_userInTeleop = entering; }
91  void InTest(bool entering) { m_userInTest = entering; }
92 
93  protected:
94  void GetData();
95 
96  private:
97  DriverStation();
98  void ReportJoystickUnpluggedError(llvm::StringRef message);
99  void ReportJoystickUnpluggedWarning(llvm::StringRef message);
100  void Run();
101  void UpdateControlWord(bool force, HAL_ControlWord& controlWord) const;
102 
103  // Joystick User Data
104  std::unique_ptr<HAL_JoystickAxes[]> m_joystickAxes;
105  std::unique_ptr<HAL_JoystickPOVs[]> m_joystickPOVs;
106  std::unique_ptr<HAL_JoystickButtons[]> m_joystickButtons;
107  std::unique_ptr<HAL_JoystickDescriptor[]> m_joystickDescriptor;
108 
109  // Joystick Cached Data
110  std::unique_ptr<HAL_JoystickAxes[]> m_joystickAxesCache;
111  std::unique_ptr<HAL_JoystickPOVs[]> m_joystickPOVsCache;
112  std::unique_ptr<HAL_JoystickButtons[]> m_joystickButtonsCache;
113  std::unique_ptr<HAL_JoystickDescriptor[]> m_joystickDescriptorCache;
114 
115  // Internal Driver Station thread
116  std::thread m_dsThread;
117  std::atomic<bool> m_isRunning{false};
118 
119  mutable std::mutex m_joystickDataMutex;
120 
121  // Robot state status variables
122  bool m_userInDisabled = false;
123  bool m_userInAutonomous = false;
124  bool m_userInTeleop = false;
125  bool m_userInTest = false;
126 
127  // Control word variables
128  mutable HAL_ControlWord m_controlWordCache;
129  mutable std::chrono::steady_clock::time_point m_lastControlWordUpdate;
130  mutable std::mutex m_controlWordMutex;
131 
132  double m_nextMessageTime = 0;
133 };
134 
135 } // namespace frc
Definition: Timer.cpp:18
static void ReportError(llvm::StringRef error)
Report an error to the DriverStation messages window.
Definition: DriverStation.cpp:49
int GetStickPOV(int stick, int pov)
Get the state of a POV on the joystick.
Definition: DriverStation.cpp:114
Base class for all sensors.
Definition: SensorBase.h:20
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:87
int GetJoystickType(int stick) const
Returns the type of joystick at a given port.
Definition: DriverStation.cpp:245
int GetStickAxisCount(int stick) const
Returns the number of axes on a given joystick port.
Definition: DriverStation.cpp:185
void GetData()
Copy data from the DS task for the user.
Definition: DriverStation.cpp:518
Provide access to the network communication data to / from the Driver Station.
Definition: DriverStation.h:28
int GetStickPOVCount(int stick) const
Returns the number of POVs on a given joystick port.
Definition: DriverStation.cpp:200
int GetStickButtons(int stick) const
The state of the buttons on the joystick.
Definition: DriverStation.cpp:140
bool IsAutonomous() const override
Check if the DS is commanding autonomous mode.
Definition: DriverStation.cpp:311
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:77
static DriverStation & GetInstance()
Return a pointer to the singleton DriverStation.
Definition: DriverStation.cpp:39
bool IsEnabled() const override
Check if the DS has enabled the robot.
Definition: DriverStation.cpp:289
double GetStickAxis(int stick, int axis)
Get the value of the axis on a joystick.
Definition: DriverStation.cpp:88
void WaitForData()
Wait until a new packet comes from the driver station.
Definition: DriverStation.cpp:457
double GetBatteryVoltage() const
Read the battery voltage.
Definition: DriverStation.cpp:504
bool IsNewControlData() const
Has a new control packet from the driver station arrived since the last time this function was called...
Definition: DriverStation.cpp:359
bool IsBrownedOut() const
Check if the system is browned out.
Definition: DriverStation.cpp:393
bool IsTest() const override
Check if the DS is commanding test mode.
Definition: DriverStation.cpp:333
bool GetJoystickIsXbox(int stick) const
Returns a boolean indicating if the controller is an xbox controller.
Definition: DriverStation.cpp:230
bool GetStickButton(int stick, int button)
The state of one joystick button.
Definition: DriverStation.cpp:156
Definition: RobotState.h:14
Alliance GetAlliance() const
Return the alliance that the driver station says it is on.
Definition: DriverStation.cpp:407
int GetStickButtonCount(int stick) const
Returns the number of buttons on a given joystick port.
Definition: DriverStation.cpp:215
bool IsDSAttached() const
Check if the DS is attached.
Definition: DriverStation.cpp:344
bool IsDisabled() const override
Check if the robot is disabled.
Definition: DriverStation.cpp:300
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:91
int GetLocation() const
Return the driver station location on the field.
Definition: DriverStation.cpp:431
std::string GetJoystickName(int stick) const
Returns the name of the joystick at the given port.
Definition: DriverStation.cpp:260
static void ReportWarning(llvm::StringRef error)
Report a warning to the DriverStation messages window.
Definition: DriverStation.cpp:59
int GetJoystickAxisType(int stick, int axis) const
Returns the types of Axes on a given joystick port.
Definition: DriverStation.cpp:275
bool IsSysActive() const
Check if the FPGA outputs are enabled.
Definition: DriverStation.cpp:381
double GetMatchTime() const
Return the approximate match time.
Definition: DriverStation.cpp:494
StringRef - Represent a constant reference to a string, i.e.
Definition: StringRef.h:42
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:82
bool IsFMSAttached() const
Is the driver station attached to a Field Management System?
Definition: DriverStation.cpp:367
bool IsOperatorControl() const override
Check if the DS is commanding teleop mode.
Definition: DriverStation.cpp:322