[u-boot.git] / drivers / led / led_pwm.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (C) 2022 VK
 * Author: Ivan Vozvakhov <[email protected]>
 */

#include <dm.h>
#include <errno.h>
#include <led.h>
#include <malloc.h>
#include <pwm.h>

#define LEDS_PWM_DRIVER_NAME	"led_pwm"

struct led_pwm_priv {
	struct udevice *pwm;
	uint period;	/* period in ns */
	uint duty;	/* duty cycle in ns */
	uint channel;	/* pwm channel number */
	bool active_low;	/* pwm polarity */
	bool enabled;
};

static int led_pwm_enable(struct udevice *dev)
{
	struct led_pwm_priv *priv = dev_get_priv(dev);
	int ret;

	ret = pwm_set_invert(priv->pwm, priv->channel, priv->active_low);
	if (ret)
		return ret;

	ret = pwm_set_config(priv->pwm, priv->channel, priv->period, priv->duty);
	if (ret)
		return ret;

	ret = pwm_set_enable(priv->pwm, priv->channel, true);
	if (ret)
		return ret;

	priv->enabled = true;

	return 0;
}

static int led_pwm_disable(struct udevice *dev)
{
	struct led_pwm_priv *priv = dev_get_priv(dev);
	int ret;

	ret = pwm_set_config(priv->pwm, priv->channel, priv->period, 0);
	if (ret)
		return ret;

	ret = pwm_set_enable(priv->pwm, priv->channel, false);
	if (ret)
		return ret;

	priv->enabled = false;

	return 0;
}

static int led_pwm_set_state(struct udevice *dev, enum led_state_t state)
{
	struct led_pwm_priv *priv = dev_get_priv(dev);
	int ret;

	switch (state) {
	case LEDST_OFF:
		ret = led_pwm_disable(dev);
		break;
	case LEDST_ON:
		ret = led_pwm_enable(dev);
		break;
	case LEDST_TOGGLE:
		ret = (priv->enabled) ? led_pwm_disable(dev) : led_pwm_enable(dev);
		break;
	default:
		ret = -ENOSYS;
	}

	return ret;
}

static enum led_state_t led_pwm_get_state(struct udevice *dev)
{
	struct led_pwm_priv *priv = dev_get_priv(dev);

	return (priv->enabled) ? LEDST_ON : LEDST_OFF;
}

static int led_pwm_probe(struct udevice *dev)
{
	struct led_pwm_priv *priv = dev_get_priv(dev);

	return led_pwm_set_state(dev, (priv->enabled) ? LEDST_ON : LEDST_OFF);
}

static int led_pwm_of_to_plat(struct udevice *dev)
{
	struct led_pwm_priv *priv = dev_get_priv(dev);
	struct ofnode_phandle_args args;
	uint def_brightness, max_brightness;
	int ret;

	ret = dev_read_phandle_with_args(dev, "pwms", "#pwm-cells", 0, 0, &args);
	if (ret)
		return ret;

	ret = uclass_get_device_by_ofnode(UCLASS_PWM, args.node, &priv->pwm);
	if (ret)
		return ret;

	priv->channel = args.args[0];
	priv->period = args.args[1];
	priv->active_low = dev_read_bool(dev, "active-low");

	def_brightness = dev_read_u32_default(dev, "u-boot,default-brightness", 0);
	max_brightness = dev_read_u32_default(dev, "max-brightness", 255);
	priv->enabled =  !!def_brightness;

	/*
	 * No need to handle pwm inverted case (active_low)
	 * because of pwm_set_invert function
	 */
	if (def_brightness < max_brightness)
		priv->duty = priv->period * def_brightness / max_brightness;
	else
		priv->duty = priv->period;

	return 0;
}

static int led_pwm_bind(struct udevice *parent)
{
	return led_bind_generic(parent, LEDS_PWM_DRIVER_NAME);
}

static const struct led_ops led_pwm_ops = {
	.set_state = led_pwm_set_state,
	.get_state = led_pwm_get_state,
};

static const struct udevice_id led_pwm_ids[] = {
	{ .compatible = "pwm-leds" },
	{ }
};

U_BOOT_DRIVER(led_pwm) = {
	.name = LEDS_PWM_DRIVER_NAME,
	.id = UCLASS_LED,
	.ops = &led_pwm_ops,
	.priv_auto = sizeof(struct led_pwm_priv),
	.probe = led_pwm_probe,
	.of_to_plat = led_pwm_of_to_plat,
};

U_BOOT_DRIVER(led_pwm_wrap) = {
	.name = LEDS_PWM_DRIVER_NAME "_wrap",
	.id = UCLASS_NOP,
	.of_match = led_pwm_ids,
	.bind = led_pwm_bind,
};
Commit	Line	Data
067cfc1c IV	1	// SPDX-License-Identifier: GPL-2.0+
	2	/*
	3	* Copyright (C) 2022 VK
	4	* Author: Ivan Vozvakhov <[email protected]>
	5	*/
	6
067cfc1c IV	7	#include <dm.h>
	8	#include <errno.h>
	9	#include <led.h>
	10	#include <malloc.h>
067cfc1c IV	11	#include <pwm.h>
	12
	13	#define LEDS_PWM_DRIVER_NAME "led_pwm"
	14
	15	struct led_pwm_priv {
	16	struct udevice *pwm;
	17	uint period; /* period in ns */
	18	uint duty; /* duty cycle in ns */
	19	uint channel; /* pwm channel number */
	20	bool active_low; /* pwm polarity */
	21	bool enabled;
	22	};
	23
	24	static int led_pwm_enable(struct udevice *dev)
	25	{
	26	struct led_pwm_priv *priv = dev_get_priv(dev);
	27	int ret;
	28
	29	ret = pwm_set_invert(priv->pwm, priv->channel, priv->active_low);
	30	if (ret)
	31	return ret;
	32
	33	ret = pwm_set_config(priv->pwm, priv->channel, priv->period, priv->duty);
	34	if (ret)
	35	return ret;
	36
	37	ret = pwm_set_enable(priv->pwm, priv->channel, true);
	38	if (ret)
	39	return ret;
	40
	41	priv->enabled = true;
	42
	43	return 0;
	44	}
	45
	46	static int led_pwm_disable(struct udevice *dev)
	47	{
	48	struct led_pwm_priv *priv = dev_get_priv(dev);
	49	int ret;
	50
	51	ret = pwm_set_config(priv->pwm, priv->channel, priv->period, 0);
	52	if (ret)
	53	return ret;
	54
	55	ret = pwm_set_enable(priv->pwm, priv->channel, false);
	56	if (ret)
	57	return ret;
	58
	59	priv->enabled = false;
	60
	61	return 0;
	62	}
	63
	64	static int led_pwm_set_state(struct udevice *dev, enum led_state_t state)
	65	{
	66	struct led_pwm_priv *priv = dev_get_priv(dev);
	67	int ret;
	68
	69	switch (state) {
	70	case LEDST_OFF:
	71	ret = led_pwm_disable(dev);
	72	break;
	73	case LEDST_ON:
	74	ret = led_pwm_enable(dev);
75	break;
76	case LEDST_TOGGLE:
77	ret = (priv->enabled) ? led_pwm_disable(dev) : led_pwm_enable(dev);
78	break;
79	default:
80	ret = -ENOSYS;
81	}
82
83	return ret;
84	}
85
86	static enum led_state_t led_pwm_get_state(struct udevice *dev)
87	{
88	struct led_pwm_priv *priv = dev_get_priv(dev);
89
90	return (priv->enabled) ? LEDST_ON : LEDST_OFF;
91	}
92
93	static int led_pwm_probe(struct udevice *dev)
94	{
95	struct led_pwm_priv *priv = dev_get_priv(dev);
067cfc1c IV	96
	97	return led_pwm_set_state(dev, (priv->enabled) ? LEDST_ON : LEDST_OFF);
	98	}
	99
	100	static int led_pwm_of_to_plat(struct udevice *dev)
	101	{
067cfc1c IV	102	struct led_pwm_priv *priv = dev_get_priv(dev);
	103	struct ofnode_phandle_args args;
	104	uint def_brightness, max_brightness;
	105	int ret;
	106
067cfc1c IV	107	ret = dev_read_phandle_with_args(dev, "pwms", "#pwm-cells", 0, 0, &args);
	108	if (ret)
	109	return ret;
	110
	111	ret = uclass_get_device_by_ofnode(UCLASS_PWM, args.node, &priv->pwm);
	112	if (ret)
	113	return ret;
	114
	115	priv->channel = args.args[0];
	116	priv->period = args.args[1];
	117	priv->active_low = dev_read_bool(dev, "active-low");
	118
	119	def_brightness = dev_read_u32_default(dev, "u-boot,default-brightness", 0);
	120	max_brightness = dev_read_u32_default(dev, "max-brightness", 255);
	121	priv->enabled = !!def_brightness;
	122
	123	/*
3708739e	124	* No need to handle pwm inverted case (active_low)
067cfc1c IV	125	* because of pwm_set_invert function
	126	*/
	127	if (def_brightness < max_brightness)
	128	priv->duty = priv->period * def_brightness / max_brightness;
	129	else
	130	priv->duty = priv->period;
	131
	132	return 0;
	133	}
	134
	135	static int led_pwm_bind(struct udevice *parent)
	136	{
d45e712f	137	return led_bind_generic(parent, LEDS_PWM_DRIVER_NAME);
067cfc1c IV	138	}
	139
	140	static const struct led_ops led_pwm_ops = {
	141	.set_state = led_pwm_set_state,
	142	.get_state = led_pwm_get_state,
	143	};
	144
	145	static const struct udevice_id led_pwm_ids[] = {
	146	{ .compatible = "pwm-leds" },
	147	{ }
	148	};
	149
	150	U_BOOT_DRIVER(led_pwm) = {
	151	.name = LEDS_PWM_DRIVER_NAME,
	152	.id = UCLASS_LED,
067cfc1c IV	153	.ops = &led_pwm_ops,
067cfc1c IV	154	.priv_auto = sizeof(struct led_pwm_priv),
067cfc1c IV	155	.probe = led_pwm_probe,
	156	.of_to_plat = led_pwm_of_to_plat,
	157	};
e3812b5b SH	158
	159	U_BOOT_DRIVER(led_pwm_wrap) = {
	160	.name = LEDS_PWM_DRIVER_NAME "_wrap",
	161	.id = UCLASS_NOP,
	162	.of_match = led_pwm_ids,
	163	.bind = led_pwm_bind,
	164	};