[app] Add command to write to a gpio

Author: engdoreis

app/commands.hh

@@ -5,6 +5,7 @@
 #pragma once
 
 #include "flash/flash.hh"
+#include "embeddedpp/gpio.hh"
 #include "visuals/progressbar.hh"
 #include "visuals/throughput.hh"
 #include <fstream>
@@ -262,4 +263,23 @@ struct LoadFile : public Commands<T> {
   }
 };
 
+template <typename T>
+  requires embeddedpp::Gpio<T>
+struct GpioWrite {
+  T& gpio;
+  uint8_t pin;
+  bool value;
+
+  GpioWrite(T& gpio, uint8_t pin, bool value) : gpio(gpio), pin(pin), value(value) {}
+
+  int run() {
+    if (is_error(gpio.set_pin(pin, value))) {
+      std::println("GPIO write failed");
+      return 1;
+    }
+    std::println("GPIO{} = {}", pin, value ? 1 : 0);
+    return 0;
+  }
+};
+
 }  // namespace commands

app/main.cc

@@ -12,6 +12,7 @@
 
 #include "ftdi/ftdi.hh"
 #include "ftdi/spi_host.hh"
+#include "ftdi/gpio.hh"
 #include "flash/flash.hh"
 #include "commands.hh"
 #include "ftdi/log.hh"
@@ -41,17 +42,22 @@ static std::vector<ftdi::DeviceInfo> scan() {
 }
 
 static std::unique_ptr<argparse::ArgumentParser>
-new_flash_command(const std::string& name, const std::string& desc) {
+new_command(const std::string& name, const std::string& desc) {
   auto cmd = std::make_unique<argparse::ArgumentParser>(name);
   cmd->add_description(desc);
-  cmd->add_argument("--interface", "-i").help("One of [spi, i2c, gpio]");
   cmd->add_argument("--traces", "-t")
       .help("Enable ftdi traces")
       .default_value(false)
       .implicit_value(true);
   cmd->add_argument("--ftdi")
       .default_value(std::string("FT4222"))
       .help("Filter ftdi chips connected to USB");
+  return cmd;
+}
+
+static std::unique_ptr<argparse::ArgumentParser>
+new_flash_command(const std::string& name, const std::string& desc) {
+  auto cmd = new_command(name, desc);
   cmd->add_argument("--clock")
       .default_value(std::size_t{15000000})
       .help("The spi clock speed in Hz.")
@@ -79,6 +85,22 @@ handle_flash_command(std::unique_ptr<argparse::ArgumentParser>& cmd) {
   exit(0);
 }
 
+static std::optional<ftdi::Gpio>
+handle_gpio_command(std::unique_ptr<argparse::ArgumentParser>& cmd) {
+  auto ftdi_filter = cmd->get<std::string>("--ftdi");
+  auto devices     = scan();
+  auto filtered    = ftdi::DeviceInfo::filter_by_description(devices, ftdi_filter);
+  if (filtered.empty()) {
+    std::print("No {} ftdi found.\n", ftdi_filter);
+    exit(0);
+  }
+  if (auto opt = ftdi::Gpio::from_device_info(filtered[0])) {
+    return opt;
+  }
+  std::println("Can't open GPIO.");
+  exit(0);
+}
+
 int main(int argc, char* argv[]) {
   std::map<std::string, Action> commands;
 
@@ -204,6 +226,19 @@ int main(int argc, char* argv[]) {
     return 0;
   };
 
+  auto gpio_write_cmd = new_command("gpio-write", "Write a value to an FTDI GPIO pin.");
+  gpio_write_cmd->add_argument("pin").help("GPIO pin number (0-3)").required().scan<'d', int>();
+  gpio_write_cmd->add_argument("value").help("Value to write (0 or 1)").required().scan<'d', int>();
+  program.add_subparser(*gpio_write_cmd);
+  commands["gpio-write"] = [&]() -> int {
+    auto pin   = static_cast<uint8_t>(gpio_write_cmd->get<int>("pin"));
+    auto value = gpio_write_cmd->get<int>("value") != 0;
+    auto gpio  = handle_gpio_command(gpio_write_cmd);
+    commands::GpioWrite(*gpio, pin, value).run();
+    gpio->close();
+    return 0;
+  };
+
   if (argc == 1) {
     std::cout << program;  // This prints the auto-generated help
     return 0;