// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/* Copyright(c) 2025  Realtek Corporation
 */

#include "main.h"
#include "debug.h"
#include "led.h"

static int rtw_led_set(struct led_classdev *led,
         enum led_brightness brightness)
{
 struct rtw_dev *rtwdev = container_of(led, struct rtw_dev, led_cdev);

 mutex_lock(&rtwdev->mutex);

 rtwdev->chip->ops->led_set(led, brightness);

 mutex_unlock(&rtwdev->mutex);

 return 0;
}

void rtw_led_init(struct rtw_dev *rtwdev)
{
 static const struct ieee80211_tpt_blink rtw_tpt_blink[] = {
  { .throughput = 0 * 1024, .blink_time = 334 },
  { .throughput = 1 * 1024, .blink_time = 260 },
  { .throughput = 5 * 1024, .blink_time = 220 },
  { .throughput = 10 * 1024, .blink_time = 190 },
  { .throughput = 20 * 1024, .blink_time = 170 },
  { .throughput = 50 * 1024, .blink_time = 150 },
  { .throughput = 70 * 1024, .blink_time = 130 },
  { .throughput = 100 * 1024, .blink_time = 110 },
  { .throughput = 200 * 1024, .blink_time = 80 },
  { .throughput = 300 * 1024, .blink_time = 50 },
 };
 struct led_classdev *led = &rtwdev->led_cdev;
 int err;

 if (!rtwdev->chip->ops->led_set)
  return;

 led->brightness_set_blocking = rtw_led_set;

 snprintf(rtwdev->led_name, sizeof(rtwdev->led_name),
   "rtw88-%s", dev_name(rtwdev->dev));

 led->name = rtwdev->led_name;
 led->max_brightness = LED_ON;
 led->default_trigger =
  ieee80211_create_tpt_led_trigger(rtwdev->hw,
       IEEE80211_TPT_LEDTRIG_FL_RADIO,
       rtw_tpt_blink,
       ARRAY_SIZE(rtw_tpt_blink));

 err = led_classdev_register(rtwdev->dev, led);
 if (err) {
  rtw_warn(rtwdev, "Failed to register the LED, error %d\n", err);
  return;
 }

 rtwdev->led_registered = true;
}

void rtw_led_deinit(struct rtw_dev *rtwdev)
{
 struct led_classdev *led = &rtwdev->led_cdev;

 if (!rtwdev->led_registered)
  return;

 rtwdev->chip->ops->led_set(led, LED_OFF);
 led_classdev_unregister(led);
}