Connect YardForce CoverUI to InputService

Author: rovo89

CMakeLists.txt

@@ -88,6 +88,7 @@ target_sources(${CMAKE_PROJECT_NAME} PRIVATE
         src/drivers/input/input_driver.cpp
         src/drivers/input/gpio_input_driver.cpp
         src/drivers/input/worx_input_driver.cpp
+        src/drivers/input/yard_force_input_driver.cpp
         $<$<CONFIG:Debug>:src/drivers/input/simulated_input_driver.cpp>
         # YardForce Cover UI Driver
         src/drivers/ui/YardForceCoverUI/yard_force_cover_ui_driver.cpp

robots/src/yardforce_robot.cpp

@@ -6,6 +6,7 @@ void YardForceRobot::InitPlatform() {
   InitMotors();
   charger_.setI2C(&I2CD1);
   power_service.SetDriver(&charger_);
+  input_service.RegisterInputDriver("yardforce", &cover_ui_driver_.GetInputDriver());
   cover_ui_driver_.Start(&UARTD7);
 }
 

services

@@ -1,7 +1,6 @@
 #ifndef GPIO_INPUT_DRIVER_HPP
 #define GPIO_INPUT_DRIVER_HPP
 
-#include <etl/vector.h>
 #include <hal.h>
 
 #include "input_driver.hpp"

src/drivers/input/gpio_input_driver.hpp

@@ -26,6 +26,10 @@ struct Input {
     struct {
       uint8_t bit;
     } worx;
+
+    struct {
+      uint8_t button_id;
+    } yardforce;
   };
 
   // State
@@ -86,20 +90,23 @@ class InputDriver {
  public:
   virtual ~InputDriver() = default;
   explicit InputDriver() = default;
+
   void AddInput(Input* input);
   void ClearInputs();
+  InputIterable Inputs() {
+    return InputIterable{inputs_head_};
+  }
+
   virtual bool OnInputConfigValue(lwjson_stream_parser_t* jsp, const char* key, lwjson_stream_type_t type,
                                   Input& input) = 0;
+
   virtual bool OnStart() {
     return true;
   };
-  virtual void OnStop(){};
+  virtual void OnStop() {};
 
- protected:
+ private:
   Input* inputs_head_ = nullptr;
-  InputIterable Inputs() {
-    return InputIterable{inputs_head_};
-  }
 };
 }  // namespace xbot::driver::input
 

src/drivers/input/input_driver.hpp

@@ -1,8 +1,6 @@
 #ifndef SIMULATED_INPUT_DRIVER_HPP
 #define SIMULATED_INPUT_DRIVER_HPP
 
-#include <etl/vector.h>
-
 #include "input_driver.hpp"
 
 namespace xbot::driver::input {

src/drivers/input/simulated_input_driver.hpp

@@ -1,7 +1,6 @@
 #ifndef WORX_INPUT_DRIVER_HPP
 #define WORX_INPUT_DRIVER_HPP
 
-#include <etl/vector.h>
 #include <hal.h>
 
 #include <services.hpp>

src/drivers/input/worx_input_driver.hpp

@@ -0,0 +1,19 @@
+#include "yard_force_input_driver.hpp"
+
+#include <ulog.h>
+
+#include <json_stream.hpp>
+
+namespace xbot::driver::input {
+
+bool YardForceInputDriver::OnInputConfigValue(lwjson_stream_parser_t *jsp, const char *key, lwjson_stream_type_t type,
+                                              Input &input) {
+  if (strcmp(key, "button") == 0) {
+    // TODO: We probably want to take a string here and map it to the ID.
+    return JsonGetNumber(jsp, type, input.yardforce.button_id);
+  }
+  ULOG_ERROR("Unknown attribute \"%s\"", key);
+  return false;
+}
+
+}  // namespace xbot::driver::input

src/drivers/input/yard_force_input_driver.cpp

@@ -0,0 +1,17 @@
+#ifndef YARDFORCE_INPUT_DRIVER_HPP
+#define YARDFORCE_INPUT_DRIVER_HPP
+
+#include <services.hpp>
+
+#include "input_driver.hpp"
+
+namespace xbot::driver::input {
+class YardForceInputDriver : public InputDriver {
+ public:
+  explicit YardForceInputDriver() = default;
+  bool OnInputConfigValue(lwjson_stream_parser_t *jsp, const char *key, lwjson_stream_type_t type,
+                          Input &input) override;
+};
+}  // namespace xbot::driver::input
+
+#endif  // YARDFORCE_INPUT_DRIVER_HPP

src/drivers/input/yard_force_input_driver.hpp

@@ -3,6 +3,7 @@
 #include <sys/unistd.h>
 #include <ulog.h>
 
+#include <json_stream.hpp>
 #include <services.hpp>
 
 #include "COBS.h"
@@ -203,8 +204,14 @@ void YardForceCoverUIDriver::ProcessPacket() {
   if (encode_decode_buf_[0] == Get_Version && size == sizeof(struct msg_get_version)) {
     board_found_ = true;
   } else if (encode_decode_buf_[0] == Get_Button && size == sizeof(struct msg_event_button)) {
-    // TODO: Send an event to the InputService.
-    // msg_event_button *msg = (struct msg_event_button *)encode_decode_buf_;
+    msg_event_button *msg = (struct msg_event_button *)encode_decode_buf_;
+    for (auto &input : input_driver_.Inputs()) {
+      if (input.yardforce.button_id == msg->button_id) {
+        const bool long_press = msg->press_duration >= 1;
+        input.InjectPress(long_press);
+        break;
+      }
+    }
   } /* else if (encode_decode_buf_[0] == Get_Emergency && size == sizeof(struct msg_event_emergency)) {
      struct msg_event_emergency *msg = (struct msg_event_emergency *)encode_decode_buf_;
      stock_ui_emergency_state = msg->state;