Added initial LED driver

Author: siliconwitch

source/application/Makefile

@@ -42,6 +42,7 @@ C_FILES += \
 	lua_libraries/display.c \
 	lua_libraries/file.c \
 	lua_libraries/imu.c \
+	lua_libraries/led.c \
 	lua_libraries/microphone.c \
 	lua_libraries/system.c \
 	lua_libraries/time.c \
@@ -85,6 +86,7 @@ C_FILES += \
 	$(LIBRARIES)/lz4/lz4.c \
 	$(LIBRARIES)/nrfx/drivers/src/nrfx_gpiote.c \
 	$(LIBRARIES)/nrfx/drivers/src/nrfx_pdm.c \
+	$(LIBRARIES)/nrfx/drivers/src/nrfx_pwm.c \
 	$(LIBRARIES)/nrfx/drivers/src/nrfx_rtc.c \
 	$(LIBRARIES)/nrfx/drivers/src/nrfx_saadc.c \
 	$(LIBRARIES)/nrfx/drivers/src/nrfx_spim.c \

source/application/lua_libraries/frame_lua_libraries.h

@@ -35,6 +35,7 @@ void lua_open_bluetooth_library(lua_State *L);
 void lua_open_camera_library(lua_State *L);
 void lua_open_display_library(lua_State *L);
 void lua_open_imu_library(lua_State *L);
+void lua_open_led_library(lua_State *L);
 void lua_open_microphone_library(lua_State *L);
 void lua_open_system_library(lua_State *L);
 void lua_open_time_library(lua_State *L);

source/application/lua_libraries/led.c

@@ -0,0 +1,93 @@
+/*
+ * This file is a part of: https://github.com/brilliantlabsAR/frame-codebase
+ *
+ * Authored by: Raj Nakarja / Brilliant Labs Ltd. (raj@brilliant.xyz)
+ *              Rohit Rathnam / Silicon Witchery AB (rohit@siliconwitchery.com)
+ *              Uma S. Gupta / Techno Exponent (umasankar@technoexponent.com)
+ *
+ * ISC Licence
+ *
+ * Copyright © 2024 Brilliant Labs Ltd.
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+ * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+ * AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+ * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+ * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+ * OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+ * PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#include "error_logging.h"
+#include "lauxlib.h"
+#include "lua.h"
+#include "nrfx_log.h"
+#include "nrfx_pwm.h"
+#include "pinout.h"
+
+static const nrfx_pwm_t pwm = NRFX_PWM_INSTANCE(0);
+
+static int lua_led_set_color(lua_State *L)
+{
+    lua_Integer red = luaL_checkinteger(L, 1);
+    lua_Integer green = luaL_checkinteger(L, 2);
+    lua_Integer blue = luaL_checkinteger(L, 3);
+
+    if (red < 0 || red > 100 || green < 0 || green > 100 || blue < 0 || blue > 100)
+    {
+        luaL_error(L, "led color must be between 0 and 100");
+    }
+
+    static nrf_pwm_values_individual_t seq_values;
+
+    seq_values.channel_0 = (uint16_t)red;
+    seq_values.channel_1 = (uint16_t)green;
+    seq_values.channel_2 = (uint16_t)blue;
+
+    nrf_pwm_sequence_t const seq = {
+        .values.p_individual = &seq_values,
+        .length = NRF_PWM_VALUES_LENGTH(seq_values),
+        .repeats = 0,
+        .end_delay = 0,
+    };
+
+    nrfx_pwm_simple_playback(&pwm, &seq, 1, NRFX_PWM_FLAG_LOOP);
+
+    return 0;
+}
+
+void lua_open_led_library(lua_State *L)
+{
+    nrfx_pwm_config_t config = NRFX_PWM_DEFAULT_CONFIG(
+        FRAME_LITE_LED_RED_PIN,
+        FRAME_LITE_LED_GREEN_PIN,
+        FRAME_LITE_LED_BLUE_PIN,
+        NRF_PWM_PIN_NOT_CONNECTED);
+
+    config.pin_inverted[0] = true;
+    config.pin_inverted[1] = true;
+    config.pin_inverted[2] = true;
+
+    config.load_mode = NRF_PWM_LOAD_INDIVIDUAL;
+    config.top_value = 100;
+
+    if (nrfx_pwm_init_check(&pwm) == false)
+    {
+        nrfx_pwm_init(&pwm, &config, NULL, NULL);
+    }
+
+    lua_getglobal(L, "frame");
+
+    lua_newtable(L);
+
+    lua_pushcfunction(L, lua_led_set_color);
+    lua_setfield(L, -2, "set_color");
+
+    lua_setfield(L, -2, "led");
+
+    lua_pop(L, 1);
+}

source/application/luaport.c

@@ -98,6 +98,7 @@ void run_lua(bool is_paired)
     lua_open_microphone_library(L);
     lua_open_imu_library(L);
     lua_open_time_library(L);
+    lua_open_led_library(L);
 
     lua_open_file_library(L, !is_paired);
 

source/application/nrfx_config.h

@@ -33,6 +33,11 @@
 #define NRFX_PDM_ENABLED 1
 #define NRFX_PDM_DEFAULT_CONFIG_IRQ_PRIORITY 5
 
+#define NRFX_PWM_ENABLED 1
+#define NRFX_PWM0_ENABLED 1
+#define NRFX_PWM1_ENABLED 1
+#define NRFX_PWM2_ENABLED 1
+
 #define NRFX_RTC_ENABLED 1
 #define NRFX_RTC1_ENABLED 1
 

tests/test_led.py

@@ -0,0 +1,26 @@
+"""
+Tests the Frame specific Lua libraries over Bluetooth.
+"""
+
+import asyncio
+from frameutils import Bluetooth
+
+
+async def main():
+    b = Bluetooth()
+
+    await b.connect(print_response_handler=lambda s: print(s))
+
+    # await b.send_lua("frame.led.set_color(100, 0, 0)")
+    await asyncio.sleep(1)
+
+    # await b.send_lua("frame.led.set_color(0, 100, 0)")
+    await asyncio.sleep(1)
+
+    await b.send_lua("frame.led.set_color(0, 0, 100)")
+    await asyncio.sleep(1)
+
+    await b.disconnect()
+
+
+asyncio.run(main())