// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2025 Marek Vasut
 *
 * Limitations:
 * - no support for offset/polarity
 * - fixed 30 kHz period
 *
 * Argon Fan HAT https://argon40.com/products/argon-fan-hat
 */

#include <linux/err.h>
#include <linux/i2c.h>
#include <linux/module.h>
#include <linux/pwm.h>

#define ARGON40_FAN_HAT_PERIOD_NS	33333	/* ~30 kHz */

#define ARGON40_FAN_HAT_REG_DUTY_CYCLE	0x80

static int argon_fan_hat_round_waveform_tohw(struct pwm_chip *chip,
					     struct pwm_device *pwm,
					     const struct pwm_waveform *wf,
					     void *_wfhw)
{
	u8 *wfhw = _wfhw;

	if (wf->duty_length_ns > ARGON40_FAN_HAT_PERIOD_NS)
		*wfhw = 100;
	else
		*wfhw = mul_u64_u64_div_u64(wf->duty_length_ns, 100, ARGON40_FAN_HAT_PERIOD_NS);

	return 0;
}

static int argon_fan_hat_round_waveform_fromhw(struct pwm_chip *chip,
					       struct pwm_device *pwm,
					       const void *_wfhw,
					       struct pwm_waveform *wf)
{
	const u8 *wfhw = _wfhw;

	wf->period_length_ns = ARGON40_FAN_HAT_PERIOD_NS;
	wf->duty_length_ns = DIV64_U64_ROUND_UP(wf->period_length_ns * *wfhw, 100);
	wf->duty_offset_ns = 0;

	return 0;
}

static int argon_fan_hat_write_waveform(struct pwm_chip *chip,
					struct pwm_device *pwm,
					const void *_wfhw)
{
	struct i2c_client *i2c = pwmchip_get_drvdata(chip);
	const u8 *wfhw = _wfhw;

	return i2c_smbus_write_byte_data(i2c, ARGON40_FAN_HAT_REG_DUTY_CYCLE, *wfhw);
}

static const struct pwm_ops argon_fan_hat_pwm_ops = {
	.sizeof_wfhw = sizeof(u8),
	.round_waveform_fromhw = argon_fan_hat_round_waveform_fromhw,
	.round_waveform_tohw = argon_fan_hat_round_waveform_tohw,
	.write_waveform = argon_fan_hat_write_waveform,
	/*
	 * The controller does not provide any way to read info back,
	 * reading from the controller stops the fan, therefore there
	 * is no .read_waveform here.
	 */
};

static int argon_fan_hat_i2c_probe(struct i2c_client *i2c)
{
	struct pwm_chip *chip = devm_pwmchip_alloc(&i2c->dev, 1, 0);
	int ret;

	if (IS_ERR(chip))
		return PTR_ERR(chip);

	chip->ops = &argon_fan_hat_pwm_ops;
	pwmchip_set_drvdata(chip, i2c);

	ret = devm_pwmchip_add(&i2c->dev, chip);
	if (ret)
		return dev_err_probe(&i2c->dev, ret, "Could not add PWM chip\n");

	return 0;
}

static const struct of_device_id argon_fan_hat_dt_ids[] = {
	{ .compatible = "argon40,fan-hat" },
	{ },
};
MODULE_DEVICE_TABLE(of, argon_fan_hat_dt_ids);

static struct i2c_driver argon_fan_hat_driver = {
	.driver = {
		.name = "argon-fan-hat",
		.probe_type = PROBE_PREFER_ASYNCHRONOUS,
		.of_match_table = argon_fan_hat_dt_ids,
	},
	.probe = argon_fan_hat_i2c_probe,
};

module_i2c_driver(argon_fan_hat_driver);

MODULE_AUTHOR("Marek Vasut <marek.vasut+renesas@mailbox.org>");
MODULE_DESCRIPTION("Argon40 Fan HAT");
MODULE_LICENSE("GPL");