Inital Commit

sources/ugreen-led-cli/main.cpp

@@ -0,0 +1,136 @@
+#include "main.h"
+#include <string>
+#include <filesystem>
+#include <fstream>
+#include <iostream>
+
+#define I2C_DEV_PATH "/sys/class/i2c-dev/"
+
+int ugreen_leds_t::start() {
+  namespace fs = std::filesystem;
+
+  if (!fs::exists(I2C_DEV_PATH))
+    return -1;
+
+  for (const auto & entry: fs::directory_iterator(I2C_DEV_PATH)) {
+    if (entry.is_directory()) {
+      std::ifstream ifs(entry.path() / "device/name");
+      std::string line;
+      std::getline(ifs, line);
+
+      if (line.rfind("SMBus I801 adapter", 0) == 0) {
+        const auto i2c_dev = "/dev/" + entry.path().filename().string();
+        return _i2c.start(i2c_dev.c_str(), UGREEN_LED_I2C_ADDR);
+      }
+    }
+  }
+
+  return -1;
+}
+
+static int compute_checksum(const std::vector < uint8_t > & data, int size) {
+  if (size < 2 || size > (int) data.size())
+    return 0;
+
+  int sum = 0;
+  for (int i = 0; i < size; ++i)
+    sum += (int) data[i];
+
+  return sum;
+}
+
+static bool verify_checksum(const std::vector < uint8_t > & data) {
+  int size = data.size();
+  if (size < 2) return false;
+  int sum = compute_checksum(data, size - 2);
+  return sum != 0 && sum == (data[size - 1] | (((int) data[size - 2]) << 8));
+}
+
+static void append_checksum(std::vector < uint8_t > & data) {
+  int size = data.size();
+  int sum = compute_checksum(data, size);
+  data.push_back((sum >> 8) & 0xff);
+  data.push_back(sum & 0xff);
+}
+
+ugreen_leds_t::led_data_t ugreen_leds_t::get_status(led_type_t id) {
+  led_data_t data {};
+  data.is_available = false;
+
+  auto raw_data = _i2c.read_block_data(0x81 + (uint8_t) id, 0xb);
+  if (raw_data.size() != 0xb || !verify_checksum(raw_data))
+    return data;
+
+  switch (raw_data[0]) {
+    case 0: data.op_mode = op_mode_t::off; break;
+    case 1: data.op_mode = op_mode_t::on; break;
+    case 2: data.op_mode = op_mode_t::blink; break;
+    case 3: data.op_mode = op_mode_t::breath; break;
+    default: return data;
+  };
+
+  data.brightness = raw_data[1];
+  data.color_r = raw_data[2];
+  data.color_g = raw_data[3];
+  data.color_b = raw_data[4];
+  int t_hight = (((int) raw_data[5]) << 8) | raw_data[6];
+  int t_low = (((int) raw_data[7]) << 8) | raw_data[8];
+  data.t_on = t_low;
+  data.t_off = t_hight - t_low;
+  data.is_available = true;
+
+  return data;
+}
+
+int ugreen_leds_t::_change_status(led_type_t id, uint8_t command, std::array < std::optional < uint8_t > , 4 > params) {
+  std::vector < uint8_t > data {
+    0x00, 0xa0, 0x01,
+    0x00, 0x00, command,
+    params[0].value_or(0x00),
+    params[1].value_or(0x00),
+    params[2].value_or(0x00),
+    params[3].value_or(0x00),
+  };
+
+  append_checksum(data);
+  data[0] = (uint8_t) id;
+  return _i2c.write_block_data((uint8_t) id, data);
+}
+
+int ugreen_leds_t::set_onoff(led_type_t id, uint8_t status) {
+  if (status >= 2) return -1;
+  return _change_status(id, 0x03, {
+    status
+  });
+}
+
+int ugreen_leds_t::_set_blink_or_breath(uint8_t command, led_type_t id, uint16_t t_on, uint16_t t_off) {
+  uint16_t t_hight = t_on + t_off;
+  uint16_t t_low = t_on;
+  return _change_status(id, command, {
+    (uint8_t)(t_hight >> 8),
+    (uint8_t)(t_hight & 0xff),
+    (uint8_t)(t_low >> 8),
+    (uint8_t)(t_low & 0xff),
+  });
+}
+
+int ugreen_leds_t::set_rgb(led_type_t id, uint8_t r, uint8_t g, uint8_t b) {
+  return _change_status(id, 0x02, { r, g, b } );
+}
+
+int ugreen_leds_t::set_brightness(led_type_t id, uint8_t brightness) {
+  return _change_status(id, 0x01, { brightness } );
+}
+
+bool ugreen_leds_t::is_last_modification_successful() {
+  return _i2c.read_byte_data(0x80) == 1;
+}
+
+int ugreen_leds_t::set_blink(led_type_t id, uint16_t t_on, uint16_t t_off) {
+  return _set_blink_or_breath(0x04, id, t_on, t_off);
+}
+
+int ugreen_leds_t::set_breath(led_type_t id, uint16_t t_on, uint16_t t_off) {
+  return _set_blink_or_breath(0x05, id, t_on, t_off);
+}