aos/input/joystick_input.h

#ifndef AOS_INPUT_JOYSTICK_INPUT_H_
#define AOS_INPUT_JOYSTICK_INPUT_H_

#include <atomic>

#include "aos/events/event-loop.h"
#include "aos/input/driver_station_data.h"

namespace aos {
namespace input {

// A class for handling joystick packet values.
// It will call RunIteration each time a new packet is received.
//
// This class automatically handles updating ::aos::joystick_state and logging
// (at INFO) button edges.
class JoystickInput {
 public:
  explicit JoystickInput(::aos::EventLoop *event_loop)
      : event_loop_(event_loop) {
    event_loop_->MakeWatcher(
        ".aos.joystick_state",
        [this](const ::aos::JoystickState &joystick_state) {
          this->HandleData(joystick_state);
        });
  }

  void Run();

 protected:
  int mode() const { return mode_; }

 private:
  void HandleData(const ::aos::JoystickState &joystick_state);

  // Subclasses should do whatever they want with data here.
  virtual void RunIteration(const driver_station::Data &data) = 0;

  static void Quit(int /*signum*/);

  static ::std::atomic<bool> run_;

  EventLoop *event_loop_;
  driver_station::Data data_;

  int mode_;
};

// Class which will proxy joystick information from UDP packets to the queues.
class JoystickProxy {
 public:
  void Run();
};

}  // namespace input
}  // namespace aos

#endif  // AOS_INPUT_JOYSTICK_INPUT_H_