Add new battery behavior

ReFil · Jan 25, 2025 · 2680c7c · 2680c7c
1 parent 548af64
commit 2680c7c
Show file tree

Hide file tree

Showing 5 changed files with 86 additions and 0 deletions.
diff --git a/app/CMakeLists.txt b/app/CMakeLists.txt
@@ -88,6 +88,7 @@ if ((NOT CONFIG_ZMK_SPLIT) OR CONFIG_ZMK_SPLIT_ROLE_CENTRAL)
 endif()
 
 target_sources_ifdef(CONFIG_ZMK_RGB_UNDERGLOW app PRIVATE src/behaviors/behavior_rgb_underglow.c)
+target_sources_ifdef(CONFIG_ZMK_RGB_UNDERGLOW app PRIVATE src/behaviors/behavior_stp_indicators.c)
 target_sources_ifdef(CONFIG_ZMK_BACKLIGHT app PRIVATE src/behaviors/behavior_backlight.c)
 
 target_sources_ifdef(CONFIG_ZMK_BATTERY_REPORTING app PRIVATE src/events/battery_state_changed.c)

diff --git a/app/dts/behaviors.dtsi b/app/dts/behaviors.dtsi
@@ -28,3 +28,4 @@
 #include <behaviors/soft_off.dtsi>
 #include <behaviors/studio_unlock.dtsi>
 #include <behaviors/mouse_keys.dtsi>
+#include <behaviors/stp_indicators.dtsi>
diff --git a/app/dts/behaviors/stp_indicators.dtsi b/app/dts/behaviors/stp_indicators.dtsi
@@ -0,0 +1,9 @@
+/ {
+    behaviors {
+        stp: behavior_stp_indicators {
+            compatible = "zmk,behavior-stp-indicators";
+            display-name = "Battery Level";
+            #binding-cells = <1>;
+        };
+    };
+};
diff --git a/app/include/dt-bindings/zmk/stp.h b/app/include/dt-bindings/zmk/stp.h
@@ -0,0 +1 @@
+#define STP_BAT 0
diff --git a/app/src/behaviors/behavior_stp_indicators.c b/app/src/behaviors/behavior_stp_indicators.c
@@ -0,0 +1,74 @@
+/*
+ * Copyright (c) 2020 The ZMK Contributors
+ *
+ * SPDX-License-Identifier: MIT
+ */
+
+#define DT_DRV_COMPAT zmk_behavior_stp_indicators
+
+#include <zephyr/device.h>
+#include <drivers/behavior.h>
+#include <zephyr/logging/log.h>
+
+#include <dt-bindings/zmk/stp.h>
+#include <zmk/rgb_underglow.h>
+#include <zmk/keymap.h>
+
+LOG_MODULE_DECLARE(zmk, CONFIG_ZMK_LOG_LEVEL);
+
+#if DT_HAS_COMPAT_STATUS_OKAY(DT_DRV_COMPAT)
+
+static const struct behavior_parameter_value_metadata param_values[] = {
+    {
+        .value = STP_BAT,
+        .display_name = "Battery command",
+        .type = BEHAVIOR_PARAMETER_VALUE_TYPE_VALUE,
+    },
+
+};
+
+static const struct behavior_parameter_metadata_set param_metadata_set[] = {{
+    .param1_values = param_values,
+    .param1_values_len = ARRAY_SIZE(param_values),
+}};
+
+static const struct behavior_parameter_metadata metadata = {
+    .sets_len = ARRAY_SIZE(param_metadata_set),
+    .sets = param_metadata_set,
+};
+
+static int behavior_stp_indicators_init(const struct device *dev) { return 0; }
+
+static int on_keymap_binding_pressed(struct zmk_behavior_binding *binding,
+                                     struct zmk_behavior_binding_event event) {
+    switch (binding->param1) {
+    case STP_BAT:
+        return zmk_rgb_underglow_select_effect(5);
+    }
+
+    return -ENOTSUP;
+}
+
+static int on_keymap_binding_released(struct zmk_behavior_binding *binding,
+                                      struct zmk_behavior_binding_event event) {
+    switch (binding->param1) {
+    case STP_BAT:
+        return zmk_rgb_underglow_select_effect(4);
+    }
+    return ZMK_BEHAVIOR_OPAQUE;
+}
+
+static const struct behavior_driver_api behavior_stp_indicators_driver_api = {
+    .binding_pressed = on_keymap_binding_pressed,
+    .binding_released = on_keymap_binding_released,
+    .locality = BEHAVIOR_LOCALITY_CENTRAL,
+
+#if IS_ENABLED(CONFIG_ZMK_BEHAVIOR_METADATA)
+    .parameter_metadata = &metadata,
+#endif // IS_ENABLED(CONFIG_ZMK_BEHAVIOR_METADATA)
+};
+
+BEHAVIOR_DT_INST_DEFINE(0, behavior_stp_indicators_init, NULL, NULL, NULL, POST_KERNEL,
+                        CONFIG_KERNEL_INIT_PRIORITY_DEFAULT, &behavior_stp_indicators_driver_api);
+
+#endif /* DT_HAS_COMPAT_STATUS_OKAY(DT_DRV_COMPAT) */