Guybrush Zephyr: Add LED control

Add PWM definitions for LEDs and copy the board LED control code for the
on/off states.

BRANCH=None
BUG=b:195137794
TEST=on guybrush, verify charging LED when charging, white LED when on,
blinking white in suspend

Signed-off-by: Diana Z <dzigterman@chromium.org>
Change-Id: I80ea849e8f1267c090a995a4b74275a88006c211
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/ec/+/3276906
Commit-Queue: Jack Rosenthal <jrosenth@chromium.org>
Reviewed-by: Jack Rosenthal <jrosenth@chromium.org>

4 files changed, 132 insertions, 0 deletions

diff --git a/zephyr/projects/guybrush/CMakeLists.txt b/zephyr/projects/guybrush/CMakeLists.txt
index 68515471f3..30e8b8c611 100644
--- a/zephyr/projects/guybrush/CMakeLists.txt
+++ b/zephyr/projects/guybrush/CMakeLists.txt
@@ -14,3 +14,6 @@ zephyr_library_sources("power_signals.c")
 zephyr_library_sources_ifdef(CONFIG_PLATFORM_EC_USBC
 	"usbc_config.c"
 	"usb_pd_policy.c")
+
+zephyr_library_sources_ifdef(CONFIG_PLATFORM_EC_LED_COMMON
+	"led.c")
diff --git a/zephyr/projects/guybrush/led.c b/zephyr/projects/guybrush/led.c
new file mode 100644
index 0000000000..6a2a2e4609
--- /dev/null
+++ b/zephyr/projects/guybrush/led.c
@@ -0,0 +1,103 @@
+/* Copyright 2021 The Chromium OS Authors. All rights reserved.
+ * Use of this source code is governed by a BSD-style license that can be
+ * found in the LICENSE file.
+ *
+ * Guybrush specific PWM LED settings.
+ */
+
+#include "common.h"
+#include "led_onoff_states.h"
+#include "led_common.h"
+#include "gpio.h"
+#include "hooks.h"
+#include "pwm.h"
+
+/* Note PWM LEDs are active low */
+#define LED_OFF_LVL	1
+#define LED_ON_LVL	0
+
+/* Define our PWM_CH values for zephyr */
+#define PWM_CH_LED_CHRG PWM_CHANNEL(DT_NODELABEL(led_charge))
+#define PWM_CH_LED_FULL PWM_CHANNEL(DT_NODELABEL(led_full))
+
+#define CPRINTS(format, args...) cprints(CC_PWM, format, ## args)
+
+__override const int led_charge_lvl_1 = 5;
+
+__override const int led_charge_lvl_2 = 97;
+
+__override struct led_descriptor
+			led_bat_state_table[LED_NUM_STATES][LED_NUM_PHASES] = {
+	[STATE_CHARGING_LVL_1]	     = {{EC_LED_COLOR_AMBER, LED_INDEFINITE} },
+	[STATE_CHARGING_LVL_2]	     = {{EC_LED_COLOR_AMBER, LED_INDEFINITE} },
+	[STATE_CHARGING_FULL_CHARGE] = {{EC_LED_COLOR_WHITE, LED_INDEFINITE} },
+	[STATE_DISCHARGE_S0]	     = {{EC_LED_COLOR_WHITE, LED_INDEFINITE} },
+	[STATE_DISCHARGE_S3]	     = {{EC_LED_COLOR_WHITE, 1 * LED_ONE_SEC},
+					{LED_OFF,	     1 * LED_ONE_SEC} },
+	[STATE_DISCHARGE_S5]         = {{LED_OFF,            LED_INDEFINITE} },
+	[STATE_BATTERY_ERROR]        = {{EC_LED_COLOR_AMBER, 1 * LED_ONE_SEC},
+					{LED_OFF,	     1 * LED_ONE_SEC} },
+	[STATE_FACTORY_TEST]         = {{EC_LED_COLOR_AMBER, 2 * LED_ONE_SEC},
+					{EC_LED_COLOR_WHITE, 2 * LED_ONE_SEC} },
+};
+
+const enum ec_led_id supported_led_ids[] = {
+	EC_LED_ID_BATTERY_LED,
+};
+
+const int supported_led_ids_count = ARRAY_SIZE(supported_led_ids);
+
+__override void led_set_color_battery(enum ec_led_colors color)
+{
+	switch (color) {
+	case EC_LED_COLOR_AMBER:
+		pwm_enable(PWM_CH_LED_CHRG, LED_ON_LVL);
+		pwm_enable(PWM_CH_LED_FULL, LED_OFF_LVL);
+		break;
+	case EC_LED_COLOR_WHITE:
+		pwm_enable(PWM_CH_LED_CHRG, LED_OFF_LVL);
+		pwm_enable(PWM_CH_LED_FULL, LED_ON_LVL);
+		break;
+	case LED_OFF:
+		pwm_enable(PWM_CH_LED_CHRG, LED_OFF_LVL);
+		pwm_enable(PWM_CH_LED_FULL, LED_OFF_LVL);
+		break;
+	default: /* Unsupported colors */
+		CPRINTS("Unsupported LED color: %d", color);
+		pwm_enable(PWM_CH_LED_CHRG, LED_OFF_LVL);
+		pwm_enable(PWM_CH_LED_FULL, LED_OFF_LVL);
+		break;
+	}
+}
+
+void led_get_brightness_range(enum ec_led_id led_id, uint8_t *brightness_range)
+{
+	if (led_id == EC_LED_ID_BATTERY_LED) {
+		brightness_range[EC_LED_COLOR_AMBER] = 1;
+		brightness_range[EC_LED_COLOR_WHITE] = 1;
+	}
+}
+
+int led_set_brightness(enum ec_led_id led_id, const uint8_t *brightness)
+{
+	if (led_id == EC_LED_ID_BATTERY_LED) {
+		if (brightness[EC_LED_COLOR_WHITE] != 0)
+			led_set_color_battery(EC_LED_COLOR_WHITE);
+		else if (brightness[EC_LED_COLOR_AMBER] != 0)
+			led_set_color_battery(EC_LED_COLOR_AMBER);
+		else
+			led_set_color_battery(LED_OFF);
+	} else {
+		CPRINTS("Unsupported LED set: %d", led_id);
+		return EC_ERROR_INVAL;
+	}
+
+	return EC_SUCCESS;
+}
+
+static void pwm_led_duty_init(void)
+{
+	pwm_set_duty(PWM_CH_LED_CHRG, 100);
+	pwm_set_duty(PWM_CH_LED_FULL, 100);
+}
+DECLARE_HOOK(HOOK_INIT, pwm_led_duty_init, HOOK_PRIO_INIT_PWM + 1);
diff --git a/zephyr/projects/guybrush/prj.conf b/zephyr/projects/guybrush/prj.conf
index 609eb044a7..5adf663ad3 100644
--- a/zephyr/projects/guybrush/prj.conf
+++ b/zephyr/projects/guybrush/prj.conf
@@ -36,6 +36,10 @@ CONFIG_SENSOR=y
 CONFIG_SENSOR_SHELL=n
 CONFIG_TACH_NPCX=y
 
+# LEDs
+CONFIG_PLATFORM_EC_LED_COMMON=y
+CONFIG_PLATFORM_EC_LED_ONOFF_STATES=y
+
 # Lid switch
 CONFIG_PLATFORM_EC_LID_ANGLE=y
 CONFIG_PLATFORM_EC_LID_ANGLE_UPDATE=y
diff --git a/zephyr/projects/guybrush/pwm.dts b/zephyr/projects/guybrush/pwm.dts
index c7b975e5fa..dd9fc94eaa 100644
--- a/zephyr/projects/guybrush/pwm.dts
+++ b/zephyr/projects/guybrush/pwm.dts
@@ -17,6 +17,16 @@
 			label = "KBLIGHT";
 			frequency = <100>;
 		};
+		led_charge: led_charge {
+			pwms = <&pwm2 0 0>;
+			label = "LED_CHARGE";
+			frequency = <100>;
+		};
+		led_full: led_full {
+			pwms = <&pwm3 0 0>;
+			label = "LED_FULL";
+			frequency = <100>;
+		};
 	};
 };
 
@@ -31,3 +41,15 @@
 	status = "okay";
 	clock-bus = "NPCX_CLOCK_BUS_LFCLK";
 };
+
+/* Amber charging LED */
+&pwm2 {
+	status = "okay";
+	clock-bus = "NPCX_CLOCK_BUS_LFCLK";
+};
+
+/* Green full LED */
+&pwm3 {
+	status = "okay";
+	clock-bus = "NPCX_CLOCK_BUS_LFCLK";
+};