/* Copyright 2019 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.
 *
 * Battery LED control for Flapjack board.
 */

#include "battery.h"
#include "charge_state.h"
#include "console.h"
#include "driver/charger/rt946x.h"
#include "hooks.h"
#include "led_common.h"
#include "util.h"

/* Define this to enable led command and debug LED */
#undef DEBUG_LED

const enum ec_led_id supported_led_ids[] = { EC_LED_ID_BATTERY_LED };

const int supported_led_ids_count = ARRAY_SIZE(supported_led_ids);

#define LED_OFF		MT6370_LED_ID_OFF
#define LED_RED		MT6370_LED_ID1
#define LED_GRN		MT6370_LED_ID2
#define LED_BLU		MT6370_LED_ID3

#define LED_MASK_OFF	0
#define LED_MASK_RED	MT6370_MASK_RGB_ISNK1DIM_EN
#define LED_MASK_GRN	MT6370_MASK_RGB_ISNK2DIM_EN
#define LED_MASK_BLU	MT6370_MASK_RGB_ISNK3DIM_EN

static enum charge_state chstate;

static void led_set_battery(void)
{
	static enum charge_state prev = PWR_STATE_UNCHANGE;
	static int battery_second;
	int red = 0, grn = 0, blu = 0;

	battery_second++;
#ifndef DEBUG_LED
	chstate = charge_get_state();
#endif
	if (chstate == prev)
		return;
	prev = chstate;

	/*
	 * Full      White
	 * Charging  Amber
	 * Error     Red
	 */
	switch (chstate) {
	case PWR_STATE_CHARGE:
		red = 0xf5;
		grn = 0xf1;
		break;
	case PWR_STATE_DISCHARGE:
		break;
	case PWR_STATE_ERROR:
		red = 0xff;
		break;
	case PWR_STATE_CHARGE_NEAR_FULL:
		red = 0xff;
		grn = 0xff;
		blu = 0xff;
		break;
	default:
		/* Other states don't alter LED behavior */
		return;
	}

	mt6370_led_set_brightness(LED_RED, red);
	mt6370_led_set_brightness(LED_GRN, grn);
	mt6370_led_set_brightness(LED_BLU, blu);
}

void led_get_brightness_range(enum ec_led_id led_id, uint8_t *brightness_range)
{
	if (led_id != EC_LED_ID_BATTERY_LED)
		return;

	brightness_range[EC_LED_COLOR_GREEN] = MT6370_LED_BRIGHTNESS_MAX;
	brightness_range[EC_LED_COLOR_RED] = MT6370_LED_BRIGHTNESS_MAX;
	brightness_range[EC_LED_COLOR_BLUE] = MT6370_LED_BRIGHTNESS_MAX;
}

int led_set_brightness(enum ec_led_id led_id, const uint8_t *brightness)
{
	if (led_id != EC_LED_ID_BATTERY_LED)
		return EC_ERROR_INVAL;

	mt6370_led_set_brightness(LED_GRN, brightness[EC_LED_COLOR_GREEN]);
	mt6370_led_set_brightness(LED_RED, brightness[EC_LED_COLOR_RED]);
	mt6370_led_set_brightness(LED_BLU, brightness[EC_LED_COLOR_BLUE]);
	return EC_SUCCESS;
}

static void flapjack_led_init(void)
{
	const enum mt6370_led_dim_mode dim = MT6370_LED_DIM_MODE_REGISTER;
	mt6370_led_set_color(LED_MASK_RED | LED_MASK_GRN | LED_MASK_BLU);
	mt6370_led_set_dim_mode(LED_RED, dim);
	mt6370_led_set_dim_mode(LED_GRN, dim);
	mt6370_led_set_dim_mode(LED_BLU, dim);
}
DECLARE_HOOK(HOOK_INIT, flapjack_led_init, HOOK_PRIO_DEFAULT);

/* Called by hook task every 1 sec */
static void led_second(void)
{
	if (led_auto_control_is_enabled(EC_LED_ID_BATTERY_LED))
		led_set_battery();
}
DECLARE_HOOK(HOOK_SECOND, led_second, HOOK_PRIO_DEFAULT);

#ifdef DEBUG_LED
static int command_led(int argc, char **argv)
{
	int val;
	char *e;

	mt6370_led_set_color(LED_MASK_RED|LED_MASK_GRN|LED_MASK_BLU);

	if (argc == 2) {
		val = strtoi(argv[1], &e, 0);
		if (*e)
			return EC_ERROR_PARAM1;
		chstate = val;
		return EC_SUCCESS;
	}

	if (argc < 4)
		return EC_ERROR_PARAM_COUNT;

	val = strtoi(argv[1], &e, 0);
	if (*e)
		return EC_ERROR_PARAM1;
	mt6370_led_set_brightness(LED_RED, val);

	val = strtoi(argv[2], &e, 0);
	if (*e)
		return EC_ERROR_PARAM2;
	mt6370_led_set_brightness(LED_GRN, val);

	val = strtoi(argv[3], &e, 0);
	if (*e)
		return EC_ERROR_PARAM3;
	mt6370_led_set_brightness(LED_BLU, val);

	return EC_SUCCESS;
}
DECLARE_CONSOLE_COMMAND(led, command_led, "<chg_state> or <R> <G> <B>", NULL);
#endif